phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +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, optional_params=None,
litellm_params=None, litellm_params=None,
logger_fn=None, logger_fn=None,
timeout=None,
): ):
exception_mapping_worked = False exception_mapping_worked = False
try: try:
@ -553,6 +554,7 @@ def completion(
aws_role_name=aws_role_name, aws_role_name=aws_role_name,
aws_session_name=aws_session_name, aws_session_name=aws_session_name,
aws_profile_name=aws_profile_name, aws_profile_name=aws_profile_name,
timeout=timeout,
) )
model = model model = model

1
litellm/main.py
View file

@ -1590,6 +1590,7 @@ def completion(
logger_fn=logger_fn, logger_fn=logger_fn,
encoding=encoding, encoding=encoding,
logging_obj=logging, logging_obj=logging,
timeout=timeout,
) )
if "stream" in optional_params and optional_params["stream"] == True: if "stream" in optional_params and optional_params["stream"] == True: