diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 10b40321e..296d8e576 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -1169,7 +1169,6 @@ async def update_key_fn(request: Request, data: UpdateKeyRequest): raise Exception("Not connected to DB!") non_default_values = {k: v for k, v in data_json.items() if v is not None} - print(f"non_default_values: {non_default_values}") response = await prisma_client.update_data(token=key, data={**non_default_values, "token": key}) return {"key": key, **non_default_values} # update based on remaining passed in values diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index 938508bcc..52a2fb6aa 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -1,5 +1,5 @@ from typing import Optional, List, Any, Literal -import os, subprocess, hashlib, importlib, asyncio, copy +import os, subprocess, hashlib, importlib, asyncio, copy, json import litellm, backoff from litellm.proxy._types import UserAPIKeyAuth from litellm.caching import DualCache @@ -147,6 +147,14 @@ class PrismaClient: return hashed_token + def jsonify_object(self, data: dict) -> dict: + db_data = copy.deepcopy(data) + + for k, v in db_data.items(): + if isinstance(v, dict): + db_data[k] = json.dumps(v) + return db_data + @backoff.on_exception( backoff.expo, Exception, # base exception to catch for the backoff @@ -193,7 +201,7 @@ class PrismaClient: try: token = data["token"] hashed_token = self.hash_token(token=token) - db_data = copy.deepcopy(data) + db_data = self.jsonify_object(data=data) db_data["token"] = hashed_token new_verification_token = await self.db.litellm_verificationtoken.upsert( # type: ignore @@ -228,7 +236,7 @@ class PrismaClient: if token.startswith("sk-"): token = self.hash_token(token=token) - db_data = copy.deepcopy(data) + db_data = self.jsonify_object(data=data) db_data["token"] = token response = await self.db.litellm_verificationtoken.update( where={ diff --git a/litellm/tests/langfuse.log b/litellm/tests/langfuse.log index 58cdb5267..e69de29bb 100644 --- a/litellm/tests/langfuse.log +++ b/litellm/tests/langfuse.log @@ -1,4 +0,0 @@ -uploading batch of 2 items -successfully uploaded batch of 2 items -uploading batch of 2 items -successfully uploaded batch of 2 items