sj\UdZddlmZmZddlmZdediZeeefe d<dZ e e d<d Z e e d <d Z e e d <d Ze e d<dZee d<de fdZedGddZeZdZe e d<dZee d<dZee d<dZe e d<dZe e d <d!e d"zdefd#Zd"S)$zConfigurable budget constants for tool result persistence. Per-tool resolution: pinned > config overrides > registry > default. ) dataclassfield)Dict read_fileinfPINNED_THRESHOLDSiDEFAULT_RESULT_SIZE_CHARSi@ DEFAULT_TURN_BUDGET_CHARSiDEFAULT_PREVIEW_SIZE_CHARSiPDEFAULT_MCP_RESULT_SIZE_CHARSmcp_MCP_TOOL_PREFIXreturnc2 ddlm}|}t|tr|dnd}t|tr.|d}|t |}|dkr|Sn#t $rYnwxYwtS)u Read ``tool_budget.mcp_result_size_chars`` from the active config. Goes through :func:`hermes_cli.config.load_config_readonly` (the sanctioned read path — raw config.yaml parsing outside owner modules is guarded by tests/hermes_cli/test_config_read_guard.py). Fully guarded: any error, missing key, or non-positive value returns the built-in default. The ``tool_budget:`` block name is shared with the wider configurable-caps proposal (#80508) so the two can merge without a key rename. r)load_config_readonly tool_budgetNmcp_result_size_chars)hermes_cli.configr isinstancedictgetint Exceptionr )rdatablockrawvalues 7/home/agent/.hermes/hermes-agent/tools/budget_config.py_configured_mcp_result_sizer's  ::::::##%%+5dD+A+AK'''t eT " " !))344CC199 L      ((sA>B BBT)frozenceZdZUdZeZeed<eZ eed<e Z eed<e Z eed<eeZeeefed<ded eezfd Zd S) BudgetConfiga|Immutable budget constants for the 3-layer tool result persistence system. Layer 2 (per-result): resolve_threshold(tool_name) -> threshold in chars. Layer 3 (per-turn): turn_budget -> aggregate char budget across all tool results in a single assistant turn. Preview: preview_size -> inline snippet size after persistence. default_result_size turn_budget preview_sizemcp_result_size)default_factorytool_overrides tool_namercZ|tvr t|S||jvr |j|S|trt |j|jSddlm}| ||j}|tdkr|St ||jS)aResolve the persistence threshold for a tool. Priority: pinned -> tool_overrides -> mcp_ prefix -> registry per-tool -> default. MCP tools (``mcp_`` prefix) get a tighter default threshold (``mcp_result_size``, 50K chars) because MCP servers return un-paginated payloads with no per-tool registry entry to constrain them. The value is additionally capped at ``default_result_size`` so a context-scaled budget for a small model still constrains MCP results the same way it constrains registry values. The registry per-tool value is capped at ``default_result_size`` so a context-scaled budget (small model) actually constrains tools that register a large fixed ``max_result_size_chars`` (web/terminal/x_search all register 100K). For the default budget this is a no-op because both equal 100K; for a scaled-down budget it prevents a per-tool registry value from re-inflating the cap past the model's window (#23767). r)registry)defaultr) rr( startswithrminr&r#tools.registryr+get_max_result_sizefloat)selfr)r+registry_values rresolve_thresholdzBudgetConfig.resolve_thresholdRs( ) ) )$Y/ / + + +&y1 1    0 0 Gt+T-EFF F++++++!55iIa5bb U5\\ ) )! !>4#;<<<N)__name__ __module__ __qualname____doc__r r#r__annotations__r r$r r%r r&rrr(rstrr1r4r5rr"r"Bs 98880K0002L#2228OS888%*U4%@%@%@NDcN@@@=3=3;======r5r"_CHARS_PER_TOKENg333333?_PER_RESULT_WINDOW_FRACTIONg333333?_PER_TURN_WINDOW_FRACTIONi@_MIN_RESULT_SIZE_CHARSi>_MIN_TURN_BUDGET_CHARScontext_lengthNct}|r|dkr"|tkrtSt|S|tz}t |t z}t |tz}ttt|t}ttt|t}t||t|S)aReturn a BudgetConfig scaled to the active model's context window. The fixed defaults (100K result / 200K turn chars) are correct for large (200K+ token) models but blind to small ones: on a 65K-token model a single tool result persisted at the 100K-char threshold, or a 200K-char turn budget (~50K tokens), can by itself approach or exceed the whole window and force an oversized request (#23767). Scaling keeps large models byte-identical to today (the proportional value is clamped to the existing defaults as a CAP) while shrinking the budget for small models proportionally to their window, floored so a usable preview always survives. r)r&)r#r$r%r&)rr DEFAULT_BUDGETr"r>rr?r@maxrAr.r rBr r )rCr& window_chars per_resultper_turns rbudget_for_context_windowrJs233O =^q00 ; ; ;! !O<<<<!$44L\$??@@J<";;<r?r@rArBrJr<r5rrMs )((((((( u'4U # ")3(((!(3((("'C'''&,s+++)S))))6 $-=-=-=-=-=-=-=-=b#&*U)))#'5'''$###$$$$#cDj#\######r5