scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

fix(bedrock_httpx.py): Fix STS region endpoint for converse flow.

Browse source
This commit is contained in:
David Manouchehri 2024-06-11 16:34:32 +00:00
parent 7faf0b9586
commit 9aee0b4a0f
No known key found for this signature in database
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/llms/bedrock_httpx.py
View file

@ -1446,7 +1446,11 @@ class BedrockConverseLLM(BaseLLM):
status_code=401,
)
sts_client = boto3.client("sts")
sts_client = boto3.client(
"sts",
region_name=aws_region_name,
endpoint_url=f"https://sts.{aws_region_name}.amazonaws.com"
)
# https://docs.aws.amazon.com/STS/latest/APIReference/API_AssumeRoleWithWebIdentity.html
# https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sts/client/assume_role_with_web_identity.html