o j-3 @s^ddlmZmZddlmZddlmZddlmZddl Z ddl Z ddl m Z m Z mZmZddlmZddlmZmZe eZed geegd d Zed geegd d ZdefddZdededdfddZdedBdefddZ d(dededededef ddZ deded ed!edef d"d#Z!ed geegd$ed%efd&d'Z"dS)))api_viewpermission_classes)AllowAny)ResponsestatusN) AgentConfigAgentVoiceConfigAgentGuardrailAgentSystemTool) connections)_q_identensure_business_cluster_tablesGETcCs ttjjddd}td|iS)z+List available agent names (for discovery).nameT)flatagents)listrobjects values_listr)requestnamesr8/var/www/html/livekitdocker/backend/apps/agents/views.py agent_lists rc Cs`ztjddddj|d}Wntjy"tdditjdYSwt|dd }id |j d |j d |j d |j d|j d|jd|jd|jd|jd|jd|jd|jd|jrct|jnd d|r|rm|jnd |rs|jnd |ry|jnd |r|jnd |r|jnd |r|jnd dniddd|jjddDddd|jjddD}t|S)z0Worker fetches full config for an agent by name. voice_config guardrails system_toolsknowledge_base)rerrorzAgent not foundrNidrmcube_exenumber mcube_gid stt_provider tts_provider llm_provider llm_model system_promptdefault_languagefirst_message_inboundfirst_message_outboundknowledge_base_idvoice)voice_idtts_model_family similarityspeed stability output_formatcSg|]}|jqSr) guardrail_key).0grrr ;z agent_config..T)enabledcSr3r)tool_key)r5trrrr7<r8) rrprefetch_relatedget DoesNotExistrrHTTP_404_NOT_FOUNDgetattrpkrr!r"r#r$r%r&r'r(r)r*r+strr-r.r/r0r1r2rfilterr)r agent_nameagentr,datarrr agent_configsl             rGreturncCs|duriSt|tr |St|ttfr|jddd}t|tr@|r@zt|}t|tr2|WSiWSt y?iYSwiS)Nutf-8replaceerrors) isinstancedictbytes bytearraydecoderBstripjsonloads Exception)valparsedrrr _json_dictAs   rXplatadvc Cst|tr t|ts dS|dp|dp|dp|d}t|trw|dp8|dp8|dp8|d }|rNt|d pBd sNt||d <|d pa|d pa|d pa|d}|rwt|dpkd swt||d<|dp|dp|d}t|tr|dp|dp|d}|rt|dpd st||d<|dp|d}|rt|dpd st||d<t|dpd sdD]}||} t| tr| r| |d<dSqdSdS)z Many tenants store STT/TTS under nested keys in `platform_settings` JSON (not copied by the flat scalar merge). Fill adv gaps used by MCube. Nsttasrspeech_to_textspeechmodel_idmodel stt_model stt_model_idlanguage language_codelocalelangstt_language_codettstext_to_speech textToSpeech tts_modelr-voiceId tts_voice_id first_message)greetingopening_message opening_line)rMrNr=rBrR) rYrZr[midrgritmtvkgvrrr_apply_nested_platform_settingsQsZ       rxrowcCs\|siSt|d}i}|D] \}}|dkrq|||<q|D]\}}|||<q#|S)u Flatten `business_id_bots` row: scalar columns plus JSON `config`, with `config` winning on collision. Supports deployments that store MCube fields as real columns (prompt, llm_model, stt_model, …) and/or inside JSON `config`. config)rXr=items)rycfgoutrvvrrr_merge_business_id_bots_rows  rrcbidbot_pkbot_nameagent_idc Cst|pd}t|p d}d||gfg}|r#|d|||gf|d|t||gf|D]L\}}z?td/}||||}|rddd|jD} tt t | |WdWSWdn1snwYWq1t y}Yq1wt d ||||iS) a8 Scalar columns + JSON `config` from livekitvoicebot_cluster.business_id_bots (per tenant/bot). Deployments vary: rows may be keyed by `bot_id` (selected cluster bot), `agent_id` (LiveKit id from `{bid}_bots`), or legacy `name`. Unknown columns are skipped without failing the whole lookup. rcz SELECT * FROM business_id_bots WHERE business_id = %s AND bot_id = %s ORDER BY updated_at DESC LIMIT 1 z SELECT * FROM business_id_bots WHERE business_id = %s AND (agent_id = %s OR CAST(agent_id AS CHAR) = %s) ORDER BY updated_at DESC LIMIT 1 z SELECT * FROM business_id_bots WHERE business_id = %s AND (name = %s OR name = %s) ORDER BY updated_at DESC LIMIT 1 clustercSg|]}|dqSrrr5crrrr7z2_fetch_business_id_bots_merged..NzQbusiness_id_bots: no row matched business_id=%s bot_id=%s bot_name=%r agent_id=%r)rBrRappendr cursorexecutefetchone descriptionrrNziprUloginfo) rrrrvariantssqlparamscurrcolsrrr_fetch_business_id_bots_mergedsP       roverriderFkeyscGsX|||fD]$}|D]}||vs||durq t||}|r(|Sq qdS)z\First non-empty among business_id_bots merge, advanced_settings JSON, then `{bid}_bots` row.Nrc)r=rBrR)rrZrFrdrvsrrr _pick_strs r business_idbot_idc Csdz t|}t|}WntytdditjdYSwt||d}i}td*}|dt |d|g| }|rQdd |j D} t t | |}Wd n1s[wYt|d pl|d pld } t|dpwd } t||| | } |s| ri}|s| stdditjdSt|d} t|r|dnd }|D]"\}}|dvrqt|t tfrqt| |d pd rq|| |<q|d}t|t r|dp|d}|r| dst|| d<t|| t| | |ddd}t|dd}d }|r|dnd }t|ttfr$|jddd}t|tr[|r[zt|}t|t rMt|dpI|dpId }Wn tyZd }Ynwt| | |dd d!d"d#}tid$|d%|d|d|d"|rt|d"pd nd d#|rt|d#pd nd d&t| | |d&d'd(t| | |d(d)d*t| | |d*d+d,t| | |d,d-d.d/t| | |d/d0d1d2t| | |d2d3d4t| | |d4d5dt| | |dd6p|d7t| | |d7d8d9t| | |d9d:d;t| | |d;d<t| | |d=d>t| | |d?d@t| | |dAdBt| | |dC|r#|dnd |r,|dDnd dES)Fa> Internal helper endpoint for MCube runtime. Reads `{business_id}_bots` on cluster DB plus `business_id_bots` (scalar columns + JSON `config`) (same database: livekitvoicebot_cluster). Returns fields the WS bridge / outbound worker need for prompts, providers, models, voice, MCube exenumber/gid. rzInvalid business_id or bot_idr_botsrzSELECT * FROM z WHERE bot_id = %s LIMIT 1cSrrrrrrrr7rz,cluster_bot_mcube_config..Nrrrcrz Bot not foundadvanced_settingsplatform_settings)Nrcr,r-rmrnr'MCUBE_SYSTEM_PROMPTpromptz\n rIrJrKroMCUBE_FIRST_MESSAGEmessage_onboardmessage_outboundmessage_inboundrrr&MCUBE_LLM_MODELr%MCUBE_LLM_PROVIDERr#MCUBE_STT_PROVIDERrbMCUBE_STT_MODEL_IDrarhMCUBE_STT_LANGUAGE_CODEr(r$MCUBE_TTS_PROVIDERrlMCUBE_TTS_MODELMCUBE_TTS_VOICE_ID tts_encodingMCUBE_TTS_ENCODING tts_chunk_msMCUBE_TTS_CHUNK_MStts_gainMCUBE_TTS_GAINplayback_pace_factorMCUBE_PLAYBACK_PACE_FACTORcheckpoint_everyMCUBE_CHECKPOINT_EVERYr! mcube_exenumr"conversation_behavior)rrr!r"rr)intrUrrHTTP_400_BAD_REQUESTrr rrr rrrNrrBr=rRrr?rXr{rMrrxrrJrOrPrQrSrT)rrrrrtablerFrryrragent_id_for_bibbibrZrYrApv voice_nestedvidr'r- voice_raw voice_obj first_msgrrrcluster_bot_mcube_configs           $ ""    r)rc)#rest_framework.decoratorsrrrest_framework.permissionsrrest_framework.responserrest_frameworkrrSloggingapps.agents.modelsrr r r django.dbr apps.cluster.dynamic_tablesr r getLogger__name__rrrGrNrXrxrrrBrrrrrrrsF       &0 G