ei8dZddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z ej eZeejZddZdd Zd dZd!dZd"dZd#dZd$dZGddZd%dZd&dZdS)'aMemory provider plugin discovery. Scans two directories for memory provider plugins: 1. Bundled providers: ``plugins/memory//`` (shipped with hermes-agent) 2. User-installed providers: ``$HERMES_HOME/plugins//`` Each subdirectory must contain ``__init__.py`` with a class implementing the MemoryProvider ABC. On name collisions, bundled providers take precedence. Only ONE provider can be active at a time, selected via ``memory.provider`` in config.yaml. Usage: from plugins.memory import discover_memory_providers, load_memory_provider available = discover_memory_providers() # [(name, desc, available), ...] provider = load_memory_provider("mnemosyne") # MemoryProvider instance ) annotationsN)Path)ListOptionalTuplereturnOptional[Path]c| ddlm}|dz }|r|ndS#t$rYdSwxYw)z8Return ``$HERMES_HOME/plugins/`` or None if unavailable.r)get_hermes_homepluginsN)hermes_constantsr is_dir Exception)r ds 9/home/div/.hermes/hermes-agent/plugins/memory/__init__.py_get_user_plugins_dirr(sd444444 O   )HHJJ(qqD( tts *- ;;pathrboolc|dz }|sdS |ddd}d|vpd|vS#t$rYdSwxYw) uHeuristic: does *path* look like a memory provider plugin? Checks for ``register_memory_provider`` or ``MemoryProvider`` in the ``__init__.py`` source. Cheap text scan — no import needed. __init__.pyFreplace)errorsNi register_memory_providerMemoryProvider)exists read_textr)r init_filesources r_is_memory_provider_dirr2s }$I     u$$I$66uu=)V3Q7G67QQ uus%A AAList[Tuple[str, Path]]ct}g}trttD]}|r|jdr1|dz sI||j| |j|ft}|rt|D]g}|r|jdr1|j|vr;t|sK| |j|fh|S)zYield ``(name, path)`` for all discovered provider directories. Scans bundled first, then user-installed. Bundled takes precedence on name collisions (first-seen wins via ``seen`` set). )_.r) set_MEMORY_PLUGINS_DIRrsortediterdirname startswithraddappendrr)seendirschilduser_dirs r_iter_provider_dirsr0Bsh D#%D!!##-/7799:: - -E<<>> UZ%:%::%F%F M)1133  HHUZ KKU+ , , , ,%&&H-H,,..// - -E<<>> UZ%:%::%F%F zT!!*511  KKU+ , , , , Kr(strct|z }|r|dz r|St}|r*||z }|rt |r|SdS)z^Resolve a provider name to its directory. Checks bundled first, then user-installed. rN)r%rrrr)r(bundledr/users rfind_provider_dirr6ds "D(G~~W}4<<>>$&&H$ ;;== 4T:: K 4r1List[Tuple[str, str, bool]]cg}tD]\}}d}|dz }|rj ddl}t|5}||pi}dddn #1swxYwY|dd}n#t $rYnwxYwd} t|} | r| }nd}n#t $rd}YnwxYw| |||f|S)zScan bundled and user-installed directories for available providers. Returns list of (name, description, is_available) tuples. Bundled providers take precedence on name collisions. plugin.yamlrN descriptionTF) r0ryamlopen safe_loadgetr_load_provider_from_dir is_availabler+) resultsr(r.desc yaml_filer<fmeta availableproviders rdiscover_memory_providersrIzs} G*,,00 eM)        )__3>>!,,2D333333333333333xx r22      .u55H "$1133 !    III  dI.//// NsGB A) B )A- -B 0A- 1B  BB(C CCOptional['MemoryProvider']c(t|}|std|dS t|}|r|Std|dS#t $r'}td||Yd}~dSd}~wwxYw)a'Load and return a MemoryProvider instance by name. Checks both bundled (``plugins/memory//``) and user-installed (``$HERMES_HOME/plugins//``) directories. Bundled takes precedence on name collisions. Returns None if the provider is not found or fails to load. z9Memory provider '%s' not found in bundled or user pluginsNz:Memory provider '%s' loaded but no provider instance foundz'Failed to load memory provider '%s': %s)r6loggerdebugr@warningr)r( provider_dirrHes rload_memory_providerrQs%T**L  PRVWWWt*<88  OSUYZZZt @$JJJtttttsA A B*B  BrOc|j}t|jvp|jtk}|rd|nd|}|dz }|sdS|t jvrt j|}ndD]}|t jvrttj}|dkr|j}|dz }|rtj |t|t|g} | rZtj | } | t j|< | j| #t $rYwxYwtj |t|t|g} | sdStj | }|t j|<|dD]} | jdkr| j} |d | } | t jvrtj | t| }|r}tj |}|t j| < |j|#t $r&}t&d | |Yd}~d}~wwxYw | j|nT#t $rG}t&d ||t j|dYd}~dSd}~wwxYwt-|d rft/} |||jr|jSn3#t $r&}t&d ||Yd}~nd}~wwxYwddlm}t9|D]Y}t;||d}t=|t>r1tA||r!||ur |cS#t $rYUwxYwZdS)uImport a provider module and extract the MemoryProvider instance. The module must have either: - A register(ctx) function (plugin-style) — we simulate a ctx - A top-level class that extends MemoryProvider — we instantiate it plugins.memory._hermes_user_memory.rN)r zplugins.memoryr )submodule_search_locationsz*.pyr#zFailed to load submodule %s: %szFailed to exec_module %s: %sregisterzregister() failed for %s: %sr)r)!r(r%parentsparentrsysmodulesr__file__ importlibutilspec_from_file_locationr2module_from_specloader exec_modulerglobstemrLrMpophasattr_ProviderCollectorrVrHagent.memory_providerrdirgetattr isinstancetype issubclass)rOr( _is_bundled module_namermodrX parent_path parent_initspec parent_modsub_filesub_name full_sub_namesub_specsub_modrP collectorr attr_nameattrs rr@r@s  D&)==kATXkAkK.9\*D***?\VZ?\?\K},I     tck!!k+&4 ! !FS[(("8nn3 Y&&"-"4K)M9 %%'' !$>AAK 0 047 4D4D3EBD!%.^%D%DT%J%J .8 F+! K33J????(!!! D!~55 Y(+L(9(9':6   4n--d33#& K %))&11 Z ZH} --}H*77X77MCK//$>AA!3x==Z'n==hGGG18CK .Z 33G<<<<$ZZZ %F WXYYYYYYYYZ  K # #C ( ( ( (    LL7a H H H KOOK . . .44444  sJB&((  B LL # # #! * )) * B B B LL7q A A A A A A A A B544444XX sIt,, tT " " z$'G'G .. tvv      4sl.E  EE&J J1 J,,J16K L"r?rr+rLrM)rBactive_provider plugin_dircli_filermrncli_modrrr help_textr;rDr<rErFrCrrPs rdiscover_plugin_cli_commandsrBs/"G  % % ' '133O #?33J H$H ??  %);;gz?PTg?gK=Hz9O9999NzetNzNzNzK.Z #+ % %k+.GG>99S]]D t{ n55d;;G'.CK $ K # #G , , ,w==  %% N>o===  .        )__3>>!,,2D333333333333333xx r22' $I"&K    W&B&B&BDII>W&6==  #&$$%        ZZZ DoWXYYYYYYYYZ Nso AH!$A)H!!H!0GF' G'F++G.F+/GH! GH!GAH!! I+I  I)rr )rrrr)rr )r(r2rr )rr7)r(r2rrJ)rOrrrJ)rr)rr)r __future__rr\importlib.utilloggingrYpathlibrtypingrrr getLoggerrrLr[rXr%rrr0r6rIrQr@rfrrrr1rrs*#""""" ((((((((((  8 $ $d8nn+     D,""""J2ddddN        (    TTTTTTr1