From 250672eddce348f0477eaa4a000281572166d7ef Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 6 Jan 2024 12:23:39 +0530 Subject: [PATCH] (ci/cd) temp fix - check if model_dump_json exists --- litellm/proxy/utils.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index 061a6c07ccc..15d45d911db 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -623,11 +623,14 @@ async def _cache_user_row(user_id: str, cache: DualCache, db: PrismaClient): if response is None: # Cache miss user_row = await db.get_data(user_id=user_id) if user_row is not None: - print_verbose(f"User Row: {user_row}") - cache_value = user_row.model_dump_json() - cache.set_cache( - key=cache_key, value=cache_value, ttl=600 - ) # store for 10 minutes + print_verbose(f"User Row: {user_row}, type = {type(user_row)}") + if hasattr(user_row, "model_dump_json") and callable( + getattr(user_row, "model_dump_json") + ): + cache_value = user_row.model_dump_json() + cache.set_cache( + key=cache_key, value=cache_value, ttl=600 + ) # store for 10 minutes return