i/
L d Z ddlZddlZddlZddlZddlmZmZmZ ej e
Z dZ dZ
dZdZdZd efd
Zefd ed ed eeef fd
Zd ed efdZd eded efdZdedededed ef
dZdedfd ededededeez dz d ef dZdefde e ded e e fd Z dS ) a Tool result persistence -- preserves large outputs instead of truncating.
Defense against context-window overflow operates at three levels:
1. **Per-tool output cap** (inside each tool): Tools like search_files
pre-truncate their own output before returning. This is the first line
of defense and the only one the tool author controls.
2. **Per-result persistence** (maybe_persist_tool_result): After a tool
returns, if its output exceeds the tool's registered threshold
(registry.get_max_result_size), the full output is written INTO THE
SANDBOX temp dir (for example /tmp/hermes-results/{tool_use_id}.txt on
standard Linux, or $TMPDIR/hermes-results/{tool_use_id}.txt on Termux)
via env.execute(). The in-context content is replaced with a preview +
file path reference. The model can read_file to access the full output
on any backend.
3. **Per-turn aggregate budget** (enforce_turn_budget): After all tool
results in a single assistant turn are collected, if the total exceeds
MAX_TURN_BUDGET_CHARS (200K), the largest non-persisted results are
spilled to disk until the aggregate is under budget. This catches cases
where many medium-sized results combine to overflow context.
N)DEFAULT_PREVIEW_SIZE_CHARS BudgetConfigDEFAULT_BUDGETzzz/tmp/hermes-resultsHERMES_PERSIST_EOF__budget_enforcement__returnc | |t | dd }t | r\ | }|r | d pd}| dS n2# t $ r%}t d| Y d}~nd}~ww xY wt S )z=Return the best temp-backed storage dir for this environment.N get_temp_dir/z/hermes-resultsz"Could not resolve env temp dir: %s)getattrcallablerstrip Exceptionloggerdebug STORAGE_DIR)envr
temp_direxcs >/home/ubuntu/.hermes/hermes-agent/tools/tool_result_storage.py_resolve_storage_dirr , s
sND99 L ! ! 8
8'<>> 8 's 3 3 :sH &77778
H
H
H A3GGGGGGGG
H s
A
A<A77A<content max_charsc t | |k r| dfS | d| }| d }||dz k r
|d|dz }|dfS )zGTruncate at last newline within max_chars. Returns (preview, has_more).FN
T)lenrfind)r r truncatedlast_nls r generate_previewr" < sg
7||y ~
#Iood##Ga lw{l+ d? c h t | vrt S dt j j dd S )z=Return a heredoc delimiter that doesn't collide with content.HERMES_PERSIST_N )HEREDOC_MARKERuuiduuid4hex)r s r _heredoc_markerr+ G s3 W$$ 3TZ\\ -bqb 1 3 33r# remote_pathc
* t | }t j | }dt j | dt j | d| d| d|
}| |d }| dd d
k S ) zJWrite content into the sandbox via env.execute(). Returns True on success.z mkdir -p z
&& cat > z << 'z'
r )timeout
returncoder r )r+ ospathdirnameshlexquoteexecuteget)r r, r marker storage_dircmdresults r _write_to_sandboxr< N s
W
%
%F'//+..K EK ,, K8P8P W]
[[b[
)
)F ::lA & &! ++r# previewhas_more
original_size file_pathc |dz }|dk r
|dz dd}n|dd}t d}|d|dd| d z
}|d
| dz
}|d z
}|d t | d
z
}|| z
}|r|dz
}|dt z
}|S )z/Build the replacement block.i z.1fz MBz KBr z This tool result was too large (,z
characters, z).
zFull output saved to: zZUse the read_file tool with offset and limit to access specific sections of this output.
zPreview (first z chars):
z
...)PERSISTED_OUTPUT_TAGr PERSISTED_OUTPUT_CLOSING_TAG)r= r> r? r@ size_kbsize_strmsgs r _build_persisted_messagerH [ s d"G$ n---- &&&&
!
%
%
%C Zm Z Z ZH Z Z ZZC 1I 1 1 11C iiC 5S\\ 5 5 55C7NC w ., . ..C Jr# tool_name tool_use_idconfig thresholdc ||n| | }|t d k r| S t | |k r| S t | }| d| d}t | |j \ } }
| t
| || rJt d||t | | t | |
t | | S n3# t $ r&} t d|| Y d} ~ nd} ~ ww xY wt d|t | | d t | d
d S ) a Layer 2: persist oversized result into the sandbox, return preview + path.
Writes via env.execute() so the file is accessible from any backend
(local, Docker, SSH, Modal, Daytona). Falls back to inline truncation
if write fails or no env is available.
Args:
content: Raw tool result string.
tool_name: Name of the tool (used for threshold lookup).
tool_use_id: Unique ID for this tool call (used as filename).
env: The active BaseEnvironment instance, or None.
config: BudgetConfig controlling thresholds and preview size.
threshold: Explicit override; takes precedence over config resolution.
Returns:
Original content if small, or replacement.
Ninfr z.txt)r z4Persisted large tool result: %s (%s, %d chars -> %s)z Sandbox write failed for %s: %szDInline-truncating large tool result: %s (%d chars, no sandbox write)z
[Truncated: tool response was rB z3 chars. Full output could not be saved to sandbox.]) resolve_thresholdfloatr r r" preview_sizer< r inforH r warning) r rI rJ r rK rL effective_thresholdr9 r, r= r> r s r maybe_persist_tool_resultrU t s 2 (1'<))&BZBZ[dBeBeeEll**
7||***&s++K 44;444K(F>D
tool_messagesc N g }d}t | D ]O\ }}| dd }t | }||z
}t |vr| ||f P||j k r| S | d d |D ]\ } }||j k r n| | }|d }| dd| }
t |t |
||d } | |k r>||z }|t | z
}| | | d< t
d
|
| | S ) a" Layer 3: enforce aggregate budget across all tool results in a turn.
If total chars exceed budget, persist the largest non-persisted results
first (via sandbox write) until under budget. Already-persisted results
are skipped.
Mutates the list in-place and returns it.
r r c | d S )Nr )xs r z%enforce_turn_budget.. s
!A$ r# T)keyreverse tool_call_idbudget_)r rI rJ r rK rL z7Budget enforcement: persisted tool result %s (%d chars)) enumerater7 r rC append turn_budgetsortrU _BUDGET_TOOL_NAMEr rR ) rV r rK
candidates
total_sizeirG r sizeidxrJ replacements r enforce_turn_budgetrl sx JJM** ) )3'')R((7||d
w . . q$i ( ( (V''' OOO555 T + + + EC i.ggnooo>> /'#
' ! ! $ J #k** *J,7M# y ) KKIT
r# ) __doc__loggingr1 r4 r( tools.budget_configr r r getLogger__name__r rC rD r r' re strr inttupleboolr" r+ r< rH rP rU listdictrl rZ r# r rx s5 0
8 $ $+ 4 # %,
5O c c SXY\^bYbSc 4S 4S 4 4 4 4
,s
,
,d
,
,
,
,
:
)$(
8 8
88 8
8 U{T!
8 8 8 8 8z
)3 3:3
3
$Z 3 3 3 3 3 3r#