From b3886b4bda7be716037c642ff93ee9aca697beb0 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 19 Feb 2024 22:53:38 -0800 Subject: [PATCH] refactor(main.py): trigger new build --- litellm/main.py | 1 + litellm/proxy/utils.py | 6 +++--- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/litellm/main.py b/litellm/main.py index ec69f5f3a5..1c6c7c2f3d 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -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 diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index bd0c0eaf77..78c1e4b639 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -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