phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(langfuse.py'): cleanup

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-12 23:20:43 -07:00
parent 375dbecb86
commit b1cf46faaa
2 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/integrations/langfuse.py
View file

@ -48,7 +48,7 @@ class LangFuseLogger:
"secret_key": self.secret_key, "secret_key": self.secret_key,
"host": self.langfuse_host, "host": self.langfuse_host,
"release": self.langfuse_release, "release": self.langfuse_release,
"debug": True, "debug": self.langfuse_debug,
"flush_interval": flush_interval, # flush interval in seconds "flush_interval": flush_interval, # flush interval in seconds
} }

2
litellm/tests/test_proxy_server.py
View file

@ -1033,9 +1033,7 @@ async def test_add_callback_via_key(prisma_client):
mock_client.assert_called() mock_client.assert_called()
mock_client.return_value.log_event.assert_called() mock_client.return_value.log_event.assert_called()
args, kwargs = mock_client.return_value.log_event.call_args args, kwargs = mock_client.return_value.log_event.call_args
print("KWARGS - {}".format(kwargs))
kwargs = kwargs["kwargs"] kwargs = kwargs["kwargs"]
print(kwargs)
assert "user_api_key_metadata" in kwargs["litellm_params"]["metadata"] assert "user_api_key_metadata" in kwargs["litellm_params"]["metadata"]
assert ( assert (
"logging" "logging"