scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

Merge pull request #2390 from BerriAI/litellm_better_debug_prisma_logs

Browse source 
[FIX] Proxy better debug prisma logs
This commit is contained in:
Ishaan Jaff 2024-03-07 16:19:26 -08:00 • committed by GitHub
commit 25a24da037
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
litellm/proxy/utils.py
View file

@ -693,6 +693,9 @@ class PrismaClient:
"""
Generic implementation of get data
"""
verbose_proxy_logger.debug(
f"PrismaClient: get_generic_data: {key}, table_name: {table_name}"
)
try:
if table_name == "users":
response = await self.db.litellm_usertable.find_first(
@ -758,6 +761,9 @@ class PrismaClient:
int
] = None, # pagination, number of rows to getch when find_all==True
):
verbose_proxy_logger.debug(
f"PrismaClient: get_data: token={token}, table_name: {table_name}, query_type: {query_type}, user_id: {user_id}, user_id_list: {user_id_list}, team_id: {team_id}, team_id_list: {team_id_list}, key_val: {key_val}"
)
try:
response: Any = None
if (token is not None and table_name is None) or (
@ -991,9 +997,11 @@ class PrismaClient:
except Exception as e:
import traceback
error_msg = f"LiteLLM Prisma Client Exception get_data: {str(e)}"
prisma_query_info = f"LiteLLM Prisma Client Exception: get_data: token={token}, table_name: {table_name}, query_type: {query_type}, user_id: {user_id}, user_id_list: {user_id_list}, team_id: {team_id}, team_id_list: {team_id_list}, key_val: {key_val}"
error_msg = prisma_query_info + str(e)
print_verbose(error_msg)
error_traceback = error_msg + "\n" + traceback.format_exc()
verbose_proxy_logger.debug(error_traceback)
asyncio.create_task(
self.proxy_logging_obj.failure_handler(
original_exception=e, traceback_str=error_traceback
@ -1020,6 +1028,7 @@ class PrismaClient:
Add a key to the database. If it already exists, do nothing.
"""
try:
verbose_proxy_logger.debug(f"PrismaClient: insert_data: {data}")
if table_name == "key":
token = data["token"]
hashed_token = self.hash_token(token=token)
@ -1152,6 +1161,9 @@ class PrismaClient:
"""
Update existing data
"""
verbose_proxy_logger.debug(
f"PrismaClient: update_data, table_name: {table_name}"
)
try:
db_data = self.jsonify_object(data=data)
if update_key_values is not None: