`gj8dZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddlmZddlmZddlmZmZmZmZmZd(d)d Zd*d+d Zd,d Zd-d.d Zd/d Zd0dZd1dZd2dZd3dZ d2dZ!d2dZ"d4dZ#dZ$dZ%dZ&d5d6dZ'd6dZ(d7dZ)d8dZ*d9dZ+d:d;dZ,dd Z/d?d!Z0d@d"Z1dAd#Z2dBd$Z3dCd%Z4dDd&Z5d>d'Z6y)EuBSetup wizard for Mem0 plugin — interactive and flag-based modes.) annotationsN)Path)Any)get_hermes_home) LLM_PROVIDERSEMBEDDER_PROVIDERSVECTOR_PROVIDERS KNOWN_DIMSvalidate_oss_configclddlm}|Dcgc]\}}|r|d|n|}}}|||||Scc}}w)z*Interactive single-select with arrow keys.r)curses_radiolist )selectedcancel_returns)hermes_cli.curses_uir)titleitemsdefaultrlabeldesc display_itemss M/root/.hermes/venv/lib/python3.12/site-packages/plugins/memory/mem0/_setup.py_curses_selectrsS5! E4#5'D6-M E=7SZ [[ s0cV|rd|dnd}|rtjjd||dtjjtjj rt j d}ntjjj}notjjd||dtjjtjjj}|xs|xsdS)zz has_oss_flags..;s VA599Q< Vs) oss_llm_keyoss_vector_pathoss_vector_urlF) parse_flagsr argvany)r7s@r has_oss_flagsr?6s;  %E V} V!U VV r-c||ntjdd}iddddddddd dd dd dd dd dddddddddddddddddddddd}idddddddddd dd dd d d d!d d"dd#dd$dd%dd&dd'dd(dd)dd*d+d,d-}d.}|t|krV||d/k(r d0|d1<|dz }n4|||vr(|dzt|kr||dz||||<|d2z }n|dz }|t|krV|S)3z7Parse CLI flags from argv. Returns dict of flag values.Nrr/rapi_keyhostoss_llmopenair9 oss_llm_model oss_llm_url oss_embeddeross_embedder_keyoss_embedder_modeloss_embedder_url oss_vectorqdrantr:r;oss_vector_hostoss_vector_portoss_vector_userF)oss_vector_passwordoss_vector_dbnameuser_iddry_runz--modez --api-keyz--hostz --oss-llmz --oss-llm-keyz--oss-llm-modelz --oss-llm-urlz--oss-embedderz--oss-embedder-keyz--oss-embedder-modelz--oss-embedder-urlz --oss-vectorz--oss-vector-pathz--oss-vector-urlz--oss-vector-hostz--oss-vector-portz--oss-vector-userrPrQrR)z--oss-vector-passwordz--oss-vector-dbnamez --user-idrz --dry-runTrS)r r=len)r=argsr7flag_mapis rr<r<@sn#4!"D2  8  r    r  B b B h 2 " 2 2!" 2#$ "+E0&Y & Y    ?   . 0  4 0   . , . .!" .#$"72)H. A c$i- 7k !#E)  FA !W QUSY%6'+AE{E(47# $ FA FA c$i- Lr-cX|jdd}t|}|jdxs|d}d|i}|jdxs|jd}|r|jdr|||d<|jd d}t|}|jd xs|d}d|i} |jd xs|jd} | r|jdr| | |d<tj|} | r| | d <|jd d} t| } t | d}| dk(rE|jdr|d|d<|jdr|j dd|d|d<n| dk(r|jdr|d|d<|jdrt|d|d<|jdr|d|d<|jdr|d|d<|jdr|d|d<||d || d | |d d!}i}|jd"r|jd#r |d#||d$<|jd"r |jd%r|d%||d$<||fS|jd"r!||k(r|jd#r |d#||d$<||fS)&zBuild OSS config dict + env_writes from parsed flags. Returns (oss_config, env_writes) where oss_config goes into mem0.json and env_writes maps env var names to secret values for .env. rCrDrE default_modelmodelrF default_url base_url_keyrGrIrJembedding_dimsrKrLdefault_configr:pathr;NurlpgvectorrMrBrNportrOuserrPpasswordrQdbname)providerconfig)llmembedder vector_store needs_keyr9env_varrH)r4rr r r dictpopint)r7llm_idllm_def llm_model llm_configllm_url embedder_id embedder_defembedder_modelembedder_config embedder_urldims vector_id vector_def vector_config oss_config env_writess rbuild_oss_configrs YYy( +FF#G /*Fgo.FI")9!5Jii &D'++m*DG7;;~..5 7>*+))NH5K%k2LYY34U _8UN'.&?O99/0SL4D4D]4SL ((88D ^45 >>. )D ,0() ,1I!),J$456MH 99& '$)*;$y$#t $rYwxYwcc}w)?uPlatform mode setup — uses the framework's schema-based flow. Delegates to the same code path the framework uses when post_setup doesn't exist, preserving the original platform onboarding experience. rAzMem0 Platform API keyT MEM0_API_KEYzhttps://app.mem0.ai)key descriptionr)requiredrmrarRUser identifier hermes-user)rrragent_idAgent identifierhermesrerankzEnable reranking for recallfalsetrue)rrrchoicesrz Configuring mem0: rrrr)Frrmrarrrr...Nrz (current: , blank to keep)r)z Get yours at rSz [dry-run] Would save config: ' [dry-run] Would write API key to .env [dry-run] No files written. platformr/rB MEM0_HOSTu- ⚠ MEM0_HOST is set in your environment (u). It overrides platform mode — remove it from ~/.hermes/.env (or unset it) or Hermes will keep routing to the self-hosted server. save_configmem0memoryrgMem0MemoryProvider.envz Memory provider: mem0! Activation saved to config.yaml Provider config savedz API keys saved to .env$ Start a new session to activate. )rrrrrrrnprintr4strlowerindexrosenvironrUr,r(hermes_cli.configrplugins.memory.mem0rr)rrhr7schemaexisting_configrprovider_configrfieldrrr is_secretrrmrac choice_itemscurrent current_idxselrmaskedr+effective_defaultrrrgs r_setup_platformrs*AT_cp~H] ^*; V+=(S)FSZhnpwgxy FO{#k1K "jj)>)>)@AO?+O!#J #$%+Elyy,))I&IIh. ))I&))I&ii 99Y C9$4).y)9J~ &  9-45QG5L5%))#w7GK3w<--/7:%mmCL,>,>,@A  2dVlKPC#*3Y]^C'*$K%+N yy 1/1BCD  ; < /0(OF!OF  zz~~k2&,,.   ;'()1 1 .#)F8Z 6!#H +64 $v-z: %& -. #$ () 23u   ,6s#M* M:* M76M7c2ddl}ddlm} |j|j ddd}|j |dt d |y#|jj$rt d |Yyt$rt d |d YywxYw) zIBest-effort reachability check for a self-hosted Mem0 server (non-fatal).rN/z/docsGETmethodtimeoutu ✓ Mem0 server reachable at u ✓ Mem0 server responding at u ⚠ Could not reach u2 — check the URL and that the server is running.) urllib.errorurllib.requestrequestRequestrstripurlopenrerror HTTPErrorr)rBurllib_urlreqreqs r_check_selfhosted_serverrRs$aooS!1 2%8oGQ' /v67 << ! !9 078 a &tf,^_`asAA'B<BBc6i}t|dz }|jr$ tj|j }t |}td|jdxs td|jdxsd}|stdtjy|jd }i}|jd r |d |d <n`tjjd d }|r*t|d kDrd|ddnd} td| dd} n tdd} | r| |d <|jdxs td|jdxsd} td|jdxsd} |jdr8td|d| d| |r td t!|td!yd"|d#<||d<| |d<| |d<d$d%lm} d&|d'd(<| |d$d)lm}|}|j%|||rt+t|d*z |t!|td+td,|td-td.|r td/td0y#t $rYRwxYw)1uSelf-hosted mode setup — point at an existing Mem0 dashboard server. For users already running the Dockerized Mem0 FastAPI server: stores the server URL (behavioral -> mem0.json) and an optional API key (secret -> .env as MEM0_API_KEY). rz* Configuring mem0 (self-hosted server): rBz,Mem0 server URL (e.g. http://localhost:8888)Nrz7 Error: a server URL is required for self-hosted mode.filerrArrrrrrzServer API key (current: rTrz'Server API key (blank if AUTH_DISABLED)rRrrrrrrSz% [dry-run] Would save config: host=z , user_id=z , agent_id=rrrr/rrrrrgrrz& Memory provider: mem0 (self-hosted)z Server: rrz API key saved to .envr)rrrrrrrnrr4r,r stderrrrrrUrrrrrr)rrhr7rrrrBr existing_keyrr+rRrrrrgs r_setup_selfhostedrbsO{#k1K "jj)>)>)@AO?+O 89 99V  6##F+3t!D  GcjjY ;;s D!#J yy%*9%5 >"zz~~nb9 25l2Ca2Gs<,-.UF5fX=MNW[\CCDQC ),J~ &ii "g?#6#6y#A#R]'G)?3F3Fz3R3^V^_H yy 6tfJwi{[cZdef  ; < & /0(OF"OF!(OI"*OJ-#)F8Z 6!#H +64 $v-z:T" 34 Jtf  -. #$ '( 23C   s#J JJc |jds t||yt|\}}t|}|r:|D] }t d|t j "t jd|jdxstjdd}|d d }|d d } |d d } |jd rt dt d|d d d|d djdddt d|d d d|d djdddt d| |r+t ddj|jt|t dy|rtt|dz |t|d|d|dt!|| | ddlm} d |d!d <| |t|t d"t d#|d d d|d djdddt d|d d d|d djdddt d$| |r t d%t d&t d't d(y))uOSS mode setup — build config from flags or interactive prompts. Non-interactive when --mode was set explicitly via flags (post_setup already resolved mode). Interactive only when mode was chosen via curses picker. _mode_from_flagNz Error: rrrRUSERrrirgrjrkrSz! [dry-run] OSS config would be:z LLM:  (rhr[r) Embedder: z Vector: z Env vars: z, rrr0rr/rRrr0rrrr! ✓ Mem0 configured (OSS mode) LLM:  Vector:  API keys saved to .env Config saved to mem0.json Provider set in config.yamlr)r4_setup_oss_interactiverr rr rexitrgetenvrkeys_run_connectivity_checksrrr_install_provider_depsrr) rrhr7rrerrorserRrqrvr|rs r _setup_ossrs 99& '{F3-e4J  ,F  4A IaS/ 3 4  ii "Fbii &FG  z *FZ(4K>*:6I yy 23 *U+J78:e;LX;V;Z;Z[bdf;g:hhijk z*5jAB"ZPZE[\dEeEiEijqsuEvDwwxyz YK()  N499Z__->#?"@A B , /04 $v-z:K%GQYbl!mn6; :-#)F8Z Z( ./ N:e,Z89JubJAWX`AaAeAefmoqAr@sst uv N9+ &' *+ )* +, 23r-ctjj|d}|szt|dz }|j r\|j j D];}|j|ds|jdddj}n|rBt|dkDrd|ddnd }tjd |d |d jStjd |d jS)z;Prompt for API key, showing masked existing value if found.rrrrrrrNrrz API key (current: z, blank to keep): z API key: ) rrr4rrrrrrr(rUr&)rrmrrrrrs r_prompt_api_keyrszz~~gr*H  $v- ??  **,779 ??gYa=1#zz#q1!4::DDFLLN % %)$5 5 [ \ JK ;<   s B9E E+*E+c  tdtdtjddtgddtjtjddd t gdd tjtd t d |dtjdd ddt ddt d|dtg dddtjt||dt||\}}|rtd|d|||dt ddStdtd||dt ddS#tj$r}td|Yd}~yd}~wt$r}td |Yd}~yd}~wwxYw)!z)Pull and start pgvector Docker container.z Pulling rrpullTxrrmz-frz Starting container 'z ' on port r%z-dz--namez-ezPOSTGRES_PASSWORD=z-pz:5432)rrcheckr$ru ✓ pgvector running on rpostgres)rBrcrdrerfzJ Warning: Container started but PostgreSQL not yet accepting connections.zD It may need a few more seconds. Config will be saved; retry later.z$ Failed to start Docker container: Nz Docker error: ) r_PGVECTOR_IMAGEr$r%r'r&_PGVECTOR_PASSWORDr)r!CalledProcessErrorr)rBrcr,r-rs rr+r++s* ?+3/0&/:&*C'// 1 $.AB&*B'// 1 &':&;:dV3OP eT ) &'9&:; TF%.   $j>P>P  R tT2.d+A .tfAdV< =d"0B$  ^ _ X Yd"0B$   ( ( 4QC89   $%s*DD*!D**E4=E E4E//E4cd}tjd}t|\}}|s|rltd t j |dgtj tj tddd t|\}}|r td n"td tdtdy|s tdy|D]j}t||rtd|dtd|d t j|xsdd|gdtj td|dly#t$r}td |Yd }~d }~wwxYw#t$r)}td|d|td|Yd }~d }~wwxYw)zEnsure Ollama is running and required models are pulled. Returns True if Ollama is ready, False if user needs to handle it manually. http://localhost:11434ollamaz/ Ollama installed but not running. Starting...serve)r!r localhosti,rru ✓ Ollama startedz Could not start Ollama: Nz Ollama not found. Install it:z1 curl -fsSL https://ollama.com/install.sh | shz" Or on macOS: brew install ollamaFz9 Warning: Ollama not reachable. Models cannot be pulled.u ✓ Model 'z ' availablez Pulling 'z"'... (this may take a few minutes)r2iX)rr$z' pulledz Warning: Could not pull 'z': z Run manually: ollama pull T) r"r# _check_ollamarr$Popenr'r)r_ollama_has_modelr%)modelsra ollama_binr,r-rr[s r_ensure_ollamarFZs #Ch'J # EB  C D 8  )%--j6H6H{E2>%c*A01 3 4 E F 6 7  IJ > S% ( M% 4 5 Kw&HI J > 6hFPS%/%7%79 eWH56 > 5 821#677 8, >3E7#aSAB4UG<== >s0AD67.s!@Q5A:0q0@sF) rrrrrrreadr4rr>r)rar[rresprmnamesrMs ` @rrCrCsnn$$uI%6u$E~~%%c1%5zz$))+&,0HHXr,BCqvr"CC[[%a( @%@@@D s$At=d?t=d@ycc}}wcc}}wcc}}w)Az+Interactive OSS setup using curses pickers.rz LLM ProviderrrZr\rlrmr>z LLM model [z]: z Ollama URL [zEmbedder Providerz embedderz Embedder model [z Vector StoreNrbrr8z PostgreSQL user [z PostgreSQL host [localhost]: r@z PostgreSQL port [5432]: 5432z" PostgreSQL database [postgres]: z' PostgreSQL password (blank if none): rTrez User ID [rz Agent ID [hermes]: rr) rCr9rErFrGrIrJrKrRrBrMrcrNrdrOrPrfrQrr0rrrrrgrrrirrhr[rrrjrrrrr)rrrkrlistr r4rr*r(r r rorrFr0rr r&rprrrrrrr`rrr r)#rrhrmr llm_itemsllm_idxrqrrrrsrurembedder_items embedder_idxrvrwrxrz vector_items vector_idxr| ollama_modelspgvector_config default_userpg_userpg_hostpg_port pg_dbname pg_passwordrRrr7rr-rs# rr r s7FSFYFYF[\FC!G*3A67\I\^Y:G -$$& ' 0FF#G!#J(Ikk-(G{gg. 0BKP -0Jwy) * M'/*B)C3GHNNPlT[\kTl .)?(@DEKKMgQXYfQgK]KcKcKefaqz#8#;<fNf!"5~qIL)..01,?K%k2L!/2N##M2LK [F%:g!6 7yADDFU+G89??AKVGBCIIKYzI!//*STZZ\KW9O.9 +k"))FM"B!C3GHNNPG96=9G,-335H#8HAHYAWz~~gi&8"=]_"}"#,(.B E#26#: #&v'>#? #26#:   z *+::+FE' (%4X%>!"$U+MJ4 $v-z:K%GQYbl!mn6; :J?"?3-#)F8Z Z( ./ N:e,Z89JubJAWX`AaAeAefmoqAr@sst uv N9+ &' *+ )* +, 23i] g$fsVV$8V*c t}t|ft|ft|ffD]:\}}|j |ij d}|s*|j |<t |D]L} td|dtjddddtj|gdd td |N|rddl }|jyy#t$rtd |d |YwxYw)z5Install all optional pip deps for selected providers.pip_depz Installing ruvpipinstallz--pythonT<)rru ✓ Installed z Warning: Could not install z#. Install manually: uv pip install rN)rrr r r4rsortedrr$r%r executabler importlibinvalidate_caches)rqrvr|depsregistryrmdeprs rrrKs UD(&14F 3T+Y79 #ll3#'' 2  HHSM  d| a a M#c* + NNuiS^^SI#R  $SE* + a ##%  a 1#6YZ]Y^_ ` as.A CC.-C.ct|j}|j} |jdddd|fS#t$r}dd|d|fcYd}~Sd}~wwxYw)z7Check that qdrant local storage parent dir is writable.TrzDirectory writable: FzCannot write to rN)r expanduserrrrf)r`prrs r_check_qdrant_pathrbsn T A XXF7 TD 1+F8444 7(1#6667sA A AAAc tjj|jddd}tjj |dy#t $r}dd |d |fcYd }~Sd }~wwxYw) z(Check Ollama is reachable via /api/tags.rrHrrr)TzOllama reachableFzOllama not reachable at rN)rrrrrr)rarrs rrArAmss<nn$$ 3'8 %B5$QsA.' <0Rs;;;ollama_base_urlr=) r4rrrrrrrrr!rA) rvsr`rar,msgrrrnris rr r s  +B vvjX%vvh#''/ffXr"&&u- (.GB C5)*  Hnn,, 3/@-IRW,X&&sA&6  z )ffXr"!#''&+">PT@UVC Ku% & .. #C wwzh&ggh#''(9;ST$C Ku% &' H;C51#FGG Hs AF00 G9GGcL ddl}t|dd}|sytd|jdddD}d}||krKdj d|D}t d |d |d t d t jd |dyy#t$rYyt$rYywxYw)z9Ensure mem0ai meets the minimum version from plugin.yaml.rN __version__c32K|]}t|ywr2)rpr5xs rr8z)_check_min_dep_version..sM1AM.r)rTrc32K|]}t|ywr2)rrs rr8z)_check_min_dep_version..s>!s1v>ru ⚠ mem0ai z installed but >=z required.z Run: uv pip install --python z 'mem0ai>=') rgetattrtuplerrrr rrVr)r installed_verinstalled_partsrequired_partsreq_strs r_check_min_dep_versionrs mT:  M 0C0CC0H!0LMM" ^ +hh>~>>G OM?2CG9JW X 3CNN3C:gYVWX Y ,      sB A5B B#B#"B#cntttjdd}|ddk(rd|d<t |||y|ddvrt |||y|ddk(rt |||ygd }td |d }|dk(rt |||y|d k(rd |d<t |||yt |||y)aJEntry point called by hermes memory setup framework. Routes on --mode (platform / selfhosted / oss); with no flag it shows an interactive picker with all three modes. Platform keeps the framework's original schema-based onboarding; selfhosted points at an existing Mem0 server; oss builds a local SDK config. rNr/r0Tr) selfhostedz self-hostedr))Platformz3Mem0 Cloud API (lightweight, just needs an API key))zSelf-hosted serverz?Connect to an existing self-hosted Mem0 server (Docker/FastAPI))z Open Sourcez1Run Mem0 locally (self-hosted LLM + vector store)z Select moderrTF)rr<r r=rrrr)rrhr7 mode_itemsmode_idxs r post_setuprs  %E V}#' ;. V}55+vu5 V} " VU3J oz1=H1}+vu5 Q#( ;. VU3r-)r)rrrzlist[tuple[str, str]]rrpreturnrp)NF)rrrz str | Noner)boolrr)rrr2)r=zlist[str] | Nonerdict[str, str])r7rrztuple[dict, dict[str, str]])rrrrrNone)rrrrnrr)rrrhrnr7rrr)rBrrr)rrrmrrrrr)r@rS)rBrrcrprz dict | None)rDz list[str]rr)rarr[rrr)r\rnrr)r)rBrrcrprrprr)rrnrr)rmrrrnrr)rrrhrnrr)rqrrvrr|rrr)r`rrtuple[bool, str])rarrr)rBrrcrprr)rrnrr)rr)7__doc__ __future__rr&rrr"rdr$r rrpathlibrtypingrhermes_constantsr_oss_providersrr r r r rr,r?r<rrrrrrrrr&r9r:r0r+rFrCr`r)rkror rrrAr!r rrrKr-rrsH"  ,\""=@<"~5. Bl4^ a M4`54p ; (*)X,^/d F4 v4r&.7<F'< &%4r-