phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +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

1
litellm/main.py
View file

@ -13,6 +13,7 @@ from functools import partial
import dotenv, traceback, random, asyncio, time, contextvars
from copy import deepcopy
import httpx
import litellm
from ._logging import verbose_logger
from litellm import ( # type: ignore

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