scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-08-28 05:25:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Merge pull request #21253 from BerriAI/refactor/langfuse-test-cleanup-dead-code

Browse source 
refactor: remove dead code from Langfuse test cleanup
This commit is contained in:
jquinter 2026-02-15 13:49:14 -03:00 • committed by GitHub
commit 4f751bf815
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -77,19 +77,12 @@ class TestLangfuseUsageDetails(unittest.TestCase):
sys.modules["langfuse"].Langfuse = self.mock_langfuse_class
# Create a fresh logger instance for each test
# Force a clean state by clearing any class-level cached state
if hasattr(LangFuseLogger, '_langfuse_clients'):
LangFuseLogger._langfuse_clients = {}
self.logger = LangFuseLogger()
# Explicitly set the Langfuse client to our mock
self.logger.Langfuse = self.mock_langfuse_client
# Ensure langfuse_sdk_version is set correctly for _supports_* methods
self.logger.langfuse_sdk_version = "3.0.0"
# Reset any cached client instances
if hasattr(self.logger, '_langfuse_client_cache'):
self.logger._langfuse_client_cache = None
# Add the log_event_on_langfuse method to the instance
def log_event_on_langfuse(
@ -132,11 +125,9 @@ class TestLangfuseUsageDetails(unittest.TestCase):
def tearDown(self):
# Clean up logger instance to prevent state leakage
if hasattr(self, 'logger'):
# Reset logger's Langfuse client
# Reset logger's Langfuse client to break any references
self.logger.Langfuse = None
# Clear any cached state
if hasattr(self.logger, '_langfuse_client_cache'):
self.logger._langfuse_client_cache = None
# Delete logger instance to ensure complete cleanup
del self.logger
self.env_patcher.stop()