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

fix(bedrock_httpx.py): Fix OIDC support.

Browse source
This commit is contained in:
David Manouchehri 2024-05-16 18:57:13 +00:00
parent c301d0c5ca
commit 0ed01f430d
No known key found for this signature in database
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/llms/bedrock_httpx.py
View file

@ -275,7 +275,14 @@ class BedrockLLM(BaseLLM):
DurationSeconds=3600,
)
return sts_response["Credentials"]
session = boto3.Session(
aws_access_key_id=sts_response["Credentials"]["AccessKeyId"],
aws_secret_access_key=sts_response["Credentials"]["SecretAccessKey"],
aws_session_token=sts_response["Credentials"]["SessionToken"],
region_name=aws_region_name,
)
return session.get_credentials()
elif aws_role_name is not None and aws_session_name is not None:
sts_client = boto3.client(
"sts",