Spaces:
Runtime error
Runtime error
File size: 760 Bytes
8018595 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
# pylint: disable=missing-docstring
import pytest
from langchain_core.prompts import PromptTemplate
from langchain_core.runnables.base import Runnable
from langchain_ollama import ChatOllama
from sentinel.llm_service import create_initial_assessment_chain, get_llm
def test_get_llm_local():
llm = get_llm("local", model="test_model")
assert isinstance(llm, ChatOllama)
def test_get_llm_invalid():
with pytest.raises(ValueError):
get_llm("invalid_provider")
def test_create_chain_returns_runnable():
# Create a simple prompt template for testing
prompt = PromptTemplate.from_template("Test prompt: {user_data}")
chain = create_initial_assessment_chain("local", "gemma3:4b", prompt)
assert isinstance(chain, Runnable)
|