scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

Fix pass through tests

Browse source
This commit is contained in:
Sameer Kankute 2026-02-26 12:55:59 +05:30
parent 34460655d7
commit eb7a6040c1
2 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/litellm_core_utils/realtime_streaming.py
View file

@ -159,7 +159,7 @@ class RealTimeStreaming:
event_type
== "conversation.item.input_audio_transcription.completed"
):
transcript = event_obj.get("transcript", "")
transcript = cast(str, event_obj.get("transcript", ""))
if transcript:
self.input_messages.append(
{"role": "user", "content": transcript}
@ -174,7 +174,7 @@ class RealTimeStreaming:
try:
if event_obj.get("type") != "response.done":
return
response = event_obj.get("response", {})
response = cast(Dict[str, Any], event_obj.get("response", {}))
for item in response.get("output", []):
if item.get("type") == "function_call":
self.tool_calls.append(

3
tests/test_litellm/proxy/pass_through_endpoints/test_llm_pass_through_endpoints.py
View file

@ -224,6 +224,7 @@ class TestVertexAIPassThroughHandler:
# Mock request
mock_request = Mock()
mock_request.state = None # Prevent Mock from returning a truthy _cached_headers
mock_request.method = "POST"
mock_request.headers = {
"Authorization": "Bearer test-creds",
@ -323,6 +324,7 @@ class TestVertexAIPassThroughHandler:
# Mock request
mock_request = Mock()
mock_request.state = None # Prevent Mock from returning a truthy _cached_headers
mock_request.method = "POST"
mock_request.headers = {
"Authorization": "Bearer test-creds",
@ -905,6 +907,7 @@ class TestVertexAIDiscoveryPassThroughHandler:
# Mock request
mock_request = Mock()
mock_request.state = None # Prevent Mock from returning a truthy _cached_headers
mock_request.method = "POST"
mock_request.headers = {
"Authorization": "Bearer test-key",