(v0) add schema.prisma

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 = {}

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("{}")
 }

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("{}")
 }