o j{@sUdZddlmZddlZddlZddlZddlZddlmZddl m Z m Z e dZ iZded <eed d Zd*ddZd+ddZd,ddZd-ddZd.dd Zd/d"d#Zd0d(d)ZdS)1aF Resolve MCube voice/LLM/STT settings from livekitvoicebot_cluster: - Per-business table `{business_id}_bots` (when present) - Shared override rows in `business_id_bots` (JSON `config`) Matches Django `cluster_bot_mcube_config` merge rules so outbound, WS bridge, and ai_worker stay consistent without calling the HTTP API. ) annotationsN)Any)_parse_mysql_cluster_url_runtime_mcube_overrideszmcube.cluster_bot_runtimez3dict[tuple[int, int], tuple[float, dict[str, str]]]_CACHECLUSTER_BOT_RUNTIME_CACHE_TTL_S30nstrreturncCs(dt|tdtdtddS)N``)r replacechr)r r^/var/www/html/livekitdocker/backend/agent_runtime/src/mcube_integration/cluster_bot_runtime.py_q_idents(rrawrdict[str, Any]cCs|duriSt|tr |St|ttfr|jddd}t|tr@|r@zt|}t|tr2|WSiWSt y?iYSwiS)Nutf-8rerrors) isinstancedictbytes bytearraydecoder stripjsonloads Exception)rprrr_json_load_maybes   r#datacCs|d}t|ttfr|jddd}t|trE|rEzt|}t|t r8t|dp3|dp3dWSWdSt yDYdSwdS)Nvoicerrrvoice_idvoiceId) getrrrrr rrr rr!)r$ voice_raw voice_objrrr_voice_id_from_row/s   " r,bidintbot_pkbot_namec CsZi}t|}|p d}z|}|ddd|pgD}|s.|WdWSd} d|vr7d} nd|vr=d} |||pB|g} |d td d td d tddtdd| d | |} | r| ddurt| d}WdW|SWdW|SWdW|S1swYW|St yt d||Y|Sw)Nr(z SELECT COLUMN_NAME FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'business_id_bots' cSsh|]}|dqS) COLUMN_NAMEr).0rrrr Isz3_fetch_business_id_bots_override.. updated_atz ORDER BY `updated_at` DESCidz ORDER BY `id` DESCz$ SELECT config FROM business_id_botsz WHERE business_idz = %s AND (namez = %s OR z = %s) z) LIMIT 1 configz?cluster_bot_runtime: business_id_bots read failed bid=%s bot=%s) r rcursorexecutefetchallrfetchoner)r#r!log exception) connr-r/r0 override_cfgname_bot name_labelcurcols order_sqlparamsr3rrr _fetch_business_id_bots_override=sb     ! ! !! rIc Cs|d}i}z>|/}|dt|dtdd|g|}|r2t|}WdW|SWdW|S1s>wYW|StyXtjd||ddY|Sw) N_botszSELECT * FROM z WHERE bot_idz = %s LIMIT 1z=cluster_bot_runtime: no row or missing table %s for bot_id=%sT)exc_info)r;r<rr>rr!r?debug)rAr-r/tabler$rErowrrr_fetch_dynamic_bot_rowfs&  $    rPrBc Cst|}|D]\}}|||<qt|dp|dpd}|dp1|dp1|dp1d}|dur8d}t||d<|dpN|dpN|d pNd}|rYt||d<t|d pi|d pit|pid}|rs||d <|S) Nr0r9r( system_promptMCUBE_SYSTEM_PROMPTprompt first_messageMCUBE_FIRST_MESSAGEmessage_outbound tts_voice_idMCUBE_TTS_VOICE_ID)ritemsr r)rr,) r$rBflatkvr0rQfirstvidrrr_flat_for_overridests6     (r_r8 int | NonerKdict[str, str]cCs|dus|dur iSz t|}t|}Wn tyiYSwt}t||f}|r9||dtkr9|dStt dd}|sEiSz ddl }ddl m }Wnt yatdiYSwdd |D} || d <z |jdi| } Wntytd iYSwzrzOt| ||} t| d p| d pd} t| ||| } t| | }t|}|rtd||t|||ft||f<|WWz| WStyYSwtytd||iYWz| WStyYSwwz| Wwty Yww)z MCube runtime string overrides from cluster `{bid}_bots` + `business_id_bots`. Empty when DB unavailable or keys missing. NrrDATABASE_CLUSTER_URLr() DictCursorz*cluster_bot_runtime: pymysql not installedcSsi|] \}}|dkr||qS) cursorclassr)r2r[r\rrr sz:get_runtime_overrides_from_cluster_bot..rdz&cluster_bot_runtime: DB connect failedr0r9z-cluster_bot_runtime: bid=%s bot_id=%s keys=%sz2cluster_bot_runtime: merge failed bid=%s bot_id=%sr)r.r!time monotonicrr) _CACHE_TTL_Srosgetenvpymysqlpymysql.cursorsrc ImportErrorr?warningrYconnectr@rPr rrIr_rinfolistkeysclose)r8rKr-bpidnowhitrHrkrcconn_kwrAr$r0rBrZoutrrr&get_runtime_overrides_from_cluster_botsv               ry)r r r r )rrr r)r$rr r )r-r.r/r.r0r r r)r-r.r/r.r r)r$rrBrr r)r8r`rKr`r ra)__doc__ __future__rrloggingrirftypingrbusiness_id_agentsrr getLoggerr?r__annotations__floatrjrhrr#r,rIrPr_ryrrrrs$         )