From 4aa097e645bbf405b45fec5f8bd4f01e61d72608 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Fri, 24 Apr 2026 14:24:39 -0700 Subject: [PATCH] fix: handle older proxy-extras that don't support use_v2_resolver kwarg --- litellm/proxy/db/prisma_client.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/litellm/proxy/db/prisma_client.py b/litellm/proxy/db/prisma_client.py index 73735796eb3..86a1adbaa4e 100644 --- a/litellm/proxy/db/prisma_client.py +++ b/litellm/proxy/db/prisma_client.py @@ -435,10 +435,16 @@ class PrismaManager: prisma_dir = PrismaManager._get_prisma_dir() - return ProxyExtrasDBManager.setup_database( - use_migrate=use_migrate, - use_v2_resolver=use_v2_resolver, - ) + try: + return ProxyExtrasDBManager.setup_database( + use_migrate=use_migrate, + use_v2_resolver=use_v2_resolver, + ) + except TypeError: + # Older proxy-extras doesn't support use_v2_resolver + return ProxyExtrasDBManager.setup_database( + use_migrate=use_migrate, + ) else: # Use prisma db push with increased timeout subprocess.run(