o i@sdZddlZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lmZdd lmZzdd lmZdd lmZWneyadd lmZdd lmZYnwedZGdddZGdddejZdS)z%[Experimental] Live Music API client.N) AsyncIterator) _api_module)_common)_live_converters) _transformers)types) BaseApiClient)set_value_by_path)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)selfrrr]/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/google/genai/live_music.py__init__-s zAsyncMusicSession.__init__promptsreturnNcsD|jjr tdddd|Di}|jtd|iIdHdS)N4Live music generation is not supported in Vertex AI.weightedPromptscSsg|] }tj|ddqS)T convert_keys)rconvert_to_dict).0promptrrr :s z:AsyncMusicSession.set_weighted_prompts.. clientContent)rvertexaiNotImplementedErrorrsendjsondumps)rrclient_content_dictrrrset_weighted_prompts1s z&AsyncMusicSession.set_weighted_promptsconfigcs@|jjr tdtj|dd}|jtd|iIdHdS)NrTrmusicGenerationConfig) rr!r"rrrr#r$r%)rr( config_dictrrrset_music_generation_configBs z-AsyncMusicSession.set_music_generation_configplayback_controlcs8|jjr tdd|ji}|jt|IdHdS)NrplaybackControl)rr!r"valuerr#r$r%)rr,playback_control_dictrrr_send_control_signalMs z&AsyncMusicSession._send_control_signalc|tjjIdHS)z0Sends playback signal to start the music stream.N)r0rLiveMusicPlaybackControlPLAYrrrrplayXzAsyncMusicSession.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)r0rr2STOPr4rrrstop`szAsyncMusicSession.stopcr1)zKReset the context (prompts retained) without stopping the music generation.N)r0rr2 RESET_CONTEXTr4rrr reset_contextgs zAsyncMusicSession.reset_contextcCs4|IdH}r|V|IdH}s dSdS)zbReceive model responses from the server. Yields: The audio chunks from the server. N)_receive)rresultrrrreceivemszAsyncMusicSession.receivecst}z |jjddIdH}Wnty"|jIdH}Ynw|r=zt|}Wntjjy<t d|wi}|j j rGt d|}tjj ||dS)NFdecodezFailed to parse response: r)responsekwargs)rLiveMusicServerMessagerrecv TypeErrorr$loadsdecoderJSONDecodeError ValueErrorrr!r"_from_response model_dump)rparameter_model raw_responserB response_dictrrrr=ws( zAsyncMusicSession._receivecs|jIdHdS)zs,            i