phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(feat) circular ref error on prisa

Browse source
This commit is contained in:
ishaan-jaff 2024-03-06 12:08:22 -08:00
parent 8a75c4c3a3
commit 48f6189760
1 changed files with 5 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

18
litellm/proxy/utils.py
View file

@ -507,24 +507,16 @@ class PrismaClient:
return hashed_token return hashed_token
def handle_circular_references(obj: Any):
try:
# Try to serialize the object using default serialization
return json.dumps(obj)
except Exception as e:
# If a TypeError is raised, return an error message JSON
error_message: dict = {
"error": "Error converting to JSON",
"error_message": str(e),
}
return json.dumps(error_message)
def jsonify_object(self, data: dict) -> dict: def jsonify_object(self, data: dict) -> dict:
db_data = copy.deepcopy(data) db_data = copy.deepcopy(data)
for k, v in db_data.items(): for k, v in db_data.items():
if isinstance(v, dict): if isinstance(v, dict):
db_data[k] = json.dumps(v) try:
db_data[k] = json.dumps(v)
except:
# This avoids Prisma retrying this 5 times, and making 5 clients
db_data[k] = "failed-to-serialize-json"
return db_data return db_data
@backoff.on_exception( @backoff.on_exception(