diff --git a/litellm/proxy/db/dynamo_db.py b/litellm/proxy/db/dynamo_db.py index fdc9ccfba1..a5c4b8371f 100644 --- a/litellm/proxy/db/dynamo_db.py +++ b/litellm/proxy/db/dynamo_db.py @@ -10,7 +10,6 @@ from typing import Any, List, Literal, Optional, Union import json from datetime import datetime from litellm._logging import verbose_proxy_logger -import aiohttp class DynamoDBWrapper(CustomDB): @@ -72,10 +71,9 @@ class DynamoDBWrapper(CustomDB): from aiodynamo.models import ReturnValues from aiodynamo.http.aiohttp import AIOHTTP from aiohttp import ClientSession - import aiohttp verbose_proxy_logger.debug("DynamoDB Wrapper - Attempting to connect") - async with ClientSession(connector=aiohttp.TCPConnector(ssl=False)) as session: + async with ClientSession() as session: client = Client(AIOHTTP(session), Credentials.auto(), self.region_name) ## User try: @@ -148,7 +146,7 @@ class DynamoDBWrapper(CustomDB): from aiodynamo.http.aiohttp import AIOHTTP from aiohttp import ClientSession - async with ClientSession(connector=aiohttp.TCPConnector(ssl=False)) as session: + async with ClientSession() as session: client = Client(AIOHTTP(session), Credentials.auto(), self.region_name) table = None if table_name == "user": @@ -181,7 +179,7 @@ class DynamoDBWrapper(CustomDB): from aiodynamo.http.aiohttp import AIOHTTP from aiohttp import ClientSession - async with ClientSession(connector=aiohttp.TCPConnector(ssl=False)) as session: + async with ClientSession() as session: client = Client(AIOHTTP(session), Credentials.auto(), self.region_name) table = None key_name = None @@ -235,7 +233,7 @@ class DynamoDBWrapper(CustomDB): from aiodynamo.http.aiohttp import AIOHTTP from aiohttp import ClientSession - async with ClientSession(connector=aiohttp.TCPConnector(ssl=False)) as session: + async with ClientSession() as session: client = Client(AIOHTTP(session), Credentials.auto(), self.region_name) table = None key_name = None