From fdcb588511ad100e0abb65d01fd4cad3c6706646 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 24 Jan 2024 13:16:34 -0800
Subject: [PATCH] (fix) stop logging messages, response in SpendLogs

---
 litellm/proxy/schema.prisma | 2 --
 litellm/proxy/utils.py      | 3 ---
 schema.prisma               | 2 --
 3 files changed, 7 deletions(-)

diff --git a/litellm/proxy/schema.prisma b/litellm/proxy/schema.prisma
index 5b9e5fc97..cbc7da399 100644
--- a/litellm/proxy/schema.prisma
+++ b/litellm/proxy/schema.prisma
@@ -53,8 +53,6 @@ model LiteLLM_SpendLogs {
   model               String   @default("")
   user                String   @default("")
   modelParameters     Json     @default("{}")// Assuming optional_params is a JSON field
-  messages            Json     @default("[]")
-  response            Json     @default("{}")
   usage               Json     @default("{}")
   metadata            Json     @default("{}")
   cache_hit           String   @default("")
diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py
index 978355568..c3e6be66f 100644
--- a/litellm/proxy/utils.py
+++ b/litellm/proxy/utils.py
@@ -908,7 +908,6 @@ def get_logging_payload(kwargs, response_obj, start_time, end_time):
     metadata = (
         litellm_params.get("metadata", {}) or {}
     )  # if litellm_params['metadata'] == None
-    messages = kwargs.get("messages")
     optional_params = kwargs.get("optional_params", {})
     call_type = kwargs.get("call_type", "litellm.completion")
     cache_hit = kwargs.get("cache_hit", False)
@@ -934,8 +933,6 @@ def get_logging_payload(kwargs, response_obj, start_time, end_time):
         "model": kwargs.get("model", ""),
         "user": kwargs.get("user", ""),
         "modelParameters": optional_params,
-        "messages": messages,
-        "response": response_obj,
         "usage": usage,
         "metadata": metadata,
     }
diff --git a/schema.prisma b/schema.prisma
index 5b9e5fc97..cbc7da399 100644
--- a/schema.prisma
+++ b/schema.prisma
@@ -53,8 +53,6 @@ model LiteLLM_SpendLogs {
   model               String   @default("")
   user                String   @default("")
   modelParameters     Json     @default("{}")// Assuming optional_params is a JSON field
-  messages            Json     @default("[]")
-  response            Json     @default("{}")
   usage               Json     @default("{}")
   metadata            Json     @default("{}")
   cache_hit           String   @default("")