epjEdZddlmZddlZddlZddlZddlZddlZddlZddl 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 l7m8Z9ddl:m;Z;ddlm?Z?m@Z@mAZAmBZBmCZCmDZDddlEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\ddl]m^Z^e j_e`Za ddlbmcZdn #e+$rdZdYnwxYwedZeedZfegeMZhe&jiZjdZkdZldZmdZndZodZpdZqdZrhdZsdZthdZuhdZvdZwdZxd Zyd!Zzd"Z{e j|d#Z}e j|d$Z~ehd%Ze j|d&Zd'Zd(Zd)Zd*Zd(ZGd+d,ZGd-d.Zdd/lmZmZeGd0d1ZGd2d3eZGd4d5ZGd6d7eZGd8d9eZGd:d;eZGd<d=eZGd>d?eZGd@dAeZGdBdCZGdDdEeZGdFdGeZGdHdIeZGdJdKeZGdLdMeZGdNdOeZGdPdQeZGdRdSeZGdTdUeZGdVdWeZGdXdYeZGdZd[eZGd\d]eZGd^d_eZGd`daeZGdbdcZGdddeZGdfdgeZGdhdieZGdjdkeZGdldmeZGdndoeZGdpdqeZGdrdsZGdtduZGdvdwZGdxdyZGdzd{ZGd|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)helpers)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 textstrreturnboolc*tj|S)z7Detect whether the text has unclosed code block fences.)_mdchunktext_has_unclosed_fencerZs ?/home/thesage/.hermes/hermes-agent/gateway/platforms/yuanbao.pyhas_unclosed_fencez$MarkdownProcessor.has_unclosed_fences/555c*tj|S)z.Detect whether the text ends with a table row.)r_text_ends_with_table_rowras rbends_with_table_rowz%MarkdownProcessor.ends_with_table_rows0666rdN max_charsintlen_fnOptional[Callable[[str], int]]tuple[str, str]c0tj|||S)zJFind the nearest paragraph boundary within max_chars; return (head, tail).rj)r_split_at_paragraph_boundary)rZrhrjs rbroz-MarkdownProcessor.split_at_paragraph_boundarys3D)FSSSSrdc*tj|S)zDDetermine whether an atomic block is a code block (starts with ```).)r_ is_fence_atomras rbrqzMarkdownProcessor.is_fence_atom%d+++rdc*tj|S)zHDetermine whether an atomic block is a table (first line starts with |).)r_ is_table_atomras rbrtzMarkdownProcessor.is_table_atomrrrd list[str]c*tj|S)z6Split text into a list of indivisible "atomic blocks".)r_split_markdown_atomsras rbsplit_into_atomsz"MarkdownProcessor.split_into_atomss,T222rdc4tj|||ddS)a( Split Markdown text into multiple chunks by max_chars. Guarantees (provided by the shared core, prefer_paragraphs mode): - 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 TF)prefer_paragraphsbalance_fences)r_split_text_fence_aware)clsrZrhrjs rbchunk_markdown_textz%MarkdownProcessor.chunk_markdown_texts-".   "     rd prev_chunk next_chunkc,tj||S)z>Infer the separator ('\n' or '\n\n') between two split chunks.)r_infer_block_separator)r~rrs rbrz'MarkdownProcessor.infer_block_separators-j*EEErdchunksc*tj|S)z=Stream-aware fence-conscious chunk merging (see shared core).)r_merge_streaming_fences)r~rs rbmerge_block_streaming_fencesz.MarkdownProcessor.merge_block_streaming_fences s.v666rdcX|s|S|d}t|dkr|S|d}|d}tjd|tjs|S|dkr|Sd|dd}|S)a Strip outer Markdown fence. When AI reply is entirely wrapped in ```markdown\n...\n```, remove the outer fence, keeping the content. Only strip when the first line is ```markdown (case-insensitive) and the last line is ```. Args: text: Text to process Returns: Text with outer fence stripped (returns original if no match)  rz^```(?:markdown|md)?\s*$z```rW)splitlenstriprematch IGNORECASEjoin)rZlines first_line last_lineinners rbstrip_outer_markdown_fencez,MarkdownProcessor.strip_outer_markdown_fence's K 4   u::>>K1X^^%% "IOO%% x3ZOO K   K %"+&& rdcfd|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 |rz^\|[\s\-:]+(\|[\s\-:]+)+\|$c3jK|].}|r|n|V/dSNr).0cells rb z.ksO** )- > $******rdz||) rr startswithendswithrrrappendreplace)rZr result_lineslinestrippedcells normalizeds rbsanitize_markdown_tablez)MarkdownProcessor.sanitize_markdown_tableMsS 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&&&rdc 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.rrdrbmarkdown_hint_system_promptz-MarkdownProcessor.markdown_hint_system_prompt|s V  rd)rZr[r\r]r)rZr[rhrirjrkr\rl)rZr[r\ruryN)rZr[rhrirjrkr\ru)rr[rr[r\r[)rrur\rurZr[r\r[r\r[)__name__ __module__ __qualname____doc__ staticmethodrcrgrorqrtrx classmethodrrrrrrrrdrbrYrYs  $666\6 777\7 26TTTT\T,,,\,,,,\,333\3 15     [ 4FFF[F 777[7 !!!\!J*'*'*'\*'\   \   rdrYc,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'rrA<r?zdict[str, dict[str, Any]]_cachezdict[str, asyncio.Lock]_locksapp_keyr[r\ 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)r~rs rbget_refresh_lockzSignManager.get_refresh_locks0 #* $ $"),..CJw z'""rdnonce 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 rbcompute_signaturezSignManager.compute_signaturesN !G+j8x ))++U\\^^W^LLVVXXXrdctjttd}|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 rbbuild_timestampzSignManager.build_timestamps< )!*<*<*>>8999rdentrydict[str, Any]r]cL|dtjz |jkS)zEDetermine whether the cache entry is valid (not expired with margin). expire_ts)timeCACHE_REFRESH_MARGIN_S)r~rs rbis_cache_validzSignManager.is_cache_valids"[!DIKK/#2LLLrdNonec8|jdS)z;Clear all per-app_key refresh locks (called on disconnect).N)rclearr~s rb clear_lockszSignManager.clear_lockss rdrictjfd|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 rb z-SignManager.purge_expired..sE   !QQUU;***Q.. ...rdN)rritemspopr)r~ expired_keysrrs @rb purge_expiredzSignManager.purge_expireds~ikk    ***,,    $ $A JNN1d # # # #<   rdr 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)./timeoutNrW)rr signaturerapplication/json) Content-Typez X-AppVersionzX-OperationSystemz X-Instance-Idz X-Bot-Versionz X-Route-EnvzSign token request: url=%s%srz (retry )r)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) rstrip TOKEN_PATHhttpx AsyncClientHTTP_TIMEOUT_Srange MAX_RETRIESsecrets token_hexrr _APP_VERSION_OPERATION_SYSTEM_YUANBAO_INSTANCE_ID _BOT_VERSIONloggerinfopost status_coderZ RuntimeErrorr Exception ValueErrorr isinstancedictRETRYABLE_CODEwarning RETRY_DELAY_Srsleep)r~rrrrurlclientattemptrrrpayloadrresponsebody result_dataexcrrrs rbfetchzSignManager.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)Ndurationrtokenrrproductsourcer(rr&r)r*r) rrrrrirrrrrr$) r~rrrrcachedremainrr&rs rb get_tokenzSignManager.get_token7s ((  c((00  ,ty{{:;;F KKK|||d{VS)zFAllow middleware instances to be called directly (duck-typing compat).N)rar_s rb__call__zInboundMiddleware.__call__s,[[g.........rdc2d|jjd|jdS)N) __class__rr[r`s rb__repr__zInboundMiddleware.__repr__s"@4>*@@$)@@@@rdN)r\r9r]r r\rr) rrrrr[r5r rardrjrrdrbrZrZs  DNNNNTTT^T////AAAAAArdrZcxeZdZdZddZeddZdddZddd 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 _middlewaresris rb__init__zInboundPipeline.__init__s"$rdNcFt|tr |j|fS||fS)zHNormalize (name, handler) or (InboundMiddleware,) into (name, callable).)rrZr[) name_or_mwhandlers rb _normalizezInboundPipeline._normalizes/ j"3 4 4 /?J. .7""rd'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 )rtror)r`rrrswhenr[hs rbusezInboundPipeline.uses=//*g66a   $4111 rdtargetr[c|||\}}tfdt|jDd}|||f}||j|n|j|||S)zEInsert a middleware before *target* (by name). Appends if not found.c36K|]\}\}}}|k|VdSrrrin_rzs rbrz-InboundPipeline.use_before.. 1VV,!YaA!v++A++++VVrdNrtnext enumeraterorinsert r`rzrrrsrwr[rxidxrs ` rb use_beforezInboundPipeline.use_befores//*g66aVVVV)D4E*F*FVVVX\]]q$ ;   $ $U + + + +   $ $S% 0 0 0 rdc|||\}}tfdt|jDd}|||f}||j|n|j|dz||S)zDInsert a middleware after *target* (by name). Appends if not found.c36K|]\}\}}}|k|VdSrrr}s rbrz,InboundPipeline.use_after..rrdNrWrrs ` rb use_afterzInboundPipeline.use_afters//*g66aVVVV)D4E*F*FVVVX\]]q$ ;   $ $U + + + +   $ $S1We 4 4 4 rdr[c8fd|jD|_|S)zRemove a middleware by name.c.g|]\}}}|k |||fSrr)rrrxwr[s rbrz*InboundPipeline.remove..s+UUU71a1PT99aAY999rdrn)r`r[s `rbremovezInboundPipeline.removes'UUUUd6GUUU rdr<c$d|jDS)zAReturn ordered list of registered middleware names (for testing).cg|]\}}}| Srr)rrrs rbrz4InboundPipeline.middleware_names..$s333gaA333rdrnris rbmiddleware_namesz InboundPipeline.middleware_names!s43!23333rdr\r9cVK|jddfd d{VdS)zKRun all middlewares in order. Each middleware receives ``(ctx, next_fn)``.rr\rcKtkra\}}}dz | |s2 |d{Vn,#t$rtd|dwxYwdSdS)NrWz'[InboundPipeline] middleware [%s] errorTexc_info)rrrerror)r[rswhen_fnchainr\indexr]s rbr]z(InboundPipeline.execute..next_fn-s#e**$$).u&gw &wws||&!'#w////////// LL!JD[_L```%$s A )A3Nr2rn)r`r\rrr]s `@@@rbexecutezInboundPipeline.execute(sa!          giirdr2rNN)r\ru)rzr[r\ru)r[r[r\ru)r\r<r\r9r\r)rrrrrprrtryrrrpropertyrrrrdrbrlrls  %%%% ###\#                444X4 rdrlcVeZdZdZdZeddZedd 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_bodyr<r\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. rPMsgTyper msg_content MsgContentrZrPr)rrrr[rloadsrr)rresultitemrPrs rbconvert_json_msg_bodyz&DecodeMiddleware.convert_json_msg_bodyIsN 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.). NrAr From_AccountrBGroupIdgroup_idrEMsgBodycallback_command to_account To_AccountrD nick_namerCmsg_seqrMsgSeqrFmsg_keyMsgKeyrGCloudCustomData bot_owner_id botOwnerIdrecall_msg_seq_listlog_exttrace_id) rrArrDrBrCrrFrErGrrr)rrrrr)rrArB msg_body_rawrEs rbparse_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 r\r9rc .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)Nrz;[%s] Push decoded but no valid message. raw hex(first64)=%sz(empty)z#[%s] Frame decoded (via=%s): len=%drE TIMTextElemrZrrz;[%s] Merged %d extra msg_body elements from aggregated pushzC[%s] Push decoded (via=%s): from=%s group=%s msg_id=%s msg_types=%srArBrFc:g|]}|ddS)rPrrres rbrz+DecodeMiddleware.handle..s& I I IqQUU:r " " I I Irdz[%s] Push payload: %s) r=rr:rrr[hexrrr?r@debug) r`r\r] data_list merged_pushr@rr?via extra_body_seps rbrazDecodeMiddleware.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.>IIIgiirdN)rr<r\r<)rrr\r)rrr\rr) rrrrr[rrrrrarrdrbrr>s D\(/ / / \/ f*444444rdrceZdZdZdZd dZdS) ExtractFieldsMiddlewarez8Extract common fields from ctx.push into ctx attributes.zextract-fieldsr\r9r\rcK|j}|dd|_|dd|_|dd|_|dd|_|dg|_|dd|_|dd|_|d{VdS) NrArrBrCrDrErFrG) r?rrArBrCrDrErFrG)r`r\r]r?s rbrazExtractFieldsMiddleware.handlesx88NB77,33,33"hh'8"==xx B// XXh++ $)V >V>V!+5gg (D *4ff $   rdrc |j}|jpi}|dkr|dpg}n8|dpd}|d}|s|r||dgng}|s"td|jdS|dpd}|d pd} |D]} | dp#t| dpd} | s>||| } | | || | || p|j | } | || || | dS) NrrrFrr)rFrz2[%s] Recall callback with empty seq_list, skippingrBrA) r:r?rrrr[rr[_find_processing_session_interrupt_for_recall_msg_content_cache_patch_transcript)r`r\rr:r?seq_listmidseqrBrA seq_entry recalled_id matched_skrecalled_contents rbrz$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 irdrrNcj|jD]\}}||kr ||jvr|cSdSr)_processing_msg_idsr_active_sessions)r:rskrs rbrz.RecallGuardMiddleware._find_processing_session;sK288::  GBk!!bG,D&D&D trd 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)rZ message_typer*internalz+[%s] Recall interrupt: msg_id=%s session=%sr) rrTEXTr_pending_messagesrrsetrrr[_processing_msg_texts_schedule_content_redact) r~r:rrrBrAwhere recall_text synth_event active_event recalled_texts rbrz+RecallGuardMiddleware._interrupt_for_recallBs=*4[%%%%9[\9[9[ ,/: , ,CH , , , #$)$$Wj,GG    2=!+./33K@@  #       A7._redactks G%5t<.>#  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)rBr[rAr[)r\r9rr[r\r)rr[r\rN) rr[rr[rBr[rAr[r\r) rr[rr[rBr[rAr[r\rr) rr[rBr[rAr[rrNr\r)rrrrr[ frozensetrrrarrrrrrr rrrdrbrrs0 D y&&"`I&&&&   \ iiii@\ #h#h#h[#hJ!B!B!B[!BJOS:d:d:d:d[:d:d:drdrc6eZdZdZdZedd Zdd ZdS)SkipSelfMiddlewarezFilter out bot's own messages.z skip-selfrAr[rrNr\r]c|r|sdS||kS)z2Detect whether the message is from the bot itself.Fr)rArs rb_is_self_referencez%SkipSelfMiddleware._is_self_references# 6 5v%%rdr\r9rcK||j|jjr-td|jj|jdS|d{VdS)Nz'[%s] Ignoring self-sent message from %s)r4rAr:_bot_idrrr[r_s rbrazSkipSelfMiddleware.handlesf  " "3#3S[5H I I  LLBCKDTVYVf g g g FgiirdN)rAr[rrNr\r]r)rrrrr[rr4rarrdrbr2r2sQ(( D&&&\& rdr2ceZdZdZdZd dZdS) ChatRoutingMiddlewarez9Determine chat_id, chat_type, chat_name from push fields.z chat-routingr\r9r\rcK|jr*d|j|_d|_|jp|j|_n)d|j|_d|_|jp|j|_|d{VdS)Nrrrr)rBrHrIrCrJrArDr_s rbrazChatRoutingMiddleware.handles > D33>33CK#CMN||_||_||_||_dSr) _dm_policy_dm_allow_from _group_policy_group_allow_from)r`r<r=r>r?s rbrpzAccessPolicy.__init__s)$+)!1rdr]ctjdddvrdStjdddvS)NGATEWAY_ALLOW_ALL_USERSr>1yestrueTYUANBAO_ALLOW_ALL_USERS)osgetenvlowerris rb_open_dm_opted_inzAccessPolicy._open_dm_opted_insQ 9. 3 3 9 9 ; ;?S S S4y2B77==??CWWWrd sender_idc|jdkrdS|jdkr||jvS|jdkr|SdS)u:Strict DM authorization — pairing does not imply access.disabledF allowlistopen)rArrBrN)r`rOs rb is_dm_allowedzAccessPolicy.is_dm_allowed s] ?j ( (5 ?k ) )??$$(;; ; ?f $ $))++ +urdct|pd}|sdS|jdkrdS|jdkr ||jvS|jdkrdS|jdkr|SdS)z?Whether a DM may reach gateway intake (pairing handshake path).rFrQrRpairingTrS)r[rrArBrN)r`rO principals rbis_dm_intake_allowedz!AccessPolicy.is_dm_intake_alloweds R((..00  5 ?j ( (5 ?k ) ) 33 3 ?i ' '4 ?f $ $))++ +urdrBc|jdkrdS|jdkr||jvS|jdkrdS|jdkr|SdS)zGPlatform-level group chat inbound filter (open / allowlist / disabled).rQFrRrVrS)rCrrDrNr`rBs rbis_group_allowedzAccessPolicy.is_group_allowed"su   + +5   , ,##%%)?? ?   * *5   ' '))++ +urdc|jSr)rAris rbr<zAccessPolicy.dm_policy.s rdc|jSr)rCris rbr>zAccessPolicy.group_policy2s !!rdN) r<r[r=rur>r[r?rur\rr\r])rOr[r\r])rBr[r\r]r) rrrrrprNrTrXr[rr<r>rrdrbr;r;s 2 2 2 2XXXX         X"""X"""rdr;ceZdZdZdZd dZdS) AccessGuardMiddlewarez.Platform-level DM/Group access control filter.z access-guardr\r9r\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) r:_access_policyrIrXrArrr[r<r[rBr>)r`r\r]r:policys rbrazAccessGuardMiddleware.handle<s+&5 =D ..s/?@@  =L#"2F4D   ]g % %**3>::  >L#.&2EgiirdNrrrrdrbr`r`7s388 Drdr`ceZdZdZdZd 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-sethomer\r9r\rcK|j}|js'||rtjdd}| p|do |jdk}|jdkrd|_|r ddlm}ddl m }m }|} | d z } || } |j | d<|| | t|j tjd<td |j|j |jn8#t&$r+} td |j| Yd} ~ nd} ~ wwxYw|d{VdS) NYUANBAO_HOME_CHANNELrrrTr)get_hermes_home)atomic_config_writeread_user_config_rawz config.yamlz=[%s] Auto-sethome: designated %s (%s) as Yuanbao home channelz[%s] Auto-sethome failed: %s)r:_auto_sethome_done_sender_may_designate_homerKrLrrIhermes_constantsrhhermes_cli.configrirjrHr[environrrr[rJrr) r`r\r]r: _cur_home _should_setrhrirj_home config_path user_configrs rbrazAutoSetHomeMiddleware.handle`s+) Tg.P.PQT.U.U T "8"==I N((22Ls}7L }$$-1* TT@@@@@@[[[[[[[[+O--E"'-"7K)=(<[(I(IK:=+K 67'' [AAA9A#hh>P>P#hwh-*:::VY 222333  : LL8 9 9 9yyrdrNc2|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.rNr{z[link: z | visit link for full content])rrrrrJSONDecodeError TypeErrorr[)rxrparsedr{s rb_format_link_understandingz3ExtractContentMiddleware._format_link_understandings**Y'' 4 Z((F)3FD)A)AK6::f%%%tDD$i0   DDD  :dC00 4>>>>>sAAA76A7rc^|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 r resourceId resourceidr) urllibparseparse_qsurlparsequeryrr[rr)rridss rb_parse_resource_idz+ExtractContentMiddleware._parse_resource_ids 2 L))&,*?*?*D*D*JKKE))L))JUYY|-D-DJC*-53s1v;;$$&&&2 5   22 sBB B,+B,rEr<c ` 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 rPrrrrZ TIMImageElemimage_info_arrayNrWrr [image|ybres:r|[image] TIMFileElem file_namefileNamefilenamez[file:|ybres: [file|ybres:[file: [file] TIMSoundElemz [voice|ybres:[voice]TIMVideoFileElem [video|ybres:[video] TIMCustomElemrz[unsupported message type] elem_typez [mention]z [chat record] TIMFaceElemr[z[emoji: z[emoji][ )rrrr<rrr[rrrrrrrrAttributeErrorr)r~rEpartselemrrrZr image_info image_urlridrfile_url sound_url video_urldata_valrxctyperaw_data face_name face_datas rb _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;=T99UUrZcr|}|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. ／rrWNrrras rb_rewrite_slash_commandz/ExtractContentMiddleware._rewrite_slash_commands; zz|| ??8 $ $ "abb>D rdList[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"}] rPrrrrNrWrrrRkindrrrrrrQr[)rrrr<rr[rr) rErefsrrPrrrrrrrefs rb_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$$$ rdctg}|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.rPrrrrrrr{rrN) rrrrrrrr[r) rEurlsrdata_strrxrr{rrs rb_extract_link_urlsz+ExtractContentMiddleware._extract_link_urlsTsN  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,FF32F3rrr>c|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``. rPrrrrrrext_mapNct|tr|sdS tj|}n#tjt f$rYdSwxYwt|}t|tr| ddkr|SdS)Nsub_typerW) rr[base64 b64decodebinasciiErrorrr/rr)valuepbrs rb _parse_valuezIExtractContentMiddleware._extract_forwarded_records.._parse_values!%-- U 4 )%00BB  3   44 .r22dD)) dhhz.B.Ba.G.GKts0A  A wexin_forward_msg_) rrrrrrrrr) rErrrrrxrrkeyrrs rb_extract_forwarded_recordsz3ExtractContentMiddleware._extract_forwarded_recordsus N' "' "DdD)) TXXj-A-A_-T-Thh}b117RGgt,, {{62..H  H--()4    vt,, K1H1HD1P1Pkk),,2Ggt,, G tt   &mmoo " " U~~&:;;%e,,%!MMMMM&  "ts?BB-,B-r\r9rcTK|||j|_||j|_||j|_||j|j |_ |d{VdSr) rrrErKrrLrrVrrArMr_s rbrazExtractContentMiddleware.handles2243E3Ecl3S3STT 99#,GG// == $ ? ? cN^ _ _giirdN)rxrr\r[)rxrr\rNrr[r\r[)rEr<r\r[r)rEr<r\r)rEr<r\r<r4)rEr<rr[r\r>r)rrrrr[rrrrrrrrrrrrarrdrbrvrvsD88 D#   \ " ? ? ?\ ?\$^0^0^0[^0@\***\*X\@8888\8trdrvc^eZdZUdZdZehdZded<eddd Z ddZ dS)PlaceholderFilterMiddlewarez>Skip pure placeholder messages (e.g. '[image]' with no media).zplaceholder-filter>[图片][文件][视频][语音]rrrrr0SKIPPABLE_PLACEHOLDERSrrZr[ media_countrir\r]cL|dkrdS|}||jvS)zEDetect whether the message is a pure placeholder (should be skipped).rF)rr)r~rZrrs rbis_skippable_placeholderz4PlaceholderFilterMiddleware.is_skippable_placeholders- ??5::<<3555rdr\r9rcK||jt|jr-td|jj|jdS|d{VdS)Nz%[%s] Skipping placeholder message: %r)rrKrrLrrr:r[r_s rbraz"PlaceholderFilterMiddleware.handlesj  ( (s3>7J7J K K  LL@#+BRTWT` a a a FgiirdN)r)rZr[rrir\r]r) rrrrr[r0rr5rrrarrdrbrrsHH D(1 333)) 6666[6rdrcteZdZUdZdZehdZded<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 /backgroundr0 ALLOWLISTrZr[r\cr|}|dr d|ddz}|S)z?Normalize full-width slash to ASCII slash and strip whitespace.rrrWNrras rbrz-OwnerCommandMiddleware._rewrite_slash_commands;zz|| ??8 $ $ "abb>D rdr?rrEr<rIrA)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)rPrrrs rbrz@OwnerCommandMiddleware._detect_owner_command..s9   uuZ  M11 111rdrWrrrZrr)maxsplitr) rrrrrrrMr[rr]) r~r?rErIrA text_elemsrZcmd_linerowner_idis_owners rb_detect_owner_commandz,OwnerCommandMiddleware._detect_owner_commands=   s} $$   B    z??a  $$1 !!-006B;;FBGG--d33""3'' %$$nnan((+1133 cm # #$$  ''77=2>>DDFF>>>h,&>Hh&&rdr\r9rc 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)r?rErIrAz;[%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-r[z4[%s] Bot owner slash command: chat=%s from=%s cmd=%s)r:rr?rErIrArrr[rH _track_taskrr#sendrOrK)r`r\r]r: matched_cmdrrs rbrazOwnerCommandMiddleware.handlesH+*.*D*D\m) +E+ + ' Xx  x  KKM ck3+;[       3 S[*wK*w*w*wxx>>>!!!    F  $8 $ $ KKF ck3+;[   !,C #CLgiirdNr) r?rrEr<rIr[rAr[r\rr) rrrrr[r0rr5rrrrrarrdrbrrs D%9&&&I \('('('[('TrdrceZdZdZdZd dZdS) BuildSourceMiddlewarez(Build SessionSource from context fields.z build-sourcer\r9r\rc K|j}||j|j|j|jpd|jp|j|jdkrdnd|_|d{VdS)Nrr)rHrIrJr user_namer)r:rrHrIrJrArDr*r`r\r]r:s rbrazBuildSourceMiddleware.handle6s+))Kmm$,)=S-= #  8 8ffd *   giirdNrrrrdrbrr1s322 D      rdrceZdZdZdZedd 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-guardrEr<rrNr\r]cv|sdS|D]}|ddkr|dpidd}|sJ tj|}n#tjtf$rYxwxYw|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. FrPrrrrrrrT)rrrrr)rErrrrxs rb _is_at_botz!GroupAtGuardMiddleware._is_at_botKs 5  Dxx ##66//52::62FFH  H--()4    zz+&&$..6::i3H3HF3R3RttuA''B?Br[c|sdS|D]}|ddkr|dpidd}|sJ tj|}n#tjtf$rYxwxYw|ddkrU|d|krrct|dd}|sdS ||} |jpd} |} |r*t||d} | r |r|d| n| } d|d| d | } d | t jtj  d d }|r||d<| | j |dS#t$r,}td|j|Yd}~dSd}~wwxYw)asWrite a group message into the session transcript without triggering the agent. This allows the model to see the full group conversation when it is eventually invoked via @bot. Messages are stored with ``role: "user"`` in the format ``[nickname|user_id]\n`` so the model can distinguish participants and their user ids. rNrFr\ is_dispatchrrr] rrT)rrrobservedr*z([%s] Failed to observe group message: %s)rrrForwardedRecordsParseMiddlewarebuild_forward_textrrrr-r.r,rrrrr[)r~r:r*rrZr\rFrMr session_entryr body_textsummary attributedrr#s rb_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)rFrMr\z6[%s] Group message observed (no @bot): chat=%s from=%s)r:rIrOrrEr6r$r*rDrArKrFrMrrr[rHr s rbrazGroupAtGuardMiddleware.handles+ =G # #C,= #dooVYVbdkdsFtFt #  ' 'S%8%LCr\rr) rrrrr[rrrrrr$rarrdrbr r Cs D\.\(    \   !%,0 *Z*Z*Z*Z*Z[*ZXrdr ceZdZdZdZd dZdS) GroupAttributionMiddlewarea Tag group @bot messages with [nickname|user_id] attribution and channel_prompt. For group messages that pass the @bot guard (i.e. the bot is mentioned), this middleware: - Builds a per-turn channel_prompt so the model knows its identity and the attribution scheme. - Rewrites ctx.raw_text to ``[nickname|user_id]\n`` to match the observed-history format. - Suppresses the runner's default ``[user_name]`` shared-thread prefix by clearing ``source.user_name``. zgroup-attributionr\r9r\rcZK|jdkr|js|j}t|j|j|_|jpd}|j p|jpd}d|d|d|j |_ |j tj |j d|_ |d{VdS)Nrrrrr)r )rIrOr:r rrEr6rWrArDrKr* dataclassesr)r`r\r]r: user_id_labelnickname_labels rbraz!GroupAttributionMiddleware.handles =G # #C,= #kG!7!S!S go""C  ,9 M 0QC4DQ NP~PP PP#,PPCLz%(0tLLL giirdNrrrrdrbr'r's9   D      rdr'ceZdZdZdZdS)YuanbaoMessageTypezYuanbao-local message subtypes; coerced back to :class:`MessageType` before leaving the adapter (see :class:`DispatchMiddleware`). chat_recordN)rrrr CHAT_RECORDrrdrbr-r-s!EE KKKrdr-c6eZdZdZdZeddZdd Zd S)ClassifyMessageTypeMiddlewarez>Determine MessageType from text content and msg_body elements.zclassify-msg-typerZr[rEr<cx|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. rrPrrrrrrrrNrr)rrCOMMANDrPHOTOVOICEVIDEODOCUMENTrrrrrrr-r/r )rZrEretyperrxs rb _classifyz'ClassifyMessageTypeMiddleware._classifys` ??3   '& & : :DHHZ,,E&&"((((&&"((((***"(((( %%"++++'' HH]339r>>vrJJ"!Z11FF,i8"""!FFF"fd++: ;0G0G40O0O-9999s?CC/.C/r\r9r\rctK||j|j|_|d{VdSr)r9rKrErPr_s rbraz$ClassifyMessageTypeMiddleware.handle s:~~clCLAA giirdN)rZr[rEr<r)rrrrr[rr9rarrdrbr1r1sQHH D   \ 8rdr1c.eZdZdZdZddZdd Zdd ZdS)QuoteContextMiddlewarez3Extract quote/reply context from cloud_custom_data.z quote-contextrGr[r\#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_*. rquoteNidrdescrDrOr) rrrrrrrr[r)r`rGrr?quote_idrAsender quote_texts rb_extract_quote_contextz-QuoteContextMiddleware._extract_quote_contextsT! : Z 122FF$i0   :: (2&$'?'?I 7###T%&& :uyy,"--3355=599V$$*++1133UYY011QUYY{5K5KQrRRXXZZBFPV=))4)))D ##s 55r\r9List[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. Nrr*rrrrWrH:z'[%s] quote transcript lookup failed: %sr[yuanbao)rQr:rr*rrrreversedrrr[_YB_RES_REF_REfinditerr partitionr_RESOLVABLE_MEDIA_KINDSrrrr)r`r\r:rLrr historyrr_contentmheadrrrrr#s rb#_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)rErGrQrRrSrSr_s rbrazQuoteContextMiddleware.handleVsk595P5PQTQf5g5g2!2%)%M%Mc%R%RRRRRRRgiirdN)rGr[r\r=)r\r9r\rFr)rrrrr[rErSrarrdrbr<r<s\== D$$$$*((((Trdr<ceZdZdZdZddZeddZe dddZ dZ eddZ eddZ dS)ruDeep-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-parser\r9r\rc DK |jr=||d{V||j|d|_nG#t$r:}t dt|jdd|Yd}~nd}~wwxYw|d{VdS)NTrz,[%s] forwarded-records deep parse failed: %sr[rI) rM_send_loading_heartbeatrrKrrrrr:)r`r\r]r#s rbraz&ForwardedRecordsParseMiddleware.handleps $ i223777777777#66s7LRUcg6hh     NN> VY77         giisAA B 0BB cK |jjj|jt d{VdS#t $rYdSwxYw)z@Best-effort RUNNING heartbeat so the user sees a loading bubble.N)r: _outbound heartbeatsend_heartbeat_oncerHr*r)r\s rbrWz7ForwardedRecordsParseMiddleware._send_loading_heartbeats +'1EE 1             DD s5; A A rmediar plain_textr[$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``. typerdoc_typermedia_idrrRrz] rz[image] N)rQdocumentrrQr[rz[file] rzz[link] rrSrz[video] rr\)rrrMr[rvrr) r~r\r] media_typerrbrrr link_titles rb _media_markerz-ForwardedRecordsParseMiddleware._media_markersii++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>>EEGGMMrdrw 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. rNrCr plainText msgContentr`rrWrZrH multimediaru[嵌套聊天记录]z c3K|]}||V dSrr)rps rbrzEForwardedRecordsParseMiddleware._walk_forward_msgs..s'$;$;1$;Q$;$;$;$;$;$;rdu…(已截断))rrrrrfrrFORWARD_MSG_TEXT_MAX_CHARS)r~rgrrCr] msg_contentsrrenderedrmcmc_typer\markerrs rb_walk_forward_msgsz2ForwardedRecordsParseMiddleware._walk_forward_msgssK0:,/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$ )$ )rdrr]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：ru 用户附言：)rDrurrLextendrrKr) r~rgr\rnicknamerrCr!rrZs rbrz2ForwardedRecordsParseMiddleware.build_forward_texts&2(6H668XY"%"8"8"F"F , , FD$ LLF--t-- . . . ,%%d+++yy  A3<--// A @#,*<*<*>*>@@ @D rdNrr4)r\rr]r[r\r^)rgrr\rh)rgrr\r9rr]r\r[) rrrrr[rarrWrrfrorurrrdrbrr\s" %D     \,.)N)N)N)N[)NX"&2)2)2)[2)l[rdrcteZdZUdZdZiZded<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_idr[r\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)r{rrr}rrKpathisfile)r~rr local_pathmimetss rb_get_cached_resourcez+MediaResolveMiddleware._get_cached_resource s 4#'' 44 =4$ D" 9;; c7 7 7   # #K 6 6 64w~~j))    # #K 6 6 644rdrrrc|sdStjjkrItjfd}|djdzD]}j|d||t jfj|<dS)zIStore download result in cache. Evicts oldest entries when over capacity.Nc*j|dS)NrH)r{)rr~s rbz=MediaResolveMiddleware._put_cached_resource... sCDWXYDZ[\D]rd)r)rr{rsortedrr)r~rrr sorted_keysrs` rb_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(((rd 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)rrrr[r)r~r:rrrhitrrs rb_append_cached_resourcez.MediaResolveMiddleware._append_cached_resource3 sz&&{33 ;5 D 5 L+z    :&&& Ttrdrctj|j}tj|d}|dvr|SdS)z$Guess image extension from URL path.rW>.heic.tiff.bmp.gif.jpg.png.jpeg.webpr)rrrrrKsplitextrM)rrexts rb_guess_image_ext_from_urlz0MediaResolveMiddleware._guess_image_ext_from_urlH sX|$$S)).gt$$Q'--// V V VJvrdc 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_idNr(rr*webrz-missing token or bot_id for resource downloadz/api/resource/v1/downloadr)rX-IDX-TokenX-Sourcer>T)rfollow_redirectsrHr)paramsrirrrrr>Nrz"resource/v1/download failed: code=rrrrrealUrlz(resource/v1/download missing url/realUrlz)resource/v1/download did not return a URL)rr_get_cached_tokenr[rr6_app_key _api_domainrrrrrr1 _app_secretraise_for_statusrrr)r:r token_datar(r*rapi_urlrrrresprrrreal_urls rb_fetch_resource_urlz*MediaResolveMiddleware._fetch_resource_urlQ 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=... rrrrN) rrrrrrrr[rrzr)r:rrr resource_idsrs rb_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&Nr)rlog_tagr fetch_urlrrrNrcK|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=%srR)rz,[%s] inbound image cache rejected: %s err=%simage/ image/jpegrSzvideo/z video/mp4rQz)[%s] inbound file cache failed: %s err=%sapplication/octet-stream)rrrr[media_download_urlMEDIA_MAX_SIZE_MBrrrrrr#rrrrrrrKrbasenamer)r~r:rrrrrr file_bytes content_typer#rrrrs rb_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 rLrTuple[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. rrrr[r[rrr\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)rrrrr[r) rrrrrr#r:r~ semaphores rb _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)rrrrrrs rbrz=MediaResolveMiddleware._resolve_media_urls..& sO66*$Xsl4h44666666rdreturn_exceptionsTNrwrz9[%s] inbound media resolve crashed: kind=%s url=%s err=%srrWz[[%s] media resolve batch: scope=media concurrency=%d total=%d ok=%d failed=%d elapsed_ms=%d) rr[rr[rr[rr[r\r)r[rrrMrNrvrrrr Semaphoremedia_resolve_concurrencyr monotonicgatherrizipr BaseExceptionrrr[rr)r~r:rLrTrUactiverrrrr_t0results _elapsed_ms_failed _filename_ridrrrrrs`` @@rb_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     ;&&rdrrF 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. rr[rrr\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)rrrrr[r) rrr fresh_urlr#r:r~rrs rbrz@MediaResolveMiddleware._resolve_ybres_refs.._resolve_oneg 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 rbrz=MediaResolveMiddleware._resolve_ybres_refs..} s8 R R4GCxll3h// R R R R R RrdrTNrwrz.[%s] %s resolve crashed: rid=%s kind=%s err=%srWz[[%s] media resolve batch: scope=ybres concurrency=%d total=%d ok=%d failed=%d elapsed_ms=%d)rr[rr[rr[r\r)rNrrrrrrrrrirrrrrr[rr)r~r:rrrrrrrrrrrrrrrrrs`` ` @@rb_resolve_ybres_refsz*MediaResolveMiddleware._resolve_ybres_refsI 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!!rdctKt|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)``.rNz.[%s] Observed-media hydration setup failed: %srrrWrHrHzobserved-mediar)rrrrrrrr[ OBSERVED_MEDIA_BACKFILL_LOOKBACKr rJrrr[r<rKrLrrMrrNr%rr,OBSERVED_MEDIA_BACKFILL_MAX_RESOLVE_PER_TURNreverser)r~r:r*rr rOr#windoworderseenrrmatchesrQrRrrrrs rb_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;;<>$''  HHTNNN"27#3#3D#9#9:: $ 5O  LL    LL    e|rdcB 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)rr%r)ru_listm_listurQrtypesrs rb_add_unique_pairsz8MediaResolveMiddleware.handle.._add_unique_pairs su'NFFFF++  1AII  A Q   rdc3K|]}|dV dS)rWNr)rrs rbrz0MediaResolveMiddleware.handle..) s'55a15555555rdrrz;[%s] observed-image hydration raised, continuing anyway: %sz.[%s] Skip placeholder after media download: %r)rrr\r)r:r rrLsumrQrSrrrIrr*rrrr[rTrUrrrKr) r`r\r]r:r own_pairs own_countr#rrrs @@@rbrazMediaResolveMiddleware.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 giirdN)rZr[rrrrr\r[r)rrrrr[rr rarrdrbrrQ sW   D\:rdrc6eZdZdZdZddZedd Zd S)DispatchMiddlewarez.Build MessageEvent and dispatch to AI handler.dispatchr\r9r\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)rrr\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/Nr)rmts rbrzMDispatchMiddleware.handle.._dispatch_inbound_event.. s/^^2==)BCC^^^^^^rd) rZrr*r* raw_messagerTrUrQrRrWrr)rrKanyrUrr7rrPr r*rFr?r<rTrQrRrWrr rrhandle_message)eventrr_skr:r\s rb_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((// / / / / / / / / /rdrz-[%s] Group message enqueued (qsize=%d) for %srrTzyuanbao-group-consumer-rrzyuanbao-inbound-rr2)r:r<r*configextrarrI _group_queues setdefaultrQueue put_nowaitrrr[qsizer#_consume_group_queue_inbound_tasksr%r&r'rF) r`r\r]ris_newqueueconsumerr(rr:s ` @@rbrazDispatchMiddleware.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 Bgiirdr:'YuanbaoAdapter'rr[cKd}|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.rONTrz3[%s] Group queue: dispatching for %s (remaining=%d)rrTg?z[%s] Group queue consumer error)r rrwait_for TimeoutErrorrrr[r$rrr exceptionr)r:r _IDLE_TIMEOUTr( dispatch_fns rbr%z'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) _receive_loopr?YuanbaoAdapter set_activer-_release_platform_lock)r`r:rrPrauthedr#s rbrSzConnectionManager.openD s -# NC  $ $%A3RV $ W W W NNA7cancelrCancelledErrorr?rr@valuesdone set_exceptionrrrrg)r`disc_excfuts rbclosezConnectionManager.close s   (  ' ' ) ) ) *********)    #'D ? # O " " $ $ $ o%%%%%%%%)    "DO =>>%,,.. , ,C88:: ,!!(+++   """ !!!           s! 2AA/ A==BBrrc :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. NFr(rrr*botrybBot) biz_iduidr*r(rF app_versionoperation_system bot_versionrz&[%s] AUTH_BIND sent (msg_id=%s uid=%s)Trz+[%s] AUTH_BIND timeout waiting for BIND_ACKrrRcmd_typerrResponsez auth-bindz$[%s] BIND_ACK received: connectId=%sz[%s] BIND_ACK missing connectIdz[%s] AUTH_BIND timeoutz[%s] AUTH_BIND error: %sr)!r;r<rr6rar[uuiduuid4r2r r rrrrr[rrirAUTH_TIMEOUT_SECONDSrr,recvrr bytearrayr-rr%_extract_connect_idr=rr-)r`rr:r(r~r*rrF auth_bytesrjdeadline remainingrawrrRrrrKr#s rbrfzConnectionManager._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.rrdNrWrrHz*[%s] AuthBindRsp error: code=%d message=%rrz$[%s] Failed to extract connectId: %s) rr&r)r(r'rrr;r[rr)r`rrfdictrmessagerKr#s rbrz%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?rrrWz[%s] PONG timeout (%d/%d)z7[%s] Heartbeat threshold exceeded, triggering reconnectz[%s] Heartbeat send failed: %s)r;_runningrrHEARTBEAT_INTERVAL_SECONDSr<r[rrr3ri create_futurerAr@rrrr[r,rBr-rrHEARTBEAT_TIMEOUT_THRESHOLDschedule_reconnectrrs)r`r:rF ping_bytesloop pong_futurer#s rbrkz!ConnectionManager._heartbeat_loop# 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.Nrz3[%s] WebSocket connection closed: code=%s reason=%sreasonrz7[%s] Close code %d is non-recoverable, NOT reconnectingz[%s] receive_loop exited: %s)r;r<rrr _handle_framerrsrc exceptionsConnectionClosedrrrr[NO_RECONNECT_CLOSE_CODESr_mark_disconnectedrr)r`r:r close_exc close_coder#s rbrlzConnectionManager._receive_loopL 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: %sNrRrrrrrFneed_ackFrrdrpingz'[%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-rrrr[rr%rAru set_resultr@rrrr<r4rr.rv_push_to_inbound)r`rr:rr#rRrrrFrrrxr ack_bytesack_excdecodeds rbrzConnectionManager._handle_frameh 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_WINDOWrr[cT 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. rrArrrBrrrH __unknown_) rrrrrrrr.r@)r`rrrArBr?s rb_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)rrFrrrrErrrr;r[rrri call_laterr_flush_inbound_buffer)r`rrexisting_timerrtimers rbrz"ConnectionManager._push_to_inbound s&&x00-11#t<<  $  ! ! # # # d* * *(*D  % S!((222 @ M S)=c)B%C%C   '))  !  &    %*S!!!rdrc|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:r=zyuanbao-pipeline-r)rFrrEr;rrr[rr9rrr#_inbound_pipeliner)r`rrr:r\s rbrz'ConnectionManager._flush_inbound_buffers   d+++(,,S"55   F- ? L#s9~~   WCCCG/  % - -c 2 2*S**        rdencoded_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) r<rrrirr@rr,shieldrr-r)r`rrrrfuturers rbsend_biz_requestz"ConnectionManager.send_biz_requests" 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;rrDrr#_reconnect_with_backoffris rbrz$ConnectionManager.schedule_reconnect9sV = ! @$*< @   < < > > ? ? ? ? ? @ @ @ @rdcK|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)rDrrr;r[ _do_reconnectris rbrz)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#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.rWrHrz#[%s] Reconnect attempt %d/%d in %dsNrVrrWrXrz![%s] Re-auth failed on attempt %drr\rr]z,[%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;rMAX_RECONNECT_ATTEMPTSrCminrrr[rrrgrr1rrrrarr[r6r,rcrdrbrer<rfrrBrhr>rurrr#rkr=r?rlrmrnr-rrr)r`r:rwaitrrpr#s rbrzConnectionManager._do_reconnectIs$-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<rr,ryWS_CLOSE_TIMEOUT_Sr-rrr;r[r)r`rIs rbrgzConnectionManager._cleanup_wssX > &rxxzz;MNNNNNNNNNNNN'    TM&(:     >s3A :B BBNr:r*r\r)r\rNr3r^r2)rrr\r])rrr\rN)rrr\r)rrr\r[)rrr\r)rr[r\r)rrrr[rrr\r)rrrrrprrIrKrMrQrSryrfrrkrlrrr5rrrDEFAULT_SEND_TIMEOUTrrrrrgrrdrbr9r9 s B B B B"X   X (((X(   X ````D!!!!BCCCCJ.%%%%R&&&&8T T T T p"!!!!++++> * * * *D    4. 11111@@@@@ ' ' ' 'JJJJXrdr9cXeZdZdZeddZedd 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:r*kwargsr r\Tuple[bytes, str, str]c KdS)zReturn (file_bytes, filename, content_type). Raises: ValueError: when file cannot be acquired (not found, empty, etc.) Nrr`r:rs rb acquire_filezMediaSendHandler.acquire_filerbrd upload_resultrr<c dS)zr file_uuidrr)rrrs rb z+MediaSendHandler.handle..s4!Q IIIqIIIrd)rrrrrZrrBrBz[%s] %s.handle() failed: %sr) _connectionrYrCrIrcancel_slow_notifierrr MessageSendervalidate_mediarr$rrr6rrrrar rrrdispatch_msg_bodyrr[rr`rrrr[)r`r:rHrrrconnrCrrrvalidation_errrrr(rrr fwd_kwargsrEgcver# handler_names rbrazMediaSendHandler.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 rbrzImageFileHandler.build_msg_bodySrrdNrrrdrbrrEs8VV 2 2 2      rdrceZdZdZdZdZdS)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 %srrrrrQr) rrr[rrrrrKrrr#)r`r:rrrrrrs rbrzFileUrlHandler.acquire_filebsz* 97<RRR); '";* * * $ $ $ $ $ $  L::j)) = s++A.Iw'' 22cK|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-rRrFrstatusz'[%s] query_group_info failed: status=%drrdr!rBz'[%s] query_group_info timeout: group=%sz [%s] query_group_info failed: %s)r;rrIr9gateway.platforms.yuanbao_protor-rrrrr[rrr0rr-r) r`rBr:encoded_decoderrr rRr-biz_datar#s rbquery_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<rroffsetrilimitcK|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. Nr3r4rr+rRrFr,r-z,[%s] get_group_member_list failed: status=%drrdr!T)members next_offset is_completer7z,[%s] get_group_member_list timeout: group=%sz%[%s] get_group_member_list failed: %s)r;rrIr:r.r-rrrrr[rrr1r _member_cacherr-r)r`rBr3r4r:r/r0rrr rRr-r1rr#s rbget_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>rHrcK|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)rrr2)r`rHrBrs rbquery_group_infoz"GroupQueryService.query_group_infosz !!(++ NLM MS]]^^, 00<<<<<<<< >9: : rdlist_allNactionr[rNc 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 membersr7findcg|]}|ddpdvsX|ddpdvs,|ddpdv|S)rxr name_cardrrrM)rrQrs rbrz;GroupQueryService.query_session_members..1sQUU:r228b??AAAAQUU;339r@@BBBBQUU9b117R>>@@@@A@@rd list_botscjg|]0}d|ddpdv.|1S)r{rxrrErrQs rbrz;GroupQueryService.query_session_members..8sB\\\QUquuZ7L7L7RPR6Y6Y6[6[-[-[q-[-[-[rdr cg|]B}|dp*|dp|ddCS)rDrxrrrrHs rbrz;GroupQueryService.query_session_members..=sJfffYZQUU;''T155+<+<TiQS@T@TfffrdzMention with @name: z, rT)r7total mentionHint)rrr;rrMr) r`rHr@r[rBrr7 mention_hintnamesrs @rbquery_session_membersz'GroupQueryService.query_session_memberssb !!(++ NLM MS]]^^, 55jAAAAAAAA ><= =**Y++ V   JJLLE"GG { " "\\'\\\G  Es7||r))ff^efffE1DIIe4D4DDLss|\\'   rdrrBr[r\r>rrrBr[r3rir4rir\r>)rHr[r\r)r?N)rHr[r@r[r[rNr\r) rrrrrpr2r;r>rOrrdrbr(r(s    @>A#####R    "!" . . . . . . . rdr(cDeZdZdZddZdd 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:r*r\rc0||_i|_i|_dSr)r;_reply_heartbeat_tasks_reply_hb_last_activerGs rbrpzHeartbeatManager.__init__Ps ?A#79"""rdrHr[ heartbeat_valric<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)rArBrZr)rArrZRUNNINGFINISHz%[%s] Reply heartbeat %s sent: chat=%sz#[%s] send_heartbeat_once failed: %s)r;rrIr6rrr8 removeprefixr7rr*rrr[r) r`rHrXr:rrBr/r status_namer#s rbr[z$HeartbeatManager.send_heartbeat_onceUse-" 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;rrIr6rVrrurrWrr#_worker)r`rHr:rexistingr(s rbrzHeartbeatManager.startss-" 7?'/? F.227;;  HMMOO 26)++D &w / F.2ikk"7+" LL ! !.W..   04#G,,,rdctK ||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)r[r*rrREPLY_HEARTBEAT_INTERVAL_SrWrrREPLY_HEARTBEAT_TIMEOUT_Sr;rrIrsrr+rVr)r`rH last_activer cancelleds rbr_zHeartbeatManager._workersF :**74HII I I I I I I I Nm$>?????????"8<>  D99;;   #))+++ "((*****rdNr)rHr[rXrir\rrHr[r\r)TrHr[rfr]r\rr2) rrrrrpr[rr_rhryrrdrbrTrTGs:::: SSSS<4444(!:!:!:!:F     ++++++rdrTc: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:r*rC'MessageSender'r\rc0||_||_i|_dSr)r;_sender_tasks)r`r:rCs rbrpzSlowResponseNotifier.__init__s  /1 rdrHr[cK||tj||d|}||j|<dS)zCStart a delayed task that notifies the user when the agent is slow.zyuanbao-slow-resp-rN)rrrr# _notifierrrr`rHr(s rbrzSlowResponseNotifier.starts] G" NN7 # #/g//    $ GrdcK 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)rrSLOW_RESPONSE_TIMEOUT_Srrr;r[rirqsend_text_chunkSLOW_RESPONSE_MESSAGErsrr)r`rHr#s rbrtzSlowResponseNotifier._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)rrrrurrrus rbrrzSlowResponseNotifier.cancelsS{w--     KKMMMMM    rdcKt|jD]*}|s|+|jdS)zCancel all slow-response tasks.N)r<rrrtrurrrrjs rbryzSlowResponseNotifier.closesc++--..  D99;;   rdN)r:r*rCror\rrkr2) rrrrrprrtrrryrrdrbrnrns 2222 $$$$ \ \ \ \ rdrncveZdZUdZehdZded<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) >rrrrrrzClassVar[frozenset] IMAGE_EXTSrwr|CHAT_DICT_MAX_SIZEr:r*r\rc||_tj|_d|_d|_t ttttd|_ dS)N)r image_filerrcr) r; collections OrderedDict _chat_locks_on_send_start_on_send_finishrrrrr_media_handlersrGs rbrpzMessageSender.__init__sq GRG^G`G`?C?C)***,,&(('))%'' = = rdr[r[rsrc||j|<dS)z1Register (or replace) a named media send handler.N)r)r`r[rss rbregister_handlerzMessageSender.register_handlers%,T"""rdrHrc||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_endrr~r<lockedrriterrr)r`rHevictedrs rb get_chat_lockzMessageSender.get_chat_locks d& & &   ( ( 1 1 1#G, , t D$; ; ;GD,--  ',3355$((---"GE C $$T$t/?*@*@%A%ABBB$+LNN!((rdNrrrrNrBrc 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|Srrrcs rbrz+MessageSender.send_text..Ds555c!ff555rdrr)r)r;rrIrrrstrip_cron_wrappertruncate_messageMAX_TEXT_CHUNKrrr[rrrxrrr)r`rHrrrBr:rlockcontent_to_sendrr~chunkr_tors rb send_textzMessageSender.send_text-s-" 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    rdrEr<c@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.NrrrrTrrr*Fr Unknown errorr)rrrsend_group_msg_bodyr\send_c2c_msg_bodyrr) r`rHrErrBrgrprrs rbrzMessageSender.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;rrZretryric |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).rrNrrrTrrrz-[%s] send_text_chunk attempt %d/%d failed: %srWz0[%s] send_text_chunk attempt %d/%d exception: %srHz>[%s] send_text_chunk max retries (%d) exceeded. Last error: %sFzMax retries exceeded: r)r;rrrsend_group_messager\send_c2c_messagerrrrr[rr[rrr) r`rHrZrrrBr: last_errorrrrrr#s rbrxzMessageSender.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 WrdcK|j}dt}t|||jpd||}||||d{VS)z(Send C2C message with arbitrary MsgBody.c2c_r)rrErArFrBN)r;r;r5r6_dispatch_encoded)r`rrErBr:rr/s rbrzMessageSender.send_c2c_msg_body"st-' '')! .B!    ++GWfEEEEEEEEErdcK|j}dt}t|||jpd||pd}||||d{VS)z*Send group message with arbitrary MsgBody.grp_r)rBrErArF ref_msg_idN)r;r;r6r6r)r`rBrErr:rr/s rbrz!MessageSender.send_group_msg_body/sz-' ''+! .B~2    ++GWfEEEEEEEEErdr/rrc 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.r,NTrFr)rrFzRequest timeout after sr)rrrrr-rrr[)r:r/rr r#s rbrzMessageSender._dispatch_encodedCs  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 rbrzMessageSender.validate_mediaRs  ZA!5!5,(,, ,$&- z??Y & &*oo,t3GThTT'TTTKTTT Ttrdry max_lengthrjrkrc|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. rncDg|]}td|Sr4) _INDICATOR_REsubrs rbrz2MessageSender.truncate_message..~s(;;;q-##B**;;;rd)rrYr)rrrj_lenrs rbrzMessageSender.truncate_messageesq} 4==J & &9 #66 Z7   <;F;;;.vvgY.rdc8|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: )rrBrfindrr)rdivider footer_prefix divider_pos footer_posr body_startr!s rbrz MessageSender.strip_cron_wrappers!!"677 N'e ll7++ ]]=11 ??j1nn k0I0IN+&  & &N 3w<</ z*,-3355wrdc<K|jdS)zCRelease chat locks (no-op for now; placeholder for future cleanup).N)rrris rbryzMessageSender.closes!      rdr)r[r[rsrr\rrHr[r\rr rHr[rr[rrNrBr[r\rr) rHr[rr[rrNrrNrr r\rrrHr[rr[rrr\r) rHr[rEr<rrNrBr[r\r)Nrr) rHr[rZr[rrNrrirBr[r\rr4)rr[rZr[rBr[r\r)rBr[rZr[rrNr\r)rZr[rBr[r\r<)rr[rEr<rBr[r\r)rBr[rEr<rrNr\r)r:r*r/rrr[r\rrrrrr[rrir\rNr)rr[rrirjrkr\r)rr[r\r[r2)rrrrr0r}r5r~rprrrrrrrxrrrcompile MULTILINErrrrrrrrrryrrdrbrrs'0i0b0b0b&c&cJcccc(,,,,,    &---- )))).#' %(%(%(%(%(V#'!%      49= ) ) ) ) ) ^#' UUUUU2#' *V*V*V*V*V\YYYYY#' NNNNN"*A2<PPK6666p F F F F F"#' FFFFF( 9 9 9\ 9GI\$15////\/<\,!!!!!!rdrceZdZUdZejZded<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. r|r~r:r*r\rc||_t||_t||_t ||j|_|j|j_|j |j_ dSr) r;rrCrTrZrn slow_notifier_handle_send_startr_handle_send_finishrrGs rbrpzOutboundManager.__init__s_ %27%;%; +;G+D+D3GQUQ\3]3]&*%< "&*&> ###rdrHr[c:|j|dS)zFCalled by MessageSender before sending: cancel slow-response notifier.Nrrrr`rHs rbrz"OutboundManager._handle_send_start !!'*****rdcVK|j|td{VdS)z=Called by MessageSender after sending: send FINISH heartbeat.N)rZr[r+rs rbrz#OutboundManager._handle_send_finishs7n00:MNNNNNNNNNNNrdNrrrrNrBrcNK|j||||d{VS)z%Send text message with auto-chunking.rN)rCr)r`rHrrrBs rbrzOutboundManager.send_texts: [**7GXR\*]]]]]]]]]rdrrr c:K|jj||fi|d{VS)rN)rCr)r`rHrrs rbrzOutboundManager.send_medias9,T[+G\LLVLLLLLLLLLrdrrrrcJK|j|||d{VS)z.Send text + media (used by send_message tool).N)rCr)r`rHrrs rbrzOutboundManager.send_directs4 [,,Wg{KKKKKKKKKrdcJK|j|d{VdS)z Start reply heartbeat (RUNNING).N)rZrrs rb start_typingzOutboundManager.start_typings4n""7+++++++++++rdFrfr]cNK|j||d{VdS)zStop reply heartbeat.rfN)rZrh)r`rHrfs rb stop_typingzOutboundManager.stop_typings9n!!'{!CCCCCCCCCCCrdcJK|j|d{VdS)zStart slow-response notifier.N)rrrs rbstart_slow_notifierz#OutboundManager.start_slow_notifiers5 &&w///////////rdc:|j|dS)zCancel slow-response notifier.Nrrs rbrz$OutboundManager.cancel_slow_notifierrrdrc6|j|S)z@Proxy to MessageSender.get_chat_lock for backward compatibility.)rCrrs rbrzOutboundManager.get_chat_locks{((111rdcollections.OrderedDictc|jjS)z>Proxy to MessageSender._chat_locks for backward compatibility.)rCrris rbrzOutboundManager._chat_lockss{&&rdrrrrrric:t|||S)z&Proxy to MessageSender.validate_media.)rr)rrrs rbrzOutboundManager.validate_medias ++J+NNNrdcK|jd{V|jd{V|jd{VdS)zShut down all sub-managers.N)rCryrZrris rbryzOutboundManager.closesk!!!!!!!!!n""$$$$$$$$$ &&(((((((((((rdrrkrr)rHr[rr[rr r\rrr)Frlr)r\rrrr2)rrrrrr~r5rprrrrrrrrrrrrrrryrrdrbrrs  )6(HHHHH????++++OOOOEI^^^^^MMMM9=LLLLL,,,,DDDDD0000++++2222'''X'GIOOOO\O ))))))rdrceZdZUdZejZdZded<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)ormzCYuanbao AI Bot adapter backed by a persistent WebSocket connection.ryrirTrTrir|REPLY_REF_MAX_ENTRIESNz$ClassVar[Optional['YuanbaoAdapter']]_active_instancer\Optional['YuanbaoAdapter']c|jS)z7Return the currently connected YuanbaoAdapter, or None.rrs rb get_activezYuanbaoAdapter.get_active s ##rdr:rc||_dS)z0Register (or clear) the active adapter instance.Nr)r~r:s rbrnzYuanbaoAdapter.set_actives 'rdrrrr 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_idrrrws_urlrrrrrPi,) ttl_secondsr<YUANBAO_DM_POLICYrVr=YUANBAO_DM_ALLOW_FROMc^g|]*}||+Srrrxs rbrz+YuanbaoAdapter.__init__..\s2#b#b#b!XYX_X_XaXa#bAGGII#b#b#brd,r>YUANBAO_GROUP_POLICYr?YUANBAO_GROUP_ALLOW_FROMc^g|]*}||+Srrrs rbrz+YuanbaoAdapter.__init__..gs2&h&h&hQ^_^e^e^g^g&hqwwyy&h&h&hrd)r<r=r>r?rgr)8superrprYUANBAOrrrrrr6DEFAULT_WS_GATEWAY_URLrbDEFAULT_API_DOMAINrrrari_DEFAULT_RESOLVE_CONCURRENCYrrmax_MIN_RESOLVE_CONCURRENCYr_MAX_RESOLVE_CONCURRENCYrr9rrrYr r&r$r:rrrr rr rrKrLrMrr;rbr( _group_queryr2r7r home_channelrHr]rrk) r`rr_extra_raw_concurrencyr<_dm_allow_from_rawr=r>_group_allow_from_rawr?_existing_homerhs rbrpzYuanbaoAdapter.__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 r( asyncio.Taskcx|j|||jj|S)z@Register a fire-and-forget task so it won't be GC'd prematurely.)r$r%r&r'rjs rbrzYuanbaoAdapter._track_tasks8 ""4((( t5=>>> rdr]cdS)zCYuanbao gates DM/group access at intake via dm_policy/group_policy.Trris rbenforces_own_access_policyz)YuanbaoAdapter.enforces_own_access_policys trdr\r9c |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. rFrTrVr) PairingStorerrRrS)rbr[rArrIrTr<gateway.pairingr+ is_approvedrrrrBr>r[rN)r`r\rcrCr+rBs rbrlz)YuanbaoAdapter._sender_may_designate_homes/ $2S%+,,2244 5 =D ##F++ t9,,888888#|~~11(2B2H&QQQ5 =G # #S^1r2288::J u"k11..z:::"f,,//1115urdF) is_reconnectr.cDK|jd{VS)zhConnect to Yuanbao WS gateway and authenticate. Delegates to ConnectionManager.open(). N)rrS)r`r.s rbrdzYuanbaoAdapter.connects/ %**,,,,,,,,,rdcVKtj|urtdd|_|||jd{V|jd{Vt|j D]*}| s| +|j |j td|jdS)z;Cancel background tasks and close the WebSocket connection.NFz[%s] Disconnected)rmrrnrrrorryrYr<r&rurrrr rrr[rjs rb disconnectzYuanbaoAdapter.disconnects#  *d 2 2  % %d + + +  !!! ##%%%$$&&&&&&&&&n""$$$$$$$$$,--  D99;;   !!###   """ '33333rdrrHr[rrrNmetadataOptional[Dict[str, Any]]rBrcNK|j||||d{VS)zCSend text message with auto-chunking. Delegates to OutboundManager.rN)rYr)r`rHrrr2rBs rbrzYuanbaoAdapter.sends:^--gwU_-`````````rdrcDK|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)r[r`rrrs rb get_chat_infozYuanbaoAdapter.get_chat_infos:   h ' ' 6#W55 5...rdr>cnK |j|d{VdS#t$rYdSwxYw)zGSend "typing" status heartbeat (RUNNING). Delegates to OutboundManager.N)rYrr)r`rHr2s rb 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)rYrrrs rbrzYuanbaoAdapter.stop_typings`  .,,W%,HH H H H H H H H H H    DD s "( 66rcK|jj}|j|d{V t ||d{V|j||j}|rV|j ||kr:|j |d|j |ddSdSdS#|j||j}|rU|j ||kr8|j |d|j |dwwwxYw)z9Wrap base class processing with a slow-response notifier.N) r*rHrYrr_process_message_backgroundrr*rrrr )r`rrrHrFrhs rbr;z*YuanbaoAdapter._process_message_backgrounds,&n00999999999 B''55e[II I I I I I I I N / / 8 8 8%F B$266{CCvMM(,,[$???*..{DAAAAA B BMM N / / 8 8 8%F B$266{CCvMM(,,[$???*..{DAAAA BMs (CA;EcFK|j|d{VS)z2Query group info (delegates to GroupQueryService).N)rr2rZs rbr>zYuanbaoAdapter.query_group_infos/&;;JGGGGGGGGGrdrrr3r4cLK|j|||d{VS)z9Query group member list (delegates to GroupQueryService).r6N)rr;)r`rBr3r4s rbget_group_member_listz$YuanbaoAdapter.get_group_member_lists:&@@TZbg@hhhhhhhhhrdi'rrZcK|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)rbrTrr DM_MAX_CHARSr)r`rrZrBrHs rbsend_dmzYuanbaoAdapter.send_dm%s"0099 Ue3STTT T t99t( ( (***+.@@D%G%%YYwYDDDDDDDDDrdrrcBK|jj|df|||d|d{VS)zKSend image message (URL). Delegates to OutboundManager via ImageUrlHandler.r)rrrNrYr)r`rHrrrr2rs rb send_imagezYuanbaoAdapter.send_image<s_/T^. [ w)           rdrcBK|jj|df|||d|d{VS)zISend local image file. Delegates to OutboundManager via ImageFileHandler.r)rrrNrC)r`rHrrrr2rs rbrzYuanbaoAdapter.send_image_fileLs_/T^. \ w:           rdrrcBK|jj|df|||d|d{VS)zISend file message (URL). Delegates to OutboundManager via FileUrlHandler.r)rrrNrC)r`rHrrrr2rs rb send_filezYuanbaoAdapter.send_file\s_/T^. Z 8           rdr$r% Optional[int]cBK|jj|df|||d|d{VS)zDSend sticker/emoji. Delegates to OutboundManager via StickerHandler.r)rr$r%NrC)r`rHr$r%rrs rb send_stickerzYuanbaoAdapter.send_stickerls`/T^. Y %*            rdrcDK|jj|df||||d|d{VS)zMSend local file (document). Delegates to OutboundManager via DocumentHandler.rc)rrrrNrC)r`rHrrrrr2rs rbrzYuanbaoAdapter.send_document|sb/T^. Z w(            rdrcvKt|j|j|j|jd{VS)zr\rrk)rr[r\rrPrQrRr4)rr[rZr[rBr[r\r)NNN)rHr[rr[rrNrrNr2r>rr r\r)rHr[rr[rrNrrNr2r>rr r\r)rHr[rr[rrNrrNr2r>rr r\r) rHr[r$rNr%rHrrNrr r\r)NNNN)rHr[rr[rrNrrNrrNr2r>rr r\r)r\r)(rrrrrrPLATFORMrr5splits_long_messagesrrrrrrnrprrr)rlrdr1rr6r8rr;r>r>r@rArDrrGrJrrrO __classcell__)rhs@rbrmrms5MMHN+.....>BAAAA$$$[$'''['iiiiiiiiiiii^X<5:------44446#'-1 a a a a a / / / /    BBBBBB4HHHH >AiiiiiLEEEEE6"&"&#'      ("&"&#'      (#'"&#'      &'+$("&      (#'!%"&#'     $             rdrmr\rc4tS)z,Delegate to ``YuanbaoAdapter.get_active()``.)rmrrrdrbget_active_adapterrUs  $ $ & &&rdr:r*rHr[rrrrcJK|j|||d{VS)z,Delegate to ``OutboundManager.send_direct``.N)rYr)r:rHrrs rbsend_yuanbao_directrWs5"..wMM M M M M M MMrdrPr) r:r*rHr[rr[rrr\r)r __future__rrrrrr)rrrloggingrKrr r urllib.parserrrrrenumrpathlibrabcr r typingr r r rrrrrsysrrcwebsockets.exceptionsr^ ImportErrorgateway.configrrgateway.platforms.baserrrrrrrgateway.platformsrr_gateway.platforms.helpersrgateway.platforms.yuanbao_mediarrrr r!r"r#r$r.r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;gateway.sessionr< getLoggerrr hermes_clir=_HERMES_VERSIONr rr[r rr rrrrerrrrrrAUTH_FAILED_CODESAUTH_RETRYABLE_CODESrbrcREPLY_REF_TTL_SrwryrrKrr0rNrrrrrrrYrr6r7rXr9rZrlrrrrr2r8r;r`rervrrrr r'r-r1r<rrzrrr2r9rrrrrrr(rTrnrrrmrUrWrrdrbrns "#"""""     2222222222########QQQQQQQQQQQQQQQQQQQQ      JJJ433333332111119999992.-----  8 $ $ 9999999OOO  s-..L L6!@?? '&&)))!  YI RZ ABB$)$>$>$>?? -.. $& /1, !K K K K K K K K Zp)p)p)p)p)p)p)p)f54444444 >)>)>)>)>)>)>) >)BAAAAAAAA6]]]]]]]]|]]]]](]]]@/"     '   NdNdNdNdNd-NdNdNdb*&-"D"D"D"D"D"D"D"D"N-2.....-...bttttt0tttl "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........bj!j!j!j!j!j!j!j!Z ])])])])])])])])@^ ^ ^ ^ ^ (^ ^ ^ L ''''59 NNNNNNNs$ B BBD""D,+D,