6jRUdZddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z ej eZ ddlmZmZmZmZmZn#e$r dZdZdZdZdZYnwxYweZiZd ed <dAd ZdBdZdCdZGddZdddDdZdEdZdddddddddFd,Z dd-dGd/Z!dHd1Z"dId2Z#dddJd3Z$dddKd6Z% dLd7dd8dMd;Z&dddNd<Z'ddd=dOd@Z(dS)Pz1WebUI bridge for Hermes persistent session goals.) annotationsN)Path)AnyDictOptional)CONTINUATION_PROMPT_TEMPLATEDEFAULT_MAX_TURNS GoalManager GoalState judge_goalzdict[str, Any] _DB_CACHEreturnintc  ddlm}t|dipi}t|tr|dini}t|tst tpdStdt |dtpdpdS#t$rt tpdcYSwxYw)zHReturn the configured /goal turn budget, defaulting to Hermes' 20 turns.r)configcfggoalsr max_turns) apirgetattr isinstancedictgetrr max Exception)_configr goals_cfgs /root/hermes-webui/api/goals.py_default_max_turnsr""s ,))))))gub))/R,6sD,A,AICGGGR(((r )T** 0(.B// /1c)-- 5F5L"MMSQSTTUUU ,,,$*+++++,sA0B-39B-- CC session_idstrc d|S)Nzgoal:)r#s r! _meta_keyr'0s :   profile_home str | Pathcpt|}t|}t|}||S ddlm}||dz }n4#t$r'}t d||Yd}~dSd}~wwxYw|t|<|S)aReturn a SessionDB pinned to *profile_home*, without reading HERMES_HOME. The upstream Hermes GoalManager persists through hermes_cli.goals.load_goal(), which resolves SessionDB from process-global HERMES_HOME. WebUI sessions are profile-scoped and can run concurrently, so the WebUI bridge uses an explicit state.db path whenever the caller provides the session's profile home. Nr) SessionDBzstate.db)db_pathz-GoalManager profile DB unavailable for %s: %s) r expanduserresolver$rr hermes_stater,rloggerdebug)r)homekeycachedr,dbexcs r! _profile_dbr84s    ( ( * * 2 2 4 4D d))C ]]3  F  ****** Ytj0 1 1 1  DdCPPPtttttIcN Is"A88 B)B$$B)ceZdZdZddd)d Zed Zd Zd*dZd+dZ d+dZ d,dZ ddd-dZ d.d/dZ ddd0d Zd*d!Zdd"d1d&Zd2d(ZdS)3_ProfileGoalManagerzHSmall WebUI-local GoalManager adapter with explicit profile persistence.r)default_max_turnsr#r$r)r*r;rcttd||_t||_t|ptpd|_ | |_ dS)NzHermes goal state unavailabler) r RuntimeErrorr#rr.r/r)rr r;_load_state)selfr#r)r;s r!__init__z_ProfileGoalManager.__init__Osu  >?? ?$ ..99;;CCEE!$%6%Q:K%Qr!R!Rjjll r(c|jSN)r?r@s r!statez_ProfileGoalManager.stateWs {r(ct|j}||jsdS |t |j}n3#t $r&}t d|Yd}~dSd}~wwxYw|sdS tj |S#t $r,}t d|j|Yd}~dSd}~wwxYw)Nz'GoalManager profile get_meta failed: %sz1GoalManager profile state parse failed for %s: %s) r8r)r#get_metar'rr1r2r from_jsonwarning)r@r6rawr7s r!r>z_ProfileGoalManager._load[s * + + :T_:4 ++i8899CC    LLBC H H H44444  4 &s++ +    NNNPTP_ad e e e44444 s/'A A9A44A9B C !CC rNonec$t|j}| |jr|dS |t |j|dS#t $r&}td|Yd}~dSd}~wwxYw)Nz'GoalManager profile set_meta failed: %s) r8r)r#set_metar'to_jsonrr1r2)r@rEr6r7s r!_savez_ProfileGoalManager._savels * + + :T_: F I KK $/22EMMOO D D D D D I I I LLBC H H H H H H H H H Is:A B)B  Bboolc4|jduo|jjdkS)Nactiver?statusrDs r! is_activez_ProfileGoalManager.is_activeus{$&I4;+=+IIr(c0|jduo |jjdvS)N)rRpausedrSrDs r!has_goalz_ProfileGoalManager.has_goalxs{$&U4;+=AU+UUr(c2|j}| |jdvrdS|jd|jd}|jdkr d|d|jS|jdkr"|jr d |jnd }d ||d|jS|jd kr d |d|jSd|jd|d|jS)Ncleared*No active goal. Set one with /goal ./z turnsrR⊙ Goal (active, ): rW — r ⏸ Goal (paused, done✓ Goal done (Goal (, )r?rT turns_usedrgoal paused_reason)r@sturnsextras r! status_linez_ProfileGoalManager.status_line{s K 9L00??<55!+555 8x  :::!&:: : 8x  12H-AO---bEAAuAAAA A 8v  7U77qv77 7666E66af666r(N)rrgr Optional[int]c|pd}|stdt|dd|rt|n|jt jd}||_|||S)Nr zgoal text is emptyrRrg)rgrTrfr created_at last_turn_at)strip ValueErrorr rr;timer?rO)r@rgrrEs r!setz_ProfileGoalManager.sets !!## 3122 2(1Mc)nnnt7My{{      5 r( user-pausedreasonc|jsdSd|j_||j_||j|jS)NrW)r?rTrhrO)r@rvs r!pausez_ProfileGoalManager.pausesA{ 4% $* ! 4;{r(T) reset_budgetryc|jsdSd|j_d|j_|r d|j_||j|jS)NrRr)r?rTrhrfrO)r@rys r!resumez_ProfileGoalManager.resumesS{ 4% $( !  '%&DK " 4;{r(cr|jdSd|j_||jd|_dS)Nr[)r?rTrOrDs r!clearz_ProfileGoalManager.clears8 ;  F&  4; r(user_initiated last_responserDict[str, Any]c |j}| |jdkr|r|jndddddddS|xjdz c_tj|_t d\}}n't |jt|pd\}}||_||_ |d kr(d |_| |d ddd |d |dS|j|j krNd |_d |jd |j d|_ | |d ddd|d|jd |j ddS| |dd| d|d|jd |j d|dS)NrRFinactiveno active goalr rTshould_continuecontinuation_promptverdictrvmessager)continuezgoal judge unavailablerbu✓ Goal achieved: rWzturn budget exhausted (r])ru⏸ Goal paused — zD turns used. Use /goal resume to keep going, or /goal clear to stop.Tu↻ Continuing toward goal (r_)r?rTrfrsrpr rgr$ last_verdict last_reasonrOrrhnext_continuation_prompt)r@rrrErrvs r!evaluate_after_turnz'_ProfileGoalManager.evaluate_after_turns  =ELH44*/9%,,T#('+%*   A!Y[[  BOGVV(S9L"5M5MNNOGV$" f  !EL JJu    #('+! 999    u . .#EL"aErOrUrXrlrtrxr{r}rrr&r(r!r:r:LsuRR^`######X"IIIIJJJJVVVV 7 7 7 7<@ .2QU: : : : : : xJJJJJJr(r:r)str | Path | Nonec tdS|rgtturYtR t||t S#t $r&}t d|Yd}~dSd}~wwxYwt|t S)N)r#r)r;z*Profile-scoped GoalManager unavailable: %s)r#r;)r _NativeGoalManagerr r:r"rr1r2)r#r)r7s r!_managerrst  '999i>S &%)"4"6"6      LLEs K K K44444  *@R@T@T U U UUsA A0 A++A0rErOptional[Dict[str, Any]]c 4|dSt|ddpdt|ddpdtt|ddpdtt|ddpdt|ddt|ddt|d dd S) Nrgr rTrfrrrrrh)rgrTrfrrrrh)rr)rEs r!_state_payloadrs }tvr**0b%2..4"'%q99>Q?? Q77<1==~t<<umT:: >>  r(T)okrEerrorkickoff_promptdecision message_key message_argsrrPactionrr str | NonerrDict[str, Any] | Nonerrlist[Any] | Nonerc t|||t|d} |r|| d<|r|| d<||| d<|r|| d<|d|D| d<| S)N)rrrrgrrrrcg|]}||SrCr&).0as r! z_payload..-sIIIa1====r(r)rPr) rrrrErrrrrbodys r!_payloadrs2hhu%% D  W 0!/ #Z*)]II<III^ Kr()default_messagerc\|d}||ddStt|ddpd}|dvr|ddStt|dd pd }tt|d d pd }tt|d dpd}|d krd |d|d|d|||gdS|dkrBuild localized-status style payload fields from a goal state.Nr\goal_status_none)rrrTr rZrfrrrgrRr^r]z turns): goal_status_active)rrrrWrhrar`r_goal_status_pausedrbrcgoal_status_donerdre)rr)r$rrqr)rErrTrfrrgrvs r!_goal_status_payloadr1sF }*;MNNN "--3 4 4 : : < B??qJqqqX^DfGfDTDTdfqqkoqq/'FDA   JJJiJJDJJ-'D9   HFGGjGG9GGGGY=  r(tuple[int, int]c|sdStjd|}|sdS t|dt|dfS#t$rYdSwxYw)z?Best-effort extraction for continuation messages like '(1/20)'.rrz\((\d+)\s*/\s*(\d+)\)r)researchrgroupr)rmatchs r! _extract_goal_turns_from_messagerVs t I. 8 8E t5;;q>>""C A$7$777 ttsAA## A10A1ct|ts|St|dpd}t|dpd}t t |ddpd}t t |ddpd}||fdkr4tt|dpd\}}|d kr i|d |gd S|d kr i|d ||gd S|dr i|d|||gd S|S)z sets a new active goal and returns kickoff_prompt so the caller can start the first normal user-role turn immediately. r Frmissing_sessionzsession_id required)rrrrrN unavailablez"Goals unavailable on this session.rTrE)rrErxru)rvno_goalz No goal set. goal_no_goal)rrrrru⏸ Goal paused: goal_paused)rrrrrEr{zNo goal to resume.u▶ Goal resumed: z6 Send a new message, or type continue, to kick it off. goal_resumed)r}stoprbr}z Goal cleared.zNo active goal. goal_cleared)rrrrErt agent_runningubAgent is running — use /goal status / pause / clear mid-run, or /stop before setting a new goal. invalid_goalzInvalid goal: u⊙ Goal set (z-turn budget): u I'll keep working until the goal is done, you pause/clear it, or the budget is exhausted. Controls: /goal status · /goal pause · /goal resume · /goal cleargoal_set)rrrrrErr&)r$rqrrlowerrrrxrgr{rPrXr}rtrrr) r#rrr)sidrtextrrEstatus_payloadhadr7s r!goal_command_payloadrsl" jB   % % ' 'C j58IShiiii 3\ 2 2 2C {5 Ostttt tzr?? " "D JJLLE H5H$$Wd++-e44GxuGGGGG   // =&*  4 44%ej//*        =,*  HUZHHH'ej//*     )))3<<>>"" '*AOO0A*-A>#w--       !6     f  fff5nNd_bNdNdeeeeeeeeef  SU_ S SUZ S S Souz2z    sH I'I<IIc&t|pd}|sdSt||}|dS t|S#t $r'}t d||Yd}~dSd}~wwxYw)zEReturn True when the session has an active standing goal to evaluate.r FrNz1goal active-state check failed for session=%s: %s)r$rqrrPrUrr1r2)r#r)rrr7s r!has_active_goalrs jB   % % ' 'C u 3\ 2 2 2C {uCMMOO$$$  H#sSSSuuuuus A B)B  B)rr)rrc t|pd}|s dddddddSt||}| dddddddS |s't t |d dd ddddd ddS|t|pd| }nQ#t $rD}td ||dddddt|j ddcYd}~Sd}~wwxYwt|tsi}| dd| dd| ddt|}t|t |d d}|S)zrs@777""""""  &&&&&&&&&&  8 $ $ #% IJJJ !   , , , ,    0eJeJeJeJeJeJeJeJPDH V V V V V V     !%&*"%)>GK""""""J        FOS666666]aNNNNNN2m!&* mmmmmmf'+. &* 88888888sA AA