diff --git a/litellm/integrations/dynamodb.py b/litellm/integrations/dynamodb.py index 143b0ec0fb..c4d409d57b 100644 --- a/litellm/integrations/dynamodb.py +++ b/litellm/integrations/dynamodb.py @@ -62,14 +62,14 @@ class DyanmoDBLogger: metadata = litellm_params.get("metadata", {}) or {} # if litellm_params['metadata'] == None messages = kwargs.get("messages") optional_params = kwargs.get("optional_params", {}) - function_name = kwargs.get("function_name", "litellm.completion") + call_type = kwargs.get("call_type", "litellm.completion") usage = response_obj["usage"] id = response_obj.get("id", str(uuid.uuid4())) # Build the initial payload payload = { "id": id, - "function_name": function_name, + "call_type": call_type, "startTime": start_time, "endTime": end_time, "model": kwargs.get("model", ""), diff --git a/litellm/utils.py b/litellm/utils.py index 1bfd9e28c8..b1e2259118 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -595,6 +595,7 @@ class Logging: "start_time": self.start_time, "stream": self.stream, "user": user, + "call_type": str(self.call_type), **self.optional_params, **additional_params }