iKdZddlZddlZddlmZmZddlmZeje Z ee e gdfZ dZ dde de d ed ee d ed ee f d Z dde de de d ee d ed df dZ dde de de ded ee d ed dfdZdS)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_llmzGenerate 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. Return ONLY the title text, nothing else. No quotes, no punctuation at the end, no prefixes.>@ user_messageassistant_responsetimeoutfailure_callback main_runtimereturnc|r |ddnd}|r |ddnd}dtddd|d|dg} td |dd || }|jd jjpd} | d } | dr| dd} t| dkr | dddz} | r| ndS#t$r~} t d| t dd|: |d| n,#t$rt ddYnwxYwYd} ~ dSd} ~ wwxYw)uGenerate 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. Nisystem)rolecontentuserzUser: z Assistant: title_generationg333333?)taskmessages max_tokens temperaturer r rz"'ztitle:PMz...zTitle generation failed: %szTitle generation tracebackT)exc_infoztitle generationz(Title generation failure_callback raised) _TITLE_PROMPTrchoicesmessagerstriplower startswithlen Exceptionloggerwarningdebug) rrr r r user_snippetassistant_snippetrresponsetitlees :/home/ubuntu/.hermes/hermes-agent/agent/title_generator.pygenerate_titler-s&*6=<%%2L4FN*4C400Bm44$]\$]$]J[$]$]^^H #%    !!$,4:AACC E"" ;;== # #H - - &!""IOO%%E u::??#2#J&E'uu4'     4a888 1D AAA  ' X  !3Q7777 X X X GRV WWWWW Xttttt s<B=C11 E9;9E45 EE4&E+(E4*E++E44E9 session_idcb|r|sdS ||}|rdSn#t$rYdSwxYwt||||}|sdS |||td|dS#t$r&}td|Yd}~dSd}~wwxYw)a!Generate 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 Nr r z Auto-generated session title: %sz&Failed to set auto-generated title: %s)get_session_titler#r-set_session_titler$r&) session_dbr.rrr r existingr*r+s r,auto_title_sessionr5Vs Z// ;;   F   (;KZf   E B$$Z777 7????? BBB =qAAAAAAAAABs$" 00 1A>> B.B))B.conversation_historyc|r|r|r|sdStd|pgD}|dkrdStjt||||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)rrN)get).0ms r, z#maybe_auto_title..s7\\qAEE&MMU[D[D[D[D[D[D[\\r0Tz auto-title)targetargskwargsdaemonname)sum threadingThreadr5start) r3r.rrr6r r user_msg_countthreads r,maybe_auto_titlerK~s Z|CU \\%9%?R\\\\\N  !*l4F G$4lSS  F LLNNNNNr>)rNN)NN)__doc__loggingrFtypingrragent.auxiliary_clientr getLogger__name__r$str BaseExceptionFailureCallbackrfloatdictr-r5listrKr>r,rYs %%%%%%%%++++++  8 $ $ C/56c26 77777/ 7  7 c] 7777~37 %B%B%B%B %B / %B  %B %B%B%B%B\37!!!! !  ! / !! !!!!!!r>