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

(feat) dynamo db - log call_type

Browse source
This commit is contained in:
ishaan-jaff 2023-12-15 17:27:48 +05:30
parent cec7fcee2c
commit 6c58fe65ff
2 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/integrations/dynamodb.py
View file

@ -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", ""),

1
litellm/utils.py
View file

@ -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
}