27 lines
829 B
Python
27 lines
829 B
Python
from nanobot.providers.litellm_provider import LiteLLMProvider
|
|
|
|
|
|
def test_litellm_provider_initializes_langsmith_flag(monkeypatch) -> None:
|
|
monkeypatch.setenv("LANGSMITH_API_KEY", "ls-test")
|
|
|
|
provider = LiteLLMProvider(default_model="openai/gpt-4o-mini")
|
|
|
|
assert provider._langsmith_enabled is True
|
|
|
|
|
|
def test_litellm_provider_build_chat_kwargs_adds_langsmith_callback(monkeypatch) -> None:
|
|
monkeypatch.setenv("LANGSMITH_API_KEY", "ls-test")
|
|
|
|
provider = LiteLLMProvider(default_model="openai/gpt-4o-mini")
|
|
kwargs, _ = provider._build_chat_kwargs(
|
|
messages=[{"role": "user", "content": "hello"}],
|
|
tools=None,
|
|
model=None,
|
|
max_tokens=128,
|
|
temperature=0.1,
|
|
reasoning_effort=None,
|
|
tool_choice=None,
|
|
)
|
|
|
|
assert kwargs["callbacks"] == ["langsmith"]
|