scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-17 23:51:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 139

fix(redis): validate signed ElastiCache IAM URL

Browse source
This commit is contained in:
eugene-yao-zocdoc 2026-08-31 14:07:04 -04:00
parent f1cd2b03ca
commit aca81c263c
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
litellm/_redis_credential_provider.py
View file

@ -4,7 +4,7 @@ import asyncio
import threading
import time
from collections.abc import Callable
from typing import TYPE_CHECKING, Any, Final, Protocol, cast
from typing import TYPE_CHECKING, Any, Final, Protocol
from urllib.parse import quote
from redis.credentials import CredentialProvider
@ -185,7 +185,10 @@ class ElastiCacheIAMCredentialProvider(CredentialProvider):
self._region,
expires=self._token_lifetime_seconds,
).add_auth(request)
return self._user_name, cast(str, request.url).removeprefix("https://")
signed_url: Final = request.url
if signed_url is None:
raise RuntimeError("Unable to generate AWS ElastiCache IAM credentials")
return self._user_name, signed_url.removeprefix("https://")
def get_credentials(self) -> tuple[str, str]:
return self._get_credentials()