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

Removed prints and added unit tests

Browse source
This commit is contained in:
Vivek Aditya 2025-02-28 21:48:13 +05:30
parent c40d45ae09
commit ed75dd61c2
3 changed files with 211 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/integrations/athina.py
View file

@ -24,6 +24,8 @@ class AthinaLogger:
"expected_response",
"user_query",
"tags",
"user_feedback",
"model_options",
"custom_attributes",
]
@ -79,11 +81,8 @@ class AthinaLogger:
# Add additional metadata keys
metadata = kwargs.get("litellm_params", {}).get("metadata", {})
if metadata:
print("additional_keys", self.additional_keys)
for key in self.additional_keys:
print("key", key)
if key in metadata:
print("key is being added", key)
data[key] = metadata[key]
response = litellm.module_level_client.post(
self.athina_logging_url,