tests/test_llm_service.py

# 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)