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

test(custom_httpx): pass the token resolver and two-argument client factory in the realtime bridge test

Browse source
This commit is contained in:
mateo-berri 2026-09-18 18:09:33 -07:00
parent 3911d62bbe
commit 17c519c40a
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/test_litellm/llms/custom_httpx/test_llm_http_handler.py
View file

@ -3780,11 +3780,17 @@ async def test_async_realtime_bridges_a_transcription_session_through_the_provid
yield script.pop(0)
speech_client = FakeSpeechClient()
async def resolve_access_token() -> str:
return "token"
provider_config = VertexChirpRealtimeConfig(
access_token="token",
resolve_access_token=resolve_access_token,
project="proj-1",
location="us",
backend_factory=lambda target: SpeechStreamingBackend(target, client_factory=lambda target: speech_client),
backend_factory=lambda target: SpeechStreamingBackend(
target, client_factory=lambda target, access_token: speech_client
),
)
audio = base64.b64encode(b"\x00\x01" * 800).decode()
client_ws = _ScriptedClientWebSocket(