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

(fix) proxy only import dynamo_db when it's being used

Browse source
This commit is contained in:
ishaan-jaff 2024-01-13 11:43:50 -08:00
parent a3d84ad2e3
commit 1b696c4129
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/utils.py
View file

@ -7,7 +7,6 @@ from litellm.proxy.hooks.parallel_request_limiter import MaxParallelRequestsHand
from litellm.proxy.hooks.max_budget_limiter import MaxBudgetLimiter from litellm.proxy.hooks.max_budget_limiter import MaxBudgetLimiter
from litellm.integrations.custom_logger import CustomLogger from litellm.integrations.custom_logger import CustomLogger
from litellm.proxy.db.base_client import CustomDB from litellm.proxy.db.base_client import CustomDB
from litellm.proxy.db.dynamo_db import DynamoDBWrapper
from fastapi import HTTPException, status from fastapi import HTTPException, status
import smtplib import smtplib
from email.mime.text import MIMEText from email.mime.text import MIMEText
@ -595,6 +594,8 @@ class DBClient:
self, custom_db_type: Literal["dynamo_db"], custom_db_args: dict self, custom_db_type: Literal["dynamo_db"], custom_db_args: dict
) -> None: ) -> None:
if custom_db_type == "dynamo_db": if custom_db_type == "dynamo_db":
from litellm.proxy.db.dynamo_db import DynamoDBWrapper
self.db = DynamoDBWrapper(database_arguments=DynamoDBArgs(**custom_db_args)) self.db = DynamoDBWrapper(database_arguments=DynamoDBArgs(**custom_db_args))
async def get_data(self, key: str, table_name: Literal["user", "key", "config"]): async def get_data(self, key: str, table_name: Literal["user", "key", "config"]):