scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-12 03:48:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 118

fix(base_aws_llm.py): fix credential caching check to see if token is set

Browse source
This commit is contained in:
Krrish Dholakia 2024-09-14 17:45:18 -07:00
parent 3f08382591
commit 13dfe1f203
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/llms/base_aws_llm.py
View file

@ -243,7 +243,10 @@ class BaseAWSLLM(BaseLLM):
credentials = session.get_credentials()
self.iam_cache.set_cache(cache_key, credentials, ttl=3600 - 60)
if (
credentials.token is None
): # don't cache if session token exists. The expiry time for that is not known.
self.iam_cache.set_cache(cache_key, credentials, ttl=3600 - 60)
return credentials