From fd0769f2ed904f70b63229aa2be2c79811f44ff3 Mon Sep 17 00:00:00 2001 From: vivek-athina <153479827+vivek-athina@users.noreply.github.com> Date: Fri, 14 Feb 2025 02:49:24 +0530 Subject: [PATCH] Added custom_attributes to additional_keys which can be sent to athina (#8518) --- docs/my-website/docs/observability/athina_integration.md | 2 +- litellm/integrations/athina.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/docs/my-website/docs/observability/athina_integration.md b/docs/my-website/docs/observability/athina_integration.md index f7c99a4a9c..4994d553c6 100644 --- a/docs/my-website/docs/observability/athina_integration.md +++ b/docs/my-website/docs/observability/athina_integration.md @@ -78,7 +78,7 @@ Following are the allowed fields in metadata, their types, and their description * `context: Optional[Union[dict, str]]` - This is the context used as information for the prompt. For RAG applications, this is the "retrieved" data. You may log context as a string or as an object (dictionary). * `expected_response: Optional[str]` - This is the reference response to compare against for evaluation purposes. This is useful for segmenting inference calls by expected response. * `user_query: Optional[str]` - This is the user's query. For conversational applications, this is the user's last message. - +* `custom_attributes: Optional[dict]` - This is a dictionary of custom attributes. This is useful for additional information about the inference. ## Using a self hosted deployment of Athina diff --git a/litellm/integrations/athina.py b/litellm/integrations/athina.py index 250b384c75..754e980c2a 100644 --- a/litellm/integrations/athina.py +++ b/litellm/integrations/athina.py @@ -23,6 +23,7 @@ class AthinaLogger: "context", "expected_response", "user_query", + "custom_attributes", ] def log_event(self, kwargs, response_obj, start_time, end_time, print_verbose):