)jdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZmZmZddlmZddlmZmZddlmZmZmZmZmZmZm Z ddl!Z!ddl"Z" ddl#Z#ddl$Z#dZ%n#e&$rd Z%dZ#YnwxYwdd l'm(Z(m)Z)dd l*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0dd l1m2Z2dd l3m4Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;ddlZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRddlSmTZTe jUeVZW ddlXmYZZn #e&$rdZZYnwxYweZZ[eZZ\e]eDZ^e!j_Z`dZadZbdZcdZddZedZfdZgdZhhdZidZjhdZkhdZldZmdZndZodZpd Zqe jrd!Zse jrd"Zteud#d$hZve jrd%Zwd&Zxd'ZyGd(d)ZzGd*d+Z{dd,lm|Z|m}Z~e|Gd-d.ZGd/d0eZGd1d2ZGd3d4eZGd5d6eZGd7d8eZGd9d:eZGd;deZGd?d@ZGdAdBeZGdCdDeZGdEdFeZGdGdHeZGdIdJeZGdKdLeZGdMdNeZGdOdPeZGdQdReZGdSdTeZGdUdVeZGdWdXeZGdYdZeZGd[d\ZGd]d^ZGd_d`eZGdadbeZGdcddeZGdedfeZGdgdheZGdidjeZGdkdlZGdmdnZGdodpZGdqdrZGdsdtZGdudve+ZddyZ dddZdS)aQ Yuanbao platform adapter. Connects to the Yuanbao WebSocket gateway, handles authentication (AUTH_BIND), heartbeat, reconnection, message receive (T05) and send (T06). Configuration in config.yaml (or via env vars): platforms: yuanbao: extra: app_id: "..." # or YUANBAO_APP_ID app_secret: "..." # or YUANBAO_APP_SECRET bot_id: "..." # or YUANBAO_BOT_ID (optional, returned by sign-token) ws_url: "wss://..." # or YUANBAO_WS_URL api_domain: "https://..." # or YUANBAO_API_DOMAIN ) annotationsN)datetimetimezone timedelta)Path)ABCabstractmethod)AnyCallableClassVarDictListOptionalTupleTF)PlatformPlatformConfig)BasePlatformAdapter MessageEvent MessageType SendResultcache_document_from_bytescache_image_from_bytes)MessageDeduplicator) download_urlget_cos_credentials upload_to_cosbuild_image_msg_bodybuild_file_msg_bodyguess_mime_typemd5_hex)CMD_TYPE_fields_to_dict _get_string _get_varint _parse_fieldsWS_HEARTBEAT_RUNNINGWS_HEARTBEAT_FINISHHERMES_INSTANCE_IDdecode_conn_msgdecode_inbound_pushdecode_query_group_info_rsp decode_get_group_member_list_rspencode_auth_bind encode_pingencode_push_ackencode_send_c2c_messageencode_send_group_messageencode_send_private_heartbeatencode_send_group_heartbeatencode_query_group_infoencode_get_group_member_list next_seq_no)build_session_key) __version__z0.0.0z0wss://bot-wss.yuanbao.tencent.com/wss/connectionzhttps://bot.yuanbao.tencent.comg>@.@$@d?>>>@r@g^@u?任务有点复杂，正在努力处理中，请耐心等待...zB\[(image|voice|video|file(?::[^|\]]*)?)\|ybres:([A-Za-z0-9_\-]+)\]z\[(\w+):[^\]]*?(/[^\]]+?)\s*\]imagefilez\s*\(\d+/\d+\)$2 c<eZdZdZeddZeddZe dddZeddZeddZ ed dZ e d!d"dZ e d#dZ e d$dZed%dZed%dZed&dZdS)'MarkdownProcessoraaEncapsulates all Markdown-related utilities for the Yuanbao platform. Provides static methods for: - Fence detection and streaming merge - Table row detection and sanitization - Paragraph-boundary splitting - Atomic-block extraction and chunk splitting - Outer markdown fence stripping - Markdown hint prompt generation textstrreturnboolcjd}|dD]}|dr| }|S)a Detect whether the text has unclosed code block fences. Scan line by line, toggling in/out state when encountering a line starting with ```. An odd number of toggles indicates an unclosed fence. Args: text: Markdown text to check Returns: Returns True if the text ends with an unclosed fence, otherwise False F ```)split startswith)rRin_fencelines @/home/wildlama/.hermes/hermes-agent/gateway/platforms/yuanbao.pyhas_unclosed_fencez$MarkdownProcessor.has_unclosed_fencesEJJt$$ ( (Du%% ('<c|}|sdS|dd}|do|dS)z Detect whether the text ends with a table row (last non-empty line starts and ends with |). Args: text: Text to check Returns: Returns True if the last non-empty line is a table row FrW|)rstriprYstriprZendswith)rRtrimmed last_lines r]ends_with_table_rowz%MarkdownProcessor.ends_with_table_rowsg++-- 5MM$''+1133 ##C((DY-?-?-D-DDr_N max_charsintlen_fnOptional[Callable[[str], int]]tuple[str, str]c|pt}|||kr|dfS|tur |d|}nQdt|}}||kr0||zdzdz}||d||kr|}n|dz }||k0|d|}|d}|dkr|d|dz||dzdfStjd} d} | |D]} | } | dkr|d| || dfS|d }|dkr|d|dz||dzdfSt|} |d| || dfS) a Find the nearest paragraph boundary split point within max_chars, return (head, tail). Split priority: 1. Blank line (paragraph boundary) 2. Newline after period/question mark/exclamation mark (Chinese and English) 3. Last newline 4. Force split at max_chars Args: text: Text to split max_chars: Maximum character count limit len_fn: Optional custom length function (e.g. UTF-16 length); defaults to built-in len Returns: (head, tail) tuple, head is the front part, tail is the back part, satisfying head + tail == text NrrC u[。！？.!?]\nrarW)lenrfindrecompilefinditerend) rRrirk_lenwindowlohimidpossentence_end_rebest_posmcuts r]split_at_paragraph_boundaryz-MarkdownProcessor.split_at_paragraph_boundarys.} 4:: " "8O 3;;*9*%FFD Br''Bw{q(4TcT ##y00BBqB r'' #2#YFll6"" 77q>4a>1 1*%899 ))&11  AuuwwHH a<<  ?DO3 3ll4   77q>4a>1 1&kkDSDz4:%%r_cP|dS)zDDetermine whether an atomic block is a code block (starts with ```).rX)lstriprZrRs r] is_fence_atomzMarkdownProcessor.is_fence_atom*s {{}}''...r_c|dd}|do|dS)zHDetermine whether an atomic block is a table (first line starts with |).rWrrb)rYrdrZre)rR first_lines r] is_table_atomzMarkdownProcessor.is_table_atom/sMZZ%%a(..00 $$S))Fj.A.A#.F.FFr_ list[str]c|d}ggd}dd}dfd }|D]}|rJ||d rtd kr d}|O|d r"|d }|||r3r|d s |||dkr |r|d r |||S)a Split text into a list of "atomic blocks", each being an indivisible logical unit: - Code block (fence): from opening ``` to closing ``` (including fence lines) - Table: consecutive |...| lines forming a whole segment - Normal paragraph: plain text segments separated by blank lines Blank lines serve as separators and are not included in any atomic block. Args: text: Markdown text to split Returns: List of atomic block strings (all non-empty) rWFr\rSrTrUc~|}|do|dS)Nrb)rdrZre)r\strippeds r]_is_table_linez:MarkdownProcessor.split_into_atoms.._is_table_lineLs6zz||H&&s++F0A0A#0F0F Fr_NonecrTd}|r|dSdS)NrW)joinrdappendclear)atomatoms current_liness r]_flush_currentz:MarkdownProcessor.split_into_atoms.._flush_currentPs` &yy//::<<'LL&&&##%%%%%  & &r_rXrpTraro)r\rSrTrUrTr)rYrrZrrrd)rRlinesr[rrr\rrs @@r]split_into_atomsz"MarkdownProcessor.split_into_atoms5s" 4  #%  G G G G & & & & & & & + +D +$$T***??5))%c-.@.@1.D.D$H"N$$$'' +   $$T****%% + % b8I)J)J%"N$$$$$T****##     %^^M"4E%F%F%"N$$$$$T**** r_c|pt}|sgS|||kr|gS||}gt}gd}d fd }|D]} || } rdnd} || z| z} | |krr|gd}d} sh| |krb|| s|| r8|t| | || | zz }|g} tD]\}}|||kr| |*||vr| |D||r| |o|}|||krW| |||\}}|s|d|||d}}|r| ||||kW|r| |t| dkrR| dg}| ddD]<}|d }|d z|z}|||kr||d <'||=|} d | DS) a Split Markdown text into multiple chunks by max_chars. Guarantees: - Each chunk <= max_chars characters (unless a single code block/table itself exceeds the limit) - Code blocks (```...```) are not split in the middle - Table rows are not split in the middle (tables output as atomic blocks) - Split at paragraph boundaries (blank lines, after periods, etc.) - Small trailing/leading chunks are merged with neighbours when possible Args: text: Markdown text to split max_chars: Max characters per chunk, default 4000 len_fn: Optional custom length function (e.g. UTF-16 length); defaults to built-in len Returns: List of text chunks after splitting (non-empty) rrTrc`r*ddSdS)Nrq)rr)chunks current_partssr] _flush_partsz;MarkdownProcessor.chunk_markdown_text.._flush_partss9 : fkk-8899999 : :r_rCrkNrprarqcg|]}||Sr.0cs r] z9MarkdownProcessor.chunk_markdown_text..s'''aQ''''r_r) rrrsetrraddr enumerater^r)clsrRrirkrxrindivisible_set current_lenrratom_lensep_len projected_lenresultidxchunk remainingheadmergedprevcombinedrrs @@r]chunk_markdown_textz%MarkdownProcessor.chunk_markdown_textrsf2} I 4:: " "6M$$T**$'EE#%   : : : : : : : . .DtDzzH(/aaaG''1H>Y..!)F2JJMM%((((F''6''''r_ prev_chunk next_chunkc|}|}|ds|drdS||r]|r-|ddnd}|dr|drdSdS)u' Infer the separator to use between two split chunks. Rules (aligned with TS markdown-stream.ts): - Previous chunk ends with code fence or next chunk starts with fence → single newline '\n' - Previous chunk ends with table row and next chunk starts with table row → single newline '\n' (continued table) - Otherwise → double newline '\n\n' (paragraph separator) Args: prev_chunk: Previous chunk next_chunk: Next chunk Returns: '\n' or '\n\n' rXrWrrorbrq)rcrrerZrhrYrd)rrr prev_trimmed next_trimmedrs r]infer_block_separatorz'MarkdownProcessor.infer_block_separators""((** !((**    ' ' <+B+B5+I+I 4  " ": . . @LT++D11!4::<<>K1X^^%% "IOO%% x3ZOO K   K %"+&& r_cfd|vr|S|d}g}|D]}|}|dr|drt jd|rJ|d}dd|D}|||dks,|dddkr||||d|S)a Table output sanitization. Handle common formatting issues in AI-generated Markdown tables: 1. Remove extra whitespace before/after table rows 2. Ensure separator rows (|---|---|) are correctly formatted 3. Remove empty table rows Args: text: Markdown text containing tables Returns: Sanitized text rbrWz^\|[\s\-:]+(\|[\s\-:]+)+\|$c3jK|].}|r|n|V/dSNrd)rcells r] z.nsO** )- > $******r_z||ro) rYrdrZrertrrrreplace)rRr result_linesr\rcells normalizeds r]sanitize_markdown_tablez)MarkdownProcessor.sanitize_markdown_tablePsS d??K 4  "$  * *Dzz||H""3'' *H,=,=c,B,B *8:HEE 2$NN3//E!$**$)***""J!'' 3333%%)9)9#r)B)B)H)H)J)Jb)P)P ''1111##D))))yy&&&r_c dS)z Markdown rendering hint (appended to system prompt). Tell AI that Yuanbao platform supports Markdown rendering, including: - Code blocks (```lang) - Tables (| col | col |) - Bold/italic aThe current platform supports Markdown rendering. You can use the following formats: - Code blocks: ```language\ncode\n``` - Tables: | col1 | col2 |\n|---|---|\n| val1 | val2 | - Bold: **text** / Italic: *text* Please use Markdown formatting when appropriate to improve readability.rrr_r]markdown_hint_system_promptz-MarkdownProcessor.markdown_hint_system_prompts V  r_)rRrSrTrUr)rRrSrirjrkrlrTrm)rRrSrTrrN)rRrSrirjrkrlrTr)rrSrrSrTrS)rrrTrrRrSrTrSrTrS)__name__ __module__ __qualname____doc__ staticmethodr^rhrrrr classmethodrrrrrrrr_r]rQrQs  \*EEE\E$26=&=&=&=&\=&B///\/GGG\G 888\8x15 k(k(k(k([k(^[B!!![!J!!!\!J*'*'*'\*'\   \   r_rQc,eZdZUdZdZdZdZdZdZdZ iZ de d <iZ d e d <e d%dZed&dZed'dZe d(dZe d)dZe d*dZe d+d,d!Ze d+d,d"Ze d+d,d#Zd$S)- SignManageraEncapsulates all sign-token related logic for the Yuanbao platform. Manages token acquisition, caching, signature computation, and automatic retry. All state (cache, locks) is kept as class-level attributes so that a single shared client serves the whole process. z/api/v5/robotLogic/sign-tokenis'rr<<r:zdict[str, dict[str, Any]]_cachezdict[str, asyncio.Lock]_locksapp_keyrSrT asyncio.Lockcd||jvrtj|j|<|j|S)zReturn (creating if needed) the per-app_key refresh lock. Must only be called from within a running event loop (async context). )rasyncioLock)rrs r]get_refresh_lockzSignManager.get_refresh_locks0 #* $ $"),..CJw z'""r_nonce timestamp app_secretc||z|z|z}tj||tjS)zCompute HMAC-SHA256 signature (aligned with TypeScript original). plain = nonce + timestamp + app_key + app_secret signature = HMAC-SHA256(key=app_secret, msg=plain).hexdigest() )hmacnewencodehashlibsha256 hexdigest)rrrrplains r]compute_signaturezSignManager.compute_signaturesN !G+j8x ))++U\\^^W^LLVVXXXr_ctjttd}|dS)zlBuild Beijing-time ISO-8601 timestamp (no milliseconds). Format: 2006-01-02T15:04:05+08:00 )hourstzz%Y-%m-%dT%H:%M:%S+08:00)rnowrrstrftime)bjtimes r]build_timestampzSignManager.build_timestamps< )!*<*<*>>8999r_entrydict[str, Any]rUcL|dtjz |jkS)zEDetermine whether the cache entry is valid (not expired with margin). expire_ts)timeCACHE_REFRESH_MARGIN_S)rrs r]is_cache_validzSignManager.is_cache_valids"[!DIKK/#2LLLr_rc8|jdS)z;Clear all per-app_key refresh locks (called on disconnect).N)rrrs r] clear_lockszSignManager.clear_lockss r_rjctjfd|jD}|D]}|j|dt |S)zRemove all expired entries from the token cache. Returns the number of entries purged. Called lazily from ``get_token()`` so that stale app_key entries don't accumulate indefinitely in long-running processes. cTg|]$\}}|ddz dk"|%S)rrget)rkvrs r]rz-SignManager.purge_expired..sE   !QQUU;***Q.. ...r_N)rritemspoprr)r expired_keysrrs @r] purge_expiredzSignManager.purge_expireds~ikk    ***,,    $ $A JNN1d # # # #<   r_ro api_domain route_envc K|d|j}tj|j4d{V}t |jdzD]P}tjd}| } | || ||} ||| | d} dttttd} |r|| d <td ||d krd |d|jd nd||| | d{V} | jdkr)| j}t)d| jd|dd | }n%#t,$r}t/d||d}~wwxYw|d}|d kr|d}t3|t4st/d|td|d|ccdddd{VS||jkrW||jkrLtd||j|dz|jt=j|jd{V'|dd}t)d|d| dddd{Vn#1d{VswxYwYt)d)zHSend sign-ticket HTTP request with auto-retry (up to MAX_RETRIES times)./timeoutNrp)rr signaturerapplication/json) Content-Typez X-AppVersionzX-OperationSystemz X-Instance-Idz X-Bot-Versionz X-Route-EnvzSign token request: url=%s%srz (retry )ro)jsonheaderszSign token API returned : z!Sign token response parse error: codedataz*Sign token response missing 'data' field: zSign token success: bot_id=%sbot_idz>Sign token retryable: code=%s, retrying in %ss (attempt=%d/%d)msgzSign token error: code=, msg=z'Sign token failed: max retries exceeded) rc TOKEN_PATHhttpx AsyncClientHTTP_TIMEOUT_Srange MAX_RETRIESsecrets token_hexrr _APP_VERSION_OPERATION_SYSTEM_YUANBAO_INSTANCE_ID _BOT_VERSIONloggerinfopost status_coderR RuntimeErrorr  Exception ValueErrorr isinstancedictRETRYABLE_CODEwarning RETRY_DELAY_Srsleep)rrrrrurlclientattemptrrrpayloadr!responsebody result_dataexcr$r%r's r]fetchzSignManager.fetchsK""3''999$S-?@@@< P< P< P< P< P< P< PF 1!455; P; P)"--//11 11%GZXX  '"!*!* %7$0):%9%1 7-6GM* 2?F{{;w;;;;;;PR "(Sw!P!PPPPPPP'3..#=D&'f(BV'f'fZ^_c`c_cZd'f'fgggY2:--//KK YYY$%N%N%NOOUXXY#v..199&??622D%dD11e()cVa)c)cdddKK ?(ASASTTTKK]< P< P< P< P< P< P< P< P< P< P< P< P< P< P`3---'CO2K2KNNX)!  "-(9:::::::::!ooeR00"#NT#N#N#N#NOOOw; P< P< P< P< P< P< P< P< P< P< P< P< P< P< P< P< P< P< P< P< P< P< P< P< P< P< P< P|DEEEs?DJ> EJ> F(E;;FB J>B J>> K Kc K||j|}|rh||rSt |dt jz }t d|t|S| |4d{V|j|}|r6||r!t|cdddd{VS| ||||d{V}|dd}|dkrt j|znt jdz} |dd|d d||d d|d d| d |j|<dddd{Vn#1d{VswxYwYt|j|S) zGet WS auth token (with cache). Return directly on cache hit without re-requesting; treat as expiring 60 seconds before actual expiry, triggering refresh. rz"Using cached token (%ds remaining)Ndurationrtokenror&productsourcerNr&rLrOrPr) rrrrrjrr5r6r=rrJ) rrrrrcachedremainr%rLrs r] get_tokenzSignManager.get_token:s ((  c((00  ,ty{{:;;F KKr.r@rr,r__annotations__rrrrrrrr rrJrTrWrr_r]rrs1JNKM N )+F**** ')F((((###[#YYY\Y:::\:MMM[M[!!![!$  GFGFGFGF[GFV  ()()()()[()X  ))))[)))r_r) dataclassfieldceZdZUdZded<eeZded<dZded <d Z d ed <d Z d ed <d Z d ed<d Z d ed<d Z d ed<eeZded<d Zd ed<d Zd ed<d Zd ed<d Zd ed<d Zd ed<d Zd ed<eeZded<dZded<dZded<dZded<dZded<dZded<eeZded <eeZded!<eeZded"<eeZded#<dZded$<dS)%InboundContextzMutable context flowing through the inbound middleware pipeline. Each middleware reads/writes fields on this context. The pipeline engine passes it to every middleware in registration order. r adapter)default_factorylist raw_framesNOptional[dict]pushrorS decoded_via from_account group_code group_namesender_nicknamemsg_bodymsg_idcloud_custom_datachat_id chat_type chat_nameraw_text media_refs Optional[str] owner_commandz Optional[Any]rPmsg_typereply_to_message_id reply_to_textquote_media_refs media_urls media_types link_urlschannel_prompt) rrrrrZdc_fieldrarbrdrerfrgrhrirjrkrlrmrnrorprqrsrPrtrurvrwrxryrzr{rr_r]r^r^sD LLLx555J5555 DKLJJOXd333H3333FGIIHx555J5555$(M''''!F    #H""""*.----#'M''''%Xd;;;;;;; x555J5555 666K6666ht444I4444%)N((((((r_r^cJeZdZUdZdZded<edd Zdd Zdd Z dS)InboundMiddlewareaAbstract base class for all inbound pipeline middlewares. Subclasses must: - Set ``name`` as a class-level attribute (used for pipeline registration and dynamic insertion/removal). - Implement ``async handle(ctx, next_fn)`` containing the middleware logic. Convention: - Call ``await next_fn()`` to pass control to the next middleware. - Return without calling ``next_fn`` to **stop** the pipeline. rorSnamectxr^next_fnr rTrc KdS)zEProcess *ctx* and optionally call *next_fn* to continue the pipeline.Nrselfrrs r]handlezInboundMiddleware.handle r_c>K|||d{VS)zFAllow middleware instances to be called directly (duck-typing compat).N)rrs r]__call__zInboundMiddleware.__call__s,[[g.........r_c2d|jjd|jdS)N) __class__rrrs r]__repr__zInboundMiddleware.__repr__s"@4>*@@$)@@@@r_N)rr^rr rTrr) rrrrrrZr rrrrr_r]r~r~s  DNNNNTTT^T////AAAAAAr_r~cxeZdZdZddZeddZdddZddd Zddd Z ddZ e ddZ ddZ dS)InboundPipelineaOnion-model middleware pipeline engine for inbound message processing. Inspired by OpenClaw's MessagePipeline (extensions/yuanbao/src/business/ pipeline/engine.ts). Supports named middlewares, conditional guards (``when``), and ``use_before`` / ``use_after`` / ``remove`` for dynamic composition. Accepts both ``InboundMiddleware`` instances (OOP style) and plain ``async def(ctx, next_fn)`` callables (functional style) for flexibility. rTrcg|_dSr _middlewaresrs r]__init__zInboundPipeline.__init__s"$r_NcFt|tr |j|fS||fS)zHNormalize (name, handler) or (InboundMiddleware,) into (name, callable).)r<r~r) name_or_mwhandlers r] _normalizezInboundPipeline._normalizes/ j"3 4 4 /?J. .7""r_'InboundPipeline'cr|||\}}|j|||f|S)uAppend a middleware to the end of the pipeline. Accepts either: - ``pipeline.use(SomeMiddleware())`` — OOP style - ``pipeline.use("name", some_fn)`` — functional style )rrr)rrrwhenrhs r]usezInboundPipeline.uses=//*g66a   $4111 r_targetrSc|||\}}tfdt|jDd}|||f}||j|n|j|||S)zEInsert a middleware before *target* (by name). Appends if not found.c36K|]\}\}}}|k|VdSrrrrn_rs r]rz-InboundPipeline.use_before..1VV,!YaA!v++A++++VVr_Nrnextrrrinsert rrrrrrrrrs ` r] use_beforezInboundPipeline.use_befores//*g66aVVVV)D4E*F*FVVVX\]]q$ ;   $ $U + + + +   $ $S% 0 0 0 r_c|||\}}tfdt|jDd}|||f}||j|n|j|dz||S)zDInsert a middleware after *target* (by name). Appends if not found.c36K|]\}\}}}|k|VdSrrrs r]rz,InboundPipeline.use_after..rr_Nrprrs ` r] use_afterzInboundPipeline.use_afters//*g66aVVVV)D4E*F*FVVVX\]]q$ ;   $ $U + + + +   $ $S1We 4 4 4 r_rc8fd|jD|_|S)zRemove a middleware by name.c.g|]\}}}|k |||fSrr)rrrwrs r]rz*InboundPipeline.remove..s+UUU71a1PT99aAY999r_r)rrs `r]removezInboundPipeline.removes'UUUUd6GUUU r_rac$d|jDS)zAReturn ordered list of registered middleware names (for testing).cg|]\}}}| Srr)rrrs r]rz4InboundPipeline.middleware_names..#s333gaA333r_rrs r]middleware_namesz InboundPipeline.middleware_names s43!23333r_rr^cVK|jddfd d{VdS)zKRun all middlewares in order. Each middleware receives ``(ctx, next_fn)``.rrTrcKtkra\}}}dz | |s2 |d{Vn,#t$rtd|dwxYwdSdS)Nrpz'[InboundPipeline] middleware [%s] errorTexc_info)rrr:r5error)rrwhen_fnchainrindexrs r]rz(InboundPipeline.execute..next_fn,s#e**$$).u&gw &wws||&!'#w////////// LL!JD[_L```%$s A )A3Nrr)rrrrrs `@@@r]executezInboundPipeline.execute'sa!          giir_rrNN)rTr)rrSrTr)rrSrTr)rTrarr^rTr)rrrrrrrrrrrpropertyrrrr_r]rrs  %%%% ###\#                444X4 r_rcVeZdZdZdZeddZedd ZddZddZ dS)DecodeMiddlewarezDecode raw inbound frames from JSON or Protobuf into ctx.push. Encapsulates JSON push parsing (aligned with TS decodeFromContent) and Protobuf decoding via ``decode_inbound_push``. decoderaw_bodyrarTcg}|pgD]}t|ts|dp|dd}|dp|di}t|tr* t j|}n#t $rd|i}YnwxYw|||pid|S)zNormalize raw JSON msg_body array to [{"msg_type": str, "msg_content": dict}]. Compatible with both PascalCase (MsgType/MsgContent) and snake_case (msg_type/msg_content) naming. rtMsgTypero msg_content MsgContentrRrtr)r<r=rrSr loadsr:r)rritemrtrs r]convert_json_msg_bodyz&DecodeMiddleware.convert_json_msg_bodyHsN T TDdD)) xx ++Ftxx 2/F/FH((=11OTXXlB5O5OK+s++ 88"&*["9"9KK 888#);"7KKK8 MMx @QrRR S S S S s B  B10B1raw_jsonr= dict | Nonec |sdS|ddp|dd}|ddp+|ddp|dd}|dgp|d g}t|}|s|s|d sdS|d d||d dp|d d|d dp|dd||dd|ddp|dd|ddp+|ddp|dd||ddp|dd|ddp|dd|dpdt|dtr+|dpiddndd S)aConvert JSON-format push to a dict with the same structure as ``decode_inbound_push``. Supports standard callback format (callback_command + from_account + msg_body) and legacy format fields (GroupId, MsgSeq, MsgKey, MsgBody, etc.). Nrfro From_AccountrgGroupIdgroup_idrjMsgBodycallback_command to_account To_Accountri nick_namerhmsg_seqrMsgSeqrkmsg_keyMsgKeyrlCloudCustomData bot_owner_id botOwnerIdrecall_msg_seq_listlog_exttrace_id) rrfrrirgrhrrkrjrlrrr)rrrr<r=)rrfrg msg_body_rawrjs r]parse_json_pushz DecodeMiddleware.parse_json_push]s 4 LL , , 0||NB//  LLr * * ,||Ir** ,||J++  LLR ( ( +||Ir** $99,GG H X\\BT5U5U 4!) -? D D(",,|R88ZHLLWY> B  B rr^rc .K|j}|sdSd}d}|D]}||j|\}}|sEtd|jj|r|ddndh|9|}|}td|jj|t||dg} | rZddd id } |dg| gz| z|d<td |jjt| |sdS||_ ||_ td |jj|j |j d d|j dd|j ddd|j dgDt d|jj|j |d{VdS)Nroz;[%s] Push decoded but no valid message. raw hex(first64)=%sz(empty)z#[%s] Frame decoded (via=%s): len=%drj TIMTextElemrRrWrz;[%s] Merged %d extra msg_body elements from aggregated pushzC[%s] Push decoded (via=%s): from=%s group=%s msg_id=%s msg_types=%srfrgrkc:g|]}|ddS)rtror res r]rz+DecodeMiddleware.handle..s& I I IqQUU:r " " I I Ir_z[%s] Push payload: %s) rbrr_r5r6rhexrrrrdredebug) rrr data_list merged_pushrer%rdvia extra_body_seps r]rzDecodeMiddleware.handles/N   F    D++CK>>ID#  MK$$&Mdhhjj#&6&6I"" !  5K$c3t99 "XXj"55 (5vtnUUD.9ooj".M.MQUPV.VYc.cK +KKU (#j//   F% Q K co HLL , , HLLr * * HLL2 & & I ICHLLR,H,H I I I      ,ck.>IIIgiir_N)rrarTra)rr=rTr)r%rrTrr) rrrrrrrrrrrr_r]rr=s D\(/ / / \/ f*444444r_rceZdZdZdZd dZdS) ExtractFieldsMiddlewarez8Extract common fields from ctx.push into ctx attributes.zextract-fieldsrr^rTrcK|j}|dd|_|dd|_|dd|_|dd|_|dg|_|dd|_|dd|_|d{VdS) Nrfrorgrhrirjrkrl) rdrrfrgrhrirjrkrl)rrrrds r]rzExtractFieldsMiddleware.handlesx88NB77,33,33"hh'8"==xx B// XXh++ $)V >V>V!+5gg (D *4ff $   r_r c |j}|jpi}|dkr|dpg}n8|dpd}|d}|s|r||dgng}|s"td|jdS|dpd}|d pd} |D]} | dp#t| dpd} | s>||| } | | || | || p|j | } | || || | dS) Nrrrkror)rkrz2[%s] Recall callback with empty seq_list, skippingrgrf) r_rdrr5rrrdrS_find_processing_session_interrupt_for_recall_msg_content_cache_patch_transcript)rrr r_rdseq_listr|seqrgrf seq_entry recalled_id matched_skrecalled_contents r]rz$RecallGuardMiddleware._handle_recalls+x~2 0 0 0xx 566<"HH((8$$*C((9%%C=@PCP337788bH  LLMw| \ \ \ Fhh|,,299;; 006B==?? ! i iI#--11XSy9Q9Q9WUW5X5XK 66w LLJ%**7J ZYeffff#*#=#A#A+#N#N &&w ZWghhhh i ir_rrrcj|jD]\}}||kr ||jvr|cSdSr)_processing_msg_idsr_active_sessions)r_rskr|s r]rz.RecallGuardMiddleware._find_processing_session:sK288::  GBk!!bG,D&D&D tr_ session_keyc |rd|nd|}d|d|d}t|tj||||d}||j|<|j|} | | t d|j ||dd |j |d } | r| ||| ||dSdS) Nzgroup zdirect chat with u_[CRITICAL — MESSAGE RECALLED] The user message that triggered your current task (message_id="z") in ud has been recalled/withdrawn by the sender. IGNORE any prior system note asking you to finish processing tool results — the original request is void. Do NOT continue the task, do NOT call more tools, do NOT reference the recalled content. Reply only with a brief acknowledgment such as "The message has been recalled." in the language the user was using.T)rR message_typerPinternalz+[%s] Recall interrupt: msg_id=%s session=%sro) rrTEXTr_pending_messagesr!rrr5r6r_processing_msg_texts_schedule_content_redact) rr_r#rrgrfwhere recall_text synth_event active_event recalled_texts r]rz+RecallGuardMiddleware._interrupt_for_recallAs=*4[%%%%9[\9[9[ ,/: , ,CH , , , #$)$$Wj,GG    2=!+./33K@@  #       A7._redactjs G%5t<rIr?r branch_labelrs r]rz'RecallGuardMiddleware._patch_transcripts!1488  F --c.?.?Ua.b.bccnCC    NNGWZ [ [ [ FFFFF  ..s33JJ    NNGWZ [ [ [ FFFFF    Eyy&&+554 6 >.>#  99V$$..599Y3G3GK[3[3["F#=LE   # F9  `((j999 BGLR]_kllll ` ` `KW\[^________ ` F ""3n{nnn!666@@BB) )     GWbcccccsG/A A=!A88A=B C !!CC 8E?? F4 !F//F4r)rgrSrfrS)rr^r rSrTr)rrSrTrr) r#rSrrSrgrSrfrSrTr) r#rSr0rSrgrSrfrSrTrr) rrSrgrSrfrSrrrrTr)rrrrr frozensetrr;rrrrrrrr+rrr_r]rrs0 D y&&"`I&&&&   \ iiii@\ #h#h#h[#hJ!B!B!B[!BJOS:d:d:d:d[:d:d:dr_rc6eZdZdZdZedd Zdd ZdS)SkipSelfMiddlewarezFilter out bot's own messages.z skip-selfrfrSr&rrrTrUc|r|sdS||kS)z2Detect whether the message is from the bot itself.Fr)rfr&s r]_is_self_referencez%SkipSelfMiddleware._is_self_references# 6 5v%%r_rr^rcK||j|jjr-td|jj|jdS|d{VdS)Nz'[%s] Ignoring self-sent message from %s)rQrfr__bot_idr5rrrs r]rzSkipSelfMiddleware.handlesf  " "3#3S[5H I I  LLBCKDTVYVf g g g Fgiir_N)rfrSr&rrrTrUr)rrrrrrrQrrr_r]rOrOsQ(( D&&&\& r_rOceZdZdZdZd dZdS) ChatRoutingMiddlewarez9Determine chat_id, chat_type, chat_name from push fields.z chat-routingrr^rTrcK|jr*d|j|_d|_|jp|j|_n)d|j|_d|_|jp|j|_|d{VdS)Nr r r r)rgrmrnrhrorfrirs r]rzChatRoutingMiddleware.handles > D33>33CK#CMN||_||_||_||_dSr) _dm_policy_dm_allow_from _group_policy_group_allow_from)rrYrZr[r\s r]rzAccessPolicy.__init__s)$+)!1r_ sender_idrUcl|jdkrdS|jdkr||jvSdS)z?Platform-level DM inbound filter (open / allowlist / disabled).disabledF allowlistT)r^rdr_)rrbs r] is_dm_allowedzAccessPolicy.is_dm_alloweds> ?j ( (5 ?k ) )??$$(;; ;tr_rgcl|jdkrdS|jdkr||jvSdS)zGPlatform-level group chat inbound filter (open / allowlist / disabled).rdFreT)r`rdrarrgs r]is_group_allowedzAccessPolicy.is_group_allowed sB   + +5   , ,##%%)?? ?tr_c|jSr)r^rs r]rYzAccessPolicy.dm_policys r_c|jSr)r`rs r]r[zAccessPolicy.group_policys !!r_N) rYrSrZrr[rSr\rrTr)rbrSrTrU)rgrSrTrUr) rrrrrrfrirrYr[rr_r]rXrXs 2 2 2 2X"""X"""r_rXceZdZdZdZd dZdS) AccessGuardMiddlewarez.Platform-level DM/Group access control filter.z access-guardrr^rTrcK|j}|j}|jdkrI||js.t d|j|j|jdSnS|jdkrH| |j s.t d|j|j |j dS|d{VdS)Nrz'[%s] DM from %s blocked by dm_policy=%sr z([%s] Group %s blocked by group_policy=%s) r__access_policyrnrfrfr5rrrYrirgr[)rrrr_policys r]rzAccessGuardMiddleware.handle!s+&5 =D ''(899  =L#"2F4D   ]g % %**3>::  >L#.&2Egiir_Nrrrr_r]rmrms388 Dr_rmceZdZdZdZd dZdS) AutoSetHomeMiddlewareaAuto-designate the first inbound conversation as Yuanbao home channel. Triggers when no home channel is configured, or when an existing group-chat home is superseded by the first DM (direct > group upgrade). Silent: writes config.yaml and env, no user-facing message. z auto-sethomerr^rTrcK|j}|js_tjdd}| p|do |jdk}|jdkrd|_|r ddlm}ddlm }ddl }|} | d z } i} | r@t| d 5} | | pi} dddn #1swxYwY|j| d<|| | t|jtjd<t"d |j|j|jn8#t*$r+} t"d |j| Yd} ~ nd} ~ wwxYw|d{VdS)NYUANBAO_HOME_CHANNELror rTr)get_hermes_home)atomic_yaml_writez config.yamlr)encodingz=[%s] Auto-sethome: designated %s (%s) as Yuanbao home channelz[%s] Auto-sethome failed: %s)r__auto_sethome_doneosgetenvrZrnhermes_constantsruutilsrvyamlexistsopen safe_loadrmrSenvironr5r6rror:r?)rrrr_ _cur_home _should_setrurvr}_home config_path user_configfrs r]rzAutoSetHomeMiddleware.handle?s2+) T "8"==I N((22Ls}7L }$$-1* TT@@@@@@777777KKK+O--E"'-"7K(*K"))++B!+@@@BA*...*;*;*ArKBBBBBBBBBBBBBBB:=+K 67%%k;???9 AD9&C > D9 CD9CA&D99 E.!E))E.Nrrrr_r]rrrr5s9 D      r_rrceZdZdZdZdZeddZedd Zedd Z e ddZ eddZ eddZ ed dZd!dZdS)"ExtractContentMiddlewarez.Extract raw text and media refs from msg_body.zextract-contentcustomr=rTrSc|dd}|dd}|r d|d|dnd|d}|g}tj}dD]j}||}|rQt|trA#hh>P>P#hwh-*:::VY 222333  : LL8 9 9 9yyr_rrc2|d}|sdS tj|}t|tr|dnd}n#tjt f$rd}YnwxYw|rt|tsdSd|dS)zNFormat elem_type 1007 (link understanding card) into bracket-placeholder text.r6Nrz[link: z | visit link for full content])rr rr<r=JSONDecodeError TypeErrorrS)rr6parsedrs r]_format_link_understandingz3ExtractContentMiddleware._format_link_understanding{s**Y'' 4 Z((F)3FD)A)AK6::f%%%tDD$i0   DDD  :dC00 4>>>>>sAAA76A7rBc^|sdS tjtj|j}|dp|dpg}|r't |dndS#t$rYdSwxYw)zExtract resourceId from Yuanbao resource URL query parameters. Args: url: Resource URL (e.g., https://...?resourceId=abc123) Returns: Resource ID string, or empty string if not found ro resourceId resourceidr) urllibparseparse_qsurlparsequeryrrSrdr:)rBridss r]_parse_resource_idz+ExtractContentMiddleware._parse_resource_ids 2 L))&,*?*?*D*D*JKKE))L))JUYY|-D-DJC*-53s1v;;$$&&&2 5   22 sBB B,+B,rjrac  g}|D]`}|dd}|di}|dkr.|dd}|r||c|dkr|d}t|tsg}d}t |d kr$t|d t r |d }n6t |d kr#t|d t r|d }t |pid pd} || } || rd | d ndq|dkr|d|d|dd} t |d pd} || } | r&|| r d| d| d nd| d (|| rd| d ndG|dkrjt |d pd} || } || rd| d nd|dkrjt |d pd}||} || rd| d nd'|dkrw|dd}|rG tj |}t|t s|d |d!}|d"kr*||dd#n|d$kr)|| |n^|d%krC| |}|r||n+|d n|d \#tj tf$r||YwxYw|d |d&kr|dd}d}|ra tj |}|d'pd}n"#tj ttf$rYnwxYw||rd(|d nd)E|r|d*|d b|rd+|ndS),a#Extract plain text content from MsgBody. - TIMTextElem -> text field - TIMImageElem -> "[image]" / "[image|ybres:RID]" - TIMFileElem -> "[file: {filename}]" / "[file:{name}|ybres:RID]" - TIMSoundElem -> "[voice]" / "[voice|ybres:RID]" - TIMVideoFileElem -> "[video]" / "[video|ybres:RID]" - TIMFaceElem -> "[emoji: {name}]" or "[emoji]" - TIMCustomElem -> try to extract data field, otherwise "[custom message]" - Multiple elems joined with spaces rtrorrrR TIMImageElemimage_info_arrayNrprrBz [image|ybres:r[image] TIMFileElem file_namefileNamefilenamez[file:|ybres:z [file|ybres:[file: [file] TIMSoundElemz [voice|ybres:[voice]TIMVideoFileElemz [video|ybres:[video] TIMCustomElemr%z[unsupported message type] elem_typez [mention] TIMFaceElemrz[emoji: z[emoji][ )rrr<rarrr=rSrdrr rrrrrAttributeErrorr)rrjpartselemrr6rRr image_info image_urlridrfile_url sound_url video_urldata_valrctyperaw_data face_name face_datas r] _extract_textz&ExtractContentMiddleware._extract_textsL /L /D!XXj"55I HH]B77GM)){{62..'LL&&&n,,#*;;/A#B#B !"2D99*')$! '((1,,Nq>QSW3X3X.!1!!4J!1r 6 6u = = CDDJJLL ,,Y77 sI3S3333 JJJJm++";;{GKK GKKXbdfLgLg4h4hiiw{{5117R88>>@@,,X66RLLX!h!A(!A!A3!A!A!A!AShbeShShShiiiiLL(!P!68!6!6!6!6QQQQn,, E 2 2 8b99??AA ,,Y77 sI3S3333 JJJJ000 E 2 2 8b99??AA ,,Y77 sI3S3333 JJJJo--";;vr22?/!%H!5!5)&$77%!LL)EFFF$ & ; 7 7 D==!LLFK)H)HIIII"d]]!LL)@)@)H)HIIII"d]]#&#A#A&#I#ID#K % T 2 2 2 2 % -I J J J J!LL)EFFF 0)</// X...../LL!=>>>>m++";;vr22 $(Jx$8$8 %.]]6%:%:%@b$G$G$I$I  0)^L P4 4444yQQQQ / ----..."'/sxxR/s+ >Q! CQ!!+RR =T  T('T(rRcr|}|dr d|ddz}|S)zNormalize input text: strip whitespace and convert full-width slash (Chinese input method) to ASCII slash so commands are recognized correctly. ／rrpNrdrZrs r]_rewrite_slash_commandz/ExtractContentMiddleware._rewrite_slash_commands; zz|| ??8 $ $ "abb>D r_List[Dict[str, str]]cg}|pgD]S}t|ts|dd}|dipi}t|ts]|dkr|d}t|tsg}d}t |dkr$t|dtr |d}n6t |dkr#t|dtr|d}t |pid pd}|r|d |d R|d krt |d pd}t |d pdpkt |dpdp5t |dpd} |r!d|d } | r| | d<|| U|S)zExtract inbound image/file references from TIM msg_body. Return example: [{"kind": "image", "url": "https://..."}, {"kind": "file", "url": "...", "name": "a.pdf"}] rtrorrrNrprrBrL)kindrBrrrrrMr)r<r=rrarrrSrdr) rjrefsrrtr6rrrrrrefs r]_extract_inbound_media_refsz4ExtractContentMiddleware._extract_inbound_media_refssl&(N" %" %DdD)) xx B//Hhh}b117RGgt,, >))#*;;/A#B#B !"2D99*')$! '((1,,Nq>QSW3X3X.!1!!4J!1r 6 6u = = CDDJJLL EKK C CDDD=((w{{5117R88>>@@ K006B77==??B7;;z228b99??AAB7;;z228b99??AA %39(*K*KC 0&/F KK$$$ r_ctg}|pgD]}t|tr|ddkr2|dpidd}|s` tj|}n#tjt f$rYwxYwt|ts|d}|dkrC|d}|r*t|tr|||d kr|d }|r tj|}t|tr|dnd }|r*t|tr||#tjt f$rYwxYw|S) zTExtract link URLs from share-card (1010) and link-understanding (1007) custom elems.rtrrr%rorrrrr6N) r<r=rr rrrrSr) rjurlsrdata_strrrrr6rs r]_extract_link_urlsz+ExtractContentMiddleware._extract_link_urls2sN  DdD)) TXXj-A-A_-T-T//52::62FFH  H--()4    fd++ JJ{++E}}zz&))&JtS11&KK%%%$ **Y//!%G!4!45?5M5MWvzz&111SW.JtS$9$9. KK--- 0)< s%(A==BB+A,FF32F3rr^rc K|||j|_||j|_||j|_|d{VdSr)rrrjrprrqrrzrs r]rzExtractContentMiddleware.handleSst2243E3Ecl3S3STT 99#,GG// == giir_N)rr=rTrS)rr=rTrrrBrSrTrS)rjrarTrSr)rjrarTr)rjrarTrar)rrrrrrrrrrrrrrrrrr_r]rrbs88 D#   \ " ? ? ?\ ?\$[0[0[0[[0z\***\*X\@r_rc^eZdZUdZdZehdZded<eddd Z ddZ dS)PlaceholderFilterMiddlewarez>Skip pure placeholder messages (e.g. '[image]' with no media).zplaceholder-filter>[图片][文件][视频][语音]rrrrrMSKIPPABLE_PLACEHOLDERSrrRrS media_countrjrTrUcL|dkrdS|}||jvS)zEDetect whether the message is a pure placeholder (should be skipped).rF)rdr)rrRrrs r]is_skippable_placeholderz4PlaceholderFilterMiddleware.is_skippable_placeholdercs- ??5::<<3555r_rr^rcK||jt|jr-td|jj|jdS|d{VdS)Nz%[%s] Skipping placeholder message: %r)rrprrrqr5rr_rrs r]rz"PlaceholderFilterMiddleware.handleksj  ( (s3>7J7J K K  LL@#+BRTWT` a a a Fgiir_N)r)rRrSrrjrTrUr) rrrrrrMrrZrrrrr_r]rrYsHH D(1 333)) 6666[6r_rcteZdZUdZdZehdZded<edd Z e ddZ ddZ dS)OwnerCommandMiddlewarezDetect bot-owner slash commands in group chat. Identifies in-group allowlisted slash commands and determines sender identity. Owner commands skip @Bot detection; non-owner attempts are rejected. z owner-command> /q/bg/btw/new/deny/stop/undo/queue/reset/retry/approve /backgroundrM ALLOWLISTrRrSrTcr|}|dr d|ddz}|S)z?Normalize full-width slash to ASCII slash and strip whitespace.rrrpNrrs r]rz-OwnerCommandMiddleware._rewrite_slash_commands;zz|| ??8 $ $ "abb>D r_rdr=rjrarnrf)Tuple[Optional[str], Optional[str], bool]c6|dks|jsdSd|pgD}t|dkrdS|ddpidd}||}|d sdS|d d}||jvrdSt|pid pd} t| o| |k} ||| fS) a2Identify allowlisted slash commands and determine sender identity. Returns (cmd, cmd_line, is_owner): - (None, None, False): Not an allowlisted command - (cmd, cmd_line, True): Owner match - (cmd, cmd_line, False): Allowlisted command but sender is not owner r )NNFcDg|]}|ddk|S)rtrr rs r]rz@OwnerCommandMiddleware._detect_owner_command..s9   uuZ  M11 111r_rprrrRror)maxsplitr) rrrrrrZrYlowerrSrdrU) rrdrjrnrf text_elemsrRcmd_liner owner_idis_owners r]_detect_owner_commandz,OwnerCommandMiddleware._detect_owner_commands=   s} $$   B    z??a  $$1 !!-006B;;FBGG--d33""3'' %$$nnan((+1133 cm # #$$  ''77=2>>DDFF>>>h,&>Hh&&r_rr^rc K|j}||j|j|j|j\}}}|rz|sxt d|j|j |j|| tj | |j d|dd|dS|r?|r=|r;t d|j|j |j|||_||_|d{VdS)N)rdrjrnrfz;[%s] Reject non-owner slash command: chat=%s from=%s cmd=%su⚠️ z6 is only available to the creator in private chat modezyuanbao-owner-cmd-denial-rz4[%s] Bot owner slash command: chat=%s from=%s cmd=%s)r_rrdrjrnrfr5r6rrm _track_taskrrAsendrsrp)rrrr_ matched_cmdrrs r]rzOwnerCommandMiddleware.handlesH+*.*D*D\m) +E+ + ' Xx  x  KKM ck3+;[       3 S[*wK*w*w*wxx>>>!!!    F  $8 $ $ KKF ck3+;[   !,C #CLgiir_Nr) rdr=rjrarnrSrfrSrTrr) rrrrrrMrrZrrrrrrr_r]rrrs D%9&&&I \('('('[('Tr_rceZdZdZdZd dZdS) BuildSourceMiddlewarez(Build SessionSource from context fields.z build-sourcerr^rTrc K|j}||j|j|j|jpd|jp|j|jdkrdnd|_|d{VdS)Nr r)rmrnror user_namer)r_rrmrnrorfrirPrrrr_s r]rzBuildSourceMiddleware.handles+))Kmm$,)=S-= #  8 8ffd *   giir_Nrrrr_r]rrs322 D      r_rceZdZdZdZedd Zedd Zedd Zed dddZ ddZ d S)GroupAtGuardMiddlewarezIn group chat, observe non-@bot messages; only reply on @Bot. Owner commands skip @Bot detection (owner doesn't need to @Bot). zgroup-at-guardrjrar&rrrTrUct|sdS|D]}|ddkr|didd}|sI tj|}n#tjtf$rYwwxYw|ddkr|d |krd SdS) aDetect whether the message @Bot. AT element format: TIMCustomElem, msg_content.data is a JSON string: {"elem_type": 1002, "text": "@xxx", "user_id": ""} Considered @Bot when elem_type == 1002 and user_id == bot_id. Frtrrr%rorrrT)rr rrr)rjr&rrrs r] _is_at_botz!GroupAtGuardMiddleware._is_at_bots 5  Dxx ##66xx r2266vrBBH  H--()4    zz+&&$..6::i3H3HF3R3RttuA&&A?>A?rSc|sdS|D]}|ddkr|didd}|sI tj|}n#tjtf$rYwwxYw|ddkrU|d|kr`` so the model can distinguish participants and their user ids. r3Nrrrb] r5rT)r4r6robservedrGz([%s] Failed to observe group message: %s)r7r8rrrrrJrKrIr9r:r5r?r) r_rPrrRrkr= session_entryr attributedrrIs r]_observe_group_messagez-GroupAtGuardMiddleware._observe_group_message(s(!1488  F Z!77??Mn1 G@^@@g@@$@@J%%\X\:::DDFF E  -&,l#  & &(      Z Z Z NNEw|UX Y Y Y Y Y Y Y Y Y ZsBB C$!C  Crr^cfK|j}|jdkr|js||j|jse|||j|jp|j |j |j pdt d|j|j|j dS|d{VdS)Nr rz6[%s] Group message observed (no @bot): chat=%s from=%s)r_rnrsrrjrSr rPrirfrprkr5r6rrmr s r]rzGroupAtGuardMiddleware.handleJs+ =G # #C,= #dooVYVbdkdsFtFt #  ' 'S%8%LC`` to match the observed-history format. - Suppresses the runner's default ``[user_name]`` shared-thread prefix by clearing ``source.user_name``. zgroup-attributionrr^rTrcZK|jdkr|js|j}t|j|j|_|jpd}|j p|jpd}d|d|d|j |_ |j tj |j d|_ |d{VdS)Nr rrrbr)r )rnrsr_r rrjrSr{rfrirprP dataclassesr)rrrr_ user_id_labelnickname_labels r]rz!GroupAttributionMiddleware.handlehs =G # #C,= #kG!7!S!S go""C  ,9 M 0QC4DQ NP~PP PP#,PPCLz%(0tLLL giir_Nrrrr_r]r#r#Ys9   D      r_r#c6eZdZdZdZedd Zdd ZdS)ClassifyMessageTypeMiddlewarez>Determine MessageType from text content and msg_body elements.zclassify-msg-typerRrSrjrarTrc2|dr tjS|D]h}|dd}|dkrtjcS|dkrtjcS|dkrtjcS|dkrtjcSitjS)z1Classify message type based on text and msg_body.rrtrorrrr) rZrCOMMANDrPHOTOVOICEVIDEODOCUMENTr()rRrjretypes r] _classifyz'ClassifyMessageTypeMiddleware._classify}s ??3   '& & , ,DHHZ,,E&&"((((&&"((((***"(((( %%"++++&r_rr^rctK||j|j|_|d{VdSr)r1rprjrtrs r]rz$ClassifyMessageTypeMiddleware.handles:~~clCLAA giir_N)rRrSrjrarTrr)rrrrrrr1rrr_r]r)r)xsQHH D   \ r_r)c.eZdZdZdZddZdd Zdd ZdS)QuoteContextMiddlewarez3Extract quote/reply context from cloud_custom_data.z quote-contextrlrSrT#Tuple[Optional[str], Optional[str]]c|sdS tj|}n#tjtf$rYdSwxYwt |t r|dnd}t |t sdSt|dpdpd}t|dpd}t|dp|dpd}|r |r|d |n|nd}||fS) zHExtract quote text context, mapping to MessageEvent.reply_to_*. rquoteNidrodescrirbr#) r rrrr<r=rrSrd)rrlrr7quote_idr9sender quote_texts r]_extract_quote_contextz-QuoteContextMiddleware._extract_quote_contextsT! : Z 122FF$i0   :: (2&$'?'?I 7###T%&& :uyy,"--3355=599V$$*++1133UYY011QUYY{5K5KQrRRXXZZBFPV=))4)))D ##s 55rr^List[Tuple[str, str, str]]c rK|jgS|j}g} t|dd}|r|jgS||j}||j}t|pgD]}|dd}|r ||jkr&|dd} t| trd| vrt | D]} | d} | d} | d \} }}| } | t vr*|| | |fnB#t$$r5}t&d t|d d |Yd}~nd}~wwxYw|S) ayLook up the quoted message in the transcript history and return any ``[kind|ybres:RID]`` anchors found in its content as ``(rid, kind, filename)`` tuples. Returns ``[]`` when ``ctx.reply_to_message_id`` is unset, when the transcript store / source is unavailable, or when the quoted message carries no resolvable media anchors. Nr3rGror6rrprC:z'[%s] quote transcript lookup failed: %sryuanbao)rur_r7rPr8r:r9reversedrr<rS_YB_RES_REF_RErvr  partitionrd_RESOLVABLE_MEDIA_KINDSrr:r5r?)rrr_rqr=rhistoryr'r|_contentrrrrrrrIs r]#_extract_media_refs_from_transcriptz:QuoteContextMiddleware._extract_media_refs_from_transcripts  " *I+13  G%5t<>MM wwqzzggajj,0NN3,?,?)a#zz||#:::&--sD(..:J:J.KLLL    NN933S          sE5EE55 F4?+F//F4rcK||j\|_|_||d{V|_|d{VdSr)r=rlrurvrHrwrs r]rzQuoteContextMiddleware.handlesk595P5PQTQf5g5g2!2%)%M%Mc%R%RRRRRRRgiir_N)rlrSrTr5)rr^rTr>r)rrrrrr=rHrrr_r]r4r4s\== D$$$$*((((Tr_r4c\eZdZUdZdZiZded<dZded<dZded <e d/dZ e d0dZ e d1dZ e d2dZe d1dZe ddddd3dZe d4d#Ze d5d'Ze d6d(Ze d7d*Ze d8d-Zd9d.ZdS):MediaResolveMiddlewarez6Resolve inbound media references to downloadable URLs.z media-resolvez+ClassVar[Dict[str, Tuple[str, str, float]]]_resource_cacheiQ ClassVar[int]_RESOURCE_CACHE_TTL_S_RESOURCE_CACHE_MAX_SIZE resource_idrSrTOptional[Tuple[str, str]]cJ|sdS|j|}|dS|\}}}tj|z |jkr|j|ddSt j|s|j|ddS||fS)zOReturn cached ``(local_path, mime)`` if still valid and file exists, else None.N)rLrrrNrrypathisfile)rrQr local_pathmimetss r]_get_cached_resourcez+MediaResolveMiddleware._get_cached_resources 4#'' 44 =4$ D" 9;; c7 7 7   # #K 6 6 64w~~j))    # #K 6 6 644r_rVrWrc|sdStjjkrItjfd}|djdzD]}j|d||t jfj|<dS)zIStore download result in cache. Evicts oldest entries when over capacity.Nc*j|dS)NrC)rL)rrs r]z=MediaResolveMiddleware._put_cached_resource.. sCDWXYDZ[\D]r_)key)rrrLrPsortedrr)rrQrVrW sorted_keysrs` r]_put_cached_resourcez+MediaResolveMiddleware._put_cached_resources  F s" # #s'C C C !4:]:]:]:]^^^K !D3#?1#D!DE 1 1#''40000,6dikk+JK(((r_rBctj|j}tj|d}|dvr|SdS)z$Guess image extension from URL path.rp>.heic.tiff.bmp.gif.jpg.png.jpeg.webprg)rrrrTrysplitextr)rBrTexts r]_guess_image_ext_from_urlz0MediaResolveMiddleware._guess_image_ext_from_url sX|$$S)).gt$$Q'--// V V VJvr_c K|}|std|d{V}t|dpd}t|dpdpd}t|dp |jp|j}|r|std|jd }d |||d }tj d d 4d{V}tdD]M} ||d|i|d{V} | j dkr| dkrt |j|j|jd{V}t|dpd}t|dp|pdpd}t|dp |jp|j}|r|sn0||d<||d<||d<0| | } | d} | dvr)td| d| ddt#| dt$r| dn| } t| pidp| pidpd}|r|ccdddd{VStddddd{Vn#1d{VswxYwYtd )!zLow-level helper: exchange a ``resourceId`` for a direct download URL. Handles token retrieval, the ``/api/resource/v1/download`` API call, and a single 401-retry with token force-refresh. Raises on failure. zmissing resource_idNrNrorPwebr&z-missing token or bot_id for resource downloadz/api/resource/v1/downloadr)rX-IDX-TokenX-Sourcer9T)rfollow_redirectsrCr)paramsr!irrprqrrr$>Nrz"resource/v1/download failed: code=r(r'r%rBrealUrlz(resource/v1/download missing url/realUrlz)resource/v1/download did not return a URL)rdr9_get_cached_tokenrSrrS_app_key _api_domainr*r+r-r8rrW _app_secretraise_for_statusr r<r=)r_rQ token_datarNrPr&api_urlr!rCrDresprEr$r%real_urls r]_fetch_resource_urlz*MediaResolveMiddleware._fetch_resource_url s"''))  6455 5"4466666666 JNN7++1r2288::Z^^H--677==??H5Z^^H--TTGDTUU[[]] PF PNOO O(CCC.   $TDIII O O O O O O OV 88 O O#ZZ{8S]dZeeeeeeee#s**w!||'2'@'@('*=w?R((""""""J w 7 7 =2>>DDFFE !9!9!LV!LuMMSSUU^Y^F !9!9!`W_!`PWP`aaggiiF &,GFO).GI&*0GJ'%%'''))++{{6**y((&aTaaUZ\^I_I_aa/9V9L9Ld.S.S`w{{6***Y` //66[4:2:J:J9:U:U[Y[\\bbdd$#OO9 O O O O O O O O O O O O O O:##MNNN; O O O O O O O O O O O O O O O O O O O O O O O O O O O>FGGGs2H=N#N## N-0N-cK tj|}n#t$r|cYSwxYwtj|j}|dp|dpg}|r't|dnd}|s|S t ||d{VS#t$r|cYSwxYw)a"Resolve Yuanbao resource placeholder to a directly fetchable real URL. Common URL patterns: https://hunyuan.tencent.com/api/resource/download?resourceId=... Direct GET returns 401; need business API: GET /api/resource/v1/download?resourceId=... rrrroN) rrrr:rrrrSrdrKr)r_rBrr resource_idsrQs r]_resolve_download_urlz,MediaResolveMiddleware._resolve_download_urlM s  \**3//FF   JJJ  %%fl33yy..O%))L2I2IOR 6BJc,q/**00222  J /CCG[YYYYYYYY Y   JJJ s$ 336 C C&%C&Nro)rlog_tagrQ fetch_urlrrrrrcLK|rA||}|*td|j||d|S t ||jd{V\}} n;#t $r.} td|j||| Yd} ~ dSd} ~ wwxYw|dkr||} t|| } n:#t$r-} td|j|| Yd} ~ dSd} ~ wwxYwtd| } | d s| d r| nd } | || | | | fS|sEtj|}t"j|jpd } t)||} n:#t $r-} td |j|| Yd} ~ dSd} ~ wwxYwt|p| pd } | || | | | fS)aDownload a Yuanbao resource and cache locally. Returns ``(local_path, mime)`` or ``None``. When *resource_id* is provided, an in-memory cache keyed by resourceId is consulted first to skip redundant downloads of the same resource within the TTL window. Nz'[%s] resource cache hit: rid=%s path=%sr max_size_mbz5[%s] inbound media download failed: kind=%s %s err=%srL)rlz,[%s] inbound image cache rejected: %s err=%simage/ image/jpegrMz)[%s] inbound file cache failed: %s err=%sapplication/octet-stream)rYr5rrmedia_download_urlMEDIA_MAX_SIZE_MBr:r?rmrr;rrZrarrrryrTbasenamer)rr_rrrrrQhit file_bytes content_typerIrlrVrWrs r]_download_and_cachez*MediaResolveMiddleware._download_and_cachef s  **;77C =L+s1v  -?w'@...(((((( $J     NNG dGS   44444   7??// ::C 3JCHHH    BL'3ttttt   #=3==11D??8,, ['3'>'>x'H'HZ||l  $ $[*d C C Ct# # @\**955F((55?I 2:yIIJJ    NN; gs   44444   y))W\W=W   j$???4sHA'' B1#BB>C D"DD-F>> G5"G00G5rqrTuple[List[str], List[str]]c LKg}g}|D]}t|dpd}t|dpd}t|dpd}|tvs|st |} |||d{V} n;#t$r.} t d|j ||| Yd} ~ .d} ~ wwxYw| || ||pdd|dd| d{V} | h| \} }| | | |||fS) zResolve inbound media refs: download to local cache, return (local_paths, mime_types). Yuanbao COS hostnames resolve to private IPs, tripping the SSRF guard in vision_tools. We download ourselves and return local cache paths. rrorBrNz8[%s] inbound media resolve failed: kind=%s url=%s err=%szplaceholder_url=PrrrrrQ)rSrrdrrErrrr:r5r?rrr)rr_rqrxryrrrBrrrrIrRrVrWs r]_resolve_media_urlsz*MediaResolveMiddleware._resolve_media_urls s!# !#  % %Cswwv,"--3355;;==Dcggenn*++1133C3776??0b117799H222#2+==cBBC "%";";GS"I"IIIIIII    NL$S  22#"*d53ss855 3F~% J   j ) ) )   t $ $ $ $;&&s'D D<#D77D<rr> log_prefixc Kg}g}|D]\}}}|tvr |||d{V} n;#t$r.} td|j|||| Yd} ~ ad} ~ wwxYw||| ||pd|d||d{V} | | \} } || || ||fS)zQResolve a list of ``(rid, kind, filename)`` ybres tuples to local paths. Nz-[%s] %s resolve failed: rid=%s kind=%s err=%sz rid=r)rErr:r5r?rrr)rr_rr media_pathsmimesrrr fresh_urlrIrRrTrWs r]_resolve_ybres_refsz*MediaResolveMiddleware._resolve_ybres_refs s]"$ #'   Cx222 "%"9"9'3"G"GGGGGGG    CL*c4   22#"*d%11C11 3F~JD$   t $ $ $ LL    E!!s7 A/$A**A/c4Kt|dd}|sggfS ||}||j}n<#t$r/}t d|j|ggfcYd}~Sd}~wwxYw|sggfStdt|tz }g}t} ||dD](} | d} t| trd| vr2t| D]} | d} | d}| d \}}}|}|t(vrc|| vrh| |||||ft|t.krnt|t.krn*|sggfS|||d d{VS) zYResolve recent observed image/file anchors from transcript into ``(local_paths, mimes)``.r3Nz.[%s] Observed-media hydration setup failed: %srr6rrprCr@zobserved-mediar)r7r8r:r9r:r5r?rmaxrr OBSERVED_MEDIA_BACKFILL_LOOKBACKrrr<rSrCrvr rDrdrErr,OBSERVED_MEDIA_BACKFILL_MAX_RESOLVE_PER_TURNr)rr_rPr=rrFrIstartorderseenr'r6rrrrrrs r]_collect_observed_mediaz.MediaResolveMiddleware._collect_observed_media so !1488 r6M !77??M++M,DEEGG    NN@ c   r6MMMMMM    r6MAs7||&FFGG,.EE566?  Cggi((Ggs++ y/G/G#,,W55  wwqzzggajj$(NN3$7$7!azz||666$;;  c4)9)9:;;;u::!MMMEN5zzIIIJ r6M,, U'7-         s/A B$A?9B?BrwcBK|||dd{VS)zResolve media anchors carried by the quoted message. ``quote_media_refs`` is a list of ``(rid, kind, filename)`` tuples produced by :class:`QuoteContextMiddleware` from the transcript. r7rN)r)rr_rws r]_resolve_quote_mediaz+MediaResolveMiddleware._resolve_quote_media- sK,, %'-         r_rr^cg}g}|j}|s||fSt|jdd}|s||fS||}t |t r|s||fSt }t|D]}| dpd }| dpd } | r| |vrntj | s|| t!tj | p |dkrdnd} || || ||fS) u^Private-chat fallback for recovering already-local quoted media. Only already-local media is handled here: by the time a turn is cached, ``PatchAnchorsMiddleware`` has rewritten resolved ``|ybres:`` anchors to ``[image: /path]`` / ``[file: name → /path]``. Unresolved anchors are an original-turn resolution failure and belong to that turn's handling, not this quote fallback — so no re-download happens here. Returns ``(local_paths, mimes)`` for media already downloaded to the local cache on its original turn, ready to inject as-is. rNrprorCrLrr)rur7r_rr<rSr_YB_LOCAL_MEDIA_RErvr rdrryrTr~rrrr) rpathsrrid_keycacherRrrrrTrWs r]_collect_quote_local_mediaz1MediaResolveMiddleware._collect_quote_local_media: s) %<  %94@@ %< yy!!$$$ D %<  EE#,,T22  AGGAJJ$"++--3355DGGAJJ$"++--D 44<<7>>$''  HHTNNN"27#3#3D#9#9:: $ 5O  LL    LL    e|r_cB K|j}g g td  fd }|||jd{V}t d|dD}|||jQ|jr+||||jd{Vn|||no|j dkrd || ||j d{Vn8#t$r+}td |j|Yd}~nd}~wwxYw |_ |_t$|j|r(td |j|jdS|d{VdS) N pair_listsrrTrc|\}}t||D]K\}}|r|vr |||LdSr)ziprr)ru_listm_listurrtypesrs r]_add_unique_pairsz8MediaResolveMiddleware.handle.._add_unique_pairsr su'NFFFF++  1AII  A Q   r_c3K|]}|dV dS)rpNr)rrs r]rz0MediaResolveMiddleware.handle..} s'55a15555555r_rr z;[%s] observed-image hydration raised, continuing anyway: %sz.[%s] Skip placeholder after media download: %r)rrrTr)r_rrrqsumrurwrrrnrrPr:r5r?rrxryrrrpr) rrrr_r own_pairs own_countrIrrrs @@@r]rzMediaResolveMiddleware.handleg s( +EE        227CNKKKKKKKK 559Q<55555 )$$$  " .# H!!(A(A'3K_(`(`"`"`"`"`"`"`aaaa "!$"A"A#"F"FGGGG ]g % % !!(D(DWcj(Y(Y"Y"Y"Y"Y"Y"YZZZZ   QL#   ' ? ? i X X  LLI7rrSrTr)rTr)rwr>rTr)rr^rTrr)rrrrrrLrZrNrPrrYrarrmrrrrrrrrrrr_r]rKrKs@@ DDFOEEEE+77777.11111   [ " K K K[ K\8H8H8H\8Ht\0$(= = = = = [= ~,',','[,'\#"#"#"[#"J/ / / [/ b    [  ***\*X;;;;;;r_rKc6eZdZdZdZedd Zdd ZdS)PatchAnchorsMiddlewarea,Replace ``[kind|ybres:RID]`` anchors in ``ctx.raw_text`` with local paths. Runs after :class:`MediaResolveMiddleware` so that ``ctx.media_urls`` / ``ctx.media_types`` are already populated with downloaded resources (own media + quote media or group-observed media). The transcript written downstream then records usable local paths for the model instead of opaque ``ybres:`` references. Only resolved media (paths starting with ``/``) are substituted; any anchor without a corresponding local resource is left untouched. z patch-anchorsrRrSr List[str]rrTcv|r|s|S|}t||D]\}}|dst|}|sn|d}|d\}} } |}|dkr|drd|d} nD|dkr=| ptj |} d | d |d} n|d| | z|| dz} |S) Nrrpr@rLrz[image: rrMru → ) rrZrCsearchr rDrdryrTrrrw) rRrrpatchedrr anchor_matchrrrr replacementlabels r]_patchzPatchAnchorsMiddleware._patch sh 4 Ke$$  DAq<<$$ )0099L %%a((D $s 3 3 D!X::<{{3<QQ giir_N)rRrSrrrrrTrSr)rrrrrrrrrr_r]rr sW   D\6r_rc6eZdZdZdZddZedd Zd S)DispatchMiddlewarez.Build MessageEvent and dispatch to AI handler.dispatchrr^rTrc Kj tj jjdd jjddd fd }jd kr jv} jtj }| |t d j|pd dd |rptj| d pd dd} j|| jjnatj|djpd} j|| jj|d{VdS)Ngroup_sessions_per_userTthread_sessions_per_userF)rrrTrc ~KtjtdjDr tjnjjjpdj tj tjj j j }r'jr jj<jpdj<jr]jrVj}j|j<t%|dkr-t|dt%|dz D]}||=|d{VdS)Nc3@K|]}|dVdS))z application/ztext/NrZ)rmts r]rzMDispatchMiddleware.handle.._dispatch_inbound_event.. s/^^2==)BCC^^^^^^r_) rRr%rPrG raw_messagerxryrurvr{ror")rrpanyryrr/rtrPrkrdrarxrurvr{r r*rrrhandle_message)eventrr_skr_rs r]_dispatch_inbound_eventz:DispatchMiddleware.handle.._dispatch_inbound_event sb \^^co^^^^^&K((z:-H// 11$'$;!/"1E  Hsz H36:+C058\5GR-c2z %cl %2$'Lcj!u::##!%[[):#e**s*:):;%%!!HH((// / / / / / / / / /r_r z-[%s] Group message enqueued (qsize=%d) for %srorNzyuanbao-group-consumer-r'rzyuanbao-inbound-rr)r_r7rPconfigextrarrn _group_queues setdefaultrQueue put_nowaitr5r6rqsizerA_consume_group_queue_inbound_tasksrrCrDrk) rrrris_newqueueconsumerrErr_s ` @@r]rzDispatchMiddleware.handle s+ J$+N$8$<$<=VX\$]$]%,^%9%=%=>XZ_%`%`     0 0 0 0 0 0 0 08 =G # # 55F)44S'-//JJE   4 5 5 5 KK? ekkmmciR"-=    K".--gs;;ECI2ss3CEE&**8444**7+A+IJJJ&''))A (?iAAD  " & &t , , ,  " "7#9#A B B Bgiir_r_'YuanbaoAdapter'r#rScKd}|j|}|sdS tj||d{V}n#tj$rYnwxYwt d|j|pddd| |d{V||j vr#tj dd{V||j v#n0#t$r#t d |jYnwxYw |j |ddS#|j |dwxYw) zIDrain the group queue one dispatch at a time, waiting for each to finish.rJNTrz3[%s] Group queue: dispatching for %s (remaining=%d)rorNg?z[%s] Group queue consumer error)rrrwait_for TimeoutErrorr5rrrr!rAr: exceptionr)r_r# _IDLE_TIMEOUTr dispatch_fns r]rz'DispatchMiddleware._consume_group_queue s %))+66  F 9 V(/(8m(\(\(\"\"\"\"\"\"\KK+E IL;#4"crc":EKKMMV%+--'''''''%)AAA%mC000000000&)AAA VVV$$%F UUUUUV V  ! % %k4 8 8 8 8 8G ! % %k4 8 8 8 8sSD5.AD5A'$D5&A''AD5)>%,,.. , ,C88:: ,!!(+++   """ !!!           s! 2AA/ A==BBr{r=c :K|j}|jdS|dd}|jp|dd}|dpd}|jp|ddpd}t t j}td ||||ttt| }|j |d{Vtd |j|| t!j} | t&z} | | z } | d kr"td|jdSt!j|j| d{V} t/| t0t2fs t5t1| } n#t6$rYwxYw| di}|dd}|dd}|t8dkri|dkrc|| }|r*||_td|j|d Std|jdS{#t j $r$td|jYdSt6$r.}td|j|d Yd}~dSd}~wwxYw)zSend AUTH_BIND and read frames until BIND_ACK is received. Returns True on success, False on failure/timeout. NFrNror&rPbotrybBot) biz_iduidrPrNrk app_versionoperation_system bot_versionrz&[%s] AUTH_BIND sent (msg_id=%s uid=%s)Trz+[%s] AUTH_BIND timeout waiting for BIND_ACKrrcmd_typerar Responsez auth-bindz$[%s] BIND_ACK received: connectId=%sz[%s] BIND_ACK missing connectIdz[%s] AUTH_BIND timeoutz[%s] AUTH_BIND error: %sr)!rrrrSrrSuuiduuid4r-r1r2r4rr5rrrr!rAUTH_TIMEOUT_SECONDSrrrecvr<r bytearrayr)r:r!_extract_connect_idrr6r)rr{r_rNr6rPrrk auth_bytesr"deadlinerrawr'rr:r rrIs r]rzConnectionManager._authenticate s% - 8 5w++o="!=!=))2U&O*..b*I*IOR TZ\\""%$.$    hmmJ''''''''' =w|VUXYYY% ,..Ezz||&::H %$uzz||3 >>LL!NPWP\]]] 5#,TX]]__iPPPPPPPPP!#y'9::)%**55CC Hwwvr**88J33hhub))x 333{8J8J!%!9!9#!>!>J!%+5( $JGLZdeee#t %F UUU$u7 %:#    LL17< @ @ @55    LL3W\3QUL V V V55555 sWA-J15AJ1G#"J1# G0-J1/G00BJ1 J1/J11/L# L,#LL decoded_msgc|dd}|sdS tt|}t|d}|dkr9t |d}t d|jj||dSt |d}|r|ndS#t$r1}t d |jj|Yd}~dSd}~wwxYw) z0Extract connectId from decoded BIND_ACK message.r%r_NrprrCz*[%s] AuthBindRsp error: code=%d message=%rrz$[%s] Failed to extract connectId: %s) rr"r%r$r#r5rrrr:r?)rrEr%fdictr$messagerrIs r]rAz%ConnectionManager._extract_connect_idY s!oofc22 4 #M$$7$788Eua((Dqyy%eQ// @M&gt$UA..J!+5:: 5    NNA4=CUWZ [ [ [44444 sA)BB C'&CCcK|j} |jrWtjtd{V|j/ t tj}t|}tj }| }||_ ||j |<|j|d{Vtd|j| tj|dd{Vd|_n#tj$r|j |d|xjdz c_td|j|jt.|jt.krYtd|j|Y|j |dd|_ dSYnwxYw|j |dd|_ n'#|j |dd|_ wxYwn8#t2$r+}td |j|Yd}~nd}~wwxYw|jUdSdS#tj$rYdSwxYw) zJSend HEARTBEAT (ping) every 30s; trigger reconnect after threshold misses.Nz[%s] PING sent (msg_id=%s)r:rrrpz[%s] PONG timeout (%d/%d)z7[%s] Heartbeat threshold exceeded, triggering reconnectz[%s] Heartbeat send failed: %s)r_runningrrAHEARTBEAT_INTERVAL_SECONDSrrSr<r=r.r! create_futurerrrr5rrrrrrr?HEARTBEAT_TIMEOUT_THRESHOLDschedule_reconnectr:r+)rr_rk ping_bytesloop pong_futurerIs r]r#z!ConnectionManager._heartbeat_loopp s-" " Vm$>?????????8#V ..F!,V!4!4J"355D262D2D2F2FK)4D&1I?c K|j} |j23d{V}t|ttfs$|t|d{VM6dS#t j$rYdStj j $r}t|dd}t d|j|t|dd|r?|tvr6td|j||n|Yd}~dSYd}~dSd}~wt&$r@}t d|j||Yd}~dSd}~wwxYw)z(Read WS frames and dispatch by cmd_type.Nr$z3[%s] WebSocket connection closed: code=%s reason=%sreasonroz7[%s] Close code %d is non-recoverable, NOT reconnectingz[%s] receive_loop exited: %s)rrr<rr@ _handle_framerr+r exceptionsConnectionClosedr7r5r?rNO_RECONNECT_CLOSE_CODESr_mark_disconnectedrNr:)rr_rD close_exc close_coderIs r]r$zConnectionManager._receive_loop s- &!X 5 5 5 5 5 5 5c!#y'9::((s4444444444&XX%    DD$5 * * * FD99J NNE j')Xr*J*J    *j,DDD ML***,,,,'')))))))))-,,,,, & & & NN97< M M M  # # % % % % % % % % % &s5A"AAA""E74E7BD** E775E22E7rDrc .K|j} t|}n9#t$r,}td|j|Yd}~dSd}~wwxYw|di}|dd}|dd}|dd}|d d } |d d } |td kr|dkrtd|j||j4|j s|j dnN|rL||j vrC|j |} | s| ddS|td kr(|dvr$td|j||dS|td kr~|rX||j vrO|j |} | s d|i} | r| | d <| | n"td|j||dS|tdkr}t d|j||t| | rp|ji t!|} |j| d{Vn8#t$r+}td|j|Yd}~nd}~wwxYw|r||j vr|j |} | sX | rt%| nd|i}| |n,#t$r}| |Yd}~nd}~wwxYwdS| rDt d|j|t| || dStd|j|||dS)z Handle a single WebSocket frame.z[%s] Failed to decode frame: %sNrr:rar rorkneed_ackFr%r_r;pingz'[%s] HEARTBEAT_ACK received (msg_id=%s)T>send_group_heartbeatsend_private_heartbeatz-[%s] Heartbeat ACK received: cmd=%s msg_id=%sz)[%s] Unmatched Response: cmd=%s msg_id=%sPushz0[%s] Push received: cmd=%s msg_id=%s data_len=%dz[%s] Failed to send PushAck: %szL[%s] WS received inbound push, decoding and dispatching: cmd=%s, data_len=%dz1[%s] Ignoring frame: cmd_type=%d cmd=%s msg_id=%s)rr)r:r5rrrr!rr- set_resultrrr6rrrr/rr*r._push_to_inbound)rrDr_r'rIrr:r rkr\r%r0r ack_bytesack_excdecodeds r]rTzConnectionManager._handle_frame s- !#&&CC    LL:GL# N N N FFFFF wwvr""88J++hhub!!(B''88J..ggfc** x + + +v LLBGLRX Y Y Y!-d6H6M6M6O6O-"--d3333 )Fd&888(,,V44xxzz)NN4((( F x + + +8 1 1  LLH',X[]c d d d F x + + + &D$666(,,V44xxzz+$d^F.)-vNN6*** ?L#v F x' ' ' KKJGLZ]_egjkogpgp q q q [DH0[ / 5 5I(-- 2222222222 [[[LL!BGLRYZZZZZZZZ[ &D$666(,,V44xxzz//?C"W"5d";";";&RVw////$///))#......../ , bL#s4yy%%d+++ F ? L(C     sD A!A  A//K L)!LL*M<< N%N  N%g?float_DEBOUNCE_WINDOWrrScT tj|d}t|trw|ddp|dd}|ddp+|ddp|dd}|r|d|Sn#t $rYnwxYw t|}|r/|ddd|ddSn#t $rYnwxYwd t|S) zLightweight decode to extract sender key for debounce grouping. Returns 'from_account:group_code' or a fallback unique key. rrfrorrgrrr@ __unknown_) r rrr<r=rr:r*r8)rrrrfrgrds r]_extract_sender_keyz%ConnectionManager._extract_sender_key sk  Z 8 899F&$'' :JJ~r226zz."55 JJ|R002zz)R002zz*b11  :*99Z999    D  &x00D V((>266UU,PR9S9SUUU V    D +BxLL***s$B2B66 CC?D DDc ||}|j|d}|r|||jvr g|j|<|j||t d|jj |t|j|tj }| |j|j|}||j|<dS)a=Debounced inbound dispatch. Buffers raw frames from the same sender within a short time window, then dispatches all buffered data as a single aggregated pipeline execution. This merges multi-part messages (e.g. image + text sent as separate WS pushes) into one pipeline run. Nz2[%s] Debounce: buffered frame for key=%s, count=%d)rjrrr*rrr5rrrrrrr! call_laterrg_flush_inbound_buffer)rrr]existing_timerrPtimers r]rbz"ConnectionManager._push_to_inbound. s&&x00-11#t<<  $  ! ! # # # d* * *(*D  % S!((222 @ M S)=c)B%C%C   '))  !  &    %*S!!!r_r]c|j|d|j|g}|sdS|j}td|j|t|t||}| tj |j |d|dS)uCFlush the debounce buffer for a given key — execute the pipeline.Nz1[%s] Debounce flush: key=%s, aggregated %d frames)r_rbzyuanbao-pipeline-r)rrrrr5r6rrrr^rrrA_inbound_pipeliner)rr]rr_rs r]rmz'ConnectionManager._flush_inbound_bufferP s   d+++(,,S"55   F- ? L#s9~~   WCCCG/  % - -c 2 2*S**        r_encoded_conn_msgreq_idrcK|jtdtj}|}||j|< |j|d{Vtjtj||d{V}||j |dS#tj $rt$rwxYw#|j |dwxYw)a Send a business-layer request and wait for the response. 1. Register a Future in pending_acks[req_id] 2. Send encoded_conn_msg (bytes) to WS 3. asyncio.wait_for(future, timeout) 4. Clean up pending_acks on timeout/exception N Not connectedr) rr9rr!rLrrrshieldrrr:)rrrrsrrPfuturers r]send_biz_requestz"ConnectionManager.send_biz_requestf s" 8 // /'))!%!3!3!5!5%+6" 1(-- 011 1 1 1 1 1 1 1"+GN6,B,BGTTTTTTTTTF   " "64 0 0 0 0 #            " "64 0 0 0 0s AB66CCC1c|jjr/|js*tj|dSdSdS)zBSchedule a reconnect only if running and not already reconnecting.N)rrJrrrA_reconnect_with_backoffrs r]rNz$ConnectionManager.schedule_reconnect sV = ! @$*< @   < < > > ? ? ? ? ? @ @ @ @r_cK|jr'td|jjdSd|_ |d{V d|_S#d|_wxYw)u?Reconnect with exponential backoff (1s, 2s, 4s, … up to 60s).z,[%s] Reconnect already in progress, skippingFTN)rr5rrr _do_reconnectrs r]rzz)ConnectionManager._reconnect_with_backoff s    LLGI[ \ \ \5! '++-------- -!&D  D  & & & &s A A$c K|j}ttD] }|dz|_t d|zd}t d|j|dzt|tj |d{V| d{V t |j |j|j|jd{V}|drt%|d|_tjt+j|jddd t0 d{V|_||d{V}|s@t d |j|dz| d{Vd |_d |_||jr2|js|j tj!|"d |j#|_|j$r2|j$s|j$ tj!|%d|j#|_$t d|j|dz|j#dS#tj&$r(t d|j|dzYtN$r0}t d|j|dz|Yd}~d}~wwxYwt (d|jt|)dS)z>Internal reconnect loop, called under the _reconnecting guard.rprCrz#[%s] Reconnect attempt %d/%d in %dsNrr&rrrz![%s] Re-auth failed on attempt %drrrrz,[%s] Reconnected on attempt %d. connectId=%sTz#[%s] Reconnect attempt %d timed outz$[%s] Reconnect attempt %d failed: %sz*[%s] Giving up after %d reconnect attemptsF)*rr-MAX_RECONNECT_ATTEMPTSrminr5r6rrrArrrWrwryrxrrrSrSrrrrrrrr?rr rr-r*rAr#rrr$rr:rrX)rr_rDwaitr{r(rIs r]r|zConnectionManager._do_reconnect s-344@ @ G'.{D $qG|R((D KK5 gk+A4   -%% % % % % % % %""$$ $ $ $ $ $ $ $5 #.#<#<$g&97;N%0$=$$ >>(++@&)*X*>&?&?GO!(!1&&*%)&'  4""" $11*========NN#F V]`aVabbb**,,,,,,,,,+,(01-'')))'20D0I0I0K0K2(//111'.':((**@d.>@@((($ ?-4?+?+?+A+A-O**,,,")"5&&((;)9;;###  BL'A+t/?tt' a a aDglT[^_T_`````   :GL'TU+WZ   8',H^    ""$$$us&DK#D(K3L= L= %L88L=c0K|j}d|_| tj|td{VdS#tj$r/t d|jj tYdSt$rYdSwxYwdS)zClose and clear the WebSocket connection, bounded by ``WS_CLOSE_TIMEOUT_S`` so an unresponsive server can't stall teardown (see the constant's definition for the full rationale).Nru>[%s] WS close handshake exceeded %.1fs — dropping connection) rrrr1WS_CLOSE_TIMEOUT_Srr5rrrr:)rrs r]rzConnectionManager._cleanup_ws sX > &rxxzz;MNNNNNNNNNNNN'    TM&(:     >s3A :B BBNr_rrTr)rTrrrXrTrUr)r{r=rTrU)rEr=rTrr)rDrrTr)rrrTrS)rrrTr)r]rSrTr)rrrrsrSrrfrTr=)rrrrrrrrrr rr1rrAr#r$rTrgrZrjrbrmDEFAULT_SEND_TIMEOUTrxrNrzr|rrr_r]rrY s B B B B"X   X (((X(   X ````D!!!!BCCCCJ.%%%%R&&&&8T T T T p"!!!!++++> * * * *D    4. 11111@@@@@ ' ' ' 'IIIIVr_rcXeZdZdZeddZedd ZddZ dddZdS)MediaSendHandleru{Abstract base class for media send strategies. Subclasses implement: - acquire_file(): how to obtain file bytes (download URL / read local) - build_msg_body(): how to build TIMxxxElem from upload result The shared flow (check ws → cancel notifier → validate → COS upload → lock → dispatch) is handled by the base handle() template method. r_rkwargsr rTTuple[bytes, str, str]c KdS)zReturn (file_bytes, filename, content_type). Raises: ValueError: when file cannot be acquired (not found, empty, etc.) Nrrr_rs r] acquire_filezMediaSendHandler.acquire_filerr_ upload_resultr=rac dS)zr file_uuidrr)rrrs r] z+MediaSendHandler.handle..Rs4!Q IIIqIIIr_)rrrrrRrrgrgz[%s] %s.handle() failed: %sr) _connection _outboundr;rrcancel_slow_notifierrr MessageSendervalidate_mediarr rvrrSrrwrxrrrrrdispatch_msg_bodyr;rSr:typerr5rr)rr_rmrrrconnr;rrrvalidation_errrr{rNr&rr fwd_kwargsrjgcverI handler_names r]rzMediaSendHandler.handles"") 7?e?dSSS S..w777P =7Ht7H88!88222222 .J,$$&& K!.!=!='*C"""K%e>JJJJ$$&&, =#J// $+#<#<#>#>>>>>>> '^^GR88NN8R00IGOIr%8#,&2%!%0 %%% '4)%!- +&|4&x0 '''!!!!!! %+\\^^ /4.!'%!-  ! /4.r<< <<== = 7zRRR *d # # "qJ " " " " " " " " " " " " " " "7##J//>;&x00@L 8\11-BBBc t|d|d|d|d|dd|dd|d Srrrs r]rzImageFileHandler.build_msg_bodyrr_Nrrr_r]rrs8VV 2 2 2      r_rceZdZdZdZdZdS)FileUrlHandleruBStrategy: send file from a URL (download → COS → TIMFileElem).cxK|d}td|j|t||jd{V\}}|d}|s<|dd}tj |pd}|r|dkrt|pd}|||fS) Nrz#[%s] FileUrlHandler: downloading %srrrrrMr) r5r6rrrrrYryrTrr)rr_rrrrrrs r]rzFileUrlHandler.acquire_filesz* 97<RRR); '";* * * $ $ $ $ $ $  L::j)) = s++A.Iw'' 22ADD0E< E<!E77E<rr"offsetrjlimitcK|j}|jjdSt|||}ddlm}||}|dd} |j||d{V} | di} | dd} | dkr#t d |j | dS| d d p| d d } | r%t| trt| } ngdd d} | r8| dr#tj| df|j|<| S#t j$r%t d|j |YdSt$$r,}t d|j |Yd}~dSd}~wwxYw)zQuery group member list via WS. Returns: Decoded dict or None on failure. Also populates adapter._member_cache. Nrrrrrrkrrz,[%s] get_group_member_list failed: status=%dr%r_rGT)members next_offset is_completerz,[%s] get_group_member_list timeout: group=%sz%[%s] get_group_member_list failed: %s)rrrr5rr)rxrr5r?rr<rr,r _member_cacherrr:)rrgrrr_rrrersrFrrrrrIs r]get_group_member_list_rawz+GroupQueryService.get_group_member_list_raw+s-   ! )4.z&PUVVVNNNNNN''""* $0AA'RXAYYYYYYYYH<<++DXXh**F{{Mw|]cdddt||FC00MHLL4M4MH PJx77 P9(CC%'$OO U&**Y// U59Y[[&BS4T%j1M#    NNI7 F>!F99F>rmr=cK|dsddiS|tdd}||d{V}|ddiS|S)zAI tool: Query current group info. No parameters needed (group_code extracted from session context). Returns group name, owner, member count, etc. r r-This command is only available in group chatsNzFailed to query group info)rZrrr)rrmrgrs r]query_group_infoz"GroupQueryService.query_group_infoTsz !!(++ NLM MS]]^^, 00<<<<<<<< >9: : r_list_allNactionrrrc K|dsddiS|tdd}||d{V}|ddiS|dg}|dkr%|r#| fd|D}n|d kr d |D}d }|r7t|d kr$d |D}dd|z}|ddt||dS)a\AI tool: Query group member list. Args: chat_id: Chat ID (extracted from session context) action: 'find' (search by name) | 'list_bots' (list bots) | 'list_all' (list all) name: Search keyword when action='find' Returns: {"members": [...], "total": int, "mentionHint": str} r rrNzFailed to query group membersrfindcg|]}|ddpdvsX|ddpdvs,|ddpdv|S)nicknamero name_cardrrr)rrrs r]rz;GroupQueryService.query_session_members..}sQUU:r228b??AAAAQUU;339r@@BBBBQUU9b117R>>@@@@A@@r_ list_botscjg|]0}d|ddpdv.|1S)r3rrorrrs r]rz;GroupQueryService.query_session_members..sB\\\QUquuZ7L7L7RPR6Y6Y6[6[-[-[q-[-[-[r_ro cg|]B}|dp*|dp|ddCS)rrrror rs r]rz;GroupQueryService.query_session_members..sJfffYZQUU;''T155+<+<TiQS@T@Tfffr_zMention with @name: z, rN)rtotal mentionHint)rZrrrrrr) rrmrrrgrr mention_hintnamesrs @r]query_session_membersz'GroupQueryService.query_session_membersbsb !!(++ NLM MS]]^^, 55jAAAAAAAA ><= =**Y++ V   JJLLE"GG { " "\\'\\\G  Es7||r))ff^efffE1DIIe4D4DDLss|\\'   r_rrgrSrTrcrr"rgrSrrjrrjrTrc)rmrSrTr=)rN)rmrSrrSrrrrTr=) rrrrrrrrr rr_r]rrs    @>A#####R    "!" . . . . . . . r_rcDeZdZdZddZdd Zdd Zdd ZdddZddZ dS)HeartbeatManagerzManages reply heartbeat (RUNNING / FINISH) lifecycle. Responsibilities: - Periodic RUNNING heartbeat sender (every 2s) - Auto-FINISH after 30s inactivity - Explicit stop with optional FINISH signal r_rrTrc0||_i|_i|_dSr)r_reply_heartbeat_tasks_reply_hb_last_activers r]rzHeartbeatManager.__init__s ?A#79"""r_rmrS heartbeat_valrjc<K|j}|j}|j|jsdS |dr/|t dd}t |j||}n,|d}t|j||}|j |d{V|tkrdnd}t d|j ||dS#t$r,} t d |j | Yd} ~ dSd} ~ wwxYw) z9Send a single heartbeat (RUNNING or FINISH), best effort.Nr )rfrg heartbeatr )rfrrRUNNINGFINISHz%[%s] Reply heartbeat %s sent: chat=%sz#[%s] send_heartbeat_once failed: %s)rrrrSrZrrr3 removeprefixr2rr&r5rrr:) rrmrr_rrgrr status_namerIs r]send_heartbeat_oncez$HeartbeatManager.send_heartbeat_oncese-" 7?'/? F S!!(++ $S]]^^4 5!()+ %11)<< 7!()+ ',,w'' ' ' ' ' ' ' ''48L'L'L))RZK LL7 k7      S S S LL> c R R R R R R R R R SsCC%% D/!DDcK|j}|j}|j|jsdS|j|}|r1|stj|j|<dStj|j|<tj | |d|}||j|<dS)zGStart or renew the Reply Heartbeat periodic sender (RUNNING, every 2s).Nzyuanbao-reply-hb-r) rrrrSrrr-rrrrA_worker)rrmr_rexistingrEs r]rzHeartbeatManager.starts-" 7?'/? F.227;;  HMMOO 26)++D &w / F.2ikk"7+" LL ! !.W..   04#G,,,r_ctK ||td{V tjtd{V|j|d}tj|z tkrn6|j j }|j n"||td{Vd}n$#tj $rd}Ynt$rd}YnwxYw|s3 ||td{Vn#t$rYnwxYw|j|d|j|ddS#|s3 ||td{Vn#t$rYnwxYw|j|d|j|dwxYw)ztBackground coroutine: send RUNNING heartbeat every 2s. 30s without renewal -> send FINISH and exit. NTrF)rr&rrAREPLY_HEARTBEAT_INTERVAL_SrrrREPLY_HEARTBEAT_TIMEOUT_Srrrr+r:r'rr)rrm last_activer cancelleds r]rzHeartbeatManager._workersF :**74HII I I I I I I I Nm$>?????????"8<>  D99;;   #))+++ "((*****r_Nr)rmrSrrjrTrrmrSrTr)TrmrSr#rUrTrr) rrrrrrrrr%r1rr_r]rrs:::: SSSS<4444(!:!:!:!:F     ++++++r_rc:eZdZdZddZdd Zdd Zdd ZddZdS)SlowResponseNotifierzManages delayed 'please wait' notifications for slow agent responses. Starts a timer per chat_id; if the agent hasn't replied within SLOW_RESPONSE_TIMEOUT_S seconds, sends a courtesy message. r_rr;'MessageSender'rTrc0||_||_i|_dSr)r_sender_tasks)rr_r;s r]rzSlowResponseNotifier.__init__s  /1 r_rmrScK||tj||d|}||j|<dS)zCStart a delayed task that notifies the user when the agent is slow.zyuanbao-slow-resp-rN)r*rrA _notifierr/rrmrEs r]rzSlowResponseNotifier.starts] G" NN7 # #/g//    $ Gr_cK tjtd{Vtd|jjtt||j |td{VdS#tj $rYdSt$r1}t d|jj|Yd}~dSd}~wwxYw)z@Wait SLOW_RESPONSE_TIMEOUT_S, then push a 'please wait' message.Nz<[%s] Agent response exceeded %ds for %s, sending wait noticez&[%s] Slow-response notifier failed: %s)rrASLOW_RESPONSE_TIMEOUT_Sr5r6rrrjr.send_text_chunkSLOW_RESPONSE_MESSAGEr+r:r)rrmrIs r]r1zSlowResponseNotifier._notifier#s \- 788 8 8 8 8 8 8 8 KKN "C(?$@$@'   ,..w8MNN N N N N N N N N N%    DD \ \ \ LLA4=CUWZ [ [ [ [ [ [ [ [ [ \sA>BC C&C  Cc|j|d}|r*|s|dSdSdS)z@Cancel the pending slow-response notifier for *chat_id*, if any.N)r/rr-r*r2s r]r*zSlowResponseNotifier.cancel1sS{w--     KKMMMMM    r_cKt|jD]*}|s|+|jdS)zCancel all slow-response tasks.N)rar/r,r-r*rr's r]r1zSlowResponseNotifier.close7sc++--..  D99;;   r_N)r_rr;r,rTrr(r) rrrrrrr1r*r1rr_r]r+r+s 2222 $$$$ \ \ \ \ r_r+cveZdZUdZehdZded<dZded<dId ZdJdZ dKdZ dLdMdZ dNdOd!Z dPdQd&Z dLdRd)Z dSdTd.ZdUdVd1Z dPdWd2Zejd3ejZdXd4ZdUdYd5Z dPdZd6Zed[d:Ze d\d]d@Ze d^d_dFZed`dGZdadHZdS)bracCore message sending dispatcher for YuanbaoAdapter. Responsibilities: - Per-chat-id lock management (serial send ordering) - Text chunk sending with retry - C2C / Group message encoding and dispatch - Media send helpers (image, file, sticker, document) - Direct send helper (text + media, used by send_message tool) >rerfrgrhrirjzClassVar[frozenset] IMAGE_EXTSrrMCHAT_DICT_MAX_SIZEr_rrTrc||_tj|_d|_d|_t ttttd|_ dS)N)r image_filerrr) r collections OrderedDict _chat_locks_on_send_start_on_send_finishrrrrr_media_handlersrs r]rzMessageSender.__init__Msq GRG^G`G`?C?C)***,,&(('))%'' = = r_rrSrrc||j|<dS)z1Register (or replace) a named media send handler.N)rC)rrrs r]register_handlerzMessageSender.register_handler`s%,T"""r_rmrc||jvr'|j||j|St|j|jkrd}t |jD]?}|j|s|j|d}n@|s9|jtt|jtj |j|<|j|S)z=Return (or create) a per-chat-id lock with safe LRU eviction.FT) r@ move_to_endrrr;ralockedrriterrr)rrmevictedr]s r] get_chat_lockzMessageSender.get_chat_lockfs d& & &   ( ( 1 1 1#G, , t D$; ; ;GD,--  ',3355$((---"GE C $$T$t/?*@*@%A%ABBB$+LNN!((r_Nror6rrrrgrc HK|j}|j}|jtdddS|jr||||}|4d{V||}|||j} t d|j t||jt| d| Dt| D]K\} } | dkr|nd} ||| | | d{V} | js| ccdddd{VSL dddd{Vn#1d{VswxYwY|jr- ||d{Vn#t"$rYnwxYwtd S) zHSend text message with auto-chunking and per-chat-id ordering guarantee.NFruTrzJ[%s] truncate_message: input=%d chars, max=%d, output=%d chunk(s) sizes=%sc,g|]}t|Srrrrs r]rz+MessageSender.send_text..s555c!ff555r_rr)r)rrrrrArKstrip_cron_wrappertruncate_messageMAX_TEXT_CHUNKr5r6rrrrr5rrBr:)rrmr6rrgr_rlockcontent_to_sendrrrr_tors r] send_textzMessageSender.send_textys-" 7?e?dSSS S   )    ( ( (!!'** " " " " " " " ""55g>>O**?G >>>>>>>K& N!L9J!L!LMM&1%6B T T !J z"")//11Cdo%%$+$;$;GZ$P$PPPPPPP $+$9$9':$N$NNNNNNN & T!R{?P!R!RSSSS T  UV V!4?I+00T    r_rjrac@K||}|4d{V|dr5|tdd}||||d{V}n3|d}||||d{V} dddd{Vn#1d{VswxYwY|dr$td|dStd |d d  S) z5Lock + dispatch an arbitrary MsgBody to C2C or group.Nr r rrTrrrGFr Unknown errorr)rKrZrrsend_group_msg_bodyrsend_c2c_msg_bodyrr) rrmrjrrgrRgrprrs r]rzMessageSender.dispatch_msg_bodys!!'** c c c c c c c c!!(++ cc(mmnn-#77XxPPPPPPPP$11)<< #55j(Wa5bbbbbbbbb  c c c c c c c c c c c c c c c c c c c c c c c c c c c ::i  Ndvzz)7L7LMMM M%vzz'?/S/STTTTsA>B11 B;>B;rrRretryrjc |K|j}d}t|D]k} |dr5|tdd} || ||d{V} n3|d} || ||d{V} | dr&td| d cS| d d}t d |j |d z||nL#t$r?} t| }t d |j |d z||Yd} ~ nd} ~ wwxYw||d z krtjd|zd{Vmtd|j ||tdd|S)zFSend a single text chunk with retry (exponential backoff: 1s, 2s, 4s).rgr Nr rrTrrfrz-[%s] send_text_chunk attempt %d/%d failed: %srpz0[%s] send_text_chunk attempt %d/%d exception: %srCz>[%s] send_text_chunk max retries (%d) exceeded. Last error: %sFzMax retries exceeded: r)rr-rZrrsend_group_messagersend_c2c_messagerrr5r?rr:rSrrAr) rrmrRrrkrgr_ last_errorrDrjrDrrIs r]r5zMessageSender.send_text_chunks#-) U|| 2 2G %%h//_!#h--..1C $ 7 7T8 L LLLLLLLCC!(!5!5i!@!@J $ 5 5j$S] 5 ^ ^^^^^^^C779%%S%dswwy?Q?QRRRRRR WWWo>> CL'A+uj    XX FL'A+uj ""mAL111111111 L L%   %/T /T/TUUUUsB5D*0022_OOPVX[}$]$]^^^{{1~~H#''(8(899E f!& 3 / 9cf+g+g h h$!!]FTbX`TbTbKc d deeeyy{{HH c$ii   ?((**D \]FTX> Z Z[[[ X OOPT~VV W W Wr_cK|j}dt}t|||jpd||}||||d{VS)z(Send C2C message with arbitrary MsgBody.c2c_ro)rrjrfrkrgN)rr6r0rS_dispatch_encoded)rrrjrgr_rsrs r]rizMessageSender.send_c2c_msg_bodyjst-' '')! .B!    ++GWfEEEEEEEEEr_cK|j}dt}t|||jpd||pd}||||d{VS)z*Send group message with arbitrary MsgBody.grp_ro)rgrjrfrk ref_msg_idN)rr6r1rSr)rrgrjrr_rsrs r]rhz!MessageSender.send_group_msg_bodywsz-' ''+! .B~2    ++GWfEEEEEEEEEr_rrrsc K |j||d{V}d|dddS#tj$rddt d d cYSt $r}dt|d cYd}~Sd}~wwxYw) zBSend pre-encoded bytes via WS and return a normalised result dict.rNTrkro)rrFzRequest timeout after sr)rrxrrrrr:rS)r_rrsrFrIs r]rzMessageSender._dispatch_encodeds  9$0AA'RXAYYYYYYYYH# Xr0J0JKK K# a a a$/_H\/_/_/_`` ` ` ` 9 9 9$s3xx88 8 8 8 8 8 8 9s!:?B B'A>8B>BrOptional[bytes]rrc|t|dkrd|S|dzdz}t||kr"t|dz dz }d|d|dd|d SdS) zMedia pre-validation: check file validity before sending/uploading. Returns: Error description (str) if validation fails, otherwise None. Nrz Empty file: izFile too large: z (z.1fzMB > zMB)rN)rrr max_bytessize_mbs r]rzMessageSender.validate_medias  ZA!5!5,(,, ,$&- z??Y & &*oo,t3GThTT'TTTKTTT Ttr_r max_lengthrkrlrc|pt}|||kr|gSt|||}d|D}|r|n|gS)aj Split a long message into chunks with table-awareness. Delegates core splitting to ``MarkdownProcessor.chunk_markdown_text`` and strips page indicators like ``(1/3)`` from the output. Falls back to ``BasePlatformAdapter.truncate_message`` for non-table content and for overall text that fits in a single chunk. rcDg|]}td|SrY) _INDICATOR_REsubrs r]rz2MessageSender.truncate_message..s(;;;q-##B**;;;r_)rrrQr)r6rrkrxrs r]rPzMessageSender.truncate_messagesq} 4==J & &9 #66 Z7   <;F;;;.vvgY.r_c8|ds|Sd}d}||}||}|dks |dks||kr|S|d|}d|vr|S|t|z}|||}|p|S)zFStrip scheduler cron header/footer wrapper for cleaner Yuanbao output.zCronjob Response: z ------------- zI To stop or manage this job, send me a new message (e.g. "stop reminder rNz (job_id: )rZrrsrrrd)r6divider footer_prefix divider_pos footer_posr body_startrGs r]rOz MessageSender.strip_cron_wrappers!!"677 N'e ll7++ ]]=11 ??j1nn k0I0IN+&  & &N 3w<</ z*,-3355wr_c<K|jdS)zCRelease chat locks (no-op for now; placeholder for future cleanup).N)r@rrs r]r1zMessageSender.closes!      r_r)rrSrrrTrrmrSrTrr rmrSr6rSrrrrgrSrTrr) rmrSrrSrrrrrrrr rTrrrmrSrHrSrYrZrTr[) rmrSrjrarrrrgrSrTr)Nrro) rmrSrRrSrrrrkrjrgrSrTrrY)rrSrRrSrgrSrTr=)rgrSrRrSrrrrTr=)rRrSrgrSrTra)rrSrjrargrSrTr=)rgrSrjrarrrrTr=)r_rrrrsrSrTr=rrrrrSrrjrTrrr)r6rSrrjrkrlrTr)r6rSrTrSr)rrrrrMr:rZr;rrErKrUrXrdrr5rnrmrtru MULTILINErvrrrirhrrrrPrOr1rr_r]rr?s'0i0b0b0b&c&cJcccc(,,,,,    &---- )))).#' %(%(%(%(%(V#'!%      49= ) ) ) ) ) ^#' UUUUU2#' *V*V*V*V*V\YYYYY#' NNNNN"*A2<PPK2222h F F F F F"#' FFFFF( 9 9 9\ 9GI\$15////\/<\,!!!!!!r_rceZdZUdZejZded<d1dZd2d Zd2d Z d3d4dZ d5dZ d6d7dZ d2dZ d8d9d"Zd2d#Zd2d$Zd:d&Zed;d(Ze dd0Zd S)?OutboundManageruOutbound coordinator that orchestrates sending, heartbeat and slow-response. Composes: - MessageSender — core text/media sending - HeartbeatManager — reply heartbeat (RUNNING / FINISH) lifecycle - SlowResponseNotifier — delayed 'please wait' notifications YuanbaoAdapter holds a single ``_outbound: OutboundManager`` and delegates all outbound operations through it. rMr;r_rrTrc||_t||_t||_t ||j|_|j|j_|j |j_ dSr) rrr;rrr+ slow_notifier_handle_send_startrA_handle_send_finishrBrs r]rzOutboundManager.__init__s_ %27%;%; +;G+D+D3GQUQ\3]3]&*%< "&*&> ###r_rmrSc:|j|dS)zFCalled by MessageSender before sending: cancel slow-response notifier.Nrr*rrms r]rz"OutboundManager._handle_send_start !!'*****r_cVK|j|td{VdS)z=Called by MessageSender after sending: send FINISH heartbeat.N)rrr'rs r]rz#OutboundManager._handle_send_finishs7n00:MNNNNNNNNNNNr_Nror6rrrrgrcNK|j||||d{VS)z%Send text message with auto-chunking.rN)r;rU)rrmr6rrgs r]rUzOutboundManager.send_text s: [**7GXR\*]]]]]]]]]r_rrr c:K|jj||fi|d{VS)rWN)r;rX)rrmrrs r]rXzOutboundManager.send_medias9,T[+G\LLVLLLLLLLLLr_rHrYrZr[cJK|j|||d{VS)z.Send text + media (used by send_message tool).N)r;rd)rrmrHrYs r]rdzOutboundManager.send_directs4 [,,Wg{KKKKKKKKKr_cJK|j|d{VdS)z Start reply heartbeat (RUNNING).N)rrrs r] start_typingzOutboundManager.start_typing!s4n""7+++++++++++r_Fr#rUcNK|j||d{VdS)zStop reply heartbeat.r#N)rr%)rrmr#s r] stop_typingzOutboundManager.stop_typing%s9n!!'{!CCCCCCCCCCCr_cJK|j|d{VdS)zStart slow-response notifier.N)rrrs r]start_slow_notifierz#OutboundManager.start_slow_notifier)s5 &&w///////////r_c:|j|dS)zCancel slow-response notifier.Nrrs r]rz$OutboundManager.cancel_slow_notifier-rr_rc6|j|S)z@Proxy to MessageSender.get_chat_lock for backward compatibility.)r;rKrs r]rKzOutboundManager.get_chat_lock1s{((111r_collections.OrderedDictc|jjS)z>Proxy to MessageSender._chat_locks for backward compatibility.)r;r@rs r]r@zOutboundManager._chat_locks5s{&&r_rrrrrrjc:t|||S)z&Proxy to MessageSender.validate_media.)rr)rrrs r]rzOutboundManager.validate_media:s ++J+NNNr_cK|jd{V|jd{V|jd{VdS)zShut down all sub-managers.N)r;r1rrrs r]r1zOutboundManager.closeAsk!!!!!!!!!n""$$$$$$$$$ &&(((((((((((r_rr(rr)rmrSrrSrr rTrrr)Fr)r)rTrrrr)rrrrrr;rZrrrrUrXrdrrrrrKrr@rrr1rr_r]rrs  )6(HHHHH????++++OOOOEI^^^^^MMMM9=LLLLL,,,,DDDDD0000++++2222'''X'GIOOOO\O ))))))r_rceZdZUdZejZdZded<dZ ded<dZ ded <d Z d ed <e dLdZ e dMdZdNfd ZdOdZedPdZdPdZdQdZ dRdSd)ZdTd+ZdUdVd-ZdWd.ZdXfd0 ZdYd1Z dZd[d6Zd7Zd\d]d:Z d^d_d=Z d^d`d?Z d^dadBZ d^dbdFZ! dcdddHZ"dedJZ#dedKZ$xZ%S)fr%zCYuanbao AI Bot adapter backed by a persistent WebSocket connection.rrjrQrNrirMREPLY_REF_MAX_ENTRIESNz$ClassVar[Optional['YuanbaoAdapter']]_active_instancerTOptional['YuanbaoAdapter']c|jS)z7Return the currently connected YuanbaoAdapter, or None.rr s r] get_activezYuanbaoAdapter.get_activeTs ##r_r_rc||_dS)z0Register (or clear) the active adapter instance.Nr)rr_s r]r&zYuanbaoAdapter.set_activeYs 'r_rrrr c <t|tj|jpi}|dpd|_|dpd|_|dpd|_ |dpt|_ |dpt d|_|dpd|_t!||_t%||_t)|_t)|_i|_d |_t3d |_i|_i|_i|_i|_|d pt?j d d!}|dpt?j dd}d|"dD}|dpt?j dd!}|dpt?j dd}d|"dD} tG|||| |_$tK||_&tN(|_)t?j dp|j*r |j*j+nd} tY| o| -d |_.dS)Napp_idrorr&ws_urlrrrrKi,) ttl_secondsrYYUANBAO_DM_POLICYrrZYUANBAO_DM_ALLOW_FROMc^g|]*}||+Srrrxs r]rz+YuanbaoAdapter.__init__..s2#b#b#b!XYX_X_XaXa#bAGGII#b#b#br_,r[YUANBAO_GROUP_POLICYr\YUANBAO_GROUP_ALLOW_FROMc^g|]*}||+Srrrs r]rz+YuanbaoAdapter.__init__..s2&h&h&hQ^_^e^e^g^g&hqwwyy&h&h&hr_)rYrZr[r\rtr )/superrrYUANBAOrrrdrwryrSDEFAULT_WS_GATEWAY_URLrDEFAULT_API_DOMAINrcrxrrrrrrrrBrrurrrr r*rryrzrrYrXror _group_queryrrrq home_channelrmrUrZrx) rrr_extrarY_dm_allow_from_rawrZr[_group_allow_from_rawr\_existing_homers r]rzYuanbaoAdapter.__init__^sR !1222#$jj228b??AA !'L!9!9!?R F F H H&,jj&:&:&Bd #ZZ11K5KRRTT !'L!9!9!O=O W WX[ \ \ & ; 7 7 =2DDFF/@.E.E*9$*?*?2558EE =?).*c::: 8:46 57"35 JJ{ # # 6y,f55 %''%%''  JJ ' ' 6y0"55 $c#b7I7O7OPS7T7T#b#b#b  JJ~ & & 9y/88 %''%%''  JJ) * * 9y3R88 'i&h:O:U:UVY:Z:Z&h&h&h*'%-    .d333I2N2N2P2P#9:: +1+> FF  ' 'B )-^(<(<(h^E^E^_gEhEhAhr_rE asyncio.Taskcx|j|||jj|S)z@Register a fire-and-forget task so it won't be GC'd prematurely.)rBrrCrDr's r]rzYuanbaoAdapter._track_tasks8 ""4((( t5=>>> r_rUcdS)zCYuanbao gates DM/group access at intake via dm_policy/group_policy.Trrs r]enforces_own_access_policyz)YuanbaoAdapter.enforces_own_access_policys tr_cDK|jd{VS)zhConnect to Yuanbao WS gateway and authenticate. Delegates to ConnectionManager.open(). N)rrrs r]rzYuanbaoAdapter.connects/ %**,,,,,,,,,r_cVKtj|urtdd|_|||jd{V|jd{Vt|j D]*}| s| +|j |j td|jdS)z;Cancel background tasks and close the WebSocket connection.NFz[%s] Disconnected)r%rr&rJrXr'rr1rrarr-r*rrr5r6rr's r] disconnectzYuanbaoAdapter.disconnects#  *d 2 2  % %d + + +  !!! ##%%%$$&&&&&&&&&n""$$$$$$$$$,--  D99;;   !!###   """ '33333r_rormrSr6rrrmetadataOptional[Dict[str, Any]]rgrcNK|j||||d{VS)zCSend text message with auto-chunking. Delegates to OutboundManager.rN)rrU)rrmr6rrrgs r]rzYuanbaoAdapter.sends:^--gwU_-`````````r_r[cDK|dr|ddS|ddS)uReturn basic chat metadata derived from the chat_id prefix. chat_id conventions: "group:" → group chat "direct:" → C2C / direct message (default) TODO (T06): fetch real chat name/member-count from Yuanbao API. r r )rrrrrs r] get_chat_infozYuanbaoAdapter.get_chat_infos:   h ' ' 6#W55 5...r_rccnK |j|d{VdS#t$rYdSwxYw)zGSend "typing" status heartbeat (RUNNING). Delegates to OutboundManager.N)rrr:)rrmrs r] send_typingzYuanbaoAdapter.send_typings[ .--g66 6 6 6 6 6 6 6 6 6    DD s & 44crK |j|dd{VdS#t$rYdSwxYw)zStop the RUNNING heartbeat loop without sending FINISH immediately. FINISH is sent by send() after actual message delivery to ensure correct ordering: RUNNING... -> message arrives -> FINISH. FrN)rrr:rs r]rzYuanbaoAdapter.stop_typing s`  .,,W%,HH H H H H H H H H H    DD s "( 66r#c(K|jj}|j|d{V t ||d{V|j|dS#|j|wxYw)z9Wrap base class processing with a slow-response notifier.N)rPrmrrr_process_message_backgroundr)rrr#rmrs r]rz*YuanbaoAdapter._process_message_backgrounds,&n00999999999 9''55e[II I I I I I I I N / / 8 8 8 8 8DN / / 8 8 8 8s (A55BcFK|j|d{VS)z2Query group info (delegates to GroupQueryService).N)rrrhs r]rzYuanbaoAdapter.query_group_info$s/&;;JGGGGGGGGGr_rr"rrcLK|j|||d{VS)z9Query group member list (delegates to GroupQueryService).rN)rr)rrgrrs r]get_group_member_listz$YuanbaoAdapter.get_group_member_list(s:&@@TZbg@hhhhhhhhhr_i'rrRcK|j|stddSt||jkr|d|jdz}d|}||||d{VS)a Actively send C2C private chat message. Args: user_id: Target user ID text: Message text (limit 10000 characters) group_code: Source group code (for group-originated DM context) Returns: SendResult FzDM access denied for this userrNz ...(truncated)r r)rorfrrr DM_MAX_CHARSr)rrrRrgrms r]send_dmzYuanbaoAdapter.send_dm4s"0099 Ue3STTT T t99t( ( (***+.@@D%G%%YYwYDDDDDDDDDr_rrcBK|jj|df|||d|d{VS)zKSend image message (URL). Delegates to OutboundManager via ImageUrlHandler.r)rrrNrrX)rrmrrrrrs r] send_imagezYuanbaoAdapter.send_imageKs_/T^. [ w)           r_rcBK|jj|df|||d|d{VS)zISend local image file. Delegates to OutboundManager via ImageFileHandler.r=)rrrNr)rrmrrrrrs r]r_zYuanbaoAdapter.send_image_file[s_/T^. \ w:           r_rrcBK|jj|df|||d|d{VS)zISend file message (URL). Delegates to OutboundManager via FileUrlHandler.r)rrrNr)rrmrrrrrs r] send_filezYuanbaoAdapter.send_fileks_/T^. Z 8           r_rr Optional[int]cBK|jj|df|||d|d{VS)zDSend sticker/emoji. Delegates to OutboundManager via StickerHandler.r)rrrNr)rrmrrrrs r] send_stickerzYuanbaoAdapter.send_sticker{s`/T^. Y %*            r_rcDK|jj|df||||d|d{VS)zMSend local file (document). Delegates to OutboundManager via DocumentHandler.r)rrrrNr)rrmrrrrrrs r]r`zYuanbaoAdapter.send_documentsb/T^. Z w(            r_r=cvKt|j|j|j|jd{VS)zBAAAA$$$[$'''['\i\i\i\i\i\iDX----44446#'-1 a a a a a / / / /    999999HHHH >AiiiiiLEEEEE6"&"&#'      ("&"&#'      (#'"&#'      &'+$("&      (#'!%"&#'     $             r_r%rTrc4tS)z,Delegate to ``YuanbaoAdapter.get_active()``.)r%rrr_r]get_active_adapterrs  $ $ & &&r_r_rrmrSrHrYrZr[cJK|j|||d{VS)z,Delegate to ``OutboundManager.send_direct``.N)rrd)r_rmrHrYs r]send_yuanbao_directrs5"..wMM M M M M M MMr_rr) r_rrmrSrHrSrYrZrTr[)r __future__rrr>r%rrr loggingryrtr/r urllib.parserr<rrrpathlibrabcrr typingr r r r rrrsysr*rwebsockets.exceptionsr ImportErrorgateway.configrrgateway.platforms.baserrrrrrgateway.platforms.helpersrgateway.platforms.yuanbao_mediarrrrrrrr rr!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6gateway.sessionr7 getLoggerrr5 hermes_clir8_HERMES_VERSIONr1r4rSr3r]r2rrrKrr>r~rrrWrMAUTH_FAILED_CODESAUTH_RETRYABLE_CODESrr REPLY_REF_TTL_Sr4r6rurCrrMrErrrrQrr[r\r|r^r~rrrrrrOrUrXrmrrrrrrr r#r)r4rKrrrrrrrrrrrrr+rrr%rrrr_r]rsG "#"""""    2222222222########GGGGGGGGGGGGGGGGGG      JJJ43333333:999990.-----  8 $ $ 9999999OOO  s-..L L6!@?? '&&)))!  YI RZ ABB$)Wf$566 -.. $& /1,^ ^ ^ ^ ^ ^ ^ ^ @p)p)p)p)p)p)p)p)f54444444 :):):):):):):) :)zAAAAAAAA6]]]]]]]]|]]]]](]]]@/"     '   NdNdNdNdNd-NdNdNdb*&-"*"*"*"*"*"*"*"*"Z-2*****-***Zuuuuu0uuun"32^^^^^.^^^B-$qqqqq.qqqh!2>$56GGGGG.GGGTEEEEE.EEEP-----.---`[9[9[9[9[9*[9[9[9|$$$$$$$$LZ Z Z Z Z Z Z Z x~=~=~=~=~=s~=~=~=B     &   6     '   4     %   4     &   .33333%333>U U U U U U U U px+x+x+x+x+x+x+x+v........bf!f!f!f!f!f!f!f!R ])])])])])])])])@e e e e e (e e e Z ''''59 NNNNNNNs$0 A;; BBDDD