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

Merge pull request #5058 from BerriAI/litellm_add_debug_statements_connecting_prisma

Browse source 
Fix - add debug statements when connecting to prisma DB
This commit is contained in:
Ishaan Jaff 2024-08-05 20:53:55 -07:00 • committed by GitHub
commit 4c4cc1ba23
3 changed files with 75 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/proxy/utils.py
View file

@ -815,7 +815,7 @@ class PrismaClient:
spend_log_transactions: List = []
def __init__(self, database_url: str, proxy_logging_obj: ProxyLogging):
print_verbose(
verbose_proxy_logger.debug(
"LiteLLM: DATABASE_URL Set in config, trying to 'pip install prisma'"
)
## init logging object
@ -844,8 +844,9 @@ class PrismaClient:
os.chdir(original_dir)
# Now you can import the Prisma Client
from prisma import Prisma # type: ignore
verbose_proxy_logger.debug("Connecting Prisma Client to DB..")
self.db = Prisma() # Client to connect to Prisma db
verbose_proxy_logger.debug("Success - Connected Prisma Client to DB")
def hash_token(self, token: str):
# Hash the string using SHA-256