From d1e8d13c4f141a4e4797ceedce0f997fbfd2d65b Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 1 Jan 2024 22:48:46 +0530 Subject: [PATCH] (fix) init_bedrock_client --- litellm/llms/bedrock.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/litellm/llms/bedrock.py b/litellm/llms/bedrock.py index 56f6b9e7d..99550280a 100644 --- a/litellm/llms/bedrock.py +++ b/litellm/llms/bedrock.py @@ -424,16 +424,16 @@ def completion( ) # use passed in BedrockRuntime.Client if provided, otherwise create a new one - client = optional_params.pop( - "aws_bedrock_client", - # only pass variables that are not None - init_bedrock_client( + client = optional_params.pop("aws_bedrock_client", None) + + # only init client, if user did not pass one + if client is None: + client = init_bedrock_client( aws_access_key_id=aws_access_key_id, aws_secret_access_key=aws_secret_access_key, aws_region_name=aws_region_name, aws_bedrock_runtime_endpoint=aws_bedrock_runtime_endpoint, - ), - ) + ) model = model provider = model.split(".")[0]