iBdZddlZddlmZmZddlmZGddeZy)a_ Vertex AI Realtime (BidiGenerateContent) config. Extends GeminiRealtimeConfig but adapts the WSS URL and auth header for the Vertex AI endpoint instead of Google AI Studio. URL pattern: wss://{location}-aiplatform.googleapis.com/ws/ google.cloud.aiplatform.v1.LlmBidiService/BidiGenerateContent Auth: OAuth2 Bearer token (not an API key). N)ListOptional)GeminiRealtimeConfigc eZdZdZdedededdfdZ ddeed ed eedefd Z dd ed ed eedefd Z ddedefdZ d edefdZ dded edeede effd Z xZS)VertexAIRealtimeConfigz Realtime config for Vertex AI (BidiGenerateContent). ``access_token`` and ``project`` must be pre-resolved by the caller (they require async I/O) and injected at construction time. access_tokenprojectlocationreturnNc.||_||_||_yN) _access_token_project _location)selfrr r s /Users/manta/Documents/Projects/TheRoad-I1/.venv/lib/python3.12/site-packages/litellm/llms/vertex_ai/realtime/transformation.py__init__zVertexAIRealtimeConfig.__init__s) !api_basemodelapi_keyc|r8|jd}|jddjdd}|dS|j}|dk(rd}n|d }d|dS) z Build the Vertex AI Live WSS endpoint URL. If *api_base* is provided it overrides the default aiplatform host, allowing enterprise / VPC-SC deployments to point at a custom gateway. /zhttps://zwss://zhttp://zws://zA/ws/google.cloud.aiplatform.v1.LlmBidiService/BidiGenerateContentglobalzaiplatform.googleapis.comz-aiplatform.googleapis.com)rstripreplacer)rrrrbaser hosts rget_complete_urlz'VertexAIRealtimeConfig.get_complete_url%ss ??3'D<< H5==iQDV\] ]>> x .DZ9:Dv^__rheaderscvt|}d|j|d<|jr|j|d<|S)u Return headers with a Bearer token for Vertex AI. ``api_key`` is intentionally ignored — Vertex AI uses OAuth2 tokens, not API keys. The token was resolved at config-construction time. zBearer Authorizationzx-goog-user-project)dictrr)rr rrs rvalidate_environmentz+VertexAIRealtimeConfig.validate_environmentCs@w-%,T-?-?,@#A ==-1]]G) *rinput_audio_formatc2dddd}|j|dS)Nzaudio/pcm;rate=16000z audio/pcmuz audio/pcma)pcm16 g711_ulaw g711_alawzapplication/octet-stream)get)rr% mime_typess rget_audio_mime_typez*VertexAIRealtimeConfig.get_audio_mime_typeYs'+%% ~~02LMMrcddlm}ddlm}dg}d|jd|j d|}|d|id d d d iiid }t jd|iS)ad Return the JSON setup message for Vertex AI Live. Vertex AI requires the fully-qualified model path: ``projects/{project}/locations/{location}/publishers/google/models/{model}`` Also enables automatic activity detection (server VAD) and output audio transcription so the proxy forwards transcript events. r)BidiGenerateContentSetup)GeminiResponseModalitiesAUDIOz projects/z /locations/z/publishers/google/models/responseModalitiesautomaticActivityDetectionFi )disabledsilenceDurationMs)rgenerationConfigrealtimeInputConfiginputAudioTranscriptionoutputAudioTranscriptionsetup)litellm.types.llms.geminir.litellm.types.llms.vertex_air/rrjsondumps)rrr.r/response_modalitiesfull_model_path setup_configs rsession_configuration_requestz4VertexAIRealtimeConfig.session_configuration_requestes GI?Fi $..)( 1  %!57J K- %),/$(*(*2 zz7L122rmessagerAc|tj|}|jddk(rgSt||||S)u Translate OpenAI realtime client messages to Vertex AI format. ``session.update`` is intentionally ignored (returns []) because Vertex AI only accepts a single ``setup`` message at the start of the connection — sending a second one causes a 1007 close error. The initial setup (sent automatically before bidirectional_forward) already includes AUDIO modality and server VAD, so there is nothing more to configure. typezsession.update)r<loadsr*supertransform_realtime_request)rrBrrA json_message __class__s rrGz1VertexAIRealtimeConfig.transform_realtime_requestsF zz'*   F #'7 7Iw1 U9  rr )r')__name__ __module__ __qualname____doc__strrrrr#r$r,rArrG __classcell__)rIs@rrrs"S"3"#"$""& `3-``# ` `D"& #   ,NcNN"33"33"3X8<    (0}  c   rr)rMr<typingrr+litellm.llms.gemini.realtime.transformationrrrrrSs$  !LP 1P r