j4UdZddlmZddlZddlZddlmZmZddlmZddl m Z  ddl Z dZ Gd d Zdad ed <d)dZd*dZd+d,dZd-dZ d. d/dZd0d1dZedk(redeZedej'dddZedededej)Zedeed ed!ej/Zed"eed#ed$ej=edej?edZ ed%e re d&dnd'ed(yy#e $rdZ ed Y+wxYw)2a memory_core.py - CyberBrain26 Memory Layer v2 Memory types: - rule: permanent agent rules (active in prompt) - preference: user preferences (active in prompt) - project_decision: valid project decisions (active in prompt) - fact: verified technical facts (active if valid) - conversation: audit history (NOT in prompt) - test: test data (NOT in prompt) - error: diagnostics (NOT in prompt unless explicit) ) annotationsN)UTCdatetime)Path)AnyTFzWARNING: ChromaDB not availablec>eZdZdZhdZhdZdddZddZ d ddZ d ddZ d dd Z dd Z d dd Z dd Z dd Z d ddZd dZd!dZy)" MemoryCorez Core memory management with ChromaDB backend. Active kinds (included in prompt): rule, preference, project_decision, fact Inactive kinds (audit only): conversation, test, error >factrule preferenceproject_decision>testerror conversationNc| tjjdd}t||_|jj ddd|_d|_i|_trStjt|j|_|j jdddi |_ytd y) N CB26_DB_PATHzworkspace/chroma_dbT)parentsexist_ok)pathcyberbrain_memory descriptionzCyberBrain26 long-term memory)namemetadataz.WARNING: Running in mock mode without ChromaDB)osenvirongetrdb_pathmkdirclient collection _mock_storageCHROMADB_AVAILABLEchromadbPersistentClientstrget_or_create_collectionprint)selfrs >/home/oem/CyberBrain26-unified/src/cyberbrain26/memory/core.py__init__zMemoryCore.__init__)s ?jjnn^5JKGG}  4$7"& &*.0 "33T\\9JKDK"kkBB(')HICDO B Cc`tjtjd}|d|S)zGenerate unique memory IDz%Y%m%d%H%M%S%f-)rnowrstrftime)r(kind timestamps r) _generate_idzMemoryCore._generate_id=s-LL%../?@ q $$r+c N||||||r|nd|r|nd|xsd| xsd| xsd| xsdd S)z>Convert metadata to flat dict for ChromaDB (no nested objects)defaultUnsorted r0statussource confidence created_at supersedes expires_at workspace_idtopic parent_topicconversation_id) r(r0r8r9r:r;r<r=r>r?r@rAs r)_metadata_to_dictzMemoryCore._metadata_to_dictBsK$$(2*(2*(5I(j(.B.4"  r+c ^|j|} tjtj } |j ||||| |||| | |  }| ||d}|j r"|j j|g|g| gn||j| <|r|j|d|S)zw Save a new memory record. Returns: Dict with 'id', 'content', 'metadata' r7idcontentr documents metadatasids supersededr8) r2rr.r isoformatrCr addr! update_memory)r(rGr0r8r9r:r<r=r>r?r@rA record_idr;rrecords r) save_memoryzMemoryCore.save_memoryZs(%%d+ \\#&002 ))!!!!%%+*  "g8L ?? OO  ")#*K  -3D  y )    z,  ? r+c~|jd}|s|ri}|r||d<|r||d<|jj|xsi}g}|jdxsg}|jdxsg}|jdxsg}t|D]B\} } | | t|kr|| nd| t|kr|| nid } |j | D|St |j j}|r|D cgc]} | d d|k(s| }} |r|D cgc]} | d d|k(s| }} |Scc} wcc} w) z)List memory records with optional filtersNr0r8)whererKrIrJr4rEr)r r enumeratelenappendlistr!values) r(r0r8rUresultsrecordsrKrIrJidoc_idrRrs r) list_memoryzMemoryCore.list_memorysb ?? &Ev$(E&M&,E(Ooo)))6<"GG++e$*C K06BI K06BI&s^ ' 6 /03y>/Ay|r01C N0B !  v&  'N4--4467G&-O:v1F$1N1OO&-S:x1HF1R1SSNPSs?D5D5D:-D:cx|j|jj|gxsi}|jdxsg}|sy|jdxsg}|jdxsg}|dt|dkDr|dndt|dkDr|ddSidS|jj|S) z"Get a specific memory record by IDNrKrKrIrJrr4rE)r rrWr!)r(rQr[rKrIrJs r) get_memoryzMemoryCore.get_memorys ?? &oo))yk):@bG++e$*C K06BI K06BI!f+.y>A+=9Q<2,/ NQ,>IaL EG  %%)))4 4r+c|j|}|sy|r||dd<|r||dd<|r||dd<|r||d<|jr)|jj|g|dg|dg|S||j|<|S)z Update an existing memory recordNrr8r:r=rG)rKrIrJ)rcr updater!)r(rQr8rGr:r=existings r)rPzMemoryCore.update_memorys??9- -3HZ  * 1;HZ  . 1;HZ  . ")HY  ?? OO " "K#I./#J/0 # -5D  y )r+c(|j|dS)zArchive a memory recordarchivedrM)rP)r(rQs r)archive_memoryzMemoryCore.archive_memorys!!)J!??r+cV|j|}|sy|jd|dz }t|d5}tj||dddd|j r|j j |gn |j|=td |d |d y #1swYSxYw) z"Delete a memory record with backupFdeleted_z.jsonwindentNrbzDeleted z (backup: )T) rcropenjsondumpr deleter!r')r(rQrf backup_filefs r) delete_memoryzMemoryCore.delete_memorys??9-llx {%%@@ +s # -q IIh! , - ?? OO " " { " 3""9-  :k]!<= - -s BB(c|jd}|Dcgc]}|dd|jvr|}}|Dchc]}|djdr|dd }}|Dcgc] }|d|vr| }}tjt j }|Dcgc]#}|djdr |dd|kDr|%}}|jdd |d |Scc}wcc}wcc}wcc}w) z Retrieve active memory context for prompt. Filters: - Only ACTIVE_KINDS (rule, preference, project_decision, fact) - Only status='active' - Excludes superseded and expired records activerMrr0r<rFr=c,|djddS)Nrr;r4)r)r_s r)z4MemoryCore.retrieve_active_context...s!J-++L"=r+T)keyreverseN)r` ACTIVE_KINDSrrr.rrNsort)r(querylimit all_recordsr_active_recordssuperseded_idsr.s r)retrieve_active_contextz"MemoryCore.retrieve_active_contextsI&&h&7 # }V$(9(99   $ }  . jM, '  & wn,   ll3))+% Z=$$\2a mL6QTW6W   =  fu%%;    sC##C(#C- (C2c|j}t|d5}tj||ddddt |S#1swYt |SxYw)z&Export all memory records to JSON filerlrmrnN)r`rqrrrsrW)r( output_pathrrvs r) export_memoryzMemoryCore.export_memory4sV&&( +s # 0q IIk1Q / 0; 0;s A  Actt|5}tj|}dddd}D]Q} |jr+|jj |dg|dg|dgn||j |d<|dz }S|S#1swYcxYw#t $r}td|dd |Yd}~d}~wwxYw) z$Import memory records from JSON fileNrrGrrFrHzImport error for z: )rqrrloadr rOr! Exceptionr')r( input_pathrvdatacountitemes r) import_memoryzMemoryCore.import_memory=s *  99Qr%r?r%r@r%rAr%rdict[str, Any]) r ryuserverifiedNNr5r6r4r4)rGr%r0r%r8r%r9r%r:r%r<rr=rr>r%r?r%r@r%rAr%rrNNr0rr8rrlist[dict[str, Any]])rQr%rdict[str, Any] | None)NNNN) rQr%r8rrGrr:rr=rrrrQr%rboolNrrrintrr)rr%rr)rr%rr)__name__ __module__ __qualname____doc__r~INACTIVE_KINDSr*r2rCrSr`rcrPrirwrrrrBr+r)r r sFL6ND(%&!  !$ .1  %(  -7          6$!%!%%!666 6  6  6666666 6t !$$$  $L5*""!%!% !!! !  !  ! !F@*!.&.&.&  .&` r+r zMemoryCore | None_memory_instancec.t tatS)z"Get or create MemoryCore singleton)rr rBr+r)get_memory_corerWs%< r+c 8tj|fi|S)zSave a memory record)rrS)rGkwargss r)rSrS_s (?  ( ( ;F ;;r+r8c8tj||S)zList memory recordsr0r8)rr`rs r)r`r`ds   ( (d6 ( BBr+c4tj|S)zDelete one memory record by ID.)rrw)rQs r)rwrwjs   * *9 55r+cx|j}|s tdtj||d||S)z+Explicitly save a persistent memory record.z!Memory content must not be empty.ry)rGr0r8r9r:)strip ValueErrorrrS)rGr0r9r:s r)rememberrosHmmoG <==   ( (  ) r+c8tj||S)z)Retrieve active memory context for promptrr)rrrs r)rrs   4 45 4 NNr+__main__z=== MemoryCore v2 Test ===z 1. Saving test record...zTest memory for CyberBrain26rry)rGr0r8z Saved: rFz 2. Listing all records...z Total: z recordsz 3. Retrieving active context...z Active: z% records (should exclude 'test' kind)z 4. Archiving test record...z Status: rzN/Az === Test complete ===)rr )rGr%rrrrr)r manualuser_confirmed) rGr%r0r%r9r%r:r%rrrr)!r __future__rrrrrrpathlibrtypingrr#r" ImportErrorr'r r__annotations__rrSr`rwrrrcorerRrrWryrircrhrBr+r)rs # "- ssn '+#*< C 6&       &O  z &'