`gj^ d Z ddlmZ ddlZddlZddlmZmZmZm Z m
Z
m Z ej dej Z edh Zddd d
d d d
ddZddZddZddZddZddZd dZ d d dZy) uV Live session context-window breakdown for UI surfaces.
Estimates how the next provider request is composed: system prompt tiers,
tool schemas, and conversation history. Uses the same rough char/4 heuristic
as ``agent.model_metadata.estimate_request_tokens_rough`` so numbers align
with compression thresholds — not exact tokenizer counts.
) annotationsN)AnyDictListOptionalSequenceTuplez(.*?
delegate_taskzvar(--context-usage-system)zvar(--context-usage-tools)zvar(--context-usage-rules)zvar(--context-usage-skills)zvar(--context-usage-mcp)z var(--context-usage-subagents)zvar(--context-usage-memory)z!var(--context-usage-conversation))
system_prompttool_definitionsrulesskillsmcpsubagent_definitionsmemory conversationc * | syt | dz dz S )Nr )len)texts J/root/.hermes/venv/lib/python3.12/site-packages/agent/context_breakdown.py_chars_to_tokensr s IMa c H | syt t j | d S )Nr F) ensure_ascii)r jsondumps)values r _json_tokensr % s DJJu5 A BBr c t | t r| j d nd }t |t r t |j d xs d S t | j d xs d S )Nfunctionname )
isinstancedictgetstr)toolfns r
_tool_namer+ + sV !+D$!7* TB"d266&>'R(( txx %2 &&r c g }g }g }| D ][ }t | }|j d r|j | 1|t v r|j | K|j | ] |||fS )Nmcp_)r+
startswithappend_SUBAGENT_TOOL_NAMES)toolsbuiltinr subagentr) r# s r _split_toolsr4 2 sr GC H !$ ??6 " JJt
)
) OOD ! NN4 ! C !!r c d}d}t | dd }|||fS t | dd r|j d xs d}t | dd r|j d xs d}||fS # t $ r Y ||fS w xY w)Nr$
_memory_store_memory_enabledTr _user_profile_enableduser)getattrformat_for_system_prompt Exception)agent memory_block
user_blockstores r _memory_blocksrA A s LJ E?D 1E }Z''
5+T 2 99(CIrL 514 8 77?E2J ##
##
s AA! ! A0/A0c ^ | }|D ] }|s|j |d } |j S )Nr$ )replacestrip)r blocksoutblocks r
_strip_blocksrH Q s7
C ) ++eR(C) 99;r c ddl m} ddlm} || }|j dd xs d}|j dd xs d}|j dd xs d}t
j
| }|r|j d nd} t | \ }
} dj d |
| fD j } t || }
t ||
| }dj d
|
|fD j }t t | d d xs g }t | \ }}} ||xs g }d
dt | fddt! | fddt | fddt | fddt! | fddt! | fddt | fdd |fg}t# d |D }t | d d }|rt% t |d d xs d nd}|rt% t |d d xs d nd}|dkD r|n|}|r%t' dt) d!t+ ||z d!z nd}|D cg c]& \ } } } | dkD rt, j | d" | | | d#( c} } } ||||t | d$d xs dd%S c c} } } w )&zAReturn a Cursor-style context usage breakdown for one live agent.r ) estimate_messages_tokens_rough)build_system_prompt_partsstabler$ contextvolatilez
c 3 & K | ] }|s| y wN .0parts r z4compute_session_context_breakdown..j s RtTd R c 3 & K | ] }|s| y wrP rQ rR s r rU z4compute_session_context_breakdown..n s $YdTXT$YrV r1 Nr z
System promptr zTool definitionsr
Rulesr Skillsr MCPr zSubagent definitionsr Memoryr Conversationc 3 ( K | ]
\ }}}| y wrP rQ )rS _tokenss r rU z4compute_session_context_breakdown.. s @\Q6&@s context_compressorcontext_lengthlast_prompt_tokensd zvar(--ui-text-tertiary))coloridlabelr_ model)
categories context_maxcontext_percent context_usedestimated_totalrg )agent.model_metadatarJ agent.system_promptrK r' _SKILLS_BLOCK_REsearchgrouprA joinrD rH listr: r4 r r sumintmaxminround_CATEGORY_COLORS) r= messagesrJ rK partsrL rM rN skills_match skills_indexr> r? memory_text system_core system_tailsystem_prompt_textr1
builtin_tools mcp_toolssubagent_toolsconversation_tokensrh rl compri
measured_usedrk rj category_idrf r_ s r !compute_session_context_breakdownr Y s
D= %e ,E
YYx
$
*Fii 2&,"GyyR(.BH#**62L,8<%%a(bL -e 4 L*++ R j/I RRXXZK 5K ,
CK $Y{K6P$YY__a .4" 5E/;E/B,M9n8 RH
/+;AKHLC&:A>C!DRSM$1A$5=?L As3l[ 83 >?@A
/9
+ UFz
*--k;TU!
#*$*,2
s !+I')r r( returnru )r r r ru )r) r& r r( )r1 zSequence[dict]r z)Tuple[List[dict], List[dict], List[dict]])r= r r zTuple[str, str])r r( rE r( r r( rP )r= r rz zOptional[List[dict]]r zDict[str, Any])__doc__
__future__r r retypingr r r r r r compileDOTALLro frozensetr0 ry r r r+ r4 rA rH r rQ r r r s # = = 2:: I299U /!23 34
)+ % <+7 C ' "
$ &*C C"C Cr