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

fix(dynamo_db.py): don't auto-create tables, allow database_type == 'dynamodb'

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-12 11:33:40 +05:30
parent aeff32ac62
commit 352f943dcf
2 changed files with 33 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/proxy_server.py
View file

@ -853,7 +853,9 @@ class ProxyConfig:
)
## dynamodb
database_type = general_settings.get("database_type", None)
if database_type is not None and database_type == "dynamo_db":
if database_type is not None and (
database_type == "dynamo_db" or database_type == "dynamodb"
):
database_args = general_settings.get("database_args", None)
custom_db_client = DBClient(
custom_db_args=database_args, custom_db_type=database_type