mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-09 00:26:01 +00:00
(feat) support bedrock timeout
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -1590,6 +1590,7 @@ def completion(
|
||||
logger_fn=logger_fn,
|
||||
encoding=encoding,
|
||||
logging_obj=logging,
|
||||
timeout=timeout,
|
||||
)
|
||||
|
||||
if "stream" in optional_params and optional_params["stream"] == True:
|
||||
|
||||
Reference in New Issue
Block a user