`gj=UdZddlmZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z m Z mZmZddlmZddlmZej(eZdZd ZdZd Zd$d%d Zd&d Zej:Zdad e d<d'dZ!edf d(dZ"d)dZ#dddddddidgddZ$dddddddidgddZ%dd digddZ&Gd!d"eZ'd*d#Z(y)+u ByteRover memory plugin — MemoryProvider interface. Persistent memory via the ByteRover CLI (``brv``). Organizes knowledge into a hierarchical context tree with tiered retrieval (fuzzy text → LLM-driven search). Local-first with optional cloud sync. Original PR #3499 by hieuntg81, adapted to MemoryProvider ABC. Requires: ``brv`` CLI installed (npm install -g byterover-cli or curl -fsSL https://byterover.dev/install.sh | sh). Config via environment variables (profile-scoped via each profile's .env): BRV_API_KEY — ByteRover API key (for cloud features, optional for local) Config via config.yaml: memory: byterover: auto_extract: false # disable automatic brv curate hooks Working directory: $HERMES_HOME/byterover/ (profile-scoped context tree) ) annotationsN)Path)AnyDictListOptional)MemoryProvider) tool_error xct|tr|S||St|ttfr t|St|tr(|j j }|dvry|dvry|S)N>1onyestrueT>0noofffalseF) isinstanceboolintfloatstrstriplower)valuedefaulttexts T/root/.hermes/venv/lib/python3.12/site-packages/plugins/memory/byterover/__init__.py _coerce_boolr"0sn%  }%#u&E{%{{}""$ - - . . NcD ddlm}|}|jdi}t|tsiS|jdi}t|tr |r t |S|jdi}t|tr t |S iS#t $rYiSwxYw)a Read ByteRover's profile-scoped memory config. New memory-provider setup stores non-secret provider settings under ``memory.``. Some users also set ``memory.provider_config`` from early docs/issues, so accept it as a compatibility fallback. r) load_configmemory byteroverprovider_config)hermes_cli.configr%getrdict Exception)r%config memory_configr( legacy_configs r!_load_plugin_configr0@s 1 8R0 -.I'++K< ot ,( (%))*;R@ mT * & & + I   I s0B.B",B BB Optional[str]_cached_brv_pathct5ttdk7rtndcdddS dddtjd}|sXt j }|dz dz dz t d|dz dz dz g}|D]}|j st|}nt5ttdk7rtndcdddS|xsdaddd|S#1swYxYw#1swY|SxYw)z?? 3 3q6223sI1A1D=#D==F(F()E5, F(5E> : F(F( F#F(#F(c"ddlm}|dz S)z:Profile-scoped working directory for the brv context tree.rget_hermes_homer')hermes_constantsrgrfs r!rNrNs0  { **r# brv_queryzSearch ByteRover's persistent knowledge tree for relevant context. Returns memories, project knowledge, architectural decisions, and patterns from previous sessions. Use for any question where past context would help.objectquerystringzWhat to search for.)type description)rm propertiesrequired)namern parameters brv_curateuStore important information in ByteRover's persistent knowledge tree. Use for architectural decisions, bug fixes, user preferences, project patterns — anything worth remembering across sessions. ByteRover's LLM automatically categorizes and organizes the memory.contentzThe information to remember. brv_statusuMCheck ByteRover status — CLI version, context tree stats, cloud sync state.ceZdZdZdddZeddZddZdZddZ ddZ d d dd Z d d dd Z d d dd Z ddZddZd dZd!dZd"dZd#dZd#dZddZy)$ByteRoverMemoryProviderz,ByteRover persistent memory via the brv CLI.Nc| t|n t|_t|jj dd|_d|_d|_d|_d|_ y)N auto_extractTr4r) r+r0_configr"r* _auto_extract_cwd _session_id _turn_count _sync_thread)selfr-s r!__init__z ByteRoverMemoryProvider.__init__sU'-'9tF|?R?T )$,,*:*:>*JDQ 8<r#cy)Nr'rs r!rqzByteRoverMemoryProvider.namesr#ctduS)z0Check if brv CLI is installed. No network calls.Nr?rs r! is_availablez$ByteRoverMemoryProvider.is_availables "$..r#c$ddddddddd d d gd gS) Napi_keyz,ByteRover API key (optional, for cloud sync)T BRV_API_KEYzhttps://app.byterover.dev)keyrnsecretenv_varurlryzAAutomatically curate completed turns and compression/memory hooksrr)rrnrchoicesrrs r!get_config_schemaz)ByteRoverMemoryProvider.get_config_schemas:!M(2  &b!"G,    r#c tt|_||_d|_t |jj ddy)NrTrC)rrNr|r}r~rrO)r session_idkwargss r! initializez"ByteRoverMemoryProvider.initializes; ' % TYYdT:r#ctsy y)Nr4z# ByteRover Memory Active. Persistent knowledge tree with hierarchical context. Use brv_query to search past knowledge, brv_curate to store important facts, brv_status to check state.rrs r!system_prompt_blockz+ByteRoverMemoryProvider.system_prompt_blocks " : r#r4)rc*|r t|jtkrytdd|jddgt|j }|dr;|j dr*|dj}t|tkDrd |Sy) zRun brv query synchronously before the agent's first LLM call. Blocks until the query completes (up to _QUERY_TIMEOUT seconds), ensuring the result is available as context before the model is called. r4rk--NrHrIrArLz## ByteRover Context )lenr_MIN_QUERY_LENrd_QUERY_TIMEOUTr|r*_MIN_OUTPUT_LEN)rrkrrbrLs r!prefetchz ByteRoverMemoryProvider.prefetchs EKKM*^; dEKKM%40 1"   ) H!5H%++-F6{_,/x88r#cy)z7No-op: prefetch() now runs synchronously at turn start.Nr)rrkrs r!queue_prefetchz&ByteRoverMemoryProvider.queue_prefetch"s r#cxjdz c_jstjdyt j t kryfd}jr6jjrjjdtj|dd _jjy) z:Curate the conversation turn in background (non-blocking).z3ByteRover sync_turn skipped (auto_extract disabled)Nc dddddd}tdd|gtjy#t$r }tj d|Yd}~yd}~wwxYw)NzUser: iz Assistant: curaterrzByteRover sync failed: %srd_CURATE_TIMEOUTr|r,loggerdebug)combinedrcassistant_contentr user_contents r!_syncz0ByteRoverMemoryProvider.sync_turn.._sync1sl =#L$$7#8 FWX]Y]F^E_`tX.+ = 8!<< =s-1 AAAg@rHTzbrv-synctargetdaemonrq) r~r{rrrrrris_alivejoin threadingThreadstart)rrrrrs``` r! sync_turnz!ByteRoverMemoryProvider.sync_turn&s A!! LLN O  |!!# $~ 5  =   !2!2!;!;!=    " "3 " /%,,J  !r#cjstjdy|dvssyfd}tj|dd}|j y)z+Mirror built-in memory writes to ByteRover.z7ByteRover memory mirror skipped (auto_extract disabled)N>addreplacec dk(rdnd}tddd|dgtjy#t$r }tj d |Yd}~yd}~wwxYw) Nuserz User profilez Agent memoryrr[z] rz"ByteRover memory mirror failed: %sr)labelrcrtrrs r!_writez7ByteRoverMemoryProvider.on_memory_write.._writeLse F*0F*:tqr'%;<+ F A1EE Fs.2 AAATz brv-memwriter)r{rrrrr)ractionrrtrts` `` r!on_memory_writez'ByteRoverMemoryProvider.on_memory_writeDsO!! LLR S  + +7  F   F4n M  r#cjstjdy|syg|ddD]f}|jdd}|jdd}t |t s8|j sI|dvsNj|d|dd hsyd jfd }tj|d d }|jy)z;Extract insights before context compression discards turns.z?ByteRover pre-compression flush skipped (auto_extract disabled)r4iNrolert>r assistantz: i c tdddgtjtj dt y#t $r }tjd|Yd}~yd}~wwxYw)Nrrz[Pre-compression context] rz,ByteRover pre-compression flush: %d messagesz*ByteRover pre-compression flush failed: %s)rdrr|rinforr,r)rcrpartsrs r!_flushz7ByteRoverMemoryProvider.on_pre_compress.._flushnsf Nt'B8*%MN+ JCPUJW N I1MM NsAA A.A))A.Tz brv-flushr) r{rrr*rrrappendrrrr) rmessagesmsgrrtrrrrs ` @@r!on_pre_compressz'ByteRoverMemoryProvider.on_pre_compressYs!! LLZ [CD> 9C7762&Dggi,G'3'GMMOH]@] vR 78  9 99U# N   F4k J  r#c$tttgSN) QUERY_SCHEMA CURATE_SCHEMA STATUS_SCHEMArs r!get_tool_schemasz(ByteRoverMemoryProvider.get_tool_schemas|sm];;r#c |dk(r|j|S|dk(r|j|S|dk(r|jStd|S)NrirsruzUnknown tool: ) _tool_query _tool_curate _tool_statusr )r tool_namer]rs r!handle_tool_callz(ByteRoverMemoryProvider.handle_tool_calls]  ###D) ) , &$$T* * , &$$& &N9+677r#c|jr8|jjr|jjdyyy)Ng$@r)rrrrs r!shutdownz ByteRoverMemoryProvider.shutdowns;   !2!2!;!;!=    " "4 " 0"> r#c|jdd}|s tdStdd|jddgt|j }|dst|jdd S|jd dj}|rt |tkrtjd d iSt |d kDr|dd dz}tjd |iS)Nrkr4zquery is requiredrrrrArBz Query failedrLrbzNo relevant memories found.i@z [... truncated]) r*r rdrrr|rrjsondumps)rr]rkrbrLs r!rz#ByteRoverMemoryProvider._tool_querys"%12 2 dEKKM%40 1"   i fjj.AB BHb)//1V6::x)FGH H v; ET]%::Fzz8V,--r#c|jdd}|s tdStdd|gt|j}|dst|jdd St j d d iS) Nrtr4zcontent is requiredrrrrArBz Curate failedrbzMemory curated successfully.)r*r rdrr|rr)rr]rtrbs r!rz$ByteRoverMemoryProvider._tool_curatess((9b)34 4 tW %#  i fjj/BC Czz8%CDEEr#ctdgd|j}|dst|jddSt j d|jddiS) NstatusrrArBzStatus check failedrLr4)rdr|r r*rr)rrbs r!rz$ByteRoverMemoryProvider._tool_statussT8*bdii@i fjj2GHI Izz8VZZ"%=>??r#r)r-zOptional[Dict[str, Any]])returnr)rr)rrrNone)rkrrrrr)rkrrrrr)rrrrrrrr)rrrrrtrrr)rList[Dict[str, Any]]rr)rr)rrr]r+rrrr)r]r+rr)__name__ __module__ __qualname____doc__rpropertyrqrrrrrrrrrrrrrrrrr#r!rwrws~6=/ ";  9;$?A Y["<*!F<81 .. F@r#rwc6|jty)z/Register ByteRover as a memory provider plugin.N)register_memory_providerrw)ctxs r!registerrs  !8!:;r#)F)rrrrrr)rzDict[str, Any])rr1)r]z List[str]rHrrIrrr+)rrr))r __future__rrloggingrPr8rUrpathlibrtypingrrrragent.memory_providerr tools.registryr getLoggerrrrrrrr"r0Lockr7r2__annotations__r?rdrNrrrrwrrr#r!rs:,#  ,,0%   8 $  > !"&-&6.<$3$3!%$3N+    h7LM I  $  >  9WX K  $ b#22F \@n\@F