diff --git a/litellm/proxy/schema.prisma b/litellm/proxy/schema.prisma
index e5a6e67bf..5ce0670b1 100644
--- a/litellm/proxy/schema.prisma
+++ b/litellm/proxy/schema.prisma
@@ -3,6 +3,10 @@ datasource client {
   url      = env("DATABASE_URL")
 }
 
+generator client {
+  provider = "prisma-client-py"
+}
+
 // Budget / Rate Limits for an org
 model LiteLLM_BudgetTable {
   budget_id String @id @default(uuid())
diff --git a/schema.prisma b/schema.prisma
index bc3c9e33c..6e2400a12 100644
--- a/schema.prisma
+++ b/schema.prisma
@@ -3,6 +3,10 @@ datasource client {
   url      = env("DATABASE_URL")
 }
 
+generator client {
+  provider = "prisma-client-py"
+}
+
 // Budget / Rate Limits for an org
 model LiteLLM_BudgetTable {
   budget_id String @id @default(uuid())