i+_ dZddlZddlmZddlmZmZmZddlm Z m Z m Z m Z m Z ddlmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZmZdd lm Z m!Z!ed Z"d Z#dd d e$de%ede$dzddfdZ&d e$de%efdZ'de e$e%effdZ(de e$e e$e fffdZ)eGddeZ*eGddeZ+eGddeZ,eGddeZ-e&de+de&d e,d!e&d"e-d#eGd$d%eZ.y)&zTooling configuration models.N)deepcopy) dataclassfieldreplace)AnyDictListMappingTuple) BaseConfig ConfigErrorConfigFieldSpec EnumOptionChildKey ensure_list optional_bool optional_strrequire_mapping require_str extend_path)enum_options_from_values)Registry RegistryError)FunctionCatalogget_function_catalog tooling_typez:All) descriptionname config_clsrreturncD|rd|ind}tj|||y)Nsummary)targetmetadata)tooling_type_registryregister)rrrr$s H/Users/bowang/.openclaw/workspace/ChatDev/entity/configs/node/tooling.pyregister_tooling_typer(s' ,7 ;'DH""4 X"Nctj|}|j}t|trt |t std|d|S)NzEntry 'z' is not a BaseConfig subclass)r%getload isinstancetype issubclassr r)rentryrs r'get_tooling_type_configr1(sK ! % %d +EJ j$ 'z*j/QgdV+IJKK r)cztjDcic]\}}||jc}}Scc}}wN)r%itemsr,rr0s r'iter_tooling_type_registrationsr60s42G2M2M2O P2O;4D%**, 2O PP Ps7c tjDcic]\}}|t|jxsi!c}}Scc}}wr3)r%r4dictr$r5s r'iter_tooling_type_metadatar94s<@U@[@[@] ^@]uD$u~~+, ,@] ^^ ^s$AceZdZUdZdZedzed<dZedzed<dZe ee fdzed<dZ e ed<de ddd dd iZed e ee fffd ZxZS)FunctionToolEntryConfigz4Schema helper used to describe per-function options.Nrr parametersT auto_fillz Function Namestrz7Function name from functions/function_calling directoryr display_name type_hintrequiredrr c .t|}t}|j}|j d}|W|j xsd}d}d}|j r|d|j d}n|s|d}ng}g}|D]\}} |t} |j| djd| ddD} t| dkDrd nd } |d } |jt| | d t| d | | rd| | dnd z|D]o\}} | D]e}|d|j}|j|j|j xsd}|jt|j||gqt|||||d<|S)Nrz Function namez (loading failed: )z" (no functions found in directory), c34K|]}|jywr3)r).0metas r' z6FunctionToolEntryConfig.field_specs..xs'Hid isz...z.pyz Load all z functions from z ()valuelabelr:z*This function does not provide a docstring)enum enum_optionsr)super field_specsr iter_modulesr+r load_errorMODULE_ALL_SUFFIXappendjoinlenrrr)clsspecscatalogmodules name_specrrP enum_values module_namemetas all_labelpreviewsuffix module_hintrHrMoption_description __class__s r'rRz#FunctionToolEntryConfig.field_specsds#%&(&&(IIf%  #//B?K48L,0K!!!, -?@R@R?SSTU !, -OP !  *1&K#.-0A/B CI&&y1"ii'HeBQi'HHG&)%j1nU"F%0M"5K ''""+"+"+CJ<7G } U>ERy#:2!O   +2 +2&K %#.-q <#**4995-1-=-=-mAm*$++&&*ii&+,> !&+2$ )' E&M  r))__name__ __module__ __qualname____doc__rr>__annotations__rr<rrr=boolr FIELD_SPECS classmethodrR __classcell__rfs@r'r;r;8s>D#*"Kt"(,JS#X%,It (Q  !KF2Do!5622r)r;ceZdZUeeeefed<dZe ed<dZ e dzed<e ddddde e dd d d d ddZedeeefdeddfdZedededzfdZededededeeefdeeeeefeff dZy)FunctionToolConfigtoolsT auto_loadNtimeoutzFunction Tool Listzlist[FunctionToolEntryConfig]z%Function tool list, at least one item)rr@rArBrchildzExecution TimeoutfloatFz Tool execution timeout (seconds)rr@rArBradvance)rsrudatapathr c t||}t|jd}|stdt |dt }g}t |D]\}}t |d|d} t|ts td| t|} | jd} t| tr| jstdt | d| j} | | d<|j| } | r%|j|j| || | |j| | fg}i}|D]w\}}t|} | jd} t| tr| jstdt |d|j| }|td | d t |d|j| }|td | d |d t |d||| <| jd d}t|t stdt |d t| }|rU|jds|j"r|j"|d<|jdst%|j&|d<|j)d d|j|zt+|d|d}|jd}|,t|t,t.fstdt |d||| t!|ndt|t,t.frt/||Sd|S)Nrsz+tools must be provided for function toolingztools[]ztool entry must be a mappingrztool name is required)r_r[r{originalz function 'z$' not found under function directoryz&' is declared multiple times (also in rDr=Tzauto_fill must be booleanrr<rtdefaultrutimeout must be numeric)rsrtrur{)rrr+r rr enumerater-r r8r>strip_extract_module_from_allextend_expand_module_all_entryrVrlrrparameters_schemapoprintrw)rYrzr{mappingrsr[expanded_toolsidxtool tool_path normalizedraw_namerr_ tool_specsseen_functionsr0 entry_pathr$previousr=mergedrt timeout_values r' from_dictzFunctionToolConfig.from_dictsH!$-GKK01K[Y]_fMgh h&(;="5)IC#DF3%q/:IdG,!"@)LLdJ!~~f-Hh,HNN4D!"9;yRX;YZZ>>#D!%Jv 66t>&)DdC( !"9;zSY;Z[[{{4(H! &JK F3&))$/H#! &LXJVWX F3$.N4 "{D9Ii.!"={:Wb?cdd*%Fzz-0X5I5I,4,@,@F=)zz,/+3H4N4N+OF<( JJ{D )   f %;"0>"';dK  I.  $Z U|-T7T99UV V)2)>d9oD,6}sEl,SE-(   Z^   r)rLc||jtsy|dtt j}|xsdSr3)endswithrUrXr)rLmodules r'rz+FunctionToolConfig._extract_module_from_alls;~~/00#/001779~r)r_r[r~c6dDcgc] }||vs| }}|r2dj|}t|td|t|d|j |}|std|dt|dg}|D]} |j d| i|f|Scc}w)N)rr<r=rEz does not support overriding rzmodule 'z+' has no functions under function directory)rWr rUrfunctions_for_modulerV) r_r[r{r~key disallowedfields functionsentriesfn_names r'rz+FunctionToolConfig._expand_module_all_entrys&Pc%OcSVZbSbc%O c YYz*F- 122OPVxXD&) 00= ;-'RSD&) 57 G NNVW-t4 5!!ds BB)rgrhrir rr>rrkrtrlrurwrr;rmrnr r staticmethodrrr rr)r'rrrrs0 S#X It GUT\ !-5?)  "#,:  %!KFM WS#X.M M AUM M ^d  !  #s(#  eDcNC'( ) r)rrc 4eZdZUeed<eeZeeefed<dZ e dzed<dZ e ed<dZ e edzed<edd d d d edddddd edddddd edddddd edddddd dZedeeefdeddfdZdefd Zy)!McpRemoteConfigserverdefault_factoryheadersNru cache_ttl tool_sourceszMCP Server URLr>Tz"HTTP(S) endpoint of the MCP serverr?zCustom Headersdict[str, str]Fz/Additional request headers (e.g. Authorization)rxzClient TimeoutrwzPer-request timeout in secondsTool Cache TTL@Seconds to cache MCP tool list; 0 disables cache for hot updateszTool Sources Filter list[str]z[Only include MCP tools whose meta.source is in this list; omit to default to ['mcp_tools'].)rrrurrrzr{r c t||}t|d|}|jd}i}|[t|tst dt |d|jDcic]\}}t|t|}}}|jd} | d} n8t| ttfr t| } nt dt |d|jdd} | d} n8t| ttfr t| } nt dt |d|jd } d}| qt| }g}t|D]S\}}t|tst d t |d |d |j}|sC|j|U|}nd g}|||| | ||Scc}}w)Nrrzheaders must be a mappingrurrrcache_ttl must be numericrz&tool_sources must be a list of stringsz tool_sources[r} mcp_tools)rrrurrr{)rrr+r-r r rr4r>rrwrrrrV)rYrzr{rr headers_rawrkvrrucache_ttl_valuertool_sources_rawrrrrr0rLs r'rzMcpRemoteConfig.from_dicths!$-Wh5kk), "$  "k73!"={4QZ?[\\2=2C2C2EF2E$!Qs1vs1v~2EGF I.  G  U| 4M*G7T99UV V!++k37  "I #u 6o.I9;t[;YZ Z";;~6)-  '!"23G$&J'0 U!%-%@#DM#a*@A %%e,1&L'=L%   KGs(F>c|jtt|jj |j f}t jt|jdjS)Nutf-8) rtuplesortedrr4ruhashlibsha1reprencode hexdigestselfpayloads r' cache_keyzMcpRemoteConfig.cache_keysZ KK &++-. / LL  ||DM009:DDFFr))rgrhrir>rkrr8rrrurwrrr rrmrnr rrrrr)r'rr6s K#D9GT#s(^9 GUT\ Iu%)L$s)d")")<  #)&I  #)8  %)Z  (.!u  A(KT5 WS#X.5 5 AR5 5 nG3Gr)rceZdZUeed<eeZeeed<dZ edzed<ee Z e eefed<dZ eed<d Zeed <dZedzed <d Zeed <edddddedddddeddddddeddddddeddddddded d d!dd d"ded d#ddd$ded d%d!dd&dd'Zed(eeefd)ed*dfd+Zd*efd,Zy)-McpLocalConfigcommandrargsNcwdenvT inherit_env$@startup_timeout wait_for_logrrzLaunch Commandr>z8Executable used to start the MCP stdio server (e.g. uvx)r? ArgumentsrFz)Command arguments, defaults to empty listzWorking Directoryz1Optional working directory for the launch commandrxzEnvironment Variablesrz0Additional environment variables for the processzInherit Parent Envrlz:Whether to start from parent env before applying overrides)rr@rArBrrryzStartup Timeoutrwz"Seconds to wait for readiness logszReady Log Patternz6Regex that marks readiness when matched against stdoutrr)rrrrrrrrrzr{r c t||}t|d|}t|jd}g}t |D]B\}}t |d|d} t |ts td| |j|Dt|d|} t|d|d } | d} |jd } | \t | tstd t |d | jD cic]\} }t| t|}} }ni}|jd d }|d }n8t |ttfr t|}ntdt |d t|d|}|jdd}|d}n8t |ttfr t|}ntdt |d|||| |t!| |||| Scc}} w)Nrrzargs[r}zargs entries must be stringsrrTrrzenv must be a mappingrrzstartup_timeout must be numericrrrr) rrrrrrrrr{)rrrr+rrr-r>r rVrrr r4rrwrl)rYrzr{rrargs_rawnormalized_argsrargarg_pathrr env_mappingrrrrrrrrs r'rzMcpLocalConfig.from_dicts!$-gy$7w{{623%'!(+HC"45Q8Hc3'!"@(KK  " "3 ' , 7E40#G]D$O  Kkk%(  "k73!"9;tU;STT.9.?.?.AB.Ada3q63q6>.ACBC $5t<  "O  U| 4#M2O?TSdAef f#G^TB !++k37  "I #u 6o.I9;t[;YZ Z [)+%  )Cs2Gc x|jt|j|jxsdtt |j j |j|j|jxsdf}tjt|jdjS)NrKr)rrrrrrr4rrrrrrrrrs r'rzMcpLocalConfig.cache_key,s LL $))  HHN &)* +        # ||DM009:DDFFr))rgrhrir>rkrlistrr rr8rrrrlrrwrrrrmrnr rrrrr)r'rrs LD1D$s)1Ct5Cc3h5K!OU!#L#*#Iu#)R   $!C  ,K  0&J  '-T +"*< (,P  %)Z  sAKF2 WS#X.2 2 AQ2 2 h G3 Gr)rfunctionzUse local Python functions)rr mcp_remotez#Connect to an HTTP-based MCP server mcp_localz.Launch and connect to a local stdio MCP serverc eZdZUeed<dZedzed<dZedzed<eddddd edd dd d d eddddd dZ e de e e effdZe deeefdeddfdZe de eefffd ZxZS) ToolingConfigr.Nconfigprefixz Tool Typer>TzfSelect a tooling adapter registered via tooling_type_registry (function, mcp_remote, mcp_local, etc.).r?z Tool PrefixFzXOptional prefix for all tools from this source to prevent name collisions (e.g. 'mcp1').rxzTool ConfigurationobjectzConfiguration block validated by the chosen tool type (Python function list, MCP server settings, local command MCP launch, etc.).)r.rrr c|tjDcic]\}}td||c}}Scc}}w)Nr)rrL)r6r4r)rYrrs r' child_routeszToolingConfig.child_routeshsG%D$E$K$K$M $M j 84 0* <$M   s8rzr{ct||}t|d|} t|}|jd}|t dt|d|j|t|d}t|d|} |||| |S#t$r>}t dt t jt|d|d}~wwxYw)Nr.ztooling.type must be one of rztooling requires config block)r{r)r.rrr{) rrr1rr rr6keysrr+rr) rYrzr{rrrexcconfig_payload config_objrs r'rzToolingConfig.from_dictos!$-"7FD9  0>J!X.  !={4QY?Z[ [)).{4QY?Z)[ gx6 ZTRR .t4S4U4Z4Z4\/].^_D&)  s B C 9CC c \t |}|jd}|rt}t }t |j }|r|dnd}|Dcic]'}||j|xsijd)}}t|||t|||d<|Scc}w)Nr.rr")rOrrP) rQrRr+r6r9rrrr) rYrZ type_spec registrationsr$ type_names default_valuer descriptionsrfs r'rRzToolingConfig.field_specss#%IIf% ;=M13Hm0023J-7JqMTMXbcXbPTD8<<#5#;"@"@"KKXbLc#%5j,O E&M  ds,B))rgrhrir>rkrr rrrmrnrrr.rr rrrRrorps@r'rrIs I $FJ $FC$J $A  "&r  "-]  !K2 T(D,<"<=  SWS#X.SSSS(Do!56r)r)/rjrcopyr dataclassesrrrtypingrrr r r entity.configs.baser r rrrrrrrrrentity.enum_optionsrutils.registryrrutils.function_catalogrrr%rUr>r.r(r1r6r9r;rrrrrrr)r'rs#1122    92H!0# O OZ Ot O  O#$z*:Qc4 3C.C)DQ_Dd38n)<$=_ ^j^ ^B XX Xv oGjoG oGd LGZLG LG\!, 5 @ JJJ Jr)