Merge remote-tracking branch 'origin/main' into better-whitebox
# Conflicts: # strix/agents/StrixAgent/strix_agent.py # strix/agents/StrixAgent/system_prompt.jinja
This commit is contained in:
@@ -1,15 +1,16 @@
|
||||
import litellm
|
||||
import pytest
|
||||
|
||||
from strix.llm.config import LLMConfig
|
||||
from strix.llm.llm import LLM
|
||||
|
||||
|
||||
def test_llm_does_not_modify_litellm_callbacks(monkeypatch) -> None:
|
||||
def test_llm_does_not_modify_litellm_callbacks(monkeypatch: pytest.MonkeyPatch) -> None:
|
||||
monkeypatch.setenv("STRIX_TELEMETRY", "1")
|
||||
monkeypatch.setenv("STRIX_OTEL_TELEMETRY", "1")
|
||||
monkeypatch.setattr(litellm, "callbacks", ["custom-callback"])
|
||||
|
||||
llm = LLM(LLMConfig(model_name="openai/gpt-5"), agent_name=None)
|
||||
llm = LLM(LLMConfig(model_name="openai/gpt-5.4"), agent_name=None)
|
||||
|
||||
assert llm is not None
|
||||
assert litellm.callbacks == ["custom-callback"]
|
||||
|
||||
Reference in New Issue
Block a user