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

refactor(main.py): trigger new build

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-19 22:53:38 -08:00
parent e2da1e8ac5
commit b3886b4bda
2 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/utils.py
View file

@ -681,11 +681,11 @@ class PrismaClient:
return response
elif table_name == "user_notification":
if query_type == "find_unique":
response = await self.db.litellm_usernotifications.find_unique(
response = await self.db.litellm_usernotifications.find_unique( # type: ignore
where={"user_id": user_id} # type: ignore
)
elif query_type == "find_all":
response = await self.db.litellm_usernotifications.find_many()
response = await self.db.litellm_usernotifications.find_many() # type: ignore
return response
except Exception as e:
print_verbose(f"LiteLLM Prisma Client Exception: {e}")
@ -795,7 +795,7 @@ class PrismaClient:
elif table_name == "user_notification":
db_data = self.jsonify_object(data=data)
new_user_notification_row = (
await self.db.litellm_usernotifications.upsert(
await self.db.litellm_usernotifications.upsert( # type: ignore
where={"request_id": data["request_id"]},
data={
"create": {**db_data}, # type: ignore