fix: remove unused VertexBase in transformation.py; apply UUID fallback in return_additional_content_done_events

This commit is contained in:
Ishaan Jaffer 2026-02-25 22:09:27 -08:00
parent cc3c8e433b
commit 58109acde6
2 changed files with 4 additions and 7 deletions

View file

@ -522,10 +522,10 @@ class GeminiRealtimeConfig(BaseRealtimeConfig):
- return response.content_part.done
- return response.output_item.done
"""
if current_output_item_id is None or current_response_id is None:
raise ValueError(
"current_output_item_id and current_response_id cannot be None for a 'done' event."
)
if current_output_item_id is None:
current_output_item_id = "item_{}".format(uuid.uuid4())
if current_response_id is None:
current_response_id = "resp_{}".format(uuid.uuid4())
returned_items: List[OpenAIRealtimeEvents] = []
delta_done_event_text = cast(Optional[str], delta_done_event.get("text"))

View file

@ -15,9 +15,6 @@ import json
from typing import List, Optional
from litellm.llms.gemini.realtime.transformation import GeminiRealtimeConfig
from litellm.llms.vertex_ai.vertex_llm_base import VertexBase
vertex_llm_base = VertexBase()
class VertexAIRealtimeConfig(GeminiRealtimeConfig):