diff --git a/litellm/llms/gemini/realtime/transformation.py b/litellm/llms/gemini/realtime/transformation.py index 51c4b9e71e7..2e0e678e69f 100644 --- a/litellm/llms/gemini/realtime/transformation.py +++ b/litellm/llms/gemini/realtime/transformation.py @@ -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")) diff --git a/litellm/llms/vertex_ai/realtime/transformation.py b/litellm/llms/vertex_ai/realtime/transformation.py index cb1238a4f67..eaa9844f108 100644 --- a/litellm/llms/vertex_ai/realtime/transformation.py +++ b/litellm/llms/vertex_ai/realtime/transformation.py @@ -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):