߲jUdZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z m Z ejeZdd d d d Zd ZehdZGddeZd[dZdZdZd\dZd]d^dZd_dZd`dZeddadZeddadZeddadZ ed dad!Z!ed"dad#Z"d$Z#d%d&d'dd(d)d*idgd+d,ed-fd.d/d'd(d)d*d(gd0d1d2d(d3d*d4d5d*d6dd7gd+d,ed8fd9d:d'd(d;d*d(dd*d?d@d'd(dAd*d4dBd*dCdDgd+dEdFgd+d,e"dGfdHdId'd(dJd*dKdLd*dMgd+d,e dNfdOdPd'd(dQd*d(dRd*d(dSd*dTgd+d,e!dUffZ$e$D]"\Z%Z&Z'e jPe%ddVe%e&e#dWe'X$ddYl m)Z)ddZl m*Z*y)buyuanbao_tools.py - 元宝平台工具集 (the "hermes-yuanbao" toolset). get_group_info / query_group_members / search_sticker / send_sticker / send_dm. Sticker flow mirrors chatbot-web's sticker-search/sticker-send: the LLM should search_sticker for a sticker_id (or pass the Chinese name), then send_sticker — never bare Unicode emoji. The active adapter singleton lives in ``gateway.platforms.yuanbao.YuanbaoAdapter.get_active``. ) annotationsN)suppress)Path)Tuple)registry tool_resultunknownuser yuanbao_aibot)rz\To @mention a user, you MUST use the format: space + @ + nickname + space (e.g. " @Alice ").>.bmp.gif.jpg.png.jpeg.webpc$eZdZdZdfd ZxZS)_YbErrorzLTool-level failure; ``payload`` is the error envelope returned to the model.c <t||d|d||_y)NF)successerror)super__init__payload)selfmsgextra __class__s 8/home/ubuntu/.hermes/hermes-agent/tools/yuanbao_tools.pyrz_YbError.__init__#s! #(3@%@ )rstr)__name__ __module__ __qualname____doc__r __classcell__)r!s@r"rr sVAAr#rcfd}|S)ueHandler decorator: ``fn(args)`` → tool_result; ``_YbError`` → its envelope; else logged envelope.cFtjfd}|S)Nc6K t|d{S7#t$r}t|jcYd}~Sd}~wt$rG}tj dttt |jcYd}~Sd}~wwxYww)Nz[yuanbao_tools] %s error)rrr Exceptionlogger exceptionr$)argskwexcfnlabels r"handlerz'_yb_tool..deco..handler+sq ?"D>22> 0"3;;// ?  !;UC"8CH#5#=#=>> ?sRB  B BABB B.deco*s%   ?  ?r#)r4r8s` r"_yb_toolr:(s  Kr#cxtt5ddlm}|j cdddS#1swYyxYw)zOLazy import to avoid ImportError when gateway.platforms.yuanbao is unavailable.rYuanbaoAdapterN)r ImportErrorgateway.platforms.yuanbaor= get_activer<s r"_get_active_adapterrA8s3 + +<((*+++ s09c4tx} td|S)Nz Yuanbao adapter is not connected)rAr)adapters r"_adapterrD@s!&((19:: Nr#namecjtt5ddlm}||dcdddS#1swYyxYw)Nr)get_session_env)rr-gateway.session_contextrG)rErGs r" _session_envrJFs1 ) );tR())) s)2cF|jd|jd|S)Nnickname nick_name)get)mdefaults r"_nickrQMs 55QUU;8 99r# group_codec~K|j|d{}| td|jdgS7#w)Nz#get_group_member_list returned Nonemembers)get_group_member_listrrN)rCrRraws r"_membersrWQs@--j9 9C {<== 779b !! :s =;$=c DK|s td|s td|jj}t||d{Dcgc]<}||j dxs|j dxsdjvr|>}}|std|d|d t |d kDr;td |d |Dcgc] }|j d dt |d"c}|d}|j d dt ||fS7cc}wcc}ww)zResolve ``name`` to (user_id, nickname) via the group member list; >1 partial match raises with ``candidates`` for disambiguation instead of guessing.z3group_code is required when user_id is not providedz-name is required when user_id is not providedNrLrMrHzNo member matching "z" found in group .r zMultiple members match "z". Please specify which one.user_idrZrL) candidatesr)rstriplowerrWrNlenrQ)rCrRrEfiltrOmatcheds r"_resolve_dm_recipientrbXs1 LMM FGG ::<   D (* ==SQ155,Hk0BHbOOQQSGS -dV3DZLPQRSS 7|a&tf,H IY`aTUAEE)R$8eAhOa    A 55B q$ //>Sbs1AD  D D AD2D %D ),D  D get_group_infoc @K|jdd}|s tdtj|d{}| tdd||jdd|jdd |jd d|jd dd d dS7]w)u9查询群基本信息（群名、群主、成员数）。rRrHgroup_code is requiredNzquery_group_info returned NoneT group_name member_countrowner_idowner_nicknamer[u+The group is called "派 (Pai)" in the app.)rrRrfrgownernote)rNrrDquery_group_info)r0rRgis r"rcrcms,+J /00z**:6 6B z788z VXAY~q1VVJ3HXZ\A]^=   7s._listingscgFFFr# list_bots>r r zNo bots found in this group.TzFound z bot(s).findrLz member(s) matching "z".zNo match for "z". All members listed below.z member(s).)rwboolrr$rTlistreturndict) rNrrWrDrQ_USER_TYPE_LABEL MENTION_HINTr_r]r^) r0rRrErprO all_membersrybotsr`rarxs @r"rnrn~sxx b1488FB3GJ XXh +F /00  J77 EE)R(eAh!%%aeeKvq9I&JIV XK 899-1XXi-GNL )RDG&Ma!F)7L*LMM9: :s4yk:DAA Dzz|!!#)KTQz]5H5H5J-J1KK DF3w<.8MdVSU"VX_` `v5QRT_`` DF3{#3"4K@+ NN/8NLsKAG F;G %AF>?:G 9 GG AG G4G8AG >G search_stickercKddlm}|jdd|jdd}} tdt d|r t |nd}||xsd| }d |xsdt||Dcgc]#}d Dcic]}||j|dc}%c}}d S#t tf$rd}YdwxYwcc}wcc}}ww) ue在内置贴纸表中按关键词模糊搜索，返回 Top-N 候选（空 query 返回前 N 条）。r)search_stickersqueryrHlimit r 2)rT) sticker_idrE description package_id)rrcountresults) !gateway.platforms.yuanbao_stickerrrNmaxminint TypeError ValueErrorr_)r0rrr safe_limitmatchessks r"rrsB88GR($((7B*?5EC%CJR@A ekr =B9B> C"B63C5B66C9B> >C send_stickercKddlm}m}m}|j ddxsdj xs t d}|s tdt}|j ddxsdj }|s|}n$|jr||ndxs||}|td |d |j||j d d|j d dxsd d{}t|ddstt|ddd||j dd|j d ddt|ddddS7]w)u$向 chat_id（缺省取当前会话 HERMES_SESSION_CHAT_ID）发送一张内置贴纸（TIMFaceElem）。 ``sticker``: 名称（如 "六六六"）或 sticker_id（如 "278"）；为空时随机发送。 ``chat_id``: ``direct:{account_id}`` / ``group:{group_code}`` / 裸 account_id。r)get_sticker_by_idget_sticker_by_nameget_random_stickerchat_idrHHERMES_SESSION_CHAT_IDz8chat_id is required (no active yuanbao session detected)stickerNzSticker not found: z:. Use search_sticker first to discover available stickers.rEreply_to)r sticker_namerrFrzsend_sticker failedTr)rrE message_idz~Sticker delivered to the chat. If you have additional text to say, reply now; otherwise end your turn without generating text.)rrrrrk) rrrrrNr]rJrrDisdigitrgetattr) r0rrrtargetrCrV sticker_objresults r"rrs\ mlhhy"%+ 2 2 4 ^ E]8^F QRRjG 88Ir " (b / / 1C (* 14(-DeM`adMe ,SG3mnoo''[__U[]_E`15*b1I1QT(SSF 69e ,wvw0EFGGF"-//,"C[__]cegMhiflD9Q   SsC1E3E4AEsend_dmc `K|jdd}|s1tdx}jdr|jddd}g}|jdxsgD]}t |t r=|j |jddt|jd d fPt |ttfsgt|d k\sv|j t|d t|dfd d l m }|j|jdd\}}|j|t|xsgz}|s |s t!dt#}|jdd} |jddxsdj%| j%} } | st'||| d{\} } | s t!dd| }d} g} |rW|j%rG|j)| ||d{} | j*s| j | j,xsd|D]\}}t/|j0j3t4vr |j6n |j8}||||d{} | j*rg| j | j,xsd| t!d| r&| j*st!dj;| d| d}| r|ddj;| dz }d| | | j<|dS7h7$7w) aSend a DM to a group member, with optional media. group_code defaults to the session's "group:" chat_id. Without ``user_id`` the member list is searched by ``name`` (partial, case-insensitive; >1 match returns candidates). media_files items are {"path", "is_voice"} dicts or (path, is_voice) pairs; ``MEDIA:`` tags in the text count too. Partial media failures are reported in ``note``, not as failure.rRrHrzgroup::r media_filespathis_voiceFrr)BasePlatformAdaptermessagez"message or media_files is requiredrErZNzCould not resolve user_idzdirect:)rRztext send failedzmedia send failedz%No deliverable text or media remainedz; z DM sent to "z" successfully.z (partial failure: )T)rrZrLrrk)rNrJ startswithsplit isinstancerappendr|r}tupler_r$gateway.platforms.baser extract_mediafilter_media_delivery_pathsrrDr]rbrrrrsuffixr^ _IMAGE_EXTSsend_image_file send_documentjoinr)r0rRrritemrembedded_mediarrCrEresolved_user_idresolved_nickname last_resulterrors media_path _is_voicesendrks r"rrs,+J l3K&LL7XXYab]]3*1- K'-2> dD !    4d488JPU;V6WX Y tUm ,Ta   DG d47m< = > ;1??TV@WXNG%AA+PTUcUigiPjBjkK ;;<<jG 88FB D+/88Ir+B+Hb*O*O*QSWS]S]S_' 4I'S]_c4d.d++ 233()*GKF7==?#OO,?? k))tyy()) +,O L&?AL.L)BL. L, L.B L.)L.,L.c:tddk(xs tduS)uNToolset availability check — True when running in a yuanbao gateway session.HERMES_SESSION_PLATFORMyuanbaoN)rJrAr9r#r"_check_yuanbaors" 1 2i ? dCVCX`dCddr#yb_query_group_infoulQuery basic info about a group (called '派/Pai' in the app), including group name, owner, and member count.objectstringz)The unique group identifier (group_code).)typer)r propertiesrequired)rEr parametersu👥yb_query_group_membersu:Query members of a group (called '派/Pai' in the app). Use this tool when you need to @mention someone, find a user by name, list bots (including Yuanbao AI), or list all members. IMPORTANT: You MUST call this tool before @mentioning any user, because you need the exact nickname to construct the @mention format.)r{rzrqufind — search a user by name (use when you need to @mention or look up someone); list_bots — list bots and Yuanbao AI assistants; list_all — list all members.)renumrzUser name to search (partial match, case-insensitive). Required for 'find'. Use the name the user mentioned in the conversation.booleanz{Set to true when you need to @mention/at someone in your reply. The response will include the exact @mention format to use.)rRrprErtrpu📋 yb_send_dmu`Send a private/direct message (DM) to a user in a group, with optional media files. This tool automatically looks up the user by name in the group member list and sends the message. Use this when someone asks to privately message / 私信 / DM a user. Supports text, images, and file attachments. You can also provide user_id directly if already known.uThe group where the target user belongs. Extract from chat_id: 'group:328306697' → '328306697'. Required when user_id is not provided.zdTarget user's display name (partial match, case-insensitive). Required when user_id is not provided.zEThe message text to send as a DM. Can be empty if only sending media.z}Target user's account ID. If provided, skips the member lookup. Usually obtained from a previous yb_query_group_members call.arrayzOptional list of media files to send along with the DM. Images (.jpg/.png/.gif/.webp/.bmp) are sent as image messages; other files are sent as document attachments.z.Absolute local file path of the media to send.z5Whether this file is a voice message (default false).)rrr)rritems)rRrErrZru✉️yb_search_stickeruiSearch the built-in Yuanbao sticker (TIM face / 表情包) catalogue by keyword. Returns the top matching candidates with sticker_id, name, and description. Use this BEFORE yb_send_sticker to discover the right sticker_id. Sticker = 贴纸 = TIM face — NOT a message reaction. Prefer sending a sticker over bare Unicode emoji when reacting/expressing emotion.uwSearch keyword (Chinese or English, e.g. '666', '比心', 'cool', '吃瓜'). Empty string returns the first N stickers.integerz8Max number of candidates to return (default 10, max 50).)rru🔍yb_send_stickeruSend a built-in sticker (TIMFaceElem / 贴纸表情) to the current Yuanbao chat. Call yb_search_sticker first if you don't know the sticker_id/name. Sticker = 贴纸 = TIM face — NOT a message reaction. CRITICAL: Whenever the user asks you to send a sticker / 贴纸 / 表情包, you MUST use this tool. DO NOT draw a PNG via execute_code / Pillow / matplotlib and then call send_image_file — that produces a fake 'sticker' image instead of a real TIM face and is the WRONG path. If no suitable sticker_id is known, call yb_search_sticker first. When the recent thread shows users sending stickers, prefer matching that tone by replying with a sticker instead of (or in addition to) text.uSticker name (e.g. '六六六', '比心', 'ok') or numeric sticker_id (e.g. '278'). Empty string sends a random built-in sticker.zvTarget chat. Defaults to the current session. Format: 'direct:{account_id}', 'group:{group_code}', or bare account_id.z5Optional ref_msg_id to quote-reply (group chat only).)rrru🎨zhermes-yuanbaoT)rEtoolsetschemar5check_fnis_asyncemoji)List)Optional)r4r$)rEr$r~r$)rH)rOrrPr$r~r$)rRr$r~r})rRr$rEr$r~zTuple[str, str])r~r)+r( __future__rr6logging contextlibrpathlibrtypingrtools.registryrr getLoggerr%r.rr frozensetrr-rr:rArDrJrQrWrbrcrnrrrr_TOOLS_schema_handler_emojiregisterrrr9r#r"rs#0   8 $ V G6 JK AyA   :"0*      !O!!OH " .> )7@7@te*A! ('R# *N   " '.-X!!)'R# !) A= !)h!*Z +:*84? * V [.b!J!!)E#!)E!)'n  !)\ !(L %--53c)" -63j-" +*0 "$71di5? @ ECL(e!!)I!*'a  6 ;B& %!!)Z !)g !)'^!(-& N S*QS j"(GXvH V_&6wPX$fr#