scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-03 02:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

test(integration): type the upstream refresh helper

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-24 08:50:43 +00:00
parent a466d8e08f
commit b403b04776
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/integration/mcp/test_mcp_kdf_rotation.py
View file

@ -18,7 +18,7 @@ from cryptography.hazmat.primitives.hashes import SHA256
from cryptography.hazmat.primitives.kdf.hkdf import HKDF
from integration._support.client import Gateway
from integration._support.mcp import McpPeer, mcp_peer, register_mcp
from integration._support.oauth_server import oauth_server
from integration._support.oauth_server import AuthorizationServer, oauth_server
from integration._support.process import owned_proxy
from pydantic import SecretStr
from typing_extensions import ReadOnly, TypedDict
@ -147,7 +147,7 @@ def _session_refresh(user_id: str, client_id: str, keys: SessionKeys) -> str:
return minted.token.get_secret_value()
def _upstream_refresh(auth) -> str:
def _upstream_refresh(auth: AuthorizationServer) -> str:
issued: Final = auth.issue("authorization_code", "kdf-client", "integration-user", "tools.read tools.call")
token: Final = issued["refresh_token"]
assert isinstance(token, str) and token, issued