fix(langfuse.py'): cleanup

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

View file

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