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 metadata = litellm_params.get("metadata", {}) or {} # if litellm_params['metadata'] == None
messages = kwargs.get("messages") messages = kwargs.get("messages")
optional_params = kwargs.get("optional_params", {}) 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"] usage = response_obj["usage"]
id = response_obj.get("id", str(uuid.uuid4())) id = response_obj.get("id", str(uuid.uuid4()))
# Build the initial payload # Build the initial payload
payload = { payload = {
"id": id, "id": id,
"function_name": function_name, "call_type": call_type,
"startTime": start_time, "startTime": start_time,
"endTime": end_time, "endTime": end_time,
"model": kwargs.get("model", ""), "model": kwargs.get("model", ""),

1
litellm/utils.py
View file

@ -595,6 +595,7 @@ class Logging:
"start_time": self.start_time, "start_time": self.start_time,
"stream": self.stream, "stream": self.stream,
"user": user, "user": user,
"call_type": str(self.call_type),
**self.optional_params, **self.optional_params,
**additional_params **additional_params
} }