From 8b599d4398bbc4fc8f3b855dd274c8c5cd09251f Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 22 Jan 2024 11:33:34 -0800
Subject: [PATCH] v0 max_budget per key

---
 litellm/proxy/schema.prisma | 1 +
 schema.prisma               | 1 +
 2 files changed, 2 insertions(+)

diff --git a/litellm/proxy/schema.prisma b/litellm/proxy/schema.prisma
index 1c73a1405a..f390d7ddc7 100644
--- a/litellm/proxy/schema.prisma
+++ b/litellm/proxy/schema.prisma
@@ -33,6 +33,7 @@ model LiteLLM_VerificationToken {
     metadata   Json  @default("{}")
     tpm_limit     BigInt?
     rpm_limit     BigInt?
+    max_budget Float    @default(0.0)
 }
 
 model LiteLLM_Config {
diff --git a/schema.prisma b/schema.prisma
index 07d4d34224..7fb0eebc41 100644
--- a/schema.prisma
+++ b/schema.prisma
@@ -33,6 +33,7 @@ model LiteLLM_VerificationToken {
     metadata   Json  @default("{}")
     tpm_limit     BigInt?
     rpm_limit     BigInt?
+    max_budget Float    @default(0.0)
 }
 
 model LiteLLM_Config {