ri^7dZddlZddlZddlZddlZddlmZddlmZ ddlm Z m Z m Z m Z mZmZmZmZmZddlmZdZGddZd Zdd efd Zd ZddefdZdefdZddZddededefdZdefdZ defdZ!de"fdZ#dS) z= Hermes Web UI -- Session model and in-memory session store. N)Path) SESSION_DIRSESSION_INDEX_FILESESSIONS SESSIONS_MAXLOCKDEFAULT_WORKSPACE DEFAULT_MODEL PROJECTS_FILEHOME)get_last_workspacecg}tjdD]v}|jdr t|jr'|g#t$rYswxYwt5tj D]Dtfd|Ds'|E dddn #1swxYwY|ddtjt#j|dd d dS) z5Rebuild the session index file for O(1) future reads.*.json_c3:K|]}|djkVdS) session_idNr).0ess '/home/ubuntu/hermes-webui/api/models.py z'_write_session_index..s.HH1q!,6HHHHHHNc|dS)N updated_atrs rz&_write_session_index.. s qrTkeyreverseF ensure_asciiindentutf-8encoding)rglobname startswithSessionloadstemappendcompact Exceptionrrvaluesanysortr write_textjsondumps)entriesprs @r_write_session_indexr:sG  h ' ' 6  S ! !+8  QV$$A -'..---    D  ,,"" , ,AHHHHHHHHH ,qyy{{+++ ,,,,,,,,,,,,,,,, LL..L===!$*W5QR"S"S"S^effffffs%AA?? B  B AC<<DDceZdZddeeedddddddddddddfdededed ed ed ed ed efdZe dZ ddZ e dZ defdZdS)r,NUntitledFrrtitlepinnedarchived project_id input_tokens output_tokensreasoning_effortc |ptjjdd|_||_t t ||_ ||_ |pg|_ |pg|_ |ptj |_|ptj |_t!| |_t!| |_| pd|_| |_| pd|_|pd|_||_||_t |pd|_dS)N rr=)uuiduuid4hexrr>strr expanduserresolve workspacemodelmessages tool_callstime created_atrboolr?r@rAprofilerBrCestimated_cost personalitystriplowerrD)selfrr>rMrNrOrRrrPr?r@rArTrBrCrUrVrDkwargss r__init__zSession.__init__%s#%= (8"(= T)__7799AACCDD  B $*$3 $3 6ll X $, (-A*/a,& #$4$: ; ; A A C C I I K Krc&t|jdz S)N.json)rrrYs rpathz Session.path?s66666rreturnctj|_|jt j|jdddtdS)NFr"r#r&r')rQrr_r5r6r7__dict__r:r^s rsavez Session.saveCs[)++  Jt}5 C C C     rc t|dz }|sdS|ditj|dS)Nr]r&r'r)rexistsr6loads read_text)clssidr9s rr-z Session.loadKsU S--- 'xxzz 4s??TZ W = =>>???rc id|jd|jd|jd|jdt |jd|jd|jd|jd |j d |j d |j d |j d |j d|jd|jd|jS)Nrr>rMrN message_countrRrr?r@rArTrBrCrUrVrD)rr>rMrNlenrOrRrr?r@rArTrBrCrUrVrDr^s rr0zSession.compactRs $/ TZ    TZ  S//  $/   $/  dk     $/  t|  D-  T/  d1  4+  5!  rr`N)__name__ __module__ __qualname__rJr r rSintr[propertyr_rc classmethodr-dictr0rrrr,r,$s"'+ 30114D uU!%t#$4! LL3LCL+/LAEL! L # L 69 L 69 LLLL477X7@@[@        rr,c t5|tvr-tj|t|cdddS dddn #1swxYwYt|}|rt5|t|<tj|t tt kr2tjdt tt k2dddn #1swxYwY|St|)NFlast) rr move_to_endr,r-rlrpopitemKeyError)rirs r get_sessionr{fs !! (??   % % %C=!!!!!!!! !!!!!!!!!!!!!!!  SA  - -HSM   % % %h--,.. e,,,,h--,.. - - - - - - - - - - - - - - -  3--s$*A  AA:A.C44C8;C8r=rDc : ddlm}|}n#t$rd}YnwxYwt|p t |p t j|t j|p t j|}t5|t|j <tj |j tttkr2tjdtttk2dddn #1swxYwY||S)Nrget_active_profile_name)rMrNrTrDFrv) api.profilesr~ ImportErrorr,r _cfgr clamp_reasoning_effortrrrrxrlrryrc)rMrNrDr~_profilers r new_sessionrusd888888**,, 3133)t)4  'T'      A ))!"Q\***(mml**  % ( ( ( ((mml**))))))))))))))) FFHHH Hs ""8A8C<<DDctjr tjtjd}d|D}t 5t jD]|j < dddn #1swxYwYt|dd}d|D}|D] dsd d<|S#t$rYnwxYwg}tjd D]d}|jd r t"|jr|U#t$rYawxYwt jD]2t+fd |Dr|3|d dd|D}|D] dsd d<|S)Nr&r'c i|] }|d| Srrrrs r z all_sessions..s;;;<!;;;rc>|dd|dfS)Nr?Frgetrs rrzall_sessions..s quuXu?U?UWXYeWf>grTrczg|]8}|dddkr|dddk6|9S)r>r<rkrrrs r z all_sessions..sNwwwAgj0I0I:0U0UZ[Z_Z_`opqZrZrtuZuZuaZuZuZurrTdefaultrrc38K|]}j|jkVdS)Nr)rxrs rrzall_sessions..s,99q|q|+999999rc2t|dd|jfS)Nr?F)getattrrrs rrzall_sessions..sGAx77Frc|g|]9}|jdkrt|jdk%|:S)r<r)r>rlrOr0rs rrz all_sessions..s? [ [ [aqw /B/Bs1:XYGYGYaiikkGYGYGYr)rrer6rfrgrrr2r0rsortedrr1rr)r*r+r,r-r.r/allr4)index index_mapresultoutr9rs @r all_sessionsrs "" J1;WMMMNNE;;U;;;I : :!**::A./iikkIal++: : : : : : : : : : : : : : : :I,,..4g4gquvvvFwwwwwF - -uuY''-#,AiLM    D  C  h ' ' 6  S ! !+8  QV$$A #**Q---    D  _  II 9999S999 9 9H3::a===HHFFPTHUUU [ [3 [ [ [F %%uuY %$AiL MsH:C-3B C-BC-BAC-- C:9C:16E(( E54E5r<fallbackc4|D]}|ddkry|dd}t|trdd|D}t |}|r |ddcS|S) z3Derive a session title from the first user message.roleusercontentr= c3K|]H}t|t|ddk0|ddVIdS)typetextr=N) isinstancertr)rr9s rrztitle_from..sZmm1Z4=P=PmUVUZUZ[aUbUbflUlUlQUU62..UlUlUlUlmmrN@)rrlistjoinrJrW)rOrmcrs r title_fromrs !! 55==F " "i$$A!T"" nHHmmmmmmmq66<<>>D !CRCy   Orr`ctjsgS tjtjdS#t $rgcYSwxYw)z;Load project list from disk. Returns list of project dicts.r&r')r rer6rfrgr1rrr load_projectsrs_   ! ! z-17CCCDDD  s&> A  A c\tjtj|ddddS)zWrite project list to disk.Fr"r#r&r'N)r r5r6r7)projectss r save_projectsrs0TZuQOOOZabbbbbbrunknownrr>rNcpt||t|||}||S)z\Create a new WebUI session populated with CLI messages. Returns the Session object. )rr>rMrNrOrT)r,r rc)rr>rOrNrTrs rimport_cli_sessionrsD $&&    AFFHHH Hrc<ddl}g} ddl}n#t$r|cYSwxYw ddlm}t |}nl#t$r_t | dttdz }YnwxYw|dz }| s|S ddlm }|}n#t$rd}YnwxYw |t|5}|j|_|} | d| D]v} | d } | d p| d } |} || | d pd tt+| dpd| dpd| d | ddd| ddd w dddn #1swxYwYn#t$rgcYSwxYw|S)aFRead CLI sessions from the agent's SQLite store and return them as dicts in a format the WebUI sidebar can render alongside local sessions. Returns empty list if the SQLite DB is missing, the sqlite3 module is unavailable, or any error occurs -- the bridge is purely additive and never crashes the WebUI. rNget_active_hermes_home HERMES_HOME.hermesstate.dbr}a SELECT s.id, s.title, s.model, s.message_count, s.started_at, s.source, MAX(m.timestamp) AS last_activity FROM sessions s LEFT JOIN messages m ON m.session_id = s.id GROUP BY s.id ORDER BY COALESCE(MAX(m.timestamp), s.started_at) DESC LIMIT 200 id last_activity started_atr>z CLI SessionrNrrkFcliT) rr>rMrNrkrRrr?r@rArT source_tagis_cli_session)ossqlite3rrrrrKrLr1getenvrJr rer~connectRow row_factorycursorexecutefetchallr/r )r cli_sessionsrr hermes_homedb_pathr~ _cli_profileconncurrowriraw_tsrTs rget_cli_sessionsrsIIIL c777777113344??AAIIKK ccc299]Cy8H4I4IJJKKVVXX``bb cJ&G >>  888888..00  & __S\\ * *" d&{D ++--C KK    ||~~  $i_-B\1B'##"% \:]!$%7%9%9!:!: \6Y%(%9%>Q"%l"3"(# %"&&"'&*%% " " " " " " " " " " " " " " " F   sh AA""A&C  C *C;; D  D "H 0CG>2 H >HH HH HHc fddl} ddl}n#t$rgcYSwxYw ddlm}t |}nl#t$r_t | dttdz }YnwxYw|dz }| sgS | t|5}|j|_|}|d|fg}|D]-} || d| d | d d . dddn #1swxYwYn#t$rgcYSwxYw|S) zRead messages for a single CLI session from the SQLite store. Returns a list of {role, content, timestamp} dicts. Returns empty list on any error. rNrrrrz SELECT role, content, timestamp FROM messages WHERE session_id = ? ORDER BY timestamp ASC rr timestamp)rrr)rrrrrrrKrLr1rrJr rerrrrrrr/) rirrrrrrrmsgsrs rget_cli_session_messagesr8s@ III  c777777113344??AAIIKK ccc299]Cy8H4I4IJJKKVVXX``bb cJ&G >>    __S\\ * * d&{D ++--C KK      D||~~   K"9~!$[!1                   KsV AA A&C C ("F A>  u __S\\ * * $d++--C KKCcV L L L KK;cV D D D KKMMME;?F FF)NNr=)r<rm)rN)$__doc__ collectionsr6rQrGpathlibr api.configconfigrrrrrrr r r r api.workspacer r:r,r{rJrrrrrrrrrrSrrrrrsC -,,,,,ggg$@ @ @ @ @ @ @ @ D     c    0%%%P  3    tcccc   3  s  S     $Q$QQQQh'T''''Ttr