chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,29 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import importlib
|
||||
|
||||
import pytest
|
||||
|
||||
pytest.importorskip("litellm")
|
||||
|
||||
|
||||
def test_litellm_logging_patch_env_var_controls_application(monkeypatch):
|
||||
"""Assert the serializer patch only applies when the env var is enabled."""
|
||||
litellm_logging = importlib.import_module("litellm.litellm_core_utils.litellm_logging")
|
||||
litellm_model = importlib.import_module("agents.extensions.models.litellm_model")
|
||||
|
||||
monkeypatch.delenv("OPENAI_AGENTS_ENABLE_LITELLM_SERIALIZER_PATCH", raising=False)
|
||||
litellm_logging = importlib.reload(litellm_logging)
|
||||
importlib.reload(litellm_model)
|
||||
|
||||
assert hasattr(
|
||||
litellm_logging,
|
||||
"_extract_response_obj_and_hidden_params",
|
||||
), "LiteLLM removed _extract_response_obj_and_hidden_params; revisit warning patch."
|
||||
assert getattr(litellm_logging, "_openai_agents_patched_serializer_warnings", False) is False
|
||||
|
||||
monkeypatch.setenv("OPENAI_AGENTS_ENABLE_LITELLM_SERIALIZER_PATCH", "true")
|
||||
litellm_logging = importlib.reload(litellm_logging)
|
||||
importlib.reload(litellm_model)
|
||||
|
||||
assert getattr(litellm_logging, "_openai_agents_patched_serializer_warnings", False) is True
|
||||
Reference in New Issue
Block a user