YjnzdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlmZmZmZddlmZddlmZddlmZmZddlmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%ddl&Z&ddl'Z' ddl(Z(ddl)Z(d Z*n#e+$rd Z*dZ(YnwxYwdd l,m-Z-m.Z.dd l/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6dd l7m8Z8ddl9m:Z;mZ>m?Z?m@Z@mAZAddlBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYddlZm[Z[e j\e]Z^ ddl_m`Zan #e+$rdZaYnwxYweaZbeaZcedeJZee&jfZgdZhdZidZjdZkdZldZmdZndZohdZpdZqhdZrhdZsdZtdZudZvd Zwd!Zxe jyd"Zze jyd#Z{e|hd$Z}e jyd%Z~d&Zd'Zd(Zd)Zd'ZGd*d+ZGd,d-Zdd.lmZmZeGd/d0ZGd1d2eZGd3d4ZGd5d6eZGd7d8eZGd9d:eZGd;deZGd?d@eZGdAdBZGdCdDeZGdEdFeZGdGdHeZGdIdJeZGdKdLeZGdMdNeZGdOdPeZGdQdReZGdSdTeZGdUdVeZGdWdXeZGdYdZeZGd[d\eZGd]d^eZGd_d`eZGdadbZGdcddZGdedfeZGdgdheZGdidjeZGdkdleZGdmdneZGdodpeZGdqdrZGdsdtZGdudvZGdwdxZGdydzZGd{d|e0ZddZ 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)Enum)Path)ABCabstractmethod)AnyCallableClassVarDictIteratorListOptionalTupleTF)PlatformPlatformConfig)BasePlatformAdapter MessageEvent MessageType SendResultcache_document_from_bytescache_image_from_bytescache_video_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_forward_msg_datadecode_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*\]>fileimagevideoz\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)rYin_fencelines NC:\Users\server\AppData\Local\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 Fr^|)rstripr`stripraendswith)rYtrimmed last_lines rdends_with_table_rowz%MarkdownProcessor.ends_with_table_rowsg++-- 5MM$''+1133 ##C((DY-?-?-D-DDrfN 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 NrrVrG u[。！？.!?]\nrhr^)lenrfindrecompilefinditerend) rYrprr_lenwindowlohimidpossentence_end_rebest_posmcuts rdsplit_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:%%rfcP|dS)zDDetermine whether an atomic block is a code block (starts with ```).r_)lstriprarYs rd is_fence_atomzMarkdownProcessor.is_fence_atom9s {{}}''...rfc|dd}|do|dS)zHDetermine whether an atomic block is a table (first line starts with |).r^rri)r`rkrarl)rY first_lines rd is_table_atomzMarkdownProcessor.is_table_atom>sMZZ%%a(..00 $$S))Fj.A.A#.F.FFrf 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) r^FrcrZr[r\c~|}|do|dS)Nri)rkrarl)rcstrippeds rd_is_table_linez:MarkdownProcessor.split_into_atoms.._is_table_line[s6zz||H&&s++F0A0A#0F0F FrfNonecrTd}|r|dSdS)Nr^)joinrkappendclear)atomatoms current_liness rd_flush_currentz:MarkdownProcessor.split_into_atoms.._flush_current_s` &yy//::<<'LL&&&##%%%%%  & &rfr_rVTrhrv)rcrZr[r\r[r)r`rrarxrk)rYlinesrbrrrcrrs @@rdsplit_into_atomsz"MarkdownProcessor.split_into_atomsDs" 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**** rfc|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) rr[rc`r*ddSdS)Nrw)rr)chunks current_partssrd _flush_partsz;MarkdownProcessor.chunk_markdown_text.._flush_partss9 : fkk-8899999 : :rfrGrrNrVrhrwcg|]}||Sr.0cs rd z9MarkdownProcessor.chunk_markdown_text..s'''aQ''''rfr) rxrsetrraddr enumeraterer)clsrYrprrr~rindivisible_set current_lenrratom_lensep_len projected_lenresultidxchunk remainingheadmergedprevcombinedrrs @@rdchunk_markdown_textz%MarkdownProcessor.chunk_markdown_textsf2} I 4:: " "6M$$T**$'EE#%   : : : : : : : . .DtDzzH(/aaaG''1H>Y..!)F2JJMM%((((F''6''''rf 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' r_r^rrvrirw)rjrrlraror`rk)rrr prev_trimmed next_trimmedrs rdinfer_block_separatorz'MarkdownProcessor.infer_block_separators""((** !((**    ' ' <+B+B5+I+I 4  " ": . . @LT++D11!4::<<>K1X^^%% "IOO%% x3ZOO K   K %"+&& rfcfd|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 rir^z^\|[\s\-:]+(\|[\s\-:]+)+\|$c3jK|].}|r|n|V/dSNrk)rcells rd z.}sO** )- > $******rfz||rv) r`rkrarlrzrrrreplace)rYr result_linesrcrcells normalizeds rdsanitize_markdown_tablez)MarkdownProcessor.sanitize_markdown_table_sS 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&&&rfc 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.rrrfrdmarkdown_hint_system_promptz-MarkdownProcessor.markdown_hint_system_prompts V  rf)rYrZr[r\r)rYrZrprqrrrsr[rt)rYrZr[rrN)rYrZrprqrrrsr[r)rrZrrZr[rZ)rrr[rrYrZr[rZr[rZ)__name__ __module__ __qualname____doc__ staticmethodrerorrrr classmethodrrrrrrrrfrdrXrXs  \*EEE\E$26=&=&=&=&\=&B///\/GGG\G 888\8x15 k(k(k(k([k(^[B!!![!J!!!\!J*'*'*'\*'\   \   rfrXc,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_keyrZr[ 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 rdget_refresh_lockzSignManager.get_refresh_locks0 #* $ $"),..CJw z'""rfnonce 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 rdcompute_signaturezSignManager.compute_signaturesN !G+j8x ))++U\\^^W^LLVVXXXrfctjttd}|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 rdbuild_timestampzSignManager.build_timestamps< )!*<*<*>>8999rfentrydict[str, Any]r\cL|dtjz |jkS)zEDetermine whether the cache entry is valid (not expired with margin). expire_ts)timeCACHE_REFRESH_MARGIN_S)rrs rdis_cache_validzSignManager.is_cache_valids"[!DIKK/#2LLLrfrc8|jdS)z;Clear all per-app_key refresh locks (called on disconnect).N)rrrs rd clear_lockszSignManager.clear_lockss rfrqctjfd|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)r rget)rkvrs rdrz-SignManager.purge_expired..sE   !QQUU;***Q.. ...rfN)r ritemspoprx)r expired_keysrrs @rd purge_expiredzSignManager.purge_expireds~ikk    ***,,    $ $A JNN1d # # # #<   rfrv 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)./timeoutNrV)rr signaturerapplication/json) Content-Typez X-AppVersionzX-OperationSystemz X-Instance-Idz X-Bot-Versionz X-Route-EnvzSign token request: url=%s%srz (retry )rv)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) rj TOKEN_PATHhttpx AsyncClientHTTP_TIMEOUT_Srange MAX_RETRIESsecrets token_hexrr _APP_VERSION_OPERATION_SYSTEM_YUANBAO_INSTANCE_ID _BOT_VERSIONloggerinfopost status_coderY RuntimeErrorr& Exception ValueErrorr isinstancedictRETRYABLE_CODEwarning RETRY_DELAY_Srsleep)rrrrrurlclientattemptrrr"payloadr'responsebody result_dataexcr*r+r-s rdfetchzSignManager.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. r z"Using cached token (%ds remaining)Ndurationrtokenrvr,productsourcerTr,rRrUrVr ) rrrr rqr r;r<rCrrP) rrrrrcachedremainr+rRr s rd get_tokenzSignManager.get_tokenIs ((  c((00  ,ty{{:;;F KKK|||d{VS)zFAllow middleware instances to be called directly (duck-typing compat).N)rrs rd__call__zInboundMiddleware.__call__s,[[g.........rfc2d|jjd|jdS)N) __class__rrrs rd__repr__zInboundMiddleware.__repr__s"@4>*@@$)@@@@rfN)rrdrr r[rr) rrrrrr`r rrrrrfrdrrs  DNNNNTTT^T////AAAAAArfrcxeZdZdZddZeddZdddZddd 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. r[rcg|_dSr _middlewaresrs rd__init__zInboundPipeline.__init__s"$rfNcFt|tr |j|fS||fS)zHNormalize (name, handler) or (InboundMiddleware,) into (name, callable).)rBrr) name_or_mwhandlers rd _normalizezInboundPipeline._normalizes/ j"3 4 4 /?J. .7""rf'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 rdusezInboundPipeline.use s=//*g66a   $4111 rftargetrZc|||\}}tfdt|jDd}|||f}||j|n|j|||S)zEInsert a middleware before *target* (by name). Appends if not found.c36K|]\}\}}}|k|VdSrrrrn_rs rdrz-InboundPipeline.use_before..1VV,!YaA!v++A++++VVrfNrnextrrrinsert rrrrrrrrrs ` rd use_beforezInboundPipeline.use_befores//*g66aVVVV)D4E*F*FVVVX\]]q$ ;   $ $U + + + +   $ $S% 0 0 0 rfc|||\}}tfdt|jDd}|||f}||j|n|j|dz||S)zDInsert a middleware after *target* (by name). Appends if not found.c36K|]\}\}}}|k|VdSrrrs rdrz,InboundPipeline.use_after..&rrfNrVrrs ` rd use_afterzInboundPipeline.use_after#s//*g66aVVVV)D4E*F*FVVVX\]]q$ ;   $ $U + + + +   $ $S1We 4 4 4 rfrc8fd|jD|_|S)zRemove a middleware by name.c.g|]\}}}|k |||fSrr)rrrwrs rdrz*InboundPipeline.remove..0s+UUU71a1PT99aAY999rfr)rrs `rdremovezInboundPipeline.remove.s'UUUUd6GUUU rfrgc$d|jDS)zAReturn ordered list of registered middleware names (for testing).cg|]\}}}| Srr)rrrs rdrz4InboundPipeline.middleware_names..6s333gaA333rfrrs rdmiddleware_namesz InboundPipeline.middleware_names3s43!23333rfrrdcVK|jddfd d{VdS)zKRun all middlewares in order. Each middleware receives ``(ctx, next_fn)``.rr[rcKtkra\}}}dz | |s2 |d{Vn,#t$rtd|dwxYwdSdS)NrVz'[InboundPipeline] middleware [%s] errorTexc_info)rxr@r;error)rrwhen_fnchainrindexrs rdrz(InboundPipeline.execute..next_fn?s#e**$$).u&gw &wws||&!'#w////////// LL!JD[_L```%$s A )A3Nrr)rrrrrs `@@@rdexecutezInboundPipeline.execute:sa!          giirfrrNN)r[r)rrZr[r)rrZr[r)r[rgrrdr[r)rrrrrrrrrrrpropertyrrrrfrdrrs  %%%% ###\#                444X4 rfrcVeZdZdZdZeddZedd 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_bodyrgr[cg}|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. r{MsgTyperv msg_content MsgContentrYr{r)rBrCrrZr&loadsr@r)rritemr{rs rdconvert_json_msg_bodyz&DecodeMiddleware.convert_json_msg_body[sN T TDdD)) xx ++Ftxx 2/F/FH((=11OTXXlB5O5OK+s++ 88"&*["9"9KK 888#);"7KKK8 MMx @QrRR S S S S s B  B10B1raw_jsonrC 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.). Nrlrv From_AccountrmGroupIdgroup_idrpMsgBodycallback_command to_account To_Accountro nick_namernmsg_seqrMsgSeqrqmsg_keyMsgKeyrrCloudCustomData bot_owner_id botOwnerIdrecall_msg_seq_listlog_exttrace_id) rrlrrormrnrrqrprrrrr)rrrrBrC)rrlrm msg_body_rawrps rdparse_json_pushz DecodeMiddleware.parse_json_pushps 4 LL , , 0||NB//  LLr * * ,||Ir** ,||J++  LLR ( ( +||Ir** $99,GG H X\\BT5U5U 4!) -? D D(",,|R88ZHLLWY> B  B rrdrc .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)Nrvz;[%s] Push decoded but no valid message. raw hex(first64)=%sz(empty)z#[%s] Frame decoded (via=%s): len=%drp TIMTextElemrYr^rz;[%s] Merged %d extra msg_body elements from aggregated pushzC[%s] Push decoded (via=%s): from=%s group=%s msg_id=%s msg_types=%srlrmrqc:g|]}|ddS)r{rvrres rdrz+DecodeMiddleware.handle..s& I I IqQUU:r " " I I Irfz[%s] Push payload: %s) rhrrer;r<rhexrxrrjrkdebug) rrr data_list merged_pushrkr+rjvia extra_body_seps rdrzDecodeMiddleware.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.>IIIgiirfN)rrgr[rg)rrCr[r)r+rr[rr) rrrrrrrrrrrrfrdrrPs D\(/ / / \/ f*444444rfrceZdZdZdZd dZdS) ExtractFieldsMiddlewarez8Extract common fields from ctx.push into ctx attributes.zextract-fieldsrrdr[rcK|j}|dd|_|dd|_|dd|_|dd|_|dg|_|dd|_|dd|_|d{VdS) Nrlrvrmrnrorprqrr) rjrrlrmrnrorprqrr)rrrrjs rdrzExtractFieldsMiddleware.handlesx88NB77,33,33"hh'8"==xx B// XXh++ $)V >V>V!+5gg (D *4ff $   rfrc |j}|jpi}|dkr|dpg}n8|dpd}|d}|s|r||dgng}|s"td|jdS|dpd}|d pd} |D]} | dp#t| dpd} | s>||| } | | || | || p|j | } | || || | dS) Nr rrqrvr)rqrz2[%s] Recall callback with empty seq_list, skippingrmrl) rerjrr;rrrkrZ_find_processing_session_interrupt_for_recall_msg_content_cache_patch_transcript)rrrrerjseq_listrseqrmrl seq_entry recalled_id matched_skrecalled_contents rdrz$RecallGuardMiddleware._handle_recall-s+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 irfr#rycj|jD]\}}||kr ||jvr|cSdSr)_processing_msg_idsr_active_sessions)rer#skrs rdrz.RecallGuardMiddleware._find_processing_sessionMsK288::  GBk!!bG,D&D&D trf 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)rY message_typerVinternalz+[%s] Recall interrupt: msg_id=%s session=%srv) rrTEXTr_pending_messagesr(rrr;r<r_processing_msg_texts_schedule_content_redact) rrer*r#rmrlwhere recall_text synth_event active_event recalled_texts rdrz+RecallGuardMiddleware._interrupt_for_recallTs=*4[%%%%9[\9[9[ ,/: , ,CH , , , #$)$$Wj,GG    2=!+./33K@@  #       A7._redact}s G%5t<r?rr@r@r;rErrArrBrCr<append_to_transcriptrrrutc isoformat) rrer#rmrlr%rDrErOrFr branch_labelrs rdrz'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)rmrZrlrZ)rrdrrZr[r)r#rZr[ry) r*rZr#rZrmrZrlrZr[r) r*rZr7rZrmrZrlrZr[rr) r#rZrmrZrlrZr%ryr[r)rrrrr frozensetrrBrrrrrrrr2rrrfrdr r  s0 D y&&"`I&&&&   \ iiii@\ #h#h#h[#hJ!B!B!B[!BJOS:d:d:d:d[:d:d:drfr c6eZdZdZdZedd Zdd ZdS)SkipSelfMiddlewarezFilter out bot's own messages.z skip-selfrlrZr,ryr[r\c|r|sdS||kS)z2Detect whether the message is from the bot itself.Fr)rlr,s rd_is_self_referencez%SkipSelfMiddleware._is_self_references# 6 5v%%rfrrdrcK||j|jjr-td|jj|jdS|d{VdS)Nz'[%s] Ignoring self-sent message from %s)rXrlre_bot_idr;rrrs rdrzSkipSelfMiddleware.handlesf  " "3#3S[5H I I  LLBCKDTVYVf g g g FgiirfN)rlrZr,ryr[r\r)rrrrrrrXrrrfrdrVrVsQ(( D&&&\& rfrVceZdZdZdZd dZdS) ChatRoutingMiddlewarez9Determine chat_id, chat_type, chat_name from push fields.z chat-routingrrdr[rcK|jr*d|j|_d|_|jp|j|_n)d|j|_d|_|jp|j|_|d{VdS)Nrrrr)rmrsrtrnrurlrors rdrzChatRoutingMiddleware.handles > D33>33CK#CMN||_||_||_||_dSr) _dm_policy_dm_allow_from _group_policy_group_allow_from)rr`rarbrcs rdrzAccessPolicy.__init__ s)$+)!1rfr\ctjdddvrdStjdddvS)NGATEWAY_ALLOW_ALL_USERSrv>1yestrueTYUANBAO_ALLOW_ALL_USERS)osgetenvlowerrs rd_open_dm_opted_inzAccessPolicy._open_dm_opted_insQ 9. 3 3 9 9 ; ;?S S S4y2B77==??CWWWrf sender_idc|jdkrdS|jdkr||jvS|jdkr|SdS)u:Strict DM authorization — pairing does not imply access.disabledF allowlistopen)rerkrfrr)rrss rd is_dm_allowedzAccessPolicy.is_dm_alloweds] ?j ( (5 ?k ) )??$$(;; ; ?f $ $))++ +urfct|pd}|sdS|jdkrdS|jdkr ||jvS|jdkrdS|jdkr|SdS)z?Whether a DM may reach gateway intake (pairing handshake path).rvFrurvpairingTrw)rZrkrerfrr)rrs principals rdis_dm_intake_allowedz!AccessPolicy.is_dm_intake_allowed%s R((..00  5 ?j ( (5 ?k ) ) 33 3 ?i ' '4 ?f $ $))++ +urfrmc|jdkrdS|jdkr||jvS|jdkrdS|jdkr|SdS)zGPlatform-level group chat inbound filter (open / allowlist / disabled).ruFrvrzrw)rgrkrhrrrrms rdis_group_allowedzAccessPolicy.is_group_allowed4su   + +5   , ,##%%)?? ?   * *5   ' '))++ +urfc|jSr)rers rdr`zAccessPolicy.dm_policy@s rfc|jSr)rgrs rdrbzAccessPolicy.group_policyDs !!rfN) r`rZrarrbrZrcrr[rr[r\)rsrZr[r\)rmrZr[r\r) rrrrrrrrxr|rrr`rbrrfrdr_r_s 2 2 2 2XXXX         X"""X"""rfr_ceZdZdZdZd dZdS) AccessGuardMiddlewarez.Platform-level DM/Group access control filter.z access-guardrrdr[rcK|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=%srz([%s] Group %s blocked by group_policy=%s) re_access_policyrtr|rlr;rrr`rrmrb)rrrrepolicys rdrzAccessGuardMiddleware.handleNs+&5 =D ..s/?@@  =L#"2F4D   ]g % %**3>::  >L#.&2EgiirfNrrrrfrdrrIs388 DrfrceZdZdZdZd 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. Runs after :class:`BuildSourceMiddleware` and :class:`GroupAtGuardMiddleware` so unaddressed group traffic is dropped before home-channel persistence. Only senders that pass strict authorization (allowlist / explicit open opt-in / pairing-store approval) may claim ``YUANBAO_HOME_CHANNEL``. Intake-only pairing forwards must not claim ``YUANBAO_HOME_CHANNEL``. z auto-sethomerrdr[rc2K|j}|jsu||r_tjdd}| p|do |jdk}|jdkrd|_|r ddlm}ddl m }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_CHANNELrvrrTr)get_hermes_home)atomic_config_writez config.yamlr)encodingz=[%s] Auto-sethome: designated %s (%s) as Yuanbao home channelz[%s] Auto-sethome failed: %s)re_auto_sethome_done_sender_may_designate_homerorprarthermes_constantsrhermes_cli.configryamlexistsrw safe_loadrsrZenvironr;r<rrur@rE)rrrre _cur_home _should_setrrr_home config_path user_configfrs rdrzAutoSetHomeMiddleware.handlersI+) Tg.P.PQT.U.U T "8"==I N((22Ls}7L }$$-1* TT@@@@@@EEEEEEKKK+O--E"'-"7K(*K"))++B!+@@@BA*...*;*;*ArKBBBBBBBBBBBBBBB:=+K 67'' [AAA96AE<C  E C$$E'C$(A&E F!E??FNrrrrfrdrrbs9   D      rfrceZdZdZdZdZeddZedd Zed d Z e d!dZ ed"dZ ed#dZ ed$dZed%d&dZd'dZdS)(ExtractContentMiddlewarez.Extract raw text and media refs from msg_body.zextract-contentcustomrCr[rZc|dd}|dd}|r d|d|dnd|d}|g}tj}dD]j}||}|rQt|trA#hh>P>P#hwh-*:::VY 222333  : LL8 9 9 9yyrfryc2|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.r=Nrz[link: z | visit link for full content])rr&rrBrCJSONDecodeError TypeErrorrZ)rr=parsedrs rd_format_link_understandingz3ExtractContentMiddleware._format_link_understandings**Y'' 4 Z((F)3FD)A)AK6::f%%%tDD$i0   DDD  :dC00 4>>>>>sAAA76A7rHc^|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 rv resourceId resourceidr) urllibparseparse_qsurlparsequeryrrZrkr@)rHridss rd_parse_resource_idz+ExtractContentMiddleware._parse_resource_ids 2 L))&,*?*?*D*D*JKKE))L))JUYY|-D-DJC*-53s1v;;$$&&&2 5   22 sBB B,+B,rprgc ` 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'|dkr|dd}|rw tj |}t|t s|d |d!}|d"kr*||dd#n|d$kr)|| |n|d%krC| |}|r||n[|d nE|d&kr*||dd'n|d #tj tf$r||YwxYw|d |d(kr|dd}d}|ra tj |}|d)pd}n"#tj ttf$rYnwxYw||rd*|d nd+u|r|d,|d |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 r{rvrrrY TIMImageElemimage_info_arrayNrVrrH [image|ybres:r[image] TIMFileElem file_namefileNamefilenamez[file:|ybres: [file|ybres:[file: [file] TIMSoundElemz [voice|ybres:[voice]TIMVideoFileElem [video|ybres:[video] TIMCustomElemr+z[unsupported message type] elem_typez [mention]z [chat record] TIMFaceElemrz[emoji: z[emoji][ )rrrBrgrxrCrZrkrr&rrrrrAttributeErrorr)rrppartselemrr=rYr image_info image_urlridrfile_url sound_url video_urldata_valrctyperaw_data face_name face_datas rd _extract_textz&ExtractContentMiddleware._extract_textsO /O /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"d]]!LLFO)L)LMMMM!LL)EFFF 0)</// X...../LL!=>>>>m++";;vr22 $(Jx$8$8 %.]]6%:%:%@b$G$G$I$I  0)^L P4 4444yQQQQ / ----..."'/sxxR/s+ >R DR+S?S;=T99UUrYcr|}|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. ／rrVNrkrars rd_rewrite_slash_commandz/ExtractContentMiddleware._rewrite_slash_command1s; zz|| ??8 $ $ "abb>D rfList[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"}] r{rvrrrNrVrrHrQkindrHrrrrrPr)rBrCrrgrxrZrkr) rprefsrr{r=rrrrrrefs rd_extract_inbound_media_refsz4ExtractContentMiddleware._extract_inbound_media_refs;sl&(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$$$ rfctg}|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.r{rrr+rvrrrrr=N) rBrCrr&rrrrZr) rpurlsrdata_strrrrr=rs rd_extract_link_urlsz+ExtractContentMiddleware._extract_link_urlshsN  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,FF32F3rvrric|pgD]Q}t|tr|ddkr2|dipi}t|ts`|dd}|sy tj|}n#tjt f$rYwxYwt|tr|ddks|dpi}t|tr|sd Sd }|D].\}} |d s|| } | | ccS/Sd S) aExtract ForwardMsgData from ext_map for elem_type 1009 (WeChat forward). The detailed chat-record payload lives in ``msg_content.ext_map`` (protobuf field 999, ``map``): - key format: ``wexin_forward_msg_[forward_msg_id]_[userid]`` - value: a **base64-encoded protobuf** ``ForwardMsgData`` (NOT JSON). Decode with base64 then ``decode_forward_msg_data`` to recover the ``sub_type`` / ``nick_name`` / ``msg`` structure. Matching strategy: take the first ``wexin_forward_msg_`` entry whose decoded payload is a valid ``ForwardMsgData`` (``sub_type == 1``). Returns the parsed ``ForwardMsgData`` dict or ``None``. r{rrr+rvrrext_mapNct|tr|sdS tj|}n#tjt f$rYdSwxYwt|}t|tr| ddkr|SdS)Nsub_typerV) rBrZbase64 b64decodebinasciiErrorrAr.rCr)valuepbr+s rd _parse_valuezIExtractContentMiddleware._extract_forwarded_records.._parse_values!%-- U 4 )%00BB  3   44 .r22dD)) dhhz.B.Ba.G.GKts0A  A wexin_forward_msg_) rBrCrr&rrrrra) rprrr=rrrrkeyrrs rd_extract_forwarded_recordsz3ExtractContentMiddleware._extract_forwarded_recordss N' "' "DdD)) TXXj-A-A_-T-Thh}b117RGgt,, {{62..H  H--()4    vt,, K1H1HD1P1Pkk),,2Ggt,, G tt   &mmoo " " U~~&:;;%e,,%!MMMMM&  "ts?BB-,B-rrdrcTK|||j|_||j|_||j|_||j|j |_ |d{VdSr) rrrprvrrwrrrrlrxrs rdrzExtractContentMiddleware.handles2243E3Ecl3S3STT 99#,GG// == $ ? ? cN^ _ _giirfN)rrCr[rZ)rrCr[ryrHrZr[rZ)rprgr[rZr)rprgr[r)rprgr[rgr_)rprgrrZr[rir)rrrrrrrrrrrrrrrrrrrfrdrrsD88 D#   \ " ? ? ?\ ?\$^0^0^0[^0@\***\*X\@8888\8trfrc^eZdZUdZdZehdZded<eddd Z ddZ dS)PlaceholderFilterMiddlewarez>Skip pure placeholder messages (e.g. '[image]' with no media).zplaceholder-filter>[图片][文件][视频][语音]rrrrrTSKIPPABLE_PLACEHOLDERSrrYrZ media_countrqr[r\cL|dkrdS|}||jvS)zEDetect whether the message is a pure placeholder (should be skipped).rF)rkr )rrYr rs rdis_skippable_placeholderz4PlaceholderFilterMiddleware.is_skippable_placeholders- ??5::<<3555rfrrdrcK||jt|jr-td|jj|jdS|d{VdS)Nz%[%s] Skipping placeholder message: %r)rrvrxrwr;rrerrs rdrz"PlaceholderFilterMiddleware.handlesj  ( (s3>7J7J K K  LL@#+BRTWT` a a a FgiirfN)r)rYrZr rqr[r\r) rrrrrrTr r`rrrrrfrdrrsHH D(1 333)) 6666[6rfrcteZdZUdZdZehdZded<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 /backgroundrT ALLOWLISTrYrZr[cr|}|dr d|ddz}|S)z?Normalize full-width slash to ASCII slash and strip whitespace.rrrVNrrs rdrz-OwnerCommandMiddleware._rewrite_slash_commands;zz|| ??8 $ $ "abb>D rfrjrCrprgrtrl)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)r{rrrs rdrz@OwnerCommandMiddleware._detect_owner_command..s9   uuZ  M11 111rfrVrrrYrvr)maxsplitr) rrxrrrar`rqrZrkr\) rrjrprtrl text_elemsrYcmd_linerowner_idis_owners rd_detect_owner_commandz,OwnerCommandMiddleware._detect_owner_commands=   s} $$   B    z??a  $$1 !!-006B;;FBGG--d33""3'' %$$nnan((+1133 cm # #$$  ''77=2>>DDFF>>>h,&>Hh&&rfrrdrc 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)rjrprtrlz;[%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)rer(rjrprtrlr;r<rrs _track_taskrrHsendrzrv)rrrre matched_cmdr%r's rdrzOwnerCommandMiddleware.handle'sH+*.*D*D\m) +E+ + ' Xx  x  KKM ck3+;[       3 S[*wK*w*w*wxx>>>!!!    F  $8 $ $ KKF ck3+;[   !,C #CLgiirfNr) rjrCrprgrtrZrlrZr[r r) rrrrrrTrr`rrrr(rrrfrdrrs D%9&&&I \('('('[('TrfrceZdZdZdZd dZdS) BuildSourceMiddlewarez(Build SessionSource from context fields.z build-sourcerrdr[rc K|j}||j|j|j|jpd|jp|j|jdkrdnd|_|d{VdS)Nrr)rsrtrur user_namer)rerrsrtrurlrorVrrrres rdrzBuildSourceMiddleware.handleJs+))Kmm$,)=S-= #  8 8ffd *   giirfNrrrrfrdr/r/Es322 D      rfr/ceZdZdZdZedd Zedd Zedd Ze d d 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-guardrprgr,ryr[r\ct|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. Fr{rrr+rvrrrT)rr&rrr)rpr,rrrs rd _is_at_botz!GroupAtGuardMiddleware._is_at_bot_s 5  Dxx ##66xx r2266vrBBH  H--()4    zz+&&$..6::i3H3HF3R3RttuA&&A?>A?rZc|sdS|D]}|ddkr|didd}|sI tj|}n#tjtf$rYwwxYw|ddkrU|d|kr`` so the model can distinguish participants and their user ids. r:Nr<Fr is_dispatchr^rri] r<rT)r;r=robservedrNz([%s] Failed to observe group message: %s)r>r?rForwardedRecordsParseMiddlewarebuild_forward_textrrrrQrRrPr@r@r;rEr)rrerVr@rYrrqrxrD session_entryr body_textsummary attributedrrOs rd_observe_group_messagez-GroupAtGuardMiddleware._observe_group_messages !1488  F Z!77??Mn1 GI  J9LL%3EMJ8< I4 4 47 4 4 4'IE^EEgEE)EEJ%%\X\:::DDFF E  -&,l#  & &(      Z Z Z NNEw|UX Y Y Y Y Y Y Y Y Y ZsB/C C>!C99C>c tK|j}|jdkr|js||j|jsl|||j|jp|j |j |j pd|j |td|j|j|j dS|d{VdS)Nr)rqrxrz6[%s] Group message observed (no @bot): chat=%s from=%s)rertrzr6rprZrLrVrorlrvrqrxr;r<rrsr2s rdrzGroupAtGuardMiddleware.handles+ =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-attributionrrdr[rcZK|jdkr|js|j}t|j|j|_|jpd}|j p|jpd}d|d|d|j |_ |j tj |j d|_ |d{VdS)Nrr<rrirD)r1)rtrzrer4r?rprZrrlrorvrV dataclassesr)rrrre user_id_labelnickname_labels rdrz!GroupAttributionMiddleware.handles =G # #C,= #kG!7!S!S go""C  ,9 M 0QC4DQ NP~PP PP#,PPCLz%(0tLLL giirfNrrrrfrdrOrOs9   D      rfrOceZdZdZdZdS)YuanbaoMessageTypezYuanbao-local message subtypes; coerced back to :class:`MessageType` before leaving the adapter (see :class:`DispatchMiddleware`). chat_recordN)rrrr CHAT_RECORDrrfrdrUrUs!EE KKKrfrUc6eZdZdZdZeddZdd Zd S)ClassifyMessageTypeMiddlewarez>Determine MessageType from text content and msg_body elements.zclassify-msg-typerYrZrprgcx|dr tjS|D] }|dd}|dkrtjcS|dkrtjcS|dkrtjcS|dkrtjcS|dkr|d pid d} tj |}n#tj tf$rd }YnwxYwt|tr'|d d krtjcS tjS)zClassify message type based on text and msg_body. Returns a base :class:`MessageType`, or a yuanbao-local :class:`YuanbaoMessageType` for platform-specific subtypes. rr{rvrrrrrrr+Nrr)rarCOMMANDrPHOTOVOICEVIDEODOCUMENTr&rrrrBrCrUrWr/)rYrpretyperrs rd _classifyz'ClassifyMessageTypeMiddleware._classify s` ??3   '& & : :DHHZ,,E&&"((((&&"((((***"(((( %%"++++'' HH]339r>>vrJJ"!Z11FF,i8"""!FFF"fd++: ;0G0G40O0O-9999s?CC/.C/rrdr[rctK||j|j|_|d{VdSr)rarvrpr{rs rdrz$ClassifyMessageTypeMiddleware.handle! s:~~clCLAA giirfN)rYrZrprgr)rrrrrrrarrrfrdrYrYsQHH D   \ 8rfrYc.eZdZdZdZddZdd Zdd ZdS)QuoteContextMiddlewarez3Extract quote/reply context from cloud_custom_data.z quote-contextrrrZr[#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_*. rquoteNidrvdescrorsr)) r&rrrrBrCrrZrk)rrrrrgquote_idrisender quote_texts rd_extract_quote_contextz-QuoteContextMiddleware._extract_quote_context+ sT! : Z 122FF$i0   :: (2&$'?'?I 7###T%&& :uyy,"--3355=599V$$*++1133UYY011QUYY{5K5KQrRRXXZZBFPV=))4)))D ##s 55rrdList[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. Nr:rNrvr=rrVrG:z'[%s] quote transcript lookup failed: %sryuanbao)r|rer>rVr?rAr@reversedrrBrZ_YB_RES_REF_REr|r partitionrk_RESOLVABLE_MEDIA_KINDSrr@r;rE)rrrerwrDrHhistoryr-r_contentrrrrrrrOs rd#_extract_media_refs_from_transcriptz:QuoteContextMiddleware._extract_media_refs_from_transcript@ s  " *I+13  G%5t<>MM wwqzzggajj,0NN3,?,?)a#zz||#:::&--sD(..:J:J.KLLL    NN933S          sE5EE55 F4?+F//F4rcK||j\|_|_||d{V|_|d{VdSr)rmrrr|r}rxr~rs rdrzQuoteContextMiddleware.handlej sk595P5PQTQf5g5g2!2%)%M%Mc%R%RRRRRRRgiirfN)rrrZr[re)rrdr[rnr)rrrrrrmrxrrrfrdrdrd& s\== D$$$$*((((TrfrdceZdZdZdZddZeddZe dddZ dZ eddZ eddZ dS)rFuDeep-parse WeChat forwarded chat records (elem_type 1009) for dispatch. Activates when a full ``ForwardMsgData`` dict is available on the current turn, carried by the current message (``ctx.forwarded_records``). Resolves media to ``[kind|ybres:RID]`` placeholders, appends downloadable refs to ``ctx.media_refs`` (for :class:`MediaResolveMiddleware`), and rewrites ``ctx.raw_text``. Group @bot turns *without* a forward on the current message rely on the eagerly-rendered summaries that :class:`GroupAtGuardMiddleware` writes to the transcript at observe time — there is no run-time summary fallback here. On any failure the middleware leaves ``ctx.raw_text`` untouched (graceful degradation, design §2.8). zforwarded-records-parserrdr[rc 8K |jr7||||j|d|_nG#t$r:}t dt|jdd|Yd}~nd}~wwxYw|d{VdS)NTrBz,[%s] forwarded-records deep parse failed: %srrq) rx_send_loading_heartbeatrGrvr@r;rEr>re)rrrrOs rdrz&ForwardedRecordsParseMiddleware.handle s $ i,,S111#66s7LRUcg6hh     NN> VY77         giis>A B 0BBcK |jjj|jt d{VdS#t $rYdSwxYw)z@Best-effort RUNNING heartbeat so the user sees a loading bubble.N)re _outbound heartbeatsend_heartbeat_oncersr)r@)rs rdr|z7ForwardedRecordsParseMiddleware._send_loading_heartbeat s +'1EE 1             DD s5; A A rvmediarC plain_textrZ$Tuple[str, Optional[Dict[str, str]]]c|ddp|dd}t|dpd}t|dpd}t|dpd}|pt|}|dkr>|r!|rd|d |d|d fSd |p|d fS|d vrE|r*|r(d|d }|r||d<d|d ||fSd|d fS|dkrB|p#t|dpd} d| d|d fS|dkr>|r!|rd|d |d|d fSd|p|d fSd|pdd |p|d fS)a:Render one ``msgContent.multimedia`` entry as a textual marker. Returns ``(marker, ref)``. Downloadable media emits a ``[kind|ybres:RID]`` marker and a ``ctx.media_refs`` ref dict when a usable RID/URL is present; otherwise a plain ``[kind] name`` marker and ``ref=None``. typervdoc_typerHmedia_idrrQrz] rz[image] N)rPdocumentr*rPrrz[file] rz[link] rrRrz[video] rr)rrkrqrZrrrj) rrr media_typerHrrrr link_titles rd _media_markerz-ForwardedRecordsParseMiddleware._media_marker sii++HuyyR/H/HOOQQWWYY %))E""(b))//11uyy,,23399;; +..4"55;;== J2EEcJJ   bs b9s99i99@@BBW]`DaDaaa7i5:77>>@@$F F 5 5 5 Gs G/5c&B&B,"+CK8c88Y88??AA3FF(Y((//1147 7   "Cc%))G*<*<*B&C&CJ/Z//#//6688$> >   bs b9s99i99@@BBW]`DaDaaa0i.30077994? ?>:(>>C,<9>>EEGGMMrfr forward_data/Iterator[Tuple[str, str, List[Dict[str, str]]]]c#Kt|tr|dndpgD]}t|ts|dd}|dd}|dgpg}g}|s|}n5g}|D]} t| ts| dd} | d kr*|| d d_| d krv| d gpgD]\} t| trE|| |\} } || | || ]| d kr|d|r||dd|Dp|}t ||jkr|d|jdz}|||fVdS)u4Walk ``ForwardMsgData['msg']`` and yield ``(sender, body, refs)``. Per-record dispatch over ``msgContent`` (text / multimedia / nested forward / fallback); ``body`` is capped at :attr:`FORWARD_MSG_TEXT_MAX_CHARS`. Media goes through :meth:`_media_marker`, always building full ``[kind|ybres:RID]`` markers; ``refs`` holds that record's downloadable ``ctx.media_refs`` entries in textual order — the order PatchAnchorsMiddleware relies on (design §2.10.6). Headers / footers are the caller's job. r-Nrkrv plainText msgContentrrrVrYrG multimediaru[嵌套聊天记录]z c3K|]}||V dSrr)rps rdrzEForwardedRecordsParseMiddleware._walk_forward_msgs.. s'$;$;1$;Q$;$;$;$;$;$;rfu…(已截断))rBrCrrrrrxFORWARD_MSG_TEXT_MAX_CHARS)rrr-rkr msg_contentsrrenderedrmcmc_typermarkerrs rd_walk_forward_msgsz2ForwardedRecordsParseMiddleware._walk_forward_msgs sK0:,/M/MWL$$U+++SW^\^$ )$ )Cc4(( WWXr**Fb11J77<44:L)+D J%#%&55B%b$//!  ffVQ//G!|| RVVFB%7%78888 A%'VVL"%=%=%C55E)%665.1.?.?$):/"/" !& V 4 4 4#&?$(KK$4$4$45!A %;<<<<%5!LL44499$;$;$;$;$;;;Iz8}}s===#$Dc&D$DEHXX(D( ( ( ( (I$ )$ )rfrCr\ch|jpd}d|dg}||D]<\}}}||d||r|j|=d|} |r8|jr| d|jz } | S)uRender ``ForwardMsgData`` into forward text. Body lines are ``发送人：正文`` with full ``[kind|ybres:RID]`` media markers preserved. When ``is_dispatch`` is true, refs are appended to ``ctx.media_refs`` for downstream resolution and a ``用户附言： {ctx.raw_text}`` footer is added; observed callers skip both since no later middleware runs. u用户u当前用户的昵称为u以下为用户的聊天记录u：r^u 用户附言：)rorrrwextendrrvrk) rrrrCnicknamerrkrMrrYs rdrGz2ForwardedRecordsParseMiddleware.build_forward_text s&2(6H668XY"%"8"8"F"F , , FD$ LLF--t-- . . . ,%%d+++yy  A3<--// A @#,*<*<*>*>@@ @D rfNrr_)rrCrrZr[r)rrCr[r)rrCrrdrCr\r[rZ) rrrrrrrr|rrrrrGrrfrdrFrFp s" %D     \,.)N)N)N)N[)NX"&2)2)2)[2)l[rfrFcteZdZUdZdZiZded<dZded<dZded <e d4dZ e d5dZ e d6dZ e d7dZe d8dZe d7dZe ddddd9d$Ze d:d(Ze d;d,Ze dd2Zd?d3ZdS)@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_idrZr[Optional[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)rrr rrropathisfile)rrr local_pathmimetss rd_get_cached_resourcez+MediaResolveMiddleware._get_cached_resource) s 4#'' 44 =4$ D" 9;; c7 7 7   # #K 6 6 64w~~j))    # #K 6 6 644rfrrrc|sdStjjkrItjfd}|djdzD]}j|d||t jfj|<dS)zIStore download result in cache. Evicts oldest entries when over capacity.Nc*j|dS)NrG)r)rrs rdz=MediaResolveMiddleware._put_cached_resource..B sCDWXYDZ[\D]rf)r)rxrrsortedrr )rrrr sorted_keysrs` rd_put_cached_resourcez+MediaResolveMiddleware._put_cached_resource; s  F s" # #s'C C C !4:]:]:]:]^^^K !D3#?1#D!DE 1 1#''40000,6dikk+JK(((rf media_paths List[str]mimesr\c||}|dS|\}}td|j||||||dS)z8Append a cached resource to output lists when available.NF'[%s] resource cache hit: rid=%s path=%sT)rr;rrr)rrerrrhitrrs rd_append_cached_resourcez.MediaResolveMiddleware._append_cached_resourceG sz&&{33 ;5 D 5 L+z    :&&& TtrfrHctj|j}tj|d}|dvr|SdS)z$Guess image extension from URL path.rV>.heic.tiff.bmp.gif.jpg.png.jpeg.webpr)rrrrrosplitextrq)rHrexts rd_guess_image_ext_from_urlz0MediaResolveMiddleware._guess_image_ext_from_url\ sX|$$S)).gt$$Q'--// V V VJvrfc 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_idNrTrvrVwebr,z-missing token or bot_id for resource downloadz/api/resource/v1/downloadr#)r$X-IDX-TokenX-Sourcer=T)r follow_redirectsrGr)paramsr'irrrrr*>Nrz"resource/v1/download failed: code=r.r-r+rHrealUrlz(resource/v1/download missing url/realUrlz)resource/v1/download did not return a URL)rkr?_get_cached_tokenrZrrZ_app_key _api_domainr0r1r3r>rr] _app_secretraise_for_statusr&rBrC)rer token_datarTrVr,api_urlr'rIrJresprKr*r+real_urls rd_fetch_resource_urlz*MediaResolveMiddleware._fetch_resource_urle 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=... rrrrvN) rrrr@rrrrZrkrr)rerHrr resource_idsrs rd_resolve_download_urlz,MediaResolveMiddleware._resolve_download_url s  \**3//FF   JJJ  %%fl33yy..O%))L2I2IOR 6BJc,q/**00222  J /CCG[YYYYYYYY Y   JJJ s$ 336 C C&%C&Nrv)rlog_tagr fetch_urlrrryrcK|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|d krRt|} t| p| 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. Nrr max_size_mbz5[%s] inbound media download failed: kind=%s %s err=%srQ)rz,[%s] inbound image cache rejected: %s err=%simage/ image/jpegrRzvideo/z video/mp4rPz)[%s] inbound file cache failed: %s err=%sapplication/octet-stream)rr;rrmedia_download_urlMEDIA_MAX_SIZE_MBr@rErrrAr"rarrrrrrorbasenamer)rrerrrrrr file_bytes content_typerOrrrrs rd_download_and_cachez*MediaResolveMiddleware._download_and_cache 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# # 7??/ ;;J":.. , 7 7 A AR {   $ $[*d C C Ct# # @\**955F((55?I 2:yIIJJ    NN; gs   44444   y))W\W=W   j$???4sHA'' B1#BB>C D"DDH I  "II rwrTuple[List[str], List[str]]c Kg}g}g}|D]}t|dpd}t|dpd}t|dpd} |tvs|st |} | r| ||r|||| | pdf|s||fStj j dfd tj } tjfd |Dd d id{V} ttj | z dz} d}t!|| D]\\}}}}}t#|t$r1t&dj||dd||dz }P||dz }X|\}}||||t&djj t/|t/||| ||fS)aResolve 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. Resolution runs with bounded concurrency (``adapter.media_resolve_concurrency``); see :meth:`_resolve_ybres_refs` for the same order-preserving / exception-isolated contract. rrvrHrrZrrr[rc K4d{V |d{V}nL#t$r?}tdj|||Yd}~dddd{VdSd}~wwxYw|||pdd|dd|d{Vcdddd{VS#1d{VswxYwYdS)Nz8[%s] inbound media resolve failed: kind=%s url=%s err=%szplaceholder_url=Prrrrr)rr@r;rErr) rrHrrrrOrer semaphores rd _resolve_onez@MediaResolveMiddleware._resolve_media_urls.._resolve_one# s!         &)&?&?&M&M M M M M M MII    NNR dC 444               !44'&.$9s3B3x99 # 5                              s6B7+B7 A4#A/B7/A440B77 CCc3>K|]\}}}}||||VdSrr)rrrHrrrs rdrz=MediaResolveMiddleware._resolve_media_urls..: sO66*$Xsl4h44666666rfreturn_exceptionsTNrrz9[%s] inbound media resolve crashed: kind=%s url=%s err=%srrVz[[%s] media resolve batch: scope=media concurrency=%d total=%d ok=%d failed=%d elapsed_ms=%d) rrZrHrZrrZrrZr[r)rZrrkrqrurrrrr Semaphoremedia_resolve_concurrencyr  monotonicgatherrqziprB BaseExceptionr;rErr<rx)rrerwrractiverrrHrr_t0results _elapsed_ms_failed _filename_ridrrrrrs`` @@rd_resolve_media_urlsz*MediaResolveMiddleware._resolve_media_urls s!# !# 24 < ++c1T9:: 474H4H % % 0 (T3 4&&-00 OL$CRC&1 ~1 % J   j ) ) )   t $ $ $ $  i L  - KK  OO     ;&&rfrrn log_prefixc `Kg}g}g}|D]A\}}} |tvr|||r)|||| fB|s||fStjjdfd t j} tjfd|Dd d id {V} tt j| z d z} d } t|| D]\\}}} }t|tr*tdj|||| dz } H|| dz } P|\}}||||tdjjt#|t#|| | ||fS)u9Resolve ``(rid, kind, filename)`` ybres tuples to local paths. Runs with bounded concurrency (``adapter.media_resolve_concurrency``) so cold-start turns with many anchors don't pay ``N × (RPC + download)`` sequentially. Output order matches input; per-rid failures are isolated. rrZrrr[rc K4d{V |d{V}nM#t$r@}tdj|||Yd}~dddd{VdSd}~wwxYw|||pdd||d{Vcdddd{VS#1d{VswxYwYdS)Nz-[%s] %s resolve failed: rid=%s kind=%s err=%sz rid=r)rr@r;rErr) rrr fresh_urlrOrerrrs rdrz@MediaResolveMiddleware._resolve_ybres_refs.._resolve_one{ s          &)&=&=gs&K&K K K K K K KII    NNG j#tS 444               !44'&.$)5555 # 5                              s6B2+B2 A5$A0B20A55*B22 B<?B<c3:K|]\}}}|||VdSrr)rrrrrs rdrz=MediaResolveMiddleware._resolve_ybres_refs.. s8 R R4GCxll3h// R R R R R RrfrTNrrz.[%s] %s resolve crashed: rid=%s kind=%s err=%srVz[[%s] media resolve batch: scope=ybres concurrency=%d total=%d ok=%d failed=%d elapsed_ms=%d)rrZrrZrrZr[r)rurrrrrr rrrqrrBrr;rErr<rx)rrerrrrrrrrrrrrrrrrrs`` ` @@rd_resolve_ybres_refsz*MediaResolveMiddleware._resolve_ybres_refs] sM "$ -/#' 1 1 Cx222**7CeLL  MM3h/ 0 0 0 0 &% %%g&GHH          (n R R R R6 R R R "        4>++c1T9:: -0-A-A   ) !S$6&-00 DL*c41 ~1 JD$   t $ $ $ LL      i L  - KK        E!!rfctKt|dd}|sggfS ||}||j}n<#t$r/}t d|j|ggfcYd}~Sd}~wwxYw|sggfS|t d}g}t} t|D]D} | d} t| trd| vr2tt| } t| D]} | d}| d}|d\}}}|}|t(vrc|| vrh| |||||ft/|t0krnt/|t0krnF||sggfS|||d d{VS) zYResolve recent observed image/file anchors from transcript into ``(local_paths, mimes)``.r:Nz.[%s] Observed-media hydration setup failed: %sr=rrVrGrpzobserved-mediar)r>r?rAr@r@r;rEr OBSERVED_MEDIA_BACKFILL_LOOKBACKrrrrrBrZrgrsr|rrtrkrurrrx,OBSERVED_MEDIA_BACKFILL_MAX_RESOLVE_PER_TURNreverser)rrerVrDrHrvrOrorderseenr-r=matchesrrrrrrs rd_collect_observed_mediaz.MediaResolveMiddleware._collect_observed_media s !1488 r6M !77??M++M,DEEGG    NN@ c   r6MMMMMM    r6M::;;<,.EEF##  Cggi((Ggs++ y/G/G>227;;<rerrBrZr_YB_LOCAL_MEDIA_REr|rrkrqrorrrr"rr) rpathsrrid_keycacherYrrrrrs rd_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|rfcB 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_listsrr[rc|\}}t||D]K\}}|r|vr |||LdSr)rrr)ru_listm_listurrtypesrs rd_add_unique_pairsz8MediaResolveMiddleware.handle.._add_unique_pairs2 su'NFFFF++  1AII  A Q   rfc3K|]}|dV dS)rVNr)rrs rdrz0MediaResolveMiddleware.handle..= s'55a15555555rfrrz;[%s] observed-image hydration raised, continuing anyway: %sz.[%s] Skip placeholder after media download: %r)rrr[r)rerrrwsumr|r~rrrtrrVr@r;rErrrrrrvr) rrrrer  own_pairs own_countrOrrrs @@@rdrzMediaResolveMiddleware.handle' 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  LLI7{{3<QQ giirfN)rYrZrrrrr[rZr)rrrrrrr.rrrfrdr&r&e sW   D\:rfr&c6eZdZdZdZddZedd Zd S)DispatchMiddlewarez.Build MessageEvent and dispatch to AI handler.dispatchrrdr[rc 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)r4r5r[rc KtjtdjDr tjn,t jtrjn tjj j pdj tj tjjjj }r'j r j j<jpdj<j r]jrVj}j|j <t)|dkr-t|dt)|dz D]}||=|d{VdS)Nc3@K|]}|dVdS))z application/ztext/Nra)rmts rdrzMDispatchMiddleware.handle.._dispatch_inbound_event.. s/^^2==)BCC^^^^^^rf) rYr,rVrN raw_messagerrr|r}rrvr()rrvanyrrr_rBr{r/rVrqrjrgrr|r}rr'r1rrxhandle_message)eventrr_skrers rd_dispatch_inbound_eventz:DispatchMiddleware.handle.._dispatch_inbound_event s \^^co^^^^^*K(( *4CL+)N)N*$)z:-H// 11$'$;!/"1%E( Hsz H36:+C058\5GR-c2z %cl %2$'Lcj!u::##!%[[):#e**s*:):;%%!!HH((// / / / / / / / / /rfrz-[%s] Group message enqueued (qsize=%d) for %srvrSzyuanbao-group-consumer-r.r*zyuanbao-inbound-r<r)rer;rVconfigextrarrt _group_queues setdefaultrQueue put_nowaitr;r<rqsizerH_consume_group_queue_inbound_tasksrrJrKrq) rrrr?is_newqueueconsumerrLr>res ` @@rdrzDispatchMiddleware.handle s+ J$+N$8$<$<=VX\$]$]%,^%9%=%=>XZ_%`%`     0 0 0 0 0 0 0 0@ =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 Bgiirfre'YuanbaoAdapter'r*rZcKd}|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.rNNTrz3[%s] Group queue: dispatching for %s (remaining=%d)rvrSg?z[%s] Group queue consumer error)rBrrwait_for TimeoutErrorr;rrrFr(rGr@ exceptionr)rer* _IDLE_TIMEOUTrJ dispatch_fns rdrGz'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)callabler\r@)r open_attrs rd is_connectedzConnectionManager.is_connectedH s 8 5DHfd33   4 I    IIKK(((   uu usA AAcK|j}ts=d}|d|dtd|j|dS|jr|js=d}|d|dtd |j|dS|j g t|j d d }|dust|r,|r"t d |jdSn#t$rYnwxYw|d |jdsdS td|j|jt"|j|j|j|jd {V}|drt+|d|_td|j|jt1jt5j|jd d dt8d {V|_ ||d {V}|s|d {VdSd|_| t1j!|_"t1j#|$d|j%|_&t1j#|'d|j%|_(td|j|j%|jtR*|dS#t0j+$rRtd|j|d {V|,YdSt$r\}td|j|d|d {V|,Yd }~dSd }~wwxYw)uOpen WebSocket connection: sign-token → WS connect → AUTH_BIND → start loops. Returns True on success, False on failure. z:Yuanbao startup failed: 'websockets' package not installedyuanbao_missing_dependencyT) retryablez$[%s] %s. Run: pip install websocketsFzJYuanbao startup failed: YUANBAO_APP_ID and YUANBAO_APP_SECRET are requiredyuanbao_missing_credentialsz[%s] %sNrwz*[%s] Already connected, skipping connect()zyuanbao-app-keyzYuanbao app keyz [%s] Fetching sign token from %srr,z[%s] Connecting to %s ping_interval ping_timeout close_timeoutrryuanbao-heartbeat-r* yuanbao-recv-z%[%s] Connected. connectId=%s botId=%sz[%s] Connection timed outz[%s] connect() failed: %sr)-r]WEBSOCKETS_AVAILABLE_set_fatal_errorr;rErrrrr^r>rqrr@_acquire_platform_lockr<rrrZ _route_envrrZrZ_ws_urlrrN websocketsconnectCONNECT_TIMEOUT_SECONDS _authenticate _cleanup_wsre_mark_connectedget_running_loop_looprH_heartbeat_loopr_r` _receive_loopraYuanbaoAdapter set_activerO_release_platform_lock)rrer-rrrauthedrOs rdrwzConnectionManager.openX s -# NC  $ $%A3RV $ W W W NNA7>%,,.. , ,C88:: ,!!(+++   """ !!!           s! 2AA/ A==BBrrCc :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. NFrTrvr,rVbotrybBot) biz_iduidrVrTrq app_versionoperation_system bot_versionrz&[%s] AUTH_BIND sent (msg_id=%s uid=%s)Trz+[%s] AUTH_BIND timeout waiting for BIND_ACKrrcmd_typerhrResponsez auth-bindz$[%s] BIND_ACK received: connectId=%sz[%s] BIND_ACK missing connectIdz[%s] AUTH_BIND timeoutz[%s] AUTH_BIND error: %sr)!r]r^rrZrrZuuiduuid4r1r7r8r:r,r;rrrrr AUTH_TIMEOUT_SECONDSrrNrecvrBr bytearrayr,r@r$_extract_connect_idr_r<rO)rrrerTrrVrrq auth_bytesrdeadlinerrawr-rrrrmrOs rdrzConnectionManager._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+rfNrVrrGz*[%s] AuthBindRsp error: code=%d message=%rrz$[%s] Failed to extract connectId: %s) rr%r(r'r&r;rr]rr@rE)rrr+fdictr*messagermrOs rdrz%ConnectionManager._extract_connect_id 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>rrrVz[%s] PONG timeout (%d/%d)z7[%s] Heartbeat threshold exceeded, triggering reconnectz[%s] Heartbeat send failed: %s)r]_runningrrGHEARTBEAT_INTERVAL_SECONDSr^rZrrr2r create_futurercrbr,r;rrrNrdrOrrEHEARTBEAT_TIMEOUT_THRESHOLDschedule_reconnectr@r)rrerq ping_bytesloop pong_futurerOs rdrz!ConnectionManager._heartbeat_loop7s-" " 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=%sreasonrvz7[%s] Close code %d is non-recoverable, NOT reconnectingz[%s] receive_loop exited: %s)r]r^rBrr _handle_framerrr exceptionsConnectionClosedr>r;rErNO_RECONNECT_CLOSE_CODESr_mark_disconnectedrr@)rrer close_exc close_coderOs rdrzConnectionManager._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** E775E22E7rrc .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: %sNrrrhrrvrqneed_ackFr+rfrpingz'[%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)r]r,r@r;rrrr$rcr set_resultrbrr<rxr^r3r,r-r_push_to_inbound)rrrer-rOrrrrqrr+rr ack_bytesack_excdecodeds rdrzConnectionManager._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_WINDOWrrZcT 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. rrlrvrrmrrrp __unknown_) r&rrrBrCrr@r-rh)rrrrlrmrjs rd_extract_sender_keyz%ConnectionManager._extract_sender_keysk  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)rrhrrrgrr;rr]rrxrr call_laterr_flush_inbound_buffer)rrrexisting_timerrtimers rdrz"ConnectionManager._push_to_inbounds&&x00-11#t<<  $  ! ! # # # d* * *(*D  % S!((222 @ M S)=c)B%C%C   '))  !  &    %*S!!!rfrc|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)rerhzyuanbao-pipeline-r*)rhrrgr]r;r<rrxrdr+rrH_inbound_pipeliner)rrrrers rdrz'ConnectionManager._flush_inbound_buffers   d+++(,,S"55   F- ? L#s9~~   WCCCG/  % - -c 2 2*S**        rfencoded_conn_msgreq_idr cK|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) r^r?rrrrbr,rNshieldrrOr@)rrrr rfuturers rdsend_biz_requestz"ConnectionManager.send_biz_request-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)r]rrfrrH_reconnect_with_backoffrs rdrz$ConnectionManager.schedule_reconnectMsV = ! @$*< @   < < > > ? ? ? ? ? @ @ @ @rfcK|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)rfr;rr]r _do_reconnectrs rdrz)ConnectionManager._reconnect_with_backoffRs    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#tL'|dS#tj($r(t d|j|dzYtR$r0}t d|j|dz|Yd}~4d}~wwxYwt *d|jt|+dS)z>Internal reconnect loop, called under the _reconnecting guard.rVrGrz#[%s] Reconnect attempt %d/%d in %dsNrxr,ryrzrz![%s] Re-auth failed on attempt %drr~r*rz,[%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),r]r3MAX_RECONNECT_ATTEMPTSreminr;r<rrrGrrr]rrrrrrZrZrNrrrrr^rrErdrr`rrrHrr_rarrrrOr@rr)rrerJwaitrrrOs rdrzConnectionManager._do_reconnect]s$-344A A G'.{D $qG|R((D KK5 gk+A4   -%% % % % % % % %""$$ $ $ $ $ $ $ $6 #.#<#<$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/?))'222tt' a a aDglT[^_T_`````   :GL'TU+WZ   8',H^    ""$$$us&DK(#EK((3M M'%MMc0K|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) r^rrNrWS_CLOSE_TIMEOUT_SrOr;rr]rr@)rrks rdrzConnectionManager._cleanup_wssX > &rxxzz;MNNNNNNNNNNNN'    TM&(:     >s3A :B BBNrerLr[r)r[ryr^rr)rrCr[r\)rrCr[ry)rrr[r)rrr[rZ)rrr[r)rrZr[r)rrrrZr rr[rC)rrrrrrrkrmrorsrwrrrrrrrr`rrrDEFAULT_SEND_TIMEOUTrrrrrrrfrdr[r[ s B B B B"X   X (((X(   X ````D!!!!BCCCCJ.%%%%R&&&&8T T T T p"!!!!++++> * * * *D    4. 11111@@@@@ ' ' ' 'JJJJXrfr[cXeZdZdZeddZedd 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. rerLkwargsr r[Tuple[bytes, str, str]c KdS)zReturn (file_bytes, filename, content_type). Raises: ValueError: when file cannot be acquired (not found, empty, etc.) Nrrrers rd acquire_filezMediaSendHandler.acquire_filerrf upload_resultrCrgc dS)zr file_uuidrr)rrrs rd z+MediaSendHandler.handle..s4!Q IIIqIIIrf)rrrrrYrrmrmz[%s] %s.handle() failed: %sr) _connectionr~rkrkrcancel_slow_notifierrr MessageSendervalidate_mediarr#rrrZrrrrrrrrdispatch_msg_bodyrArZr@rrr;rr)rrersrrrconnrkrrrvalidation_errrrrTr,rr fwd_kwargsrpgcverO handler_names rdrzMediaSendHandler.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<s8NN 2 2 2      rfrceZdZdZdZdZdS)ImageFileHandleruLStrategy: send image from a local file path (read → COS → TIMImageElem).cK|d}tj|std|td|j|t|d5}|}dddn #1swxYwYtj |pd}t|pd}|||fS)N image_pathFile not found: z![%s] ImageFileHandler: reading %srbrr) rorrrAr;r<rrwreadrr")rrerr(rrrrs rdrzImageFileHandler.acquire_file\s . w~~j)) >< <<== = 7zRRR *d # # "qJ " " " " " " " " " " " " " " "7##J//>;&x00@L 8\11-BBBc t|d|d|d|d|dd|dd|d Srr"rs rdrzImageFileHandler.build_msg_bodygr#rfNr$rrfrdr&r&Ys8VV 2 2 2      rfr&ceZdZdZdZdZdS)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 %srrrrrPr) r;r<rrrrr`rorrr")rrerrrrrrs rdrzFileUrlHandler.acquire_filevsz* 97<RRR); '";* * * $ $ $ $ $ $  L::j)) = s++A.Iw'' 22s rdrzStickerHandler.build_msg_bodys            zz.11 ZZ --  #)),77G !G|!G!GHHH))'22 2  #&&*=== =((**G))'22 2rfNr)rrrrrrrrrfrdr;r;sLKK:::33333rfr;cDeZdZdZddZdd Z dddZddZ dd dZdS)!GroupQueryServiceaNEncapsulates all group query operations (both low-level WS calls and higher-level AI-tool-facing wrappers). Responsibilities: - Low-level WS encode/decode for group info and member list queries - Chat-id parsing, error wrapping and result filtering for AI tools - Member cache population on the adapter rerLr[rc||_dSr)r]ris rdrzGroupQueryService.__init__s  rfrmrZricK|j}|jjdSt|}ddlm}||}|dd} |j||d{V}|di}|dd} | dkr#t d|j | dS|d d p|d d } | r$t| trt| Sd |iS#tj$r%t d |j |YdSt $r,} t d|j | Yd} ~ dSd} ~ wwxYw)zQuery group info via WS (group name, owner, member count, etc.). Returns: Decoded dict or None on failure. Nrr,rrqrstatusz'[%s] query_group_info failed: status=%dr+rfrMrmz'[%s] query_group_info timeout: group=%sz [%s] query_group_info failed: %s)r]r rkr8gateway.platforms.yuanbao_protor,rrr;rErrBrr/rrOr@) rrmreencoded_decoderrrLrrMbiz_datarOs rdquery_group_info_rawz&GroupQueryService.query_group_info_raws -   ! )4)*55NNNNNN''""* $0AA'RXAYYYYYYYYH<<++DXXh**F{{H',X^___t||FC00MHLL4M4MH =Jx77 =28<<< *- -#    NNDglT^ _ _ _44    NN=w|S Q Q Q44444 s,A5D>ADD0E< E<!E77E<rr(offsetrqlimitcK|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. NrSrTrrKrrqrLrMz,[%s] get_group_member_list failed: status=%dr+rfrMT)members next_offset is_completerWz,[%s] get_group_member_list timeout: group=%sz%[%s] get_group_member_list failed: %s)r]r rkr9rNr,rrr;rErrBrr0r  _member_cacherrOr@)rrmrSrTrerOrPrrrLrrMrQrrOs rdget_group_member_list_rawz+GroupQueryService.get_group_member_list_raws-   ! )4.z&PUVVVNNNNNN''""* $0AA'RXAYYYYYYYYH<<++DXXh**F{{Mw|]cdddt||FC00MHLL4M4MH PJx77 P9(CC%'$OO U&**Y// U59Y[[&BS4T%j1M#    NNI7 F>!F99F>rsrCcK|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. rr-This command is only available in group chatsNzFailed to query group info)rarxrR)rrsrmrs rdquery_group_infoz"GroupQueryService.query_group_infosz !!(++ NLM MS]]^^, 00<<<<<<<< >9: : rflist_allNactionrryc 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} rrr]NzFailed to query group membersrWfindcg|]}|ddpdvsX|ddpdvs,|ddpdv|S)rrv name_cardrrrq)rrrs rdrz;GroupQueryService.query_session_members..EsQUU:r228b??AAAAQUU;339r@@BBBBQUU9b117R>>@@@@A@@rf list_botscjg|]0}d|ddpdv.|1S)rrrvrerrs rdrz;GroupQueryService.query_session_members..LsB\\\QUquuZ7L7L7RPR6Y6Y6[6[-[-[q-[-[-[rfrv cg|]B}|dp*|dp|ddCS)rdrrrvrrhs rdrz;GroupQueryService.query_session_members..QsJfffYZQUU;''T155+<+<TiQS@T@TfffrfzMention with @name: z, rS)rWtotal mentionHint)rarxr[rrqr) rrsr`rrmrrW mention_hintnamesrs @rdquery_session_membersz'GroupQueryService.query_session_members*sb !!(++ NLM MS]]^^, 55jAAAAAAAA ><= =**Y++ V   JJLLE"GG { " "\\'\\\G  Es7||r))ff^efffE1DIIe4D4DDLss|\\'   rfrrmrZr[rirr(rmrZrSrqrTrqr[ri)rsrZr[rC)r_N)rsrZr`rZrryr[rC) rrrrrrRr[r^rorrfrdrHrHs    @>A#####R    "!" . . . . . . . rfrHcDeZdZdZddZdd 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 rerLr[rc0||_i|_i|_dSr)r]_reply_heartbeat_tasks_reply_hb_last_activeris rdrzHeartbeatManager.__init__ds ?A#79"""rfrsrZ heartbeat_valrqc<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)rlrmrr)rlrrRUNNINGFINISHz%[%s] Reply heartbeat %s sent: chat=%sz#[%s] send_heartbeat_once failed: %s)r]r rkrZrarxr7 removeprefixr6r,r)r;rrr@) rrsrxrerrmrOr status_namerOs rdrz$HeartbeatManager.send_heartbeat_onceise-" 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*) r]r rkrZrvrrr rwrrH_worker)rrsrerexistingrLs rdr)zHeartbeatManager.starts-" 7?'/? F.227;;  HMMOO 26)++D &w / F.2ikk"7+" LL ! !.W..   04#G,,,rfctK ||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)rrGREPLY_HEARTBEAT_INTERVAL_Srwrr REPLY_HEARTBEAT_TIMEOUT_Sr]r rkrr@r*rvr)rrs last_activer cancelleds rdrzHeartbeatManager._workersF :**74HII I I I I I I I Nm$>?????????"8<>  D99;;   #))+++ "((*****rfNr)rsrZrxrqr[rrsrZr[r)TrsrZrr\r[rr) rrrrrrr)rrrrrfrdrtrt[s:::: SSSS<4444(!:!:!:!:F     ++++++rfrtc: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. rerLrk'MessageSender'r[rc0||_||_i|_dSr)r]_sender_tasks)rrerks rdrzSlowResponseNotifier.__init__s  /1 rfrsrZcK||tj||d|}||j|<dS)zCStart a delayed task that notifies the user when the agent is slow.zyuanbao-slow-resp-r*N)rrrH _notifierrrrsrLs rdr)zSlowResponseNotifier.starts] G" NN7 # #/g//    $ GrfcK 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)rrGSLOW_RESPONSE_TIMEOUT_Sr;r<r]rrqrsend_text_chunkSLOW_RESPONSE_MESSAGErr@r)rrsrOs rdrzSlowResponseNotifier._notifiers \- 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)rrrrrs rdrzSlowResponseNotifier.cancelsS{w--     KKMMMMM    rfcKt|jD]*}|s|+|jdS)zCancel all slow-response tasks.N)rgrrrrrrs rdrzSlowResponseNotifier.closesc++--..  D99;;   rfN)rerLrkrr[rrr) rrrrrr)rrrrrfrdrrs 2222 $$$$ \ \ \ \ rfrcveZdZUdZehdZded<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)br acCore 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) >rrrrrrzClassVar[frozenset] IMAGE_EXTSrrCHAT_DICT_MAX_SIZErerLr[rc||_tj|_d|_d|_t ttttd|_ dS)N)r image_filerrr>) r] collections OrderedDict _chat_locks_on_send_start_on_send_finishrr&r/r6r;_media_handlersris rdrzMessageSender.__init__sq GRG^G`G`?C?C)***,,&(('))%'' = = rfrrZrrc||j|<dS)z1Register (or replace) a named media send handler.N)r)rrrs rdregister_handlerzMessageSender.register_handler(s%,T"""rfrsrc||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_endrxrrglockedrriterrr)rrsevictedrs rd get_chat_lockzMessageSender.get_chat_lock.s d& & &   ( ( 1 1 1#G, , t D$; ; ;GD,--  ',3355$((---"GE C $$T$t/?*@*@%A%ABBB$+LNN!((rfNrvr=rryrmrc 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.NFrTrzJ[%s] truncate_message: input=%d chars, max=%d, output=%d chunk(s) sizes=%sc,g|]}t|Srrxrs rdrz+MessageSender.send_text..Xs555c!ff555rfrr )r)r]r rkrrrstrip_cron_wrappertruncate_messageMAX_TEXT_CHUNKr;r<rrxrrrrr@)rrsr=rrmrerlockcontent_to_sendrrrr_tors rd send_textzMessageSender.send_textAs-" 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    rfrprgc@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.Nrrr rTrrrNFr Unknown errorr)rrarxsend_group_msg_bodyr|send_c2c_msg_bodyrr) rrsrprrmrgrprrs rdrzMessageSender.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;rrYretryrqc |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).rrNrr rTrrrz-[%s] send_text_chunk attempt %d/%d failed: %srVz0[%s] send_text_chunk attempt %d/%d exception: %srGz>[%s] send_text_chunk max retries (%d) exceeded. Last error: %sFzMax retries exceeded: r)r]r3rarxsend_group_messager|send_c2c_messagerrr;rErr@rZrrGr) rrsrYrrrmre last_errorrJrrrrOs rdrzMessageSender.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 WrfcK|j}dt}t|||jpd||}||||d{VS)z(Send C2C message with arbitrary MsgBody.c2c_rv)rrprlrqrmN)r]r:r4rZ_dispatch_encoded)rrrprmrerrOs rdrzMessageSender.send_c2c_msg_body2st-' '')! .B!    ++GWfEEEEEEEEErfcK|j}dt}t|||jpd||pd}||||d{VS)z*Send group message with arbitrary MsgBody.grp_rv)rmrprlrq ref_msg_idN)r]r:r5rZr)rrmrprrerrOs rdrz!MessageSender.send_group_msg_body?sz-' ''+! .B~2    ++GWfEEEEEEEEErfrOrrc 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.rLNTrqrv)rrFzRequest timeout after sr)r rrrrOrr@rZ)rerOrrLrOs rdrzMessageSender._dispatch_encodedSs  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)r)rrr max_bytessize_mbs rdrzMessageSender.validate_mediabs  ZA!5!5,(,, ,$&- z??Y & &*oo,t3GThTT'TTTKTTT Ttrfr max_lengthrrrsrc|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|Sr_) _INDICATOR_REsubrs rdrz2MessageSender.truncate_message..s(;;;q-##B**;;;rf)rxrXr)r=rrrr~rs rdrzMessageSender.truncate_messageusq} 4==J & &9 #66 Z7   <;F;;;.vvgY.rfc8|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: )rarbryrxrk)r=divider footer_prefix divider_pos footer_posr body_startrMs rdrz MessageSender.strip_cron_wrappers!!"677 N'e ll7++ ]]=11 ??j1nn k0I0IN+&  & &N 3w<</ z*,-3355wrfc<K|jdS)zCRelease chat locks (no-op for now; placeholder for future cleanup).N)rrrs rdrzMessageSender.closes!      rfr)rrZrrr[rrsrZr[rr rsrZr=rZrryrmrZr[rr) rsrZrrZrryrryrr r[rrrsrZrrZrrr[r) rsrZrprgrryrmrZr[r)Nrrv) rsrZrYrZrryrrqrmrZr[rr_)rrZrYrZrmrZr[rC)rmrZrYrZrryr[rC)rYrZrmrZr[rg)rrZrprgrmrZr[rC)rmrZrprgrryr[rC)rerLrOrrrZr[rCrrrrrZrrqr[ryr)r=rZrrqrrrsr[r)r=rZr[rZr)rrrrrTrr`rrrrrrrrrrrrzr{ MULTILINErrrrrrrrrrrrfrdr r 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////\/<\,!!!!!!rfr ceZdZUdZejZded<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. rrrerLr[rc||_t||_t||_t ||j|_|j|j_|j |j_ dSr) r]r rkrtrr slow_notifier_handle_send_startr_handle_send_finishrris rdrzOutboundManager.__init__s_ %27%;%; +;G+D+D3GQUQ\3]3]&*%< "&*&> ###rfrsrZc:|j|dS)zFCalled by MessageSender before sending: cancel slow-response notifier.Nrrrrss rdrz"OutboundManager._handle_send_start !!'*****rfcVK|j|td{VdS)z=Called by MessageSender after sending: send FINISH heartbeat.N)rrr*r s rdrz#OutboundManager._handle_send_finishs7n00:MNNNNNNNNNNNrfNrvr=rryrmrcNK|j||||d{VS)z%Send text message with auto-chunking.r N)rkr)rrsr=rrms rdrzOutboundManager.send_texts: [**7GXR\*]]]]]]]]]rfrrr c:K|jj||fi|d{VS)rN)rkr)rrsrrs rdrzOutboundManager.send_medias9,T[+G\LLVLLLLLLLLLrfrrrrcJK|j|||d{VS)z.Send text + media (used by send_message tool).N)rkr)rrsrrs rdrzOutboundManager.send_directs4 [,,Wg{KKKKKKKKKrfcJK|j|d{VdS)z Start reply heartbeat (RUNNING).N)rr)r s rd start_typingzOutboundManager.start_typings4n""7+++++++++++rfFrr\cNK|j||d{VdS)zStop reply heartbeat.rN)rr)rrsrs rd stop_typingzOutboundManager.stop_typings9n!!'{!CCCCCCCCCCCrfcJK|j|d{VdS)zStart slow-response notifier.N)rr)r s rdstart_slow_notifierz#OutboundManager.start_slow_notifiers5 &&w///////////rfc:|j|dS)zCancel slow-response notifier.Nrr s rdr z$OutboundManager.cancel_slow_notifierr rfrc6|j|S)z@Proxy to MessageSender.get_chat_lock for backward compatibility.)rkrr s rdrzOutboundManager.get_chat_locks{((111rfcollections.OrderedDictc|jjS)z>Proxy to MessageSender._chat_locks for backward compatibility.)rkrrs rdrzOutboundManager._chat_lockss{&&rfrrrrrrqc:t|||S)z&Proxy to MessageSender.validate_media.)r r)rrrs rdrzOutboundManager.validate_medias ++J+NNNrfcK|jd{V|jd{V|jd{VdS)zShut down all sub-managers.N)rkrrrrs rdrzOutboundManager.close sk!!!!!!!!!n""$$$$$$$$$ &&(((((((((((rfrrrr)rsrZrrZrr r[rrr)Frr)r[rrrr)rrrrr rr`rrrrrrrrrr rrrrrrrrfrdrrs  )6(HHHHH????++++OOOOEI^^^^^MMMM9=LLLLL,,,,DDDDD0000++++2222'''X'GIOOOO\O ))))))rfrceZdZUdZejZdZded<dZ dZ ded<dZ d ed <d Z d ed <e dSdZe dTdZdUfd ZdVdZedWdZdXd Zd!d"dYd$ZdZd%Z d[d\d0Zd]d2Zd^d_d4Zd`d5Zdafd7 Zdbd8Z dcddd=Zd>ZdedfdAZ dgdhdDZ dgdidFZ! dgdjdIZ" dgdkdMZ# dldmdOZ$dndQZ%dndRZ&xZ'S)orzCYuanbao AI Bot adapter backed by a persistent WebSocket connection.rrqrTrSrirREPLY_REF_MAX_ENTRIESNz$ClassVar[Optional['YuanbaoAdapter']]_active_instancer[Optional['YuanbaoAdapter']c|jS)z7Return the currently connected YuanbaoAdapter, or None.rrs rd get_activezYuanbaoAdapter.get_actives ##rfrerc||_dS)z0Register (or clear) the active adapter instance.Nr!)rres rdrzYuanbaoAdapter.set_active"s 'rfr@rrr c &t|tj|jpi}|dpd|_|dpd|_|dpd|_ |dpt|_ |dpt d|_|dpd|_ t!|d t"}n#t$t&f$r t"}YnwxYwt)t*t-t.||_t3||_t7||_t;|_t;|_i|_ d |_!tEd |_#i|_$i|_%i|_&i|_'|d ptQj)dd*}|dptQj)dd}d|+dD}|dptQj)dd*}|dptQj)dd} d| +dD} tY|||| |_-t]||_/t`1|_2tQj)dp|j3r |j3j4nd} tk| o| 6d |_7dS)Napp_idrvrr,ws_urlrrrrrOi,) ttl_secondsr`YUANBAO_DM_POLICYrzraYUANBAO_DM_ALLOW_FROMc^g|]*}||+Srrrxs rdrz+YuanbaoAdapter.__init__..ls2#b#b#b!XYX_X_XaXa#bAGGII#b#b#brf,rbYUANBAO_GROUP_POLICYrcYUANBAO_GROUP_ALLOW_FROMc^g|]*}||+Srrr+s rdrz+YuanbaoAdapter.__init__..ws2&h&h&hQ^_^e^e^g^g&hqwwyy&h&h&hrf)r`rarbrcrr)8superrrYUANBAOrArrkrrrZDEFAULT_WS_GATEWAY_URLrDEFAULT_API_DOMAINrjrrrq_DEFAULT_RESOLVE_CONCURRENCYrrAmax_MIN_RESOLVE_CONCURRENCYr_MAX_RESOLVE_CONCURRENCYrr[r rr~rrHrIrZrrrrBr'r1rrorprqr`r_rrH _group_queryrTrYr home_channelrsr\rar) rr@r_extra_raw_concurrencyr`_dm_allow_from_rawrarb_group_allow_from_rawrc_existing_homers rdrzYuanbaoAdapter.__init__'s !1222#$jj228b??AA !'L!9!9!?R F F H H&,jj&:&:&Bd #ZZ11K5KRRTT !'L!9!9!O=O W WX[ \ \ & ; 7 7 =2DDFF  <"6::.IKg#h#hii  :& < < <;    <.1 $ (*: ; ;/ / & /@.E.E*9$*?*?2558EE =?).*c::: 8:46 57"35 JJ{ # # 9y,i88 %''%%''  JJ ' ' 6y0"55 $c#b7I7O7OPS7T7T#b#b#b  JJ~ & & <y/;; %''%%''  JJ) * * 9y3R88 'i&h:O:U:UVY:Z:Z&h&h&h*'%-    .d333I2N2N2P2P#9:: +1+> FF  ' 'B )-^(<(<(h^E^E^_gEhEhAhs(E..F F rL asyncio.Taskcx|j|||jj|S)z@Register a fire-and-forget task so it won't be GC'd prematurely.)rIrrJrKrs rdr+zYuanbaoAdapter._track_tasks8 ""4((( t5=>>> rfr\cdS)zCYuanbao gates DM/group access at intake via dm_policy/group_policy.Trrs rdenforces_own_access_policyz)YuanbaoAdapter.enforces_own_access_policys trfrrdc |j}t|jpd}|sdS|jdkrW||rdS|jdkr3ddlm}| tj j |SdS|jdkrmt|j pd}|sdS|jd kr||S|jd kr|SdSdS) zTrue when the sender may persist ``YUANBAO_HOME_CHANNEL``. Intake-only pairing forwards are excluded until the sender is on the strict allowlist, has explicit open-world opt-in, or is approved in the pairing store. rvFrTrzr) PairingStorerrvrw)rrZrlrkrtrxr`gateway.pairingrE is_approvedrr2rrmrbrrr)rrrrkrErms rdrz)YuanbaoAdapter._sender_may_designate_homes/ $2S%+,,2244 5 =D ##F++ t9,,888888#|~~11(2B2H&QQQ5 =G # #S^1r2288::J u"k11..z:::"f,,//1115urfF) is_reconnectrHcDK|jd{VS)zhConnect to Yuanbao WS gateway and authenticate. Delegates to ConnectionManager.open(). N)r rw)rrHs rdrzYuanbaoAdapter.connects/ %**,,,,,,,,,rfcVKtj|urtdd|_|||jd{V|jd{Vt|j D]*}| s| +|j |j td|jdS)z;Cancel background tasks and close the WebSocket connection.NFz[%s] Disconnected)rrrrrrr rr~rgrHrrrrBr;r<rrs rd disconnectzYuanbaoAdapter.disconnects#  *d 2 2  % %d + + +  !!! ##%%%$$&&&&&&&&&n""$$$$$$$$$,--  D99;;   !!###   """ '33333rfrvrsrZr=rrymetadataOptional[Dict[str, Any]]rmrcNK|j||||d{VS)zCSend text message with auto-chunking. Delegates to OutboundManager.r N)r~r)rrsr=rrLrms rdr,zYuanbaoAdapter.sends:^--gwU_-`````````rfrcDK|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. rr)rrrr8r s rd get_chat_infozYuanbaoAdapter.get_chat_infos:   h ' ' 6#W55 5...rfricnK |j|d{VdS#t$rYdSwxYw)zGSend "typing" status heartbeat (RUNNING). Delegates to OutboundManager.N)r~rr@)rrsrLs rd 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)r~rr@r s rdrzYuanbaoAdapter.stop_typings`  .,,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)rVrsr~rr1_process_message_backgroundr )rr=r*rsrs rdrUz*YuanbaoAdapter._process_message_background s,&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)r9rRr~s rdr^zYuanbaoAdapter.query_group_infos/&;;JGGGGGGGGGrfrr(rSrTcLK|j|||d{VS)z9Query group member list (delegates to GroupQueryService).rVN)r9r[)rrmrSrTs rdget_group_member_listz$YuanbaoAdapter.get_group_member_lists:&@@TZbg@hhhhhhhhhrfi'rrYcK|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)rr )rrxrrx DM_MAX_CHARSr,)rrrYrmrss rdsend_dmzYuanbaoAdapter.send_dm(s"0099 Ue3STTT T t99t( ( (***+.@@D%G%%YYwYDDDDDDDDDrfrrcBK|jj|df|||d|d{VS)zKSend image message (URL). Delegates to OutboundManager via ImageUrlHandler.r)rrrNr~r)rrsrrrrLrs rd send_imagezYuanbaoAdapter.send_image?s_/T^. [ w)           rfr(cBK|jj|df|||d|d{VS)zISend local image file. Delegates to OutboundManager via ImageFileHandler.r)rrr(Nr])rrsr(rrrLrs rdrzYuanbaoAdapter.send_image_fileOs_/T^. \ w:           rfrrcBK|jj|df|||d|d{VS)zISend file message (URL). Delegates to OutboundManager via FileUrlHandler.r)rrrNr])rrsrrrrLrs rd send_filezYuanbaoAdapter.send_file_s_/T^. Z 8           rfrDrE Optional[int]cBK|jj|df|||d|d{VS)zDSend sticker/emoji. Delegates to OutboundManager via StickerHandler.r>)rrDrENr])rrsrDrErrs rd send_stickerzYuanbaoAdapter.send_stickeros`/T^. Y %*            rfr8cDK|jj|df||||d|d{VS)zMSend local file (document). Delegates to OutboundManager via DocumentHandler.r)rrr8rNr])rrsr8rrrrLrs rdrzYuanbaoAdapter.send_documentsb/T^. Z w(            rfrCcvKt|j|j|j|jd{VS)zBAAAA$$$[$'''['iiiiiiiiiiii^X<5:------44446#'-1 a a a a a / / / /    999999HHHH >AiiiiiLEEEEE6"&"&#'      ("&"&#'      (#'"&#'      &'+$("&      (#'!%"&#'     $             rfrr[rc4tS)z,Delegate to ``YuanbaoAdapter.get_active()``.)rr"rrfrdget_active_adapterros  $ $ & &&rfrerLrsrZrrrrcJK|j|||d{VS)z,Delegate to ``OutboundManager.send_direct``.N)r~r)rersrrs rdsend_yuanbao_directrqs5"..wMM M M M M M MMrfrjr) rerLrsrZrrZrrr[r)r __future__rrrrrrQrrr&loggingrorzr5r  urllib.parserrrrrenumrpathlibrabcr r typingr r r rrrrrsysr0rwebsockets.exceptionsr ImportErrorgateway.configrrgateway.platforms.baserrrrrrrgateway.platforms.helpersrgateway.platforms.yuanbao_mediarrrrr r!r"r#rNr$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:gateway.sessionr; getLoggerrr; hermes_clir<_HERMES_VERSIONr7r:rZr9rr8r3r4rrrrrrrrAUTH_FAILED_CODESAUTH_RETRYABLE_CODESrrREPLY_REF_TTL_Srrr{rsrrTrurr r r5r7r8rXrrarbrrdrrrrrr rVr\r_rrrrrr/r4rOrUrYrdrFrr&r1rTr[rrr&r/r6r;rHrtrr rrrorqrrfrdrs "#"""""     2222222222########QQQQQQQQQQQQQQQQQQQQ      JJJ43333333:999992.-----  8 $ $ 9999999OOO  s-..L L6!@?? '&&)))!  YI RZ ABB$)$>$>$>?? -.. $& /1, !^ ^ ^ ^ ^ ^ ^ ^ @p)p)p)p)p)p)p)p)f54444444 >)>)>)>)>)>)>) >)BAAAAAAAA6]]]]]]]]|]]]]](]]]@/"     '   NdNdNdNdNd-NdNdNdb*&-"D"D"D"D"D"D"D"D"N-200000-000fttttt0tttl "32^^^^^.^^^B-$~~~~~.~~~B!2>        $$$$$$5$$$NGGGGG.GGGTkkkkk&7kkk\D D D D D .D D D N/////.///d_9_9_9_9_9*_9_9_9D%%%%%%%%N[ [ [ [ [ [ [ [ z~=~=~=~=~=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 ])])])])])])])])@Q Q Q Q Q (Q Q Q r ''''59 NNNNNNNs$ B BBDD&%D&