wehub-resource-sync
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
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
|