phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) init_bedrock_client

Browse source
This commit is contained in:
ishaan-jaff 2024-01-01 22:48:46 +05:30
parent e0dfbdcd68
commit d1e8d13c4f
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/llms/bedrock.py
View file

@ -424,16 +424,16 @@ def completion(
) )
# use passed in BedrockRuntime.Client if provided, otherwise create a new one # use passed in BedrockRuntime.Client if provided, otherwise create a new one
client = optional_params.pop( client = optional_params.pop("aws_bedrock_client", None)
"aws_bedrock_client",
# only pass variables that are not None # only init client, if user did not pass one
init_bedrock_client( if client is None:
client = init_bedrock_client(
aws_access_key_id=aws_access_key_id, aws_access_key_id=aws_access_key_id,
aws_secret_access_key=aws_secret_access_key, aws_secret_access_key=aws_secret_access_key,
aws_region_name=aws_region_name, aws_region_name=aws_region_name,
aws_bedrock_runtime_endpoint=aws_bedrock_runtime_endpoint, aws_bedrock_runtime_endpoint=aws_bedrock_runtime_endpoint,
), )
)
model = model model = model
provider = model.split(".")[0] provider = model.split(".")[0]