diff --git a/litellm/proxy/schema.prisma b/litellm/proxy/schema.prisma index e24a9d5f3..7ffeab9ba 100644 --- a/litellm/proxy/schema.prisma +++ b/litellm/proxy/schema.prisma @@ -174,6 +174,7 @@ model LiteLLM_SpendLogs { completion_tokens Int @default(0) startTime DateTime // Assuming start_time is a DateTime field endTime DateTime // Assuming end_time is a DateTime field + completionStartTime DateTime? // Assuming completionStartTime is a DateTime field model String @default("") api_base String @default("") user String @default("") diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index f12cd4b43..ffff87c51 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -1862,6 +1862,7 @@ def get_logging_payload( metadata = ( litellm_params.get("metadata", {}) or {} ) # if litellm_params['metadata'] == None + completion_start_time = kwargs.get("completion_start_time", end_time) call_type = kwargs.get("call_type") cache_hit = kwargs.get("cache_hit", False) usage = response_obj["usage"] @@ -1910,6 +1911,7 @@ def get_logging_payload( "cache_hit": cache_hit, "startTime": start_time, "endTime": end_time, + "completionStartTime": completion_start_time, "model": kwargs.get("model", ""), "user": kwargs.get("litellm_params", {}) .get("metadata", {}) diff --git a/schema.prisma b/schema.prisma index e24a9d5f3..7ffeab9ba 100644 --- a/schema.prisma +++ b/schema.prisma @@ -174,6 +174,7 @@ model LiteLLM_SpendLogs { completion_tokens Int @default(0) startTime DateTime // Assuming start_time is a DateTime field endTime DateTime // Assuming end_time is a DateTime field + completionStartTime DateTime? // Assuming completionStartTime is a DateTime field model String @default("") api_base String @default("") user String @default("")