`gjeRdZddlmZddlmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl m Z mZddlmZmZddlmZddlmZmZmZmZej2eZdd Zdd Zdd Zdd Zdd Z ddZ!ddZ" d ddZ#ddZ$eGddZ%GddZ&y) uACP session manager — maps ACP sessions to Hermes AIAgent instances. Sessions are persisted to the shared SessionDB (``~/.hermes/state.db``) so they survive process restarts and appear in ``session_search``. When the editor reconnects after idle/restart, the ``load_session`` / ``resume_session`` calls find the persisted session in the database and restore the full conversation history. ) annotations)get_hermes_homeN)datetimetimezone) dataclassfield)Lock)AnyDictListOptionalc0ddlm}|t|S)aTranslate Windows ACP cwd values when Hermes itself is running in WSL. Windows ACP clients can launch ``hermes acp`` inside WSL while still sending editor workspaces as Windows drive paths (``E:\Projects``) or ``\\wsl.localhost\`` UNC paths. Store and execute against the POSIX form so agents, tools, and persisted ACP sessions all agree on the usable workspace. Native Linux/macOS keeps the original cwd unchanged. r)translate_cwd_for_wsl_backend)hermes_constantsrstr)cwdrs F/root/.hermes/venv/lib/python3.12/site-packages/acp_adapter/session.py_translate_acp_cwdrs? (S 22cJt|xsdj}|sd}tjj |}ddlm}||}||}n2tjd|rd|djd|dd}tjj|S) N.r)windows_path_to_wslz^/mnt/[A-Za-z]/z/mnt//) rstripospath expanduserrrrematchlowernormpath)rrawexpandedr translateds r_normalize_cwd_for_comparer'+s cjS/   !C ww!!#&H5$X.J $h /8A;,,./q!"? 77  H %%rct|xsdj}|r|St|xsdj}|r|Stjj t|xsdj d}|xsdS)Nz/\z New thread)rrrrbasenamerstrip)titlepreviewrexplicit preview_textleafs r_build_session_titler1>ss5;B%%'Hw}"%++-L 77  C rN11%8 9D  <rc|yt|tr|jr|S tjt |t jjS#t$rYywxYw)N)tz) isinstancerrr fromtimestampfloatrutc isoformat Exception)values r_format_updated_atr;Is\ }%%++- %%eElx||DNNPP s;A## A/.A/c| tdSt|ttfr t|St|j }|s tdS t j |jddjS#t$r+ t|cYS#t$rtdcYcYSwxYwwxYw)Nz-infZz+00:00) r6r4intrrr fromisoformatreplace timestampr9)r:r$s r_updated_at_sort_keyrBTs }V}%#u&U| e*   C V}!%%ckk#x&@AKKMM ! !:  !=  !!s02B B= B B= B94B=8B99B=cpt|}|jdtjt |i|y)zBest-effort human-readable output sink for ACP stdio sessions. ACP reserves stdout for JSON-RPC frames, so any incidental CLI/status output from AIAgent must be redirected away from stdout. Route it to stderr instead. fileN)dict setdefaultsysstderrprint)argskwargss r_acp_stderr_printrLes. &\F fcjj) 46rc|sy ddlm}||dt|iy#t$rtj ddYywxYw)aKBind a task/session id to the editor's working directory for tools. Zed can launch Hermes from a Windows workspace while the ACP process runs inside WSL. In that case ACP sends cwd as e.g. ``E:\Projects\POTI``; local tools need the WSL mount equivalent or subprocess creation fails before the command can run. Nr)register_task_env_overridesrz(Failed to register ACP task cwd overrideTexc_info)tools.terminal_toolrNrr9loggerdebug)task_idrrNs r_register_task_cwdrUpsJ PC#Ge5G5L-MN P ?$ OPs AAcg}t|xsdgD]}|s||vs |j|t|xsgD] }d|}|s ||vs|j|"|S)zGReturn ACP toolsets plus explicit MCP server toolsets for this session. hermes-acpzmcp-)listappend)toolsetsmcp_server_namesr%name server_name toolset_names r_expand_acp_enabled_toolsetsr_s HX/,0" D( OOD !",23* k]+  %s)rrrrrrrrgrdrcopydeepcopyrjrr}r|rUrrRr)rrerroriginalnew_idrfrs r fork_sessionzSessionManager.fork_sessions %##J/  TZZ\"  ..(D!  %(..9KX^^MM("2"23"*  ZZ +%*DNN6 " +63' e 1:vF  + +s DD%c|r t|nd}|j}i}|4 |jddD]}t||t |d< |j5t|jj}g}|jjD] }t|j} | d kr|rt|j|k7r:|j!|j"i} t%d |jD| j!d xsd } |j'|j"|j|j(| t+| j!d | |jt-| j!dxs'| j!dxst/j.d# ddd|j1D]\} }| vr t3|j!dxsd } | d kr/d}|j!d}|r& t5j6|j!dd}|rt||k7r{j'| ||j!dxsd | t+|j!d |j!d |t-|j!dxs|j!ddj=dd|S#t $rt jddYwxYw#1swYSxYw#t4j8t:f$rYwxYw)zCReturn lightweight info dicts for all sessions (memory + database).Nacpisourcelimitidz#Failed to load ACP sessions from DBTrOrc3K|]p}|jddk(rZt|jdxsdjr.t|jdxsdjryw)roleusercontentr)N)rrr).0msgs r z/SessionManager.list_sessions..)sc776?f4SWWY=O=USU9V9\9\9^CGGI.4"5;;=sA6A8r-r)r, last_active started_at)rerrg history_lenr, updated_at message_countr model_configrrgc6t|jdS)Nr)rBr)items rz.SessionManager.list_sessions..Ws&:488L;Q&Rr)keyreverse)r'_get_dblist_sessions_richrErr9rRrSr}setr|keysvalueslenrjrrrenextrYrgr1r;timeitemsr>jsonloadsJSONDecodeError TypeErrorsort)rrnormalized_cwdrpersisted_rowsrowseen_idsresultssr persistedr-sidr session_cwdmcs r list_sessionszSessionManager.list_sessionss S00T0J?C59#YN3s4y>2? ZZ 4>>..01HG^^**, !!))n !#!&@&G>&Y*..q||R@ #$99 MM),2 &'ll uu!"'2!5immG6LgWXW\W\!]&8%MM-8fIMM,M!K(B"&**R."4"4UC"@K"<["I^"[ NN!")/R,-cggg.> @RT_`01G1`377S_K`a  ! 2  R\` a} S BT R S  X,,i8s02K3$E$L&%L'3 LLL$'MMct|}|j|}|y||_t|||j ||S)zBUpdate the working directory for a session and its tool overrides.N)rrrrUr)rrerrs r update_cwdzSessionManager.update_cwdZsF %  , = :s+ e rc|j5t|jj}|jj dddD]}t ||j | |j}|= |jdd}|D]#}|d}t ||j|%yy#1swY|xYw#t$rtjddYywxYw) zPRemove all sessions (memory and database) and clear task-specific cwd overrides.Nri'rrz&Failed to cleanup ACP sessions from DBTrO) r}rXr|rclearrbrrsearch_sessionsdelete_sessionr9rRrS)r session_idsrerrowsrrs rcleanupzSessionManager.cleanupes ZZ #t~~2245K NN " #& /J J '  " ": . /\\^ > V))e)D+Cd)C#C(%%c*+  # # V EPT U Vs>C ;CC C54C5c|j5|jj|}ddd|j|yy#1swYxYw)zPersist the current state of a session to the database. Called by the server after prompt completion, slash commands that mutate history, and model switches. N)r}r|rrrs r save_sessionzSessionManager.save_sessionysK ZZ 3NN&&z2E 3   MM%   3 3rc|j |jS ddlm}t}||dz |_|jS#t$rt j ddYywxYw) aLazily initialise and return the SessionDB instance. Returns ``None`` if the DB is unavailable (e.g. import error in a minimal test environment). Note: we resolve ``HERMES_HOME`` dynamically rather than relying on the module-level ``DEFAULT_DB_PATH`` constant, because that constant is evaluated at import time and won't reflect env-var changes made later (e.g. by the test fixture ``_isolate_hermes_home``). Nr) SessionDBzstate.db)db_pathz)SessionDB unavailable for ACP persistenceTrO)r hermes_staterrr9rRrS)rr hermes_homes rrzSessionManager._get_dbsm    ($$ $  .)+K )+ 2J KD $$ $  LLDtL T s,A A*)A*c|j}|y|jrt|jnd}d|ji}t |j dd}t |j dd}t |j dd}t |tr#|jr|j|d<t |tr#|jr|j|d<t |tr#|jr|j|d<tj|} |j|j} | ,|j|jd|d|jin |j|j|||j } t | d d} | duxr| |uxrt#t | d d } | sE |j%|j} |j'|j|j(| yy#t$rtj!dd YwxYw#t$rd } Y]wxYw#t$r%tj+d|jd YywxYw)zWrite session state to the database. Creates the session record if it doesn't exist, then replaces all stored messages with the current in-memory history. Nrproviderbase_urlapi_moder)rerrgrz%Failed to update ACP session metadataTrO _session_db_session_db_createdF) active_onlyz Failed to persist ACP session %s)rrgrrrrfr4rrdumpsrrerupdate_session_metar9rRrSrlhas_archived_messagesreplace_messagesrjwarning)rrr model_str session_metarrrcwd_jsonexistingrfagent_dbagent_owns_persistence has_archiveds rrzSessionManager._persistsV \\^ : ). C $ uyy) 5;; D95;; D95;; D9 h $)9'/~~'7L $ h $)9'/~~'7L $ h $)9'/~~'7L $::l+; `~~e&6&67H!!$// #"'!3 "Y**5+;+;XyQ(KKEumT:H$GNG(=uEF # *)#%#;#;EH(1I7!GGG;:G;>#H%$H%(!I  I Ic|j}|y |j|S#t$rtj d|dYywxYw)z?Delete a session from the database. Returns True if it existed.Fz'Failed to delete ACP session %s from DBTrO)rrr9rRrS)rrers rrz SessionManager._delete_persistedAsQ \\^ : $$Z0 0  LLBJY]L ^ s&!A  A )rgrrrc H|j|jSddlm}ddlm}ddlm} |} | jd} d} d} t| tr0t| jdxs| } | jd} n0t| tr | jr| j} | jd xsijDcgc]+\}}t|tr|jd d d ur|-}}}d tdg|d ||j|xs| d} | |xs| }|j|jd|xs|jd|xs|jd|jd|jdt!|jdxsgdt)|||di|}||_t,|_|Scc}}w#t"$rt$j'dd YPwxYw)Nr)AIAgent) load_config)resolve_runtime_providerrgr)defaultr mcp_serversenabledTFrrW)r[)platformenabled_toolsets quiet_modere session_dbrg) requestedrrapi_keycommandrJ)rrrr r rJz7ACP session falling back to default provider resolutionrOrx)r~ run_agentrhermes_cli.configrhermes_cli.runtime_providerrrr4rErrrr_rupdaterXr9rRrSrUrrL _print_fn)rrerrgrrrrrrconfig model_cfg default_modelconfig_providerr\cfgconfigured_mcp_serversrKruntimerfs rrzSessionManager._make_agentNs    *&&( (%1HJJw'   i & i 8 IMJM'mmJ7O  3 'IOO,=%OO-M%jj7=2DDF" cc4(CGGIt,DE,Q " "  <!7!$,,.+m   c.9K9^_G MM ' J 7 ( CGKK ,C ( CGKK ,C&{{95&{{95 V!4!:;   :s+!&! , U" < c LLR]aL b cs!0G8> H! H!NN)r)rrreturnrd)rerrOptional[SessionState])rerrrl)rerrrrr)N)r str | NonerrirNone)rerrr)rrdrr) rerrrrgrrrrrrr)rsrtrurvrrrrrrrrrrrrrrrxrrrzrzs  * )%r&s#,  '(,,   8 $ 3&& !"P$"&)-&$M  & &  & ddr