mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-07 02:59:05 +00:00
fix: use inspect.signature instead of broad TypeError to detect use_v2_resolver support
This commit is contained in:
parent
fe7ca53950
commit
8f1f7eb157
1 changed files with 5 additions and 2 deletions
|
|
@ -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,
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue