From db0489d6037bb4c6fd1f0460a2d0eed76cc3efe1 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sat, 2 Mar 2024 21:11:25 -0800
Subject: [PATCH] fix(proxy/utils.py): add linting fix

---
 litellm/proxy/utils.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py
index c92f75f0e..fc90c3b7b 100644
--- a/litellm/proxy/utils.py
+++ b/litellm/proxy/utils.py
@@ -1142,7 +1142,10 @@ class PrismaClient:
                 )
                 _data: dict = {}
                 if response is not None:
-                    _data = response.dict()
+                    try:
+                        _data = response.model_dump()  # type: ignore
+                    except Exception as e:
+                        _data = response.dict()
                 return {"token": token, "data": _data}
             elif (
                 user_id is not None