From 8f1f7eb1579885c3d5c7d774c9dd0b768b70c006 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Fri, 24 Apr 2026 18:19:23 -0700 Subject: [PATCH] fix: use inspect.signature instead of broad TypeError to detect use_v2_resolver support --- litellm/proxy/db/prisma_client.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/db/prisma_client.py b/litellm/proxy/db/prisma_client.py index 86a1adbaa4e..e3b56d62dd1 100644 --- a/litellm/proxy/db/prisma_client.py +++ b/litellm/proxy/db/prisma_client.py @@ -435,12 +435,15 @@ class PrismaManager: prisma_dir = PrismaManager._get_prisma_dir() - try: + import inspect + + _sig = inspect.signature(ProxyExtrasDBManager.setup_database) + if "use_v2_resolver" in _sig.parameters: return ProxyExtrasDBManager.setup_database( use_migrate=use_migrate, use_v2_resolver=use_v2_resolver, ) - except TypeError: + else: # Older proxy-extras doesn't support use_v2_resolver return ProxyExtrasDBManager.setup_database( use_migrate=use_migrate,