phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix prisma issues

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-13 19:17:01 -07:00
parent 2de276cb44
commit 353b470cbc
1 changed files with 0 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

28
litellm/proxy/utils.py
View file

@ -14,7 +14,6 @@ from datetime import datetime, timedelta
from email.mime.multipart import MIMEMultipart from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText from email.mime.text import MIMEText
from functools import wraps from functools import wraps
from pathlib import Path
from typing import TYPE_CHECKING, Any, List, Literal, Optional, Tuple, Union from typing import TYPE_CHECKING, Any, List, Literal, Optional, Tuple, Union
import backoff import backoff
@ -816,17 +815,6 @@ class PrismaClient:
org_list_transactons: dict = {} org_list_transactons: dict = {}
spend_log_transactions: List = [] spend_log_transactions: List = []
def ensure_prisma_has_writable_dirs(self, path: str | Path) -> None:
import stat
for root, dirs, _ in os.walk(path):
for directory in dirs:
dir_path = os.path.join(root, directory)
os.makedirs(dir_path, exist_ok=True)
os.chmod(
dir_path, os.stat(dir_path).st_mode | stat.S_IWRITE | stat.S_IEXEC
)
def __init__(self, database_url: str, proxy_logging_obj: ProxyLogging): def __init__(self, database_url: str, proxy_logging_obj: ProxyLogging):
verbose_proxy_logger.debug( verbose_proxy_logger.debug(
"LiteLLM: DATABASE_URL Set in config, trying to 'pip install prisma'" "LiteLLM: DATABASE_URL Set in config, trying to 'pip install prisma'"
@ -858,22 +846,6 @@ class PrismaClient:
# Now you can import the Prisma Client # Now you can import the Prisma Client
from prisma import Prisma # type: ignore from prisma import Prisma # type: ignore
verbose_proxy_logger.debug("Connecting Prisma Client to DB..") verbose_proxy_logger.debug("Connecting Prisma Client to DB..")
import importlib.util
# Get the location of the 'prisma' package
package_name = "prisma"
spec = importlib.util.find_spec(package_name)
print("spec = ", spec) # noqa
if spec and spec.origin:
print("spec origin= ", spec.origin) # noqa
_base_prisma_package_dir = os.path.dirname(spec.origin)
print("base prisma package dir = ", _base_prisma_package_dir) # noqa
else:
raise ImportError(f"Package {package_name} not found.")
# Use the package directory in your method call
self.ensure_prisma_has_writable_dirs(path=_base_prisma_package_dir)
self.db = Prisma() # Client to connect to Prisma db self.db = Prisma() # Client to connect to Prisma db
verbose_proxy_logger.debug("Success - Connected Prisma Client to DB") verbose_proxy_logger.debug("Success - Connected Prisma Client to DB")