
    uZi                     P    d Z ddlZddlmZmZ ddlmZmZ ddlm	Z	  G d d      Z
y)   数据查询Skill    N)AnyDict)SkillContextSkillResult)get_settingsc                   ^    e Zd ZdZdZdZd Zdedeee	f   de
defd	Zdeee	f   de
defd
Zy)DataQuerySkillr   
data_queryz1.0.0c                 H    d| j                   i| _        dddddidg| _        y )Nquery_employee_infou   查询.*员工d   case_sensitiveF)pattern
capabilityprioritymatch_config)_query_employee_infocapabilitiestriggers)selfs    H/home/jang/Projects/ai-researcher/src/agent/skills/data_query/handler.py__init__zDataQuerySkill.__init__   s8    !4#<#<

 -3!15 9	
    r   paramscontextreturnc                    K   | j                   j                  |      }|st        d|        |||       d{   S 7 w)u   执行数据查询zUnknown capability: N)r   get
ValueError)r   r   r   r   handlers        r   executezDataQuerySkill.execute   sG      ##''
33J<@AAVW----s   8A?Ac           	         K   |j                  dd      }d|dddddd	d
dgd}t        |ddddddgdddd      S w)u   查询员工信息
department    emp_001u   张三u   高级工程师)idnametitleemp_002u   李四u	   工程师)countr$   	employeesg      ?r   r   zhr.company.com/api/v1/employeesGET)skillr   endpointmethodapiF)source_type	cache_hitr   )data
confidencesourcesmetadataexecution_time_ms)r   r   )r   r   r   r$   	mock_datas        r   r   z#DataQuerySkill._query_employee_info)   sy      ZZb1
 $ (=NO ([I
	 %3=	  &+?
 	
s   <>N)__name__
__module____qualname____doc__r)   versionr   strr   r   r   r   r"   r    r   r   r
   r
   	   sp    DG
.. S#X. 	.
 
.
S#X
 
 
	
r   r
   )r>   httpxtypingr   r   src.agent.executorr   r   
src.configr   r
   rA   r   r   <module>rF      s       8 #>
 >
r   