phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) SpendLogs stop logging model params

Browse source
This commit is contained in:
ishaan-jaff 2024-01-26 10:13:34 -08:00
parent 4634f7b1c1
commit b9fc2c3735
4 changed files with 0 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/proxy/_types.py
View file

@ -349,7 +349,6 @@ class LiteLLM_SpendLogs(LiteLLMBase):
startTime: Union[str, datetime, None] startTime: Union[str, datetime, None]
endTime: Union[str, datetime, None] endTime: Union[str, datetime, None]
user: Optional[str] = "" user: Optional[str] = ""
modelParameters: Optional[Json] = {}
usage: Optional[Json] = {} usage: Optional[Json] = {}
metadata: Optional[Json] = {} metadata: Optional[Json] = {}
cache_hit: Optional[str] = "False" cache_hit: Optional[str] = "False"

1
litellm/proxy/schema.prisma
View file

@ -54,7 +54,6 @@ model LiteLLM_SpendLogs {
endTime DateTime // Assuming end_time is a DateTime field endTime DateTime // Assuming end_time is a DateTime field
model String @default("") model String @default("")
user String @default("") user String @default("")
modelParameters Json @default("{}")// Assuming optional_params is a JSON field
usage Json @default("{}") usage Json @default("{}")
metadata Json @default("{}") metadata Json @default("{}")
cache_hit String @default("") cache_hit String @default("")

2
litellm/proxy/utils.py
View file

@ -1059,7 +1059,6 @@ def get_logging_payload(kwargs, response_obj, start_time, end_time):
metadata = ( metadata = (
litellm_params.get("metadata", {}) or {} litellm_params.get("metadata", {}) or {}
) # if litellm_params['metadata'] == None ) # if litellm_params['metadata'] == None
optional_params = kwargs.get("optional_params", {})
call_type = kwargs.get("call_type", "litellm.completion") call_type = kwargs.get("call_type", "litellm.completion")
cache_hit = kwargs.get("cache_hit", False) cache_hit = kwargs.get("cache_hit", False)
usage = response_obj["usage"] usage = response_obj["usage"]
@ -1090,7 +1089,6 @@ def get_logging_payload(kwargs, response_obj, start_time, end_time):
"endTime": end_time, "endTime": end_time,
"model": kwargs.get("model", ""), "model": kwargs.get("model", ""),
"user": kwargs.get("user", ""), "user": kwargs.get("user", ""),
"modelParameters": optional_params,
"usage": usage, "usage": usage,
"metadata": metadata, "metadata": metadata,
"cache_key": cache_key, "cache_key": cache_key,

1
schema.prisma
View file

@ -57,7 +57,6 @@ model LiteLLM_SpendLogs {
endTime DateTime // Assuming end_time is a DateTime field endTime DateTime // Assuming end_time is a DateTime field
model String @default("") model String @default("")
user String @default("") user String @default("")
modelParameters Json @default("{}")// Assuming optional_params is a JSON field
usage Json @default("{}") usage Json @default("{}")
metadata Json @default("{}") metadata Json @default("{}")
cache_hit String @default("") cache_hit String @default("")