iAtdZddlmZmZmZddlmZmZmZm Z ddl m Z ddl m Z mZddlmZmZmZddlmZmZmZmZmZmZmZmZmZmZeGdd eZeGd d eZeGd d eZ eGddeZ!eGddeZ"eGddeZ#eGddeZ$y)z)Memory-related configuration dataclasses.) dataclassfieldreplace)AnyDictListMapping)AgentExecFlowStage)enum_options_forenum_options_from_values)SchemaLookupErrorget_memory_store_schemaiter_memory_store_schemas) BaseConfig ConfigErrorConfigFieldSpecChildKey ensure_list optional_dict optional_strrequire_mapping require_str extend_pathc &eZdZUeed<eed<dZedzed<dZedzed<eeZ e ee fed<e de ee fd ed dfd Zedd d dddeddd dddeddd ddddeddd ddddeddddidd d!Zy)"EmbeddingConfigprovidermodelNapi_keybase_urldefault_factoryparamsdatapathreturnct||}t|d|}t|d|}t|d|}t|d|}t|d|xsi}|||||||S)Nrrrrr")rrrrr"r$)rrrr) clsr#r$mappingrrrrr"s G/Users/bowang/.openclaw/workspace/ChatDev/entity/configs/node/memory.py from_dictzEmbeddingConfig.from_dict$sp!$-w D9GWd3w 48T:w$7=2HE7X^dkoppzEmbedding ProviderstrTopenaizEmbedding provider)name display_name type_hintrequireddefault descriptionzEmbedding Modelztext-embedding-3-smallzEmbedding model namezAPI KeyFzAPI keyz ${API_KEY}r.r/r0r1r3r2advancezBase URLzCustom Base URLz ${BASE_URL}zCustom Parameterszdict[str, Any]z(Embedding parameters (temperature, etc.)r.r/r0r1r2r3r5)rrrrr")__name__ __module__ __qualname__r,__annotations__rrrdictr"rr classmethodr r*r FIELD_SPECSr+r)rrsM JGS4ZHcDj"48FDcN8qWS#X.qqARqq$-,  !*,.  #"!  $#)! ",&B G,Kr+rc eZdZUeed<dZeedzed<dZeed<dZ eed<e de ee fd ed dfd Z ed d d ddedddddedddddddeddd dddddZy)FileSourceConfig source_pathN file_typesT recursiveutf-8encodingr#r$r%c t||}t|d|}|jd}d}|]t|}g}t |D]@\} } t | t stdt|d| d|j| B|}|jdd} t | tstdt|dt|d |xsd } |||| | | S) Nr$rBz"file_types entries must be stringsz file_types[]rCTzrecursive must be booleanrErD)rArBrCrEr$) rrgetr enumerate isinstancer,rrappendboolr) r'r#r$r( file_pathfile_types_valuerBitems normalizedidxitemrecursive_valuerEs r)r*zFileSourceConfig.from_dictds!$-6 ";;|4'+  ' 01E$&J&u- T!$,%&JKX\`klokppq^rLsuu!!$'.$J!++k48/409;t[;YZ ZT:EgyZ?emtxyyr+zFile/Directory Pathr,z$Path to file/directory to be indexedr.r/r0r1r3zFile Type Filterz list[str]Fz4List of file type suffixes to limit (e.g. .md, .txt)zRecursive SubdirectoriesrLz-Whether to include subdirectories recursivelyr6z File EncodingzEncoding used to read files)r$rBrCrE)r7r8r9r,r:rBrrCrLrEr<r rr*rr=r>r+r)r@r@]s#'JS D 'ItHczWS#X.zzASzz, .>  &+!N  %3G $(5 1!Kr+r@c eZdZUdZedzed<dZedzed<ede ee fdeddfdZ e ddd d d d e dddd dedZ y)SimpleMemoryConfigN memory_path embeddingr#r$r%ct||}t|d|}d}d|vr)|d$tj|dt |d}||||S)NrWrXr$)rWrXr$)rrrr*r)r'r#r$r(rW embedding_cfgs r)r*zSimpleMemoryConfig.from_dictsf!$-"7M4@  ' !gk&:&F+55gk6JQ\]acnQo5pM{m$OOr+zMemory File Pathr,FzSimple memory file pathTr.r/r0r1r3r5Embedding Configurationrz Optional embedding configurationr.r/r0r1r3child)rWrX)r7r8r9rWr,r:rXrr<r rr*rr=r>r+r)rVrVs"Kt"(,I%,PWS#X.PPAUPP'+1  %2':!  Kr+rVc eZdZUdZedzed<eeZe e ed<dZ e dzed<e deeefdeddfd Zedd d d d deddddde edddd de dZy)FileMemoryConfigN index_pathr file_sourcesrXr#r$r%c t||}t|jd}|stdt |dg}t |D]9\}}|j tj|t |d|d;t|d|}| t|d|}d} d|vr)|d$tj|dt |d} |||| | S) Nrcz,file_sources must contain at least one entryz file_sources[rGrZrbrWrX)rbrcrXr$) rrrHrrrIrKr@r*rr) r'r#r$r( sources_rawsourcesrQrRrbr[s r)r*zFileMemoryConfig.from_dicts!$-!'++n"=> LkZ^`nNop p*,";/IC NN+55dTUbcfbgghSiAj5k l0"'<>  %g}dCJ ' !gk&:&F+55gk6JQ\]acnQo5pMjw-^bccr+z Index Pathr,FzVector index storage pathTr\zFile Source Listzlist[FileSourceConfig]zList of file sources to ingestr^r]rzEmbedding used for file memory)rbrcrX)r7r8r9rbr,r:rlistrcrr@rXrr<r rr*rr=r>r+r)raras!Jd !+0+FL$'(F(,I%,dWS#X.ddASdd(&%3  (+.8"  %2'8!  #Kr+rac eZdZUdZedzed<dZeed<ede ee fdeddfdZ e dd d d d d de dddd ddddZ y)BlackboardMemoryConfigNrW max_itemsr#r$r%ct||}t|d|}|jdd}t|tr|dkrt dt |d||||S)NrWrkrjrz$max_items must be a positive integer)rWrkr$)rrrHrJintrr)r'r#r$r(rWmax_items_values r)r*z BlackboardMemoryConfig.from_dictsc!$-"7M4@ !++k48/3/?a3GDkRVXcFde e{oDQQr+zBlackboard Pathr,FzqJSON path for blackboard memory writing. Pass 'auto' to auto-create in working directory, valid for this run onlyautoTr4z Maximum Itemsrmz:Maximum number of memory items to retain (trimmed by time)r6)rWrk)r7r8r9rWr,r:rkrmr<r rr*rr=r>r+r)riris"Kt"IsRWS#X.RRAYRR'*L %(T Kr+ric eZdZUeed<eed<dZedzed<edeee fdeddfdZ defd Z e dd d d d e ddd d de dddd ddZ edeeeeffdZedeee fffd ZxZS)MemoryStoreConfigr.typeNconfigr#r$r%cht||}t|d|}t|d|} t|}d|vs|dt dt |d|j j|dt |d}|||||S#t$r }t d|dt |d|d}~wwxYw) Nr.rrzunsupported memory store type ''rsz"memory store requires config blockrZ)r.rrrsr$)rrrr rr config_clsr*) r'r#r$r(r. store_typeschemaexc config_objs r)r*zMemoryStoreConfig.from_dict s!$-7FD1 &$7  s,Z8F 7 "gh&7&?BKPTV^D_` `&&001BUY[cId0e :jtLL! s ? |1M{[_agOhior r ss B B1B,,B1cr|js tdt|jd|jS)Nzmemory store payload missingrs)rsrrr$)selfs r)require_payloadz!MemoryStoreConfig.require_payload0s-{{^_ _{{r+z Store Namer,TzUnique name of the memory storerTz Store TypezMemory store typezStore ConfigurationobjectznSchema required by the selected store type (simple/file/blackboard/etc.), following that type's required keys.)r.rrrsctjDcic]\}}td||jc}}Scc}}w)Nrs)rvalue)rrOrrv)r'r.rxs r) child_routeszMemoryStoreConfig.child_routesMsK!: ; A A C C f 84 0&2C2C C C   s"Ac 0t|}|jd}|rmt}t |j }|j Dcic]\}}||j}}}t||t||d|d<|Scc}}w)NrrT)preserve_label_case)enum enum_options) super field_specsrHrrgkeysrOsummaryrr ) r'specs type_spec registrationsnamesr.rx descriptions __class__s r)rzMemoryStoreConfig.field_specsTs#%IIf% 57M++-.EEREXEXEZZ\T6D&..0EZL[#5e\_cdE&M  \sB)r7r8r9r,r:rsrr<r rr*r}rr=rrrrrr __classcell__)rs@r)rqrqs I I $FJ $ MWS#X. M MAT M M  %9   %+  ".I  K0 T(D,<"<=   Do!56  r+rqceZdZUeed<dZeedzed<dZe ed<dZ e ed<dZ e ed <dZe ed <ed eeefd ed dfdZedddddedddddeDcgc]}|j(c}}eeedddddddeddddddded dd ddd!ded d"d ddd#dd$Zycc}}w)%MemoryAttachmentConfigr.Nretrieve_stagetop_ksimilarity_thresholdTreadwriter#r$r%c t||}t|d|}|jd}d}|=t|}g}t |D] \} } |j t | "|}|jdd} t| tr| dkrtd t|d|jd d }t|ttfstd t|d |jd d}t|tstdt|d |jdd}t|tstdt|d|||| t||||S#t$rE} tdt D cgc]} | jncc} wc} t|d| d| d} ~ wwxYw)Nr.rz&retrieve_stage entries must be one of zretrieve_stage[rGrrrz top_k must be a positive integerrrz$similarity_threshold must be numericrTzread must be booleanrzwrite must be boolean)r.rrrrrr$)rrrHrrIrKr ValueErrorrrrrJrmfloatrL)r'r#r$r(r. stages_rawstages stage_listparsedrQrRrystage top_k_valuethreshold_value read_value write_values r)r*z MemoryAttachmentConfig.from_dictms!$-7FD1[[!12 26  !$Z0J/1F&z2 TMM"4T":;3Fkk'1- +s+{a/?@+dT[B\] ]!++&Similarity threshold (-1 means no similarity threshold filter)z Allow ReadrLz,Whether to read this memory during executionz Allow Writez4Whether to write back to this memory after execution)r.rrrrr)r7r8r9r,r:rrr rrmrrrrLrr<r rr*rrr r=).0rs00r)rrdsU I6:ND+,t3:E3N"&%&D$E4+ WS#X.+ + AY+ + \ &=  *!)0g+=>+=%%+++=>)*<= ! 5 !0'/X!  %F !&N Y5K?s B>rN)%__doc__ dataclassesrrrtypingrrrr entity.enumsr entity.enum_optionsr r schema_registryr rrentity.configs.baserrrrrrrrrrrr@rVrarirqrr>r+r)rs/11+++J    =j= =@ =z= =@  B 2z2 2j  Z   F F F FR kZk kr+