`gjk0dZddlmZddlZddlmZddlmZmZm Z eje Z dZ ddd d d Zd ZdOdZ dP dQdZdRdSdZ dT dUdZehdZ dV dWdZddlmZmZdZdZdZdZdZdZdZ ejBde dddd d d!d"id gd#d$eed%d&'ejBd(e d(d)dd d!d"d gd*d+d,d d-d"d.d/d"d0d dgd#d$eed%d1'ejBd2e d2d3dd d4d"d d5d"d d6d"d d7d"d8d9dd d:d"d.d;d"d<d=gd#d>d?gd#d$eed%d@'ejBdAe dAdBdd dCd"dDdEd"dFgd#d$eed%dG'ejBdHe dHdIdd dJd"d dKd"d dLd"dMgd#d$eed%dN'y)Xu yuanbao_tools.py - 元宝平台工具集 提供以下工具函数，供 hermes-agent 的 "hermes-yuanbao" toolset 使用： - get_group_info : 查询群基本信息（群名、群主、成员数） - query_group_members : 查询群成员（按名搜索、列举 bot、列举全部） - search_sticker : 按关键词搜索内置贴纸（返回候选列表，含 sticker_id/name/description） - send_sticker : 向当前会话或指定 chat_id 发送贴纸（TIMFaceElem） - send_dm : 发送私聊消息（按昵称查找用户并发送） 对齐 chatbot-web/yuanbao-openclaw-plugin 的 sticker-search/sticker-send 行为： LLM 应先用 search_sticker 找到合适的 sticker_id（或直接传中文 name），再用 send_sticker 发送。不要在文本中夹杂裸的 Unicode emoji 当作贴纸。 The active adapter singleton lives in ``gateway.platforms.yuanbao`` and is accessed via ``get_active_adapter()``. ) annotationsN)Path)ListOptionalTuplec< ddlm}|S#t$rYywxYw)zOLazy import to avoid ImportError when gateway.platforms.yuanbao is unavailable.rget_active_adapterN)gateway.platforms.yuanbaor ImportErrorr s F/root/.hermes/venv/lib/python3.12/site-packages/tools/yuanbao_tools.py_get_active_adapterrs%@!## s  unknownuser yuanbao_aibot)rz\To @mention a user, you MUST use the format: space + @ + nickname + space (e.g. " @Alice "). group_codec K|sdddSt}|dddS |j|d{}|dddSd||jdd |jd d |jd d |jd d dddS7W#t$r-}tj ddt |dcYd}~Sd}~wwxYww)u9查询群基本信息（群名、群主、成员数）。Fgroup_code is requiredsuccesserrorN Yuanbao adapter is not connectedzquery_group_info returned NoneT group_name member_countrowner_idowner_nicknameuser_idnicknameu+The group is called "派 (Pai)" in the app.)rrrrownernotez$[yuanbao_tools] get_group_info error)rquery_group_infoget Exceptionlogger exceptionstr)radaptergiexcs r get_group_infor05s  +CDD!#G +MNN5++J7 7 :$/OP P$&&r2FF>1566*b1FF#3R8B  8 5?@ 3s8445sOCB B B CA B CB C"B;5C6C;CCactioncK|sdddSt}|dddS |j|d{}|dddS|jdgDcgc]l}|jdd |jd |jd d tj|jd |jd dddn}}|sdddS|rdtini}|dk(r3|Dcgc] }|d dvs |} }| sdddSddt | d| d|S|dk(r|ri|j j} |Dcgc]}| |d jvs|} }| rddt | d|d| d|Sdd|d|d|Sddt |d|d|Sddt |d|d|S7xcc}wcc}wcc}w#t$r-} tjd dt| dcYd} ~ Sd} ~ wwxYww)!u 统一的群成员查询工具（对齐 TS query_session_members）。 action: - find : 按昵称模糊搜索 - list_bots : 列出 bot 和元宝 AI - list_all : 列出全部成员 FrrNr#get_group_member_list returned Nonemembersr#rr$ nick_name user_typerolerr)r#r$r7zNo members found in this group. mention_hint list_bots>rrzNo bots found in this group.TzFound z bot(s).)rmsgr4findz member(s) matching "z".zNo match for "z". All members listed below.z member(s).z)[yuanbao_tools] query_group_members error) rget_group_member_listr(_USER_TYPE_LABEL MENTION_HINTlenstriplowerr)r*r+r,) rr1namementionr-rawm all_membershintbotsfiltmatchedr/s r query_group_membersrKRs  +CDD!#G +MNNB511*== ;$/TU UWWY+   55B/EE*aeeK.DE(,,EE+quuVQ'78)    $/PQ Q18 -b [ *Q!ai;P.PAQDQ#(3QRRD {(3   V zz|))+&1STQz]=P=P=R5R1SS#'!'G ~5J4&PRS#*  %+D61MN*  K 01=&  C ,-[9"    q>  "RT: 5DE 3s8445sG3F:F( F:G3F:A1F+F: G3 F: F0,F00F:8G39F: G3)F:7F5F5F:1G32 F:=G3>F:G3F:'G3(F:+F:: G0"G+%G0&G3+G00G3c Kddlm} tdtd|r t |nd} ||xsd|}d |xsdt||Dcgc]I}|jd d|jdd|jdd|jdddKc}dS#t t f$rd}YwxYw#t$r-}tjdd t|d cYd }~Sd }~wwxYwcc}ww)u 在内置贴纸表中按关键词模糊搜索，返回 Top-N 候选。 返回每条候选的 sticker_id / name / description / package_id， 供 LLM 选择后传给 send_sticker。空 query 时返回前 N 条。 r)search_stickersr2 r)limitz$[yuanbao_tools] search_sticker errorFrNT sticker_idrB description package_id)rQrBrRrS)rquerycountresults) !gateway.platforms.yuanbao_stickerrMmaxminint TypeError ValueErrorr)r*r+r,r?r()rTrPrM safe_limitmatchesr/ss r search_stickerr`sBC%CJR@A 5!%+2Z@ "W   eeL"5fb) uu]B7eeL"5     z "  5?@ 3s8445 s]C8#B#B:C8AC3 C8#B74C86B77C8: C0"C+%C0&C8+C00C8cKddlm}ddlm}m}m}|xsdj xs |dd}|sdddSt}|dd dS|xsdj } d } | s|} n"| jr|| } | || } | dd | d dS |j|| jd d|xsd d {} t| ddr6d|| jdd| jd ddt| dd ddSdt| dddS7W#t$r-} tjddt| dcYd } ~ Sd } ~ wwxYww)u 向 chat_id（缺省取当前会话）发送一张内置贴纸（TIMFaceElem）。 Args: sticker: 贴纸名称（如 "六六六"）或 sticker_id（如 "278"）。为空时随机发送一张。 chat_id: 目标会话；缺省时使用当前会话上下文（HERMES_SESSION_CHAT_ID）。 格式：``direct:{account_id}`` / ``group:{group_code}`` / 或裸 account_id。 reply_to: 群聊场景的引用消息 ID（可选）。 Returns: ``{"success": bool, ...}`` rget_session_env)get_sticker_by_idget_sticker_by_nameget_random_stickerrHERMES_SESSION_CHAT_IDFz8chat_id is required (no active yuanbao session detected)rNrzSticker not found: z:. Use search_sticker first to discover available stickers.rB)chat_id sticker_namereply_toz"[yuanbao_tools] send_sticker errorrTrQ)rQrB message_idz~Sticker delivered to the chat. If you have additional text to say, reply now; otherwise end your turn without generating text.)rrhstickerrkr&rzsend_sticker failed)gateway.session_contextrcrWrdrerfr@risdigit send_stickerr(r)r*r+r,getattr) rlrhrjrcrdrerftargetr-rD sticker_objresultr/s r roros 8 m " " $ U8PRT(UF O  "#G +MNN =b   !C"&K (* ;;=+C0K  -c2K*3'2PQ  5++$4%,  vy%()oolB?#3"&,=U  *?@ ) 5=> 3s8445sIBE+DDDAED E&"EE EEE>.bmp.gif.jpg.png.jpeg.webpc K|s|sdddSt}|dddS|r|jnd}|j}|sK|sdddS|sdddS |j|d{}|dd dS|jd g} |jj } | D cgc]<} | | jd xs| jd xsdj vr| >} } | s dd |d|ddSt | dkDrL| D cgc]7} | jdd| jd | jd dd9} } dd|d| dS| djdd}| djd | djd |}|sdddSd|}d}g} |rW|jrG|j|||d{}|js|j|jxsd|xsgD]\}}t|jj }|t vr|j#|||d{}n|j%|||d{}|jrw|j|jxsd|dddS|r"| |jsddj'|dSd|||j(d |d!d"}|r |d#xxd$dj'|d%z cc<|S7cc} wcc} w#t $r-}tjddt|dcYd}~Sd}~wwxYw7y77#t $r-}tjd&dt|dcYd}~Sd}~wwxYww)'a Send a DM (private chat message) to a group member, with optional media. Workflow: 1. If user_id is provided, send directly. 2. Otherwise, search the group member list by name to resolve user_id. 3. Send text via adapter.send_dm(), then iterate media_files by extension. Args: group_code: The group where the target user belongs. name: Target user's nickname (partial match, case-insensitive). message: The message text to send. user_id: (Optional) If already known, skip the member lookup. media_files: (Optional) List of (file_path, is_voice) tuples to send after the text message. Images are sent via send_image_file; everything else via send_document. Fz"message or media_files is requiredrNrrz3group_code is required when user_id is not providedz-name is required when user_id is not providedr3r4r$r5zNo member matching "z" found in group .rr#r"zMultiple members match "z". Please specify which one.)rr candidatesrz+[yuanbao_tools] send_dm member lookup errorzCould not resolve user_idzdirect:rztext send failedzmedia send failedz%No deliverable text or media remainedz; Tz DM sent to "z" successfully.)rr#r$rkr&r&z (partial failure: )z[yuanbao_tools] send_dm error)rr@r<r(rAr?r)r*r+r,send_dmrappendrrsuffix _IMAGE_EXTSsend_image_file send_documentjoinrk)rrBmessager# media_filesr-resolved_user_idresolved_nicknamerDr4rIrErJr|r/rh last_resulterrors media_path _is_voiceextrss r rr"s0 ; +OPP!#G +MNN*1w}}r  $/de e$/^_ _$ 955jAAC{#(3XYYggi,G::<%%'D"AEE*-I{1CIrPPRRG $3D69J:,VWX7|a%  $%55B#7$%EE*aeeK6L$M  %7v=YZ", 'qz~~i<  ' z71:>>+W[;\ ]   +FGG()*GKF"5 w}} '0@'V` aaK&& k//E3EF&1%6B H !J z"))//1Ck!$+$;$;GZ\f$;$gg $+$9$9':Zd$9$ee && k//F3FG H  $/VW W {*+2E2E$tyy/@A A')%00"#4"5_E    6N 3DIIf4E3FaH HN WB  9   J K$s3x8 8 9bhe( 589 3s8445sANL 'K<( L 2N34L 'AK?(L 7N8L CG!!(+$]]3215 '-2IK> dD !    4d488JPU;V6WX Y tUm ,Ta   DG d47m< = >hhy"%G:1??HNG>*%AA+NK W488FB+?B''4  +   *sBF05F A?F0 F0B9F0F.F0 F+'F0*F++F0c Ktt|jdd|jddd{S7w)NrTrrPrOrTrP)rr`r(rs r _handle_yb_search_stickerrs@ ^hhw#hhw# s6AA Ac Ktt|jdd|jdd|jddd{S7w)Nrlrrhrjrlrhrj)rror(rs r _handle_yb_send_stickerrsN \B'B'*b) sAA A Azhermes-yuanbaoyb_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).)typerR)r propertiesrequired)rBrR parametersTu👥)rBtoolsetschemahandlercheck_fnis_asyncemojiyb_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;r9rufind — 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.)renumrRzUser 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.ru📋 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)rrRitemsru✉️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).ru🔍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).ru🎨)rr,returnr)rrF) rr,r1r,rBr,rCrrr)rrO)rTr,rPrZrr)rrr)rlr,rhr,rjr,rr)rN) rr,rBr,rr,r#r,rz Optional[List[Tuple[str, bool]]]rr)"__doc__ __future__rloggingpathlibrtypingrrr getLogger__name__r*rr=r>r0rKr`ro frozensetrrtools.registryrrrrrrrr_TOOLSETregisterr rs$#((   8 $!V G6 5> W5W5 W5 W5 W5  W5t!JK K KK K^JK 48 x5x5 x5x5 x5 2 x5  x5~1- $N   % =$#N &   " (   /4 ! ( T%#N %=9 %d&V+:&x0?  * V +   c2h   F%A%A%#j %X$H !))1/_% )2/f) '&,H  71di5 ? @    MGT  # a%E&#]   6 &   C"J  ! !%V%c%#Z(- & N $   [.r