[j ;"dZddlZddlZddlmZmZddlmZeje Z ee e gdfZ ee gdfZegefZdZdZde fdZdefd Z dd e d e d eed ee dedeedee fdZdZ dde d e d e d ee dedeedeeddfdZ dde d e d e d ee dedeedeeddfdZ dde d e d e ded ee dedeedeeddfdZdS)zAuto-generate short session titles from the first user/assistant exchange. Runs asynchronously after the first response is delivered so it never adds latency to the user-facing reply. N)CallableOptional)call_llma8Generate a short, descriptive title (3-7 words) for a conversation that starts with the following exchange. The title should capture the main topic or intent. Write the title in the same language the user is writing in. Return ONLY the title text, nothing else. No quotes, no punctuation at the end, no prefixes.aGenerate a short, descriptive title (3-7 words) for a conversation that starts with the following exchange. The title should capture the main topic or intent. Write the title in {language}. Return ONLY the title text, nothing else. No quotes, no punctuation at the end, no prefixes.returnc ddlm}t|pidpididdS#t $rYdSwxYw)zDReturn configured title language, or empty string to match the user.r load_config auxiliarytitle_generationlanguage)hermes_cli.configr strgetstrip Exceptionrs 7/home/ice/.hermes/hermes-agent/agent/title_generator.py_title_languager+s 111111kmm!r & &{ 3 3 9r S#R ( ( SR    %''   rrsA-A00 A>=A>c" ddlm}ddlm}|}|dpidpi}||ddS#t $r t d d YdSwxYw) z=Return whether automatic session title generation is enabled.r)load_config_readonly)is_truthy_valuer r enabledT)defaultz'Failed to read title_generation.enabledexc_info)rrutilsrrrloggerdebug)rrconfig title_configs r_auto_title_enabledr!9s  ;:::::))))))%%'' ;//52::;MNNTRT |// ::DIIII  > NNNttsA!A$$&B B user_messageassistant_responsetimeoutfailure_callback main_runtimeruntime_validatorcvtstddS|T |stddSn,#t$rtddYnwxYw|r |ddnd}|r |ddnd}t }|rt | nt} d | d d d |d|d g} td| dd||} | j dj j pd} ddl m } | d| }|d}|dr|dd}t#|dkr |dddz}|r|ndS#t$r~}td|tdd|: |d|n,#t$rtddYnwxYwYd}~dSd}~wwxYw)uAGenerate a session title from the first exchange. Uses the main runtime's model when available, falling back to the auxiliary LLM client (cheapest/fastest available model). Returns the title string or None on failure. ``failure_callback`` is invoked with ``(task, exception)`` when the auxiliary call raises — the caller typically wires this to ``AIAgent._emit_auxiliary_failure`` so the user sees a warning instead of silently accumulating untitled sessions. ``runtime_validator`` is called right before the LLM request. If it returns False (e.g. the user's model was switched since the background thread captured its runtime snapshot), the call is skipped silently — no request is sent, so a stale title request can't reload a model the runtime already unloaded (#19027). =A> CF00 H8:9H34 HH3&H*'H3)H**H33H8ctddfd} ||S#t$r6tdd}|||}|r||kr||cYSwxYw)uPersist a generated title, recovering from duplicate-title collisions. The write goes through ``set_auto_title_if_empty`` (predicate + write in one transaction) so a manual ``/title`` set while LLM generation was in flight is never overwritten — a plain ``set_session_title`` fallback keeps older stores working. ``set_session_title`` raises ValueError when the title would collide with another session (the unique-title index). Rather than swallow it and leave the session untitled (#50537), append a #N suffix via get_next_title_in_lineage() when the store supports lineage dedup; otherwise re-raise so the caller can decide. Returns the title actually persisted, or None when a concurrent manual title won the race (nothing was written). set_auto_title_if_emptyNc*|stddS|S|}|durtdd|S)Nz\Skipping auto-generated session title because a title was set while generation was in flightFzsession z not found when storing title)rrset_session_title RuntimeError)tok atomic_fn session_db session_ids r_setz$_persist_session_title.._sets  9Z++  =tH  ) )*a 8 8 ;;D:DDD get_next_title_in_lineage)getattr ValueError)rPrQrErR next_title_fndedupedrOs`` @r_persist_session_titlerYs $=tDDI$ tE{{  ,GNN  -&& 'U** tG}}s (=A('A(rQtitle_callbackc d t||||||||dS#t$r}td|tdd|: |d|n2#t$rtddYnwxYwYd}~dSYd}~dSYd}~dSd}~wwxYw) uwGenerate and set a session title if one doesn't already exist. Called in a background thread after the first exchange completes. Silently skips if: - session_db is None - session already has a title (user-set or previously auto-generated) - title generation fails - runtime_validator returns False (model was switched) Never lets an exception escape: this is a daemon-thread target, and an escaping exception would spray a raw traceback into the user's terminal via the default threading excepthook. The canonical trigger is the post-``hermes update`` stale-module window, where this function's lazy imports read NEW source from disk while already-cached modules (``agent.portal_tags`` etc.) are still the OLD version — the resulting ImportError repeats on every auto-title attempt until the long-running process restarts. r%r&rZr'zeAuto-title failed (harmless; if this started after an update, restart the running Hermes process): %szAuto-title tracebackTrNr6z"Auto-title failure_callback raised)_auto_title_sessionrrr@r) rPrQr"r#r%r&rZr'rFs rauto_title_sessionr^s68R    -%)/   R R R  6     +d ;;;  ' R  !3Q7777 R R R AD QQQQQ R ( ' ' ' ' '777777QQQQQQ Rs8 B/9B* A,+B*,&BB*BB**B/cx|r|sdS ||}|rdSn#t$rYdSwxYwddlm} ddlm} |} ||p|} n#t$rYnwxYw| | | ||t|||||} | sdS t||| } | dSt d| |; || dS#t$r t dd YdSwxYwdS#t$r&}t d |Yd}~dSd}~wwxYw) u9Body of :func:`auto_title_session` — see its docstring.Nr)set_accounting_context)set_conversation_context)r%r&r'z Auto-generated session title: %szAuto-title callback failedTrz&Failed to set auto-generated title: %s) get_session_titleragent.aux_accountingr`agent.portal_tagsraget_conversation_rootrGrYrr)rPrQr"r#r%r&rZr'existingr`raconversation_idrE persistedrFs rr]r] s Z// ;;   F  <;;;;;:::::: O $:::FFT*      _---:z222 )!+    E  B*:z5II   F 7CCC  % Jy))))) J J J 9D IIIIII J & % BBB =qAAAAAAAAABsZ" 00A A'&A'D /D C&DD DD D9D44D9conversation_historyc *|r|r|r|sdStd|pgD} | dkrdStstddSt jt ||||f||||ddd} | dS) uFire-and-forget title generation after the first exchange. Only generates a title when: - This appears to be the first user→assistant exchange - No title is already set Nc3LK|]}|ddkdV dS)r+r-N)r).0ms r z#maybe_auto_title..bs7\\qAEE&MMU[D[D[D[D[D[D[\\rSr)r\Tz auto-title)targetargskwargsdaemonname)sumr!rr threadingThreadr^start) rPrQr"r#rir%r&rZr'user_msg_countthreads rmaybe_auto_titler|Js" Z|CU \\%9%?R\\\\\N   STTT  !*l4F G 0(,!2       F LLNNNNNrS)NNNN)__doc__loggingrwtypingrragent.auxiliary_clientr getLogger__name__rr BaseExceptionFailureCallback TitleCallbackboolRuntimeValidatorr9r7rr!floatdictrGrYr^r]listr|rSrrs= %%%%%%%%++++++  8 $ $ C/56#% BH%cc     T( $2648 VVVVe_V/ V  V  01 Vc]VVVVr,,,h37.2484R4R4R4R 4R / 4R  4R]+4R 014R 4R4R4R4Rx37.248>B>B>B>B >B / >B  >B]+>B 01>B >B>B>B>BN37.248.... .  . / ..]+. 01. ......rS