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

Added tags to additional keys that can be sent to athina

Browse source
This commit is contained in:
Vivek Aditya 2025-02-26 21:00:56 +05:30
parent 07c0d26e68
commit c40d45ae09
2 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/integrations/athina.py
View file

@ -23,6 +23,7 @@ class AthinaLogger:
"context",
"expected_response",
"user_query",
"tags",
"custom_attributes",
]
@ -78,10 +79,12 @@ 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,
headers=self.headers,