tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-09 02:24:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #21434 from BerriAI/fix/langfuse-otel-sys-modules-leak

Browse Source 
fix: restore sys.modules after stub injection in langfuse otel test
This commit is contained in:
jquinter
2026-02-17 22:32:12 -03:00
committed by GitHub
parent 24fcc9da7c 32922449a3
commit 93bbc27c53
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_litellm/integrations/test_langfuse_otel.py
+6 -2
View File
@@ -157,8 +157,12 @@ class TestLangfuseOtelIntegration:
stub_module.LangFuseLogger = StubLFLogger # type: ignore
# Register stub in sys.modules so import inside method succeeds
sys.modules["litellm.integrations.langfuse.langfuse"] = stub_module # type: ignore
# Register stub in sys.modules so import inside method succeeds.
# Use monkeypatch so the real module is restored after the test runs,
# preventing sys.modules corruption that would break patch() targets in
# later tests (the patch would hit the stub while the real module's
# globals remain unpatch-ed).
monkeypatch.setitem(sys.modules, "litellm.integrations.langfuse.langfuse", stub_module) # type: ignore
kwargs = {"litellm_params": {"metadata": {"foo": "bar"}}}
extracted = LangfuseOtelLogger._extract_langfuse_metadata(kwargs)