o Th;@s dZddlZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z dd l m Z dd lmZdd lmZdd lmZdd lmZzddlmZddlmZWneymddlmZddlmZYnwedZGdddZGdddejZdS)z%[Experimental] Live Music API client.N) AsyncIterator) _api_module)_common) _transformers)types) BaseApiClient)set_value_by_path)_live_converters)_Content_to_mldev)_Content_to_vertex)ClientConnection)connectzgoogle_genai.live_musicc@seZdZdZdedefddZdeej ddfd d Z d ej ddfd d Z dej ddfddZdddZdddZdddZdddZdeejfddZdejfddZdddZdS) AsyncMusicSessionz![Experimental] AsyncMusicSession. api_client websocketcCs||_||_dS)N) _api_client_ws)selfrrrZ/home/air/segue/gemini/backup/venv/lib/python3.10/site-packages/google/genai/live_music.py__init__/s zAsyncMusicSession.__init__promptsreturnNcsB|jjr tdtjd|id}|jtd|iIdHdS)N4Live music generation is not supported in Vertex AI.weighted_prompts from_object clientContent) rvertexaiNotImplementedErrorlive_converters _LiveMusicClientContent_to_mldevrsendjsondumps)rrclient_content_dictrrrset_weighted_prompts5s z&AsyncMusicSession.set_weighted_promptsconfigcs>|jjr tdtj|d}|jtd|iIdHdS)NrrmusicGenerationConfig) rrr r!#_LiveMusicGenerationConfig_to_mldevrr#r$r%)rr( config_dictrrrset_music_generation_configAs z-AsyncMusicSession.set_music_generation_configplayback_controlcs>|jjr tdtjd|id}|jt|IdHdS)Nrr-r) rrr r! _LiveMusicClientMessage_to_mldevrr#r$r%)rr-playback_control_dictrrr_send_control_signalMsz&AsyncMusicSession._send_control_signalc|tjjIdHS)z0Sends playback signal to start the music stream.N)r0rLiveMusicPlaybackControlPLAYrrrrplayYzAsyncMusicSession.playcr1)z2Sends a playback signal to pause the music stream.N)r0rr2PAUSEr4rrrpause]r6zAsyncMusicSession.pausecr1)zSends a playback signal to stop the music stream. Resets the music generation context while retaining the current config. N)r0rr2STOPr4rrrstopaszAsyncMusicSession.stopcr1)zKReset the context (prompts retained) without stopping the music generation.N)r0rr2 RESET_CONTEXTr4rrr reset_contexths zAsyncMusicSession.reset_contextcCs4|IdH}r|V|IdH}s dSdS)zbReceive model responses from the server. Yields: The audio chunks from the server. N)_receive)rresultrrrreceivenszAsyncMusicSession.receivecst}z |jjddIdH}Wnty"|jIdH}Ynw|r=zt|}Wntjjy<t d|wi}|j j rGt dt |}tjj||dS)NFdecodezFailed to parse response: r)responsekwargs)rLiveMusicServerMessagerrecv TypeErrorr$loadsdecoderJSONDecodeError ValueErrorrrr r!"_LiveMusicServerMessage_from_mldev_from_response model_dump)rparameter_model raw_responserB response_dictrrrr=xs, zAsyncMusicSession._receivecs|jIdHdS)zs0              j