o i}-@sddlZddlZddlmZddlZddlmmZddl m Z m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7zddl8m9Z9Wne:yddlm9Z9Ynwdej;dej;fdd ZfZ?Gd!d"d"e Z@Gd#d$d$e ZAdS)%N)JSONDecodeError)EventEmitterMixin EventType)construct_type)AgentV1AgentAudioDone)AgentV1AgentStartedSpeaking)AgentV1AgentThinking)AgentV1ConversationText) AgentV1Error)AgentV1FunctionCallRequest)AgentV1InjectAgentMessage)AgentV1InjectUserMessage)AgentV1InjectionRefused)AgentV1KeepAlive)AgentV1PromptUpdated)"AgentV1ReceiveFunctionCallResponse)AgentV1SendFunctionCallResponse)AgentV1Settings)AgentV1SettingsApplied)AgentV1SpeakUpdated)AgentV1UpdatePrompt)AgentV1UpdateSpeak)AgentV1UserStartedSpeaking)AgentV1Warning)AgentV1Welcome)WebSocketClientProtocolobjreturncCsRt|trdd|DSt|trdd|DSt|tr'|r't|S|S)a Recursively convert float values that are whole numbers to int. Workaround for Fern-generated models that type integer API fields (like sample_rate) as float, causing JSON serialization to produce values like 44100.0 instead of 44100. The Deepgram API rejects float representations of integer fields. See: https://github.com/deepgram/internal-api-specs/issues/205 cSsi|] \}}|t|qS_sanitize_numeric_types).0kvr r S/home/ubuntu/.local/lib/python3.10/site-packages/deepgram/agent/v1/socket_client.py 2sz+_sanitize_numeric_types..cSsg|]}t|qSr r!)r#itemr r r& 4sz+_sanitize_numeric_types..) isinstancedictitemslistfloat is_integerint)rr r r&r"&s  r"cseZdZdeffdd ZddZddZded d fd d Zde d d fd dZ de d d fddZ de d d fddZded d fddZd"dejed d fddZded d fddZded d fddZd efddZdejd d fddZdejd d fd d!ZZS)#AsyncV1SocketClient websocketct||_dSNsuper__init__ _websocketselfr2 __class__r r&r7N  zAsyncV1SocketClient.__init__cCsB|j2z3dHW}t|tr|Vqttt|dVq6dSNtype_object_r8r*bytesrV1SocketClientResponsejsonloadsr:messager r r& __aiter__Rs  zAsyncV1SocketClient.__aiter__c s|tjdIdHzaz*|j2z#3dHW}t|tr|}n t|}tt |d}|tj |IdHq6Wnt yR}z|tj |IdHWYd}~nd}~wwW|tj dIdHdSW|tj dIdHdS|tj dIdHwaH Start listening for messages on the websocket connection. Emits events in the following order: - EventType.OPEN when connection is established - EventType.MESSAGE for each message received - EventType.ERROR if an error occurs - EventType.CLOSE when connection is closed Nr?) _emit_asyncrOPENr8r*rCrErFrrDMESSAGE ExceptionERRORCLOSEr: raw_messageparsed json_dataexcr r r&start_listeningYs$      .z#AsyncV1SocketClient.start_listeningrHrNc||IdHdSzt Send a message to the websocket connection. The message will be sent as a AgentV1Settings. N _send_modelrGr r r& send_settingsqz!AsyncV1SocketClient.send_settingscrWzw Send a message to the websocket connection. The message will be sent as a AgentV1UpdateSpeak. NrYrGr r r&send_update_speakxr\z%AsyncV1SocketClient.send_update_speakcrWz} Send a message to the websocket connection. The message will be sent as a AgentV1InjectUserMessage. NrYrGr r r&send_inject_user_messager\z,AsyncV1SocketClient.send_inject_user_messagecrWz~ Send a message to the websocket connection. The message will be sent as a AgentV1InjectAgentMessage. NrYrGr r r&send_inject_agent_messager\z-AsyncV1SocketClient.send_inject_agent_messagecrWz Send a message to the websocket connection. The message will be sent as a AgentV1SendFunctionCallResponse. NrYrGr r r&send_function_call_responser\z/AsyncV1SocketClient.send_function_call_responsecs||ptIdHdSzu Send a message to the websocket connection. The message will be sent as a AgentV1KeepAlive. NrZrrGr r r&send_keep_alivesz#AsyncV1SocketClient.send_keep_alivecrWzx Send a message to the websocket connection. The message will be sent as a AgentV1UpdatePrompt. NrYrGr r r&send_update_promptr\z&AsyncV1SocketClient.send_update_promptcrWzj Send a message to the websocket connection. The message will be sent as a bytes. N_sendrGr r r& send_mediar\zAsyncV1SocketClient.send_mediacs6|jIdH}t|tr|St|}tt|dS)B Receive a message from the websocket connection. Nr?r8recvr*rCrErFrrDr:datarTr r r&rps    zAsyncV1SocketClient.recvrrcs,t|tr t|}|j|IdHdSz= Send a message to the websocket connection. Nr*r+rEdumpsr8sendr:rrr r r&rls  zAsyncV1SocketClient._sendcs|t|IdHdSzD Send a Pydantic model to the websocket connection. Nrlr"r+rwr r r&rZszAsyncV1SocketClient._send_modelr4)__name__ __module__ __qualname__rr7rIrVrr[rr^rr`rrbrrdtypingOptionalrrgrrirCrmrDrpAnyrlrZ __classcell__r r r;r&r1Ms r1cseZdZdejffdd ZddZddZded d fd d Z de d d fd dZ de d d fddZ ded d fddZded d fddZd"dejed d fddZded d fddZded d fddZd efddZdejd d fddZdejd d fd d!ZZS)#V1SocketClientr2cr3r4r5r9r;r r&r7r=zV1SocketClient.__init__ccs8|jD]}t|tr|Vqttt|dVqdSr>rBrGr r r&__iter__s   zV1SocketClient.__iter__c Cs|tjdzPz"|jD]}t|tr|}n t|}tt |d}|tj |q Wnt yC}z |tj |WYd}~nd}~wwW|tj ddSW|tj ddS|tj dwrJ)_emitrrLr8r*rCrErFrrDrMrNrOrPrQr r r&rVs"      "zV1SocketClient.start_listeningrHrNcC||dSrXrYrGr r r&r[zV1SocketClient.send_settingscCrr]rYrGr r r&r^rz V1SocketClient.send_update_speakcCrr_rYrGr r r&r`rz'V1SocketClient.send_inject_user_messagecCrrarYrGr r r&rbrz(V1SocketClient.send_inject_agent_messagecCrrcrYrGr r r&rdrz*V1SocketClient.send_function_call_responsecCs||ptdSrerfrGr r r&rg szV1SocketClient.send_keep_alivecCrrhrYrGr r r&rirz!V1SocketClient.send_update_promptcCrrjrkrGr r r&rmrzV1SocketClient.send_mediacCs.|j}t|tr |St|}tt|dS)rnr?rorqr r r&rps    zV1SocketClient.recvrrcCs$t|tr t|}|j|dSrsrtrwr r r&rl(s  zV1SocketClient._sendcCs|t|dSrxryrwr r r&rZ0szV1SocketClient._send_modelr4) rzr{r|websockets_sync_connection Connectionr7rrVrr[rr^rr`rrbrrdr}r~rrgrrirCrmrDrprrlrZrr r r;r&rs r)BrEr} json.decoderr websocketswebsockets.sync.connectionsync connectionr core.eventsrrcore.unchecked_base_modelrtypes.agent_v1agent_audio_doner$types.agent_v1agent_started_speakingr types.agent_v1agent_thinkingr types.agent_v1conversation_textr types.agent_v1errorr #types.agent_v1function_call_requestr "types.agent_v1inject_agent_messager!types.agent_v1inject_user_messagertypes.agent_v1injection_refusedrtypes.agent_v1keep_alivertypes.agent_v1prompt_updatedr,types.agent_v1receive_function_call_responser)types.agent_v1send_function_call_responsertypes.agent_v1settingsrtypes.agent_v1settings_appliedrtypes.agent_v1speak_updatedrtypes.agent_v1update_promptrtypes.agent_v1update_speakr#types.agent_v1user_started_speakingrtypes.agent_v1warningrtypes.agent_v1welcomerwebsockets.legacy.clientr ImportErrorrr"UnionrCrDr1rr r r r&sl                        u