`gj.jpdZddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z ddl mZddlmZej"eZdZdZd Zd Zd Zd-d Zd.d ZddddddddddddddgddZdddddddidgddZd d!ddd"ddd#dd$d%d&gddZd'd(dd%dd)did%gddZGd*d+eZd/d,Z y)0u*Mem0 memory plugin — MemoryProvider interface. Server-side LLM fact extraction, semantic search, and automatic deduplication via the Mem0 Platform API (cloud) or OSS (self-hosted) via Memory. Original PR #2933 by kartik-mem0, adapted to MemoryProvider ABC. Configuration ------------- Secret (lives in $HERMES_HOME/.env or the environment): MEM0_API_KEY — Mem0 Platform API key (required for platform mode) MEM0_HOST — Base URL of a self-hosted Mem0 server. When set, the plugin talks to that server directly over HTTP (X-API-Key auth) instead of the cloud API. Behavioral settings (live in $HERMES_HOME/mem0.json, set via `hermes memory setup`): mode — Backend mode: "platform" (default) or "oss" host — Self-hosted Mem0 server URL (alt: MEM0_HOST env var). When set, routes to the self-hosted HTTP backend. user_id — Canonical user identifier. When set, it is applied uniformly across every gateway (CLI, Telegram, Slack, Discord, …) so the same human gets one merged memory store. When unset, the gateway-native id (e.g. Telegram numeric id, Discord snowflake) is used instead. agent_id — Agent identifier (default: hermes) The matching MEM0_MODE / MEM0_USER_ID / MEM0_AGENT_ID environment variables are still read as a backward-compatible fallback, but mem0.json is the canonical home for these non-secret settings. ) annotationsN)AnyDictList)MemoryProvider) tool_errorx)MemoryNotFoundErrorValidationError hermes-userct|j}|tvryt|j }d|vxs d|vxsd|vS)zUTrue for user-caused errors (bad ID, not found) that should NOT trip circuit breaker.T404z not foundz valid uuid)type__name___CLIENT_ERROR_TYPESstrlower)excetypeerr_strs O/root/.hermes/venv/lib/python3.12/site-packages/plugins/memory/mem0/__init__.py_is_client_errorr@sL I  E ###hnnG G  P{g5 P9PPcddlm}tjj ddtjj ddtjj ddtjj dd id }tjj d }|r||d <|d z }|j rb t j|jd}|j|jDcic]\}}||dk7r||c}}|S|Scc}}w#t$rY|SwxYw)a+Load config from env vars, with $HERMES_HOME/mem0.json overrides. Environment variables provide defaults; mem0.json (if present) overrides individual keys. This avoids a silent failure when the JSON file exists but is missing fields like ``api_key`` that the user set in ``.env``. r)get_hermes_home MEM0_MODEplatform MEM0_API_KEY MEM0_HOST MEM0_AGENT_IDhermes)modeapi_keyhostagent_idoss MEM0_USER_IDuser_id mem0.jsonzutf-8)encoding) hermes_constantsrosenvirongetexistsjsonloads read_textupdateitems Exception)rconfig env_user_id config_pathfile_cfgkvs r _load_configr?Ms1 {J7::>>."5 {B/JJNN?H= F**..0K'y!#k1K zz+"7"7"7"IJH MMHNN,<9DAqmRa49 : M6M 9   M s%AD1 D+ D1+D11 D>=D> mem0_searchuSearch the user's memories by meaning; returns facts ranked by relevance. Use this before answering any question that may depend on what you know about the user (preferences, facts, history, people, projects, past decisions). For multi-part or multi-hop questions, call it several times — vary the wording and run follow-up searches on what earlier results reveal; one search is rarely enough.objectstringzWhat to search for.)r descriptionintegerz#Max results (default: 10, max: 50).booleanzBRerank results for relevance (default: false, platform mode only).)querytop_krerankrF)r propertiesrequired)namerC parametersmem0_addu(Store a durable fact about the user, verbatim (no LLM extraction). Call this the moment the user states a lasting preference, correction, decision, or personal detail worth recalling on future turns — don't wait to be asked to remember. Skip transient chit-chat and facts you've already stored.contentzThe fact to store. mem0_updateuReplace the text of an existing memory by its ID (take the ID from a mem0_search result). Use when a stored fact has changed or was wrong — correct it in place instead of adding a duplicate.zMemory UUID to update.zNew text content.) memory_idtextrPrQ mem0_deletezDelete a memory by its ID (take the ID from a mem0_search result). Use when a stored fact is obsolete or the user asks you to forget it; prefer mem0_update if the fact merely changed.zMemory UUID to delete.ceZdZdZdZeddZddZdZdZ ddZ dZ dd Z d d Z d Zd Zd!d Zd"dZd"dZddZd#dZd$dZd%dZddd&dZddd'dZd(dZd)dZdZd*dZy)+Mem0MemoryProviderzMem0 memory with server-side extraction and semantic search. Supports Platform API (cloud) and OSS (self-hosted) modes via MEM0_MODE. cd|_d|_d|_d|_d|_t |_d|_d|_d|_ d|_ d|_ d|_ d|_ d|_d|_d|_t#j$|_t#j$|_t#j$|_d|_y)Nrr!r$Fclirg)_config_backend_mode_api_key_host_DEFAULT_USER_ID_user_id _agent_id_rerank_default_channel _sync_thread_prefetch_thread_prefetch_query_prefetch_result_prefetch_done_consecutive_failures_breaker_open_until threadingLock _breaker_lock _sync_lock_prefetch_lock_atexit_registeredselfs r__init__zMem0MemoryProvider.__init__s     ( !$   $! "#%&"#& &^^-#..*'nn."'rcy)Nmem0rns rrKzMem0MemoryProvider.namesrct}|jdd}|dk(r*t|jdijdSt|jdxs|jdS)Nr%rr) vector_storer&r')r?r1bool)rocfgr%s r is_availablezMem0MemoryProvider.is_availablesdnwwvz* 5=r*..~>? ?CGGI&9#''&/::rcddl}ddlm}||dz }i}|jr! |j|j }|j|ddlm }|||dy#t $rY.wxYw)z'Write config to $HERMES_HOME/mem0.json.rN)Pathr,)atomic_json_writei)r%) r3pathlibrzr2r4r5r8r6utilsr{)rovalues hermes_homer3rzr;existingr{s r save_configzMem0MemoryProvider.save_configsw ;'+5      %4::k&;&;&=> ++xe<   s A,, A87A8c t}|jdd}|dk7}ddd|ddd d d d d ddddddddddddddgdgS)Nr%rr)r&zMem0 Platform API keyTr zhttps://app.mem0.ai)keyrCsecretrJenv_varurlr'z3Self-hosted Mem0 server URL (leave blank for cloud)Fr")rrCrJrr+zUser identifierr)rrCdefaultr(zAgent identifierr$rHzEnable reranking for recallfalsetrue)rrCrchoices)r?r1)rorwr%api_key_requireds rget_config_schemaz$Mem0MemoryProvider.get_config_schemasnwwvz*5=.EQUcsAOXm n+`nsAL M.?M Z/Ah W-JW^lrt{k| }   rc"ddlm}|||y)N) post_setup)_setupr)rorr9rs rrzMem0MemoryProvider.post_setups&;'rc ddlm}|dd |jdk(r(ddlm}||jjdiS|jr#dd lm }||j|jSdd lm }||jS#t$rYt$rYwxYw#t$r;}tjd |j|t||_Yd}~yd}~wwxYw) Nr)ensurez memory.mem0F)promptr)r) OSSBackend)SelfHostedBackend)PlatformBackendz/Mem0 backend failed to initialize (%s mode): %s)tools.lazy_depsr ImportErrorr8rYrXrrWr1r[rrZrloggererrorr _init_error)ro _lazy_ensurerrres r_create_backendz"Mem0MemoryProvider._create_backend s   > u 5 zzU"0!$,,"2"25""=>>zz7( CC 1"4==1 1       LLJDJJXY Z"1vD  s:B6B* .B*9B* B'B'&B'* C.31C))C.c|j5|jtkr dddytj|j k\rd|_ dddy dddy#1swYyxYw)zBReturn True if the circuit breaker is tripped (too many failures).NFrT)rjrf_BREAKER_THRESHOLDtime monotonicrgrns r_is_breaker_openz#Mem0MemoryProvider._is_breaker_open%sm    )),>>  ~~4#;#;;-.*        sA')A'A''A0c|d|}|jdk(rjt|j}d|vsd|vsd|vrE|jj dij di}|d|j dd d z }|S) Nz: r) connectionrefusedtimeoutruz (check that provider vector storez is running))rYrrrWr1)roprefixrmsgrvss r _format_errorz Mem0MemoryProvider._format_error/s3%  :: #hnn&Gw&)w*>)wBV\\%%eR044^RHrvvj.'I&J,WW rcT|j5d|_dddy#1swYyxYw)Nr)rjrfrns r_record_successz"Mem0MemoryProvider._record_success8s(    +)*D & + + +s'c|j5|xjdz c_|j}|tk\r!tjt z|_nd}dddtk\rrd}|jdk(rD|jjdijdi}|jdd}d|d }tjd |t |yy#1swYxYw) Nrrr!r)rurunknown Check that your z' vector store is running and reachable.zXMem0 circuit breaker tripped after %d consecutive failures. Pausing API calls for %ds.%s) rjrfrrr_BREAKER_COOLDOWN_SECSrgrYrWr1rwarning)rocounthintrrs r_record_failurez"Mem0MemoryProvider._record_failure<s      & &! + &..E**+/>>+;>T+T(   & &DzzU"\\%%eR044^RH66*i8*8*4[\ NN/-t  '  s ACC(c 4t|_|jjdd|_|jjdd|_|jjdd|_|jjd}|t k(rd}|xs|jdxst |_|jjdd|_|jjd d }t|tr|jd vn t||_ |jdxsd |_|j|_|j r4|j"s't%j&|j(d |_yyy)Nr%rr&r!r'r+r(r$rHF)r1yesrVT)r?rWr1rYrZr[r\r]r^ isinstancerrrvr_r`rrXrmatexitregister_shutdown_backend)ro session_idkwargs configured_rrs r initializezMem0MemoryProvider.initializePs@#~ \\%%fj9  ((B7 \\%%fb1 \\%%i0 ) )J"Ofjj&;O?O ))*h? llx/3=c33GCIIK/ /TRUY  :.7% ,,. ==!8!8 OOD22 3&*D #"9=rcd|jiS)Nr+)r]rns r _read_filtersz Mem0MemoryProvider._read_filtersss 4==))rc:|jrd|jiSiS)Nchannel)r`rns r_write_metadataz"Mem0MemoryProvider._write_metadata{s.2]] 4==)BBrc|jdk(rd}n|jrd}nd}|jdk(r|jsdnd}d|d |jd |S) Nr)zOSS (self-hosted)zself-hosted (HTTP API)zplatform (cloud API)rz Rerank is available on search.r!z# Mem0 Memory Active. Mode: z. User: u. You have persistent memory of this user from past conversations. You should call mem0_search before answering anything that could depend on prior context (the user's preferences, facts, history, people, projects, or earlier decisions) — do not rely on the chat window alone, and do not assume you have no memory. For multi-part or multi-hop questions, run several searches with different wording/angles and follow-up searches on what the first results surface; one search is rarely enough. Keep searching until you have every fact the question needs before you answer. Tools: mem0_search to find memories, mem0_add to store facts, mem0_update and mem0_delete to manage by ID.)rYr[r])ro mode_label rerank_notes rsystem_prompt_blockz&Mem0MemoryProvider.system_prompt_blocksp :: ,J ZZ1J/J<@JJ*zCMem0MemoryProvider._start_prefetch.._run..s9RqBqc(9RszMem0 prefetch failed: %sT) searchrr1joinrr8rrdebugrlrcrdre)bodyresultsrlinesrbackendrFros r_runz0Mem0MemoryProvider._start_prefetch.._runsD <!..4#5#5#7r%)8?}"Yxx,YY- 9RE9R0RRD$$&$$ /''50,0D)*.D' / /Z <$$& 7;; < / /s:+CB=B=/CC>=C C; &C66C;>DTz mem0-prefetchtargetdaemonrK) rXrrlrcrerbis_aliverdrhThreadstart)rorFrtrs`` @rrz"Mem0MemoryProvider._start_prefetchs -1F1F1H --   (##u,&& ( (((T-B-B-K-K-M  ( ( $)D $&D !"'D  ( /$   DO L   &$%D ! &  ? ( (: & &s#C:!'C:C:D:DDr!)rc&|j|}||S|j||j5|j|k(r |jnd}dddr|j t |j|}||Sy#1swY7xYw)zDRecall memories for the CURRENT question with a short hot-path wait.Nrr!)rrrlrcrbr_PREFETCH_WAIT_SECS)rorFrcachedthreads rprefetchzMem0MemoryProvider.prefetchs..u5  M U#   V.2.B.Be.KT**QUF V  KK 3K 4..u5  M V Vs BBcjjryfd}j5jr6jj rjj djr$jj r dddyt j|dd_jjdddy#1swYyxYw)zESend the turn to Mem0 for server-side fact extraction (non-blocking).NcBj}|y ddddg}|j|jjdj j y#t $r0}jtjd|Yd}~yd}~wwxYw)NuserrolerN assistantTr+r(infermetadatazMem0 sync failed: %s) rXaddr]r^rrr8rrr)rmessagesrassistant_contentro user_contents r_syncz+Mem0MemoryProvider.sync_turn.._syncsmmG :# =(5FG  MM!^^!113 $$& :$$&5q99 :sAA%% B.&BB@rTz mem0-syncr) rXrrkrarrrhrr)rorrrrs``` r sync_turnzMem0MemoryProvider.sync_turns == D$9$9$;  :*__ &  T%6%6%?%?%A!!&&s&3  T%6%6%?%?%A  & & !* 0 0dQ\ ]D     # # % & & &sA*C''7C''C0c.ttttgSr) SEARCH_SCHEMA ADD_SCHEMA UPDATE_SCHEMA DELETE_SCHEMArns rget_tool_schemasz#Mem0MemoryProvider.get_tool_schemassz=-HHrc  |jt|dd}d}|jdk(rD|jj dij di}|j dd}d|d }t j d d |d |iS|jrmd }|jdk(rE|jj dij di}|d|j dddz }t j d |iS|dk(rS|j dd} | s tdS tdtt|j ddd} |j dt|dd} t| tr| jdv} n t| } |jj!| |j#| | } |j%| st j ddiS| Dcgc]7}|j d|j dd|j d d!d"9}}t j |t'|d#S|d%k(r|j d&d}|s td'S |jj1d(|d)g|j2|j4d|j7*}|j%t|t8r|j d+nd}|jdk(s |j:rd,nd-}t j ||d.S|d0k(r|j d1d}|j d2d}|s td3S|s td4S |jj=||}|j%t j |S|d7k(r`|j d1d}|s td3S |jj?|}|j%t j |Std9|Scc}w#t($r@}t+|s|j-t|j/d$|cYd}~Sd}~wwxYw#t($r5}|j-t|j/d/|cYd}~Sd}~wwxYw#t($rS}t+|rtd5|cYd}~S|j-t|j/d6|cYd}~Sd}~wwxYw#t($rS}t+|rtd5|cYd}~S|j-t|j/d8|cYd}~Sd}~wwxYw):Nrz unknown errorr!r)rurrz Check that z is running and reachable.rzMem0 backend not initialized: .zWMem0 temporarily unavailable (multiple consecutive failures). Will retry automatically.rz is running.r@rFz!Missing required parameter: queryrrGr2rHr_F)r0norrzNo relevant memories found.idrscorer)rrr)rrz Search failedrMrNz#Missing required parameter: contentrrrevent_idz Fact stored.zFact queued for storage.)rrzFailed to storerOrPrQz%Missing required parameter: memory_idz Missing required parameter: textzMemory not found: z Update failedrRz Delete failedzUnknown tool: ) rXgetattrrYrWr1r3dumpsrrmaxminintrrrrvrrrlenr8rrrrr]r^rdictr[r6delete)ro tool_nameargsrerrrrrrrFrG rerank_rawrHrrr7rrNrrrPrQs rhandle_tool_callz#Mem0MemoryProvider.handle_tool_calls == $ ?CDzzU"\\%%eR044^RH66*n=%hZ/IJ::w*HQtf(UVW W  "kCzzU"\\%%eR044^RH*266*n+M*Nl[[::wn- -  %HHWb)E!"EFF JAs3txx'<#=rBC!XXh>OQV0WX j#.'--/7KKF!*-F--..ud>P>P>RZ_hn.o$$&::x1N&OPP?FH:;!"d quuXr7J#$55!#46HHzzec%j"IJJ * $hhy"-G!"GHH L**$9: MM!^^!113 +$$&5?5M6::j1SW)-u)< nQkzzSh"GHH - 'b1I88FB'D!"IJJ!"DEE J--i>$$&zz&))- 'b1I!"IJJ J--i8$$&zz&))N9+677yH J'*((*!$"4"4_a"HII J, L$$&!$"4"45F"JKK L J#A&%(:9+&FGG$$&!$"4"4_a"HII  J J#A&%(:9+&FGG$$&!$"4"4_a"HII  Js/CP4?P4

 *R93R>9R> T T"T(*TTT U<)U7U<*U71U<7U<c |jr"|jjd|_yy#t$rYywxYwr)rXcloser8rns rrz$Mem0MemoryProvider._shutdown_backendbs> }} ##% $    s -1 ==c|j|jfD](}|s|js|jd*|j y)Nrr)rbrarrr)rors rshutdownzMem0MemoryProvider.shutdownjsI''):):; $AQZZ\s# $  rN)returnr)r&rv)rrr9rr&None)rrrr8r&r)rrr&r')r&zDict[str, Any])rrrrr&r')rFrr&z str | None)rFrr&r')rFrrrr&r)rrrrrrr&r')r&zList[Dict[str, Any]])rrrrr&rr&r')r __module__ __qualname____doc__rppropertyrKrxrrrrrrrrrrrrrrrrrr r!rr%rsrrrTrTs (.; =  (6+(!+F*C  :&#J9; Y[!&FI\8|!rrTc6|jty)z*Register Mem0 as a memory provider plugin.N)register_memory_providerrT)ctxs rrrqs  !3!56r)rr8r&rv)r&rr()!r+ __future__rrr3loggingr/rhrtypingrrragent.memory_providerrtools.registryr getLoggerrrrrrrr\rr?rr r r rTrrsrrr6sg@#   ""0%   8 $@!Q P  G&7LM'8]^(9}~ I ,   9MN K  &  N "*;ST%6IJ !&)  $  D  (;ST !M  *m!m!` 7r