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

(feat) support bedrock timeout

Browse source
This commit is contained in:
ishaan-jaff 2024-02-09 14:36:43 -08:00
parent 61031fe380
commit 896fd393db
2 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/bedrock.py
View file

@ -526,6 +526,7 @@ def completion(
optional_params=None,
litellm_params=None,
logger_fn=None,
timeout=None,
):
exception_mapping_worked = False
try:
@ -553,6 +554,7 @@ def completion(
aws_role_name=aws_role_name,
aws_session_name=aws_session_name,
aws_profile_name=aws_profile_name,
timeout=timeout,
)
model = model