From b9fc2c37352eb2b6f9ab7b69414a9afc303c6cb0 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 26 Jan 2024 10:13:34 -0800 Subject: [PATCH] (fix) SpendLogs stop logging model params --- litellm/proxy/_types.py | 1 - litellm/proxy/schema.prisma | 1 - litellm/proxy/utils.py | 2 -- schema.prisma | 1 - 4 files changed, 5 deletions(-) diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index 13c086285..ff49b08c6 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -349,7 +349,6 @@ class LiteLLM_SpendLogs(LiteLLMBase): startTime: Union[str, datetime, None] endTime: Union[str, datetime, None] user: Optional[str] = "" - modelParameters: Optional[Json] = {} usage: Optional[Json] = {} metadata: Optional[Json] = {} cache_hit: Optional[str] = "False" diff --git a/litellm/proxy/schema.prisma b/litellm/proxy/schema.prisma index f06d42ba5..5da42bd38 100644 --- a/litellm/proxy/schema.prisma +++ b/litellm/proxy/schema.prisma @@ -54,7 +54,6 @@ model LiteLLM_SpendLogs { endTime DateTime // Assuming end_time is a DateTime field model String @default("") user String @default("") - modelParameters Json @default("{}")// Assuming optional_params is a JSON field usage Json @default("{}") metadata Json @default("{}") cache_hit String @default("") diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index 713f117ca..d4b8b30c3 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -1059,7 +1059,6 @@ def get_logging_payload(kwargs, response_obj, start_time, end_time): metadata = ( litellm_params.get("metadata", {}) or {} ) # if litellm_params['metadata'] == None - optional_params = kwargs.get("optional_params", {}) call_type = kwargs.get("call_type", "litellm.completion") cache_hit = kwargs.get("cache_hit", False) usage = response_obj["usage"] @@ -1090,7 +1089,6 @@ def get_logging_payload(kwargs, response_obj, start_time, end_time): "endTime": end_time, "model": kwargs.get("model", ""), "user": kwargs.get("user", ""), - "modelParameters": optional_params, "usage": usage, "metadata": metadata, "cache_key": cache_key, diff --git a/schema.prisma b/schema.prisma index 72d14e13b..ffb4932ee 100644 --- a/schema.prisma +++ b/schema.prisma @@ -57,7 +57,6 @@ model LiteLLM_SpendLogs { endTime DateTime // Assuming end_time is a DateTime field model String @default("") user String @default("") - modelParameters Json @default("{}")// Assuming optional_params is a JSON field usage Json @default("{}") metadata Json @default("{}") cache_hit String @default("")