scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-20 00:11:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 76

fix(auth_v2): type DbIdentityStore cache as UserApiKeyCache for auth_checks calls

Browse source
This commit is contained in:
Claude 2026-06-13 21:56:56 +00:00
parent 86ac48701c
commit 1f6506ffc4
No known key found for this signature in database
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/auth_v2/resolvers.py
View file

@ -37,10 +37,10 @@ from litellm.repositories.team_repository import TeamRepository
from litellm.repositories.user_repository import UserRepository
if TYPE_CHECKING:
from litellm.caching.caching import DualCache
from litellm.models.team import LiteLLM_TeamTable
from litellm.models.user import LiteLLM_UserTable
from litellm.proxy.auth_v2.authorization import Role
from litellm.proxy.common_utils.user_api_key_cache import UserApiKeyCache
from litellm.proxy.utils import PrismaClient
@ -89,7 +89,7 @@ class DbIdentityStore(IdentityStore):
the composition root can build once the proxy DB is connected.
"""
def __init__(self, prisma_client: "PrismaClient", cache: "DualCache") -> None:
def __init__(self, prisma_client: "PrismaClient", cache: "UserApiKeyCache") -> None:
self._prisma = prisma_client
self._cache = cache