From 8cce2f28caf5a9a31d86a9b40bf51f60df67e6bf Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 15 Jan 2024 12:07:20 -0800 Subject: [PATCH] Revert "(fix) proxy only import dynamo_db when it's being used" This reverts commit 8d3ca785ae0f862a9695250c87c274d261da5a7b. --- litellm/proxy/utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index c55453423bf..4224437f55e 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -7,6 +7,7 @@ from litellm.proxy.hooks.parallel_request_limiter import MaxParallelRequestsHand from litellm.proxy.hooks.max_budget_limiter import MaxBudgetLimiter from litellm.integrations.custom_logger import CustomLogger from litellm.proxy.db.base_client import CustomDB +from litellm.proxy.db.dynamo_db import DynamoDBWrapper from fastapi import HTTPException, status import smtplib from email.mime.text import MIMEText @@ -594,8 +595,6 @@ class DBClient: self, custom_db_type: Literal["dynamo_db"], custom_db_args: dict ) -> None: if custom_db_type == "dynamo_db": - from litellm.proxy.db.dynamo_db import DynamoDBWrapper - self.db = DynamoDBWrapper(database_arguments=DynamoDBArgs(**custom_db_args)) async def get_data(self, key: str, table_name: Literal["user", "key", "config"]):