From de7663ead3e31bf550ec8e1a8828f73d6591e788 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 22 Jan 2024 12:20:38 -0800
Subject: [PATCH] (fix) make max_budget optional for LiteLLM_VerificationToken

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

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