i% dZddlmZmZddlmZmZmZddlm Z ddl m Z m Z ddl mZmZmZmZmZmZmZddlmZddlmZdd lmZmZed Zd d d edeeded zdd fdZd edeefdZ deeeeffdZ!deeeeefffdZ"eGddeZ#eGddeZ$eGddeZ%y )z1Subgraph node configuration and registry helpers.) dataclassreplace)AnyDictMapping)LogLevel)enum_options_forenum_options_from_values) BaseConfig ConfigErrorConfigFieldSpecChildKeyrequire_mapping require_str extend_path) EdgeConfig)MemoryStoreConfig)Registry RegistryErrorsubgraph_sourceN) descriptionname config_clsrreturncD|rd|ind}tj|||y)z/Register a subgraph source configuration class.summaryN)targetmetadata)subgraph_source_registryregister)rrrrs I/Users/bowang/.openclaw/workspace/ChatDev/entity/configs/node/subgraph.pyregister_subgraph_sourcer"s',7 ;'DH%%d:%Qctj|}|j}t|trt |t std|d|S)NzEntry 'z' is not a BaseConfig subclass)rgetload isinstancetype issubclassr r)rentryrs r!get_subgraph_source_configr+%sK $ ( ( .EJ j$ 'z*j/QgdV+IJKK r#cztjDcic]\}}||jc}}Scc}}wN)ritemsr&rr*s r!"iter_subgraph_source_registrationsr0-s42J2P2P2R S2R;4D%**, 2R SS Ss7c tjDcic]\}}|t|jxsi!c}}Scc}}wr-)rr.dictrr/s r!iter_subgraph_source_metadatar31s<@X@^@^@` a@`uD$u~~+, ,@` aa as$Ac^eZdZUeed<edeeefdeddfdZde ddddd iZ y ) SubgraphFileConfig file_pathdatapathrcHt||}t|d|}|||S)Nr8)r6r8)rr)clsr7r8mappingr6s r! from_dictzSubgraphFileConfig.from_dict9s)!$-6 YT22r#zSubgraph File PathstrTz@Subgraph file path (relative to yaml_instance/ or absolute path)r display_name type_hintrequiredrN) __name__ __module__ __qualname__r=__annotations__ classmethodrrr<r FIELD_SPECSr#r!r5r55sZN3WS#X.33AU33 -Z  Kr#r5c&eZdZUeeefed<edeeefdeddfdZ d;dZ e ddd d d e d dd dd e dddde jje Dcgc]}|jc}}dee e dddddde dddd d e dd d!d d"e#e d$d%d&dd'e#e d(d)d*did+e d,d-d dd. e d/d0d dd1 e d2d3d4dd5 e d6d7d4dd8 d9 Zedeee fffd: ZxZScc}}w)<SubgraphInlineConfiggraphr7r8rc@t||}|t||S)N)rKr8)rr2)r:r7r8r;s r!r<zSubgraphInlineConfig.from_dictNs!$-gT22r#cd|jvr tdt|jdd|jvr tdt|jdy)Nnodesz!subgraph config must define nodesedgesz!subgraph config must define edges)rKr rr8selfs r!validatezSubgraphInlineConfig.validateSsS $** $A;tyyZaCbc c $** $A;tyyZaCbc c %r#idz Subgraph IDr=TzSubgraph identifierr>rzSubgraph DescriptionFzsDescribe the subgraph's responsibility, trigger conditions, and success criteria so reviewers know when to call it. log_levelz Log Levelz enum:LogLevelzSubgraph runtime log level)rr?r@rAdefaultenumr enum_optionsis_majority_votingzMajority Votingboolz2Whether to perform majority voting on node results)rr?r@rArUrrNz Node Listz list[Node]z2Subgraph node list, must contain at least one noderOz Edge Listzlist[EdgeConfig]zSubgraph edge list)rr?r@rArchildmemoryz Memory Storeszlist[MemoryStoreConfig]ztProvide a list of memory stores if this subgraph needs dedicated stores; leave empty to inherit parent graph stores.vars Variableszdict[str, Any]z"Variables passed to subgraph nodes organization Organizationz%Subgraph organization/team identifierinitial_instructionzInitial Instructionz"Subgraph level initial instructionstartz Start Nodezstr | list[str]z\Start node ID list (entry list executed at subgraph start; not recommended to edit manually)endzEnd NodezEnd node ID list (used to collect final subgraph output, not part of execution logic). This is an ordered list: earlier nodes are checked first; the first with output becomes the subgraph output, otherwise continue down the list.) rSrrTrXrNrOr[r\r^r`rarbcvt|}|jd}|rddlm}t |||d<|S)NrNr)Node)rZ)super field_specsr%entity.configs.node.noderdr)r:specs nodes_specrd __class__s r!rfz SubgraphInlineConfig.field_specss9#%YYw'  5$Zt3WS#X.33AW33d&-  '/N  %$%MM'''/0x#))x04)(3  .%*L  !$"L  !$(,  "(/O#   $&<  ('?   /&.<  !%'v  #'@  k\K|Do!56U1s!D rJceZdZUeed<dZedzed<edeee fdeddfdZ ddZ e dd d d d e dddd d dZ edeeeeffdZedeee fffd ZxZS)SubgraphConfigr(Nconfigr7r8rct||}t|d|}d|vr|drtdt|dd|vs|dtdt|d t |}|j|dt|d}||||S#t $r>}tdt tjt|d|d}~wwxYw) Nr(r\z6vars is only allowed at root level (DesignConfig.vars)rsz.subgraph configuration requires 'config' blockzsubgraph.type must be one of )r8)r(rsr8) rrr rr+rlistr0keysr<)r:r7r8r; source_typerexc config_objs r!r<zSubgraphConfig.from_dicts!$-!'648 W VXcdhjpXqr r 7 "gh&7&?NP[\`bjPkl l 3K@J  ))'(*;+dT\B])^  JTBB /5W5Y5^5^5`0a/bcD&)  s B C9CCc|js tdt|jdt |jdr|jj yy)Nzsubgraph config missingrsrR)rsr rr8hasattrrRrPs r!rRzSubgraphConfig.validatesF{{7TYYPX9YZ Z 4;; + KK " ,r#zSubgraph Source Typer=TzURegistered subgraph source such as 'config' or 'file' (see subgraph_source_registry).r>zSubgraph ConfigurationobjectuvPayload interpreted by the chosen type—for example inline graph schema for 'config' or file path payload for 'file'.)r(rsc|tjDcic]\}}td||c}}Scc}}w)Nrs)fieldrm)r0r.r)r:rrs r! child_routeszSubgraphConfig.child_routessG%G$H$N$N$P $P j 84 0* <$P   s8c Lt|}|jd}|r|t}t }t |j }|Dcic]'}||j|xsijd)}}t||t||d|d<|Scc}w)Nr(rT)preserve_label_case)rVrW) rerfr%r0r3rurvrr ) r:rh type_spec registrationsrnamesr descriptionsrjs r!rfzSubgraphConfig.field_specss#%IIf% >@M46H++-.ELQLQDx||D)/R44Y??E $5e\_cdE&M  s,B!rk)rBrCrDr=rErsr rFrrr<rRr rGrrr(rrfrn)rjs@r!rrrrs I $FJ $CWS#X.CCAQCC(# /o  "1Q  K" T(D,<"<=  Do!56r#rr)&__doc__ dataclassesrrtypingrrr entity.enumsrentity.enum_optionsr r entity.configs.baser r r rrrrentity.configs.edge.edgerentity.configs.node.memoryrutils.registryrrrr=r(r"r+r0r3r5rJrrrHr#r!rs*7*%%!J082$$56# R  RZ  Rt R  RST*-=TDd:6F1F,GTbtCc3h,?'@b  ( t:t tn GZG Gr#