From 4821fa9201af5ee3da1cf337e3ba34c642ae6eee Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 18 Jan 2024 10:04:34 -0800
Subject: [PATCH] (v0) add schema.prisma

---
 litellm/proxy/_types.py     | 16 ++++++++++++++++
 litellm/proxy/schema.prisma | 15 +++++++++++++++
 schema.prisma               | 15 +++++++++++++++
 3 files changed, 46 insertions(+)

diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py
index 6e8588314..220bf2f91 100644
--- a/litellm/proxy/_types.py
+++ b/litellm/proxy/_types.py
@@ -194,6 +194,7 @@ class DynamoDBArgs(LiteLLMBase):
     user_table_name: str = "LiteLLM_UserTable"
     key_table_name: str = "LiteLLM_VerificationToken"
     config_table_name: str = "LiteLLM_Config"
+    spend_table_name: str = "LiteLLM_SpendLogs"
 
 
 class ConfigGeneralSettings(LiteLLMBase):
@@ -312,3 +313,18 @@ class LiteLLM_UserTable(LiteLLMBase):
         if values.get("models") is None:
             values.update({"models", []})
         return values
+
+
+class LiteLLM_SpendLogs(LiteLLMBase):
+    id: str
+    call_type: str
+    startTime: Union[str, None]
+    endTime: Union[str, None]
+    model: str = ""
+    user: str = ""
+    modelParameters: Dict = {}
+    messages: List[str] = []
+    call_cost: float = 0.0
+    response: Dict = {}
+    usage: Dict = {}
+    metadata: Dict = {}
diff --git a/litellm/proxy/schema.prisma b/litellm/proxy/schema.prisma
index aa45a8818..d2e338bd4 100644
--- a/litellm/proxy/schema.prisma
+++ b/litellm/proxy/schema.prisma
@@ -31,4 +31,19 @@ model LiteLLM_VerificationToken {
 model LiteLLM_Config {
   param_name String @id
   param_value Json?
+}
+
+model LiteLLM_SpendLogs {
+  request_id          String @unique
+  call_type           String
+  startTime           DateTime // Assuming start_time is a DateTime field
+  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
+  messages            Json     @default("[]")
+  spend               Float    @default(0.0)
+  response            Json     @default("{}")
+  usage               Json     @default("{}")
+  metadata            Json     @default("{}")
 }
\ No newline at end of file
diff --git a/schema.prisma b/schema.prisma
index 704ada42c..df2c1d0b4 100644
--- a/schema.prisma
+++ b/schema.prisma
@@ -31,4 +31,19 @@ model LiteLLM_VerificationToken {
 model LiteLLM_Config {
   param_name String @id
   param_value Json?
+}
+
+model LiteLLM_SpendLogs {
+  id                  String @unique
+  call_type           String
+  startTime           DateTime // Assuming start_time is a DateTime field
+  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
+  messages            Json     @default("[]")
+  call_cost           Float    @default(0.0)
+  response            Json     @default("{}")
+  usage               Json     @default("{}")
+  metadata            Json     @default("{}")
 }
\ No newline at end of file