scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

Merge pull request #21214 from BerriAI/fix/langfuse-test-isolation

Browse source 
Fix: Langfuse test isolation to prevent flaky failures
This commit is contained in:
jquinter 2026-02-15 12:42:33 -03:00 • committed by GitHub
commit bf9cf7ad45
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_litellm/integrations/test_langfuse.py
View file

@ -39,7 +39,7 @@ class TestLangfuseUsageDetails(unittest.TestCase):
self.mock_langfuse_trace = MagicMock()
self.mock_langfuse_generation = MagicMock()
self.mock_langfuse_generation.trace_id = "test-trace-id"
# Mock span method for trace (used by log_provider_specific_information_as_span and _log_guardrail_information_as_span)
self.mock_langfuse_span = MagicMock()
self.mock_langfuse_span.end = MagicMock()
@ -109,7 +109,6 @@ class TestLangfuseUsageDetails(unittest.TestCase):
response_obj=response_obj,
level=level,
litellm_call_id=kwargs.get("litellm_call_id", None),
print_verbose=True, # Add the missing parameter
)
# Bind the method to the instance
@ -125,7 +124,7 @@ class TestLangfuseUsageDetails(unittest.TestCase):
def tearDown(self):
self.env_patcher.stop()
self.langfuse_module_patcher.stop()
self.langfuse_module_patcher.stop() # patch.dict automatically restores sys.modules
def test_langfuse_usage_details_type(self):
"""Test that LangfuseUsageDetails TypedDict is properly defined with the correct fields"""