OpenViking Tool Stub 设计:从 head+tail 截断到类型化确定性摘要的演进与实践
【免费下载链接】OpenVikingSelf-evolving Context Database for AI Agents. Unify Agent Memory, Knowledge RAG and Skills.项目地址: https://gitcode.com/GitHub_Trending/op/OpenViking
导读
本文以 OpenViking 仓库中的 tool-stub-design.md 设计文档为核心骨架,结合 tool_result_synopsis.py、tool_result_store.py、session.py 等源码实现,系统讲解 OpenViking 的tool stub能力:如何识别大体积 tool output、如何将其外置到ToolResultStore、如何用类型化规则摘要替代简单截断生成 preview stub、以及如何通过read/search/list回溯原文。读完本文,你将掌握这一闭环的端到端工作流程、八种摘要类型的识别与处理规则、Stub 文本结构、配置参数及测试边界,能够直接用于排查上下文占用问题或扩展新的摘要类型。
一、概述:什么是 Tool Stub
在 Agent 会话中,工具(tool)的输出往往体积巨大,比如一段 JSON API 响应、一份 CSV 导出、一段日志或一段源码。如果把这些原始输出全部塞进会话上下文,会快速挤占上下文窗口,导致 token 浪费、成本上升、模型注意力分散。
OpenViking 原生已经支持tool result preview:原有链路能在 session 写入阶段把过大的 tool outputexternalize(外置),并在ToolPart中留下一个 preview stub,同时保留 ref 供后续回溯。
本次 Tool Stub 工作的重点不是新建 externalize 机制,而是在现有能力上优化 preview 的生成方式:从偏head + tail的直接截断,升级为按内容类型输出更稳定、更可读的规则化摘要(deterministic synopsis)。
保持不变的四个基础能力
- 哪些 tool output 需要 externalize,仍由 session 写入阶段决定;
- 原始内容仍写入
ToolResultStore; ToolPart仍保留 stub 和tool_output_ref;- 原文回溯方式仍是
read/search/list。
本次变化集中在 preview 生成层
- 在 session 写入阶段识别哪些 tool output 需要 externalize;
- 原始输出写入 session 下的 tool result store;
- preview 从简单截断优化为基于内容和 MIME 的 deterministic synopsis;
- 把原始
ToolPart.tool_output替换成 stub 文本,并保留tool_output_ref; - 后续通过
read/search/list工具按 ref 回溯原文。
需要特别强调:text类型只做规则化摘要,不接 LLM。当前版本不引入 LLM 摘要,只做 deterministic 的规则化摘要,以保证稳定、低成本、可测试。
二、设计目标
- 在保留 OV 原生 externalize 和 ref 回溯链路的前提下,优化 preview 的可读性;
- 减少大 tool output 对上下文窗口的占用;
- 把原有偏
head + tail的截断 preview,升级为按类型输出的规则化摘要; - 对常见文本型输出给出稳定、可读的 deterministic synopsis;
- 保留原始输出,支持按 ref 精确回溯;
- 当前版本不做 LLM 摘要,只做 deterministic 的规则化摘要。
三、端到端流程
1. 选择哪些输出需要 externalize
入口在 session.py 的_externalize_large_tool_output_group()。
当前按两类条件触发 externalize:
- 单个 tool output 超过
threshold_chars; - 同一个 assistant turn 中多个 tool output 的总 inline 体积超过
assistant_turn_inline_budget_chars(聚合预算)。
命中后会进入 externalization 流程。触发阈值仍沿用 OV 原有配置;规则化摘要本身不改变"什么时候 externalize"。
从源码结构看,选择逻辑的细节是:先收集当前消息组中的所有ToolPart(p.tool_output非空),计算整个组的原始体积总和group_original_chars;然后遍历找出单个输出超过threshold_chars的 part(reason 记为single_threshold);其余 part 按体积从大到小排序,只要"选中后的预计 inline 体积仍不低于assistant_turn_inline_budget_chars"就继续选中(reason 记为turn_budget),直到聚合体积降到预算以下或没有可选项为止。
2. 外置原始结果并替换 ToolPart
入口在 session.py 的_externalize_tool_part()。这一步会:
- 计算原始输出的
sha256摘要,把原始tool_output写入ToolResultStore; - 调用 preview 生成逻辑产出 stub 文本;
- 用 stub 替换
ToolPart.tool_output; - 把原始结果的 ref 写入
ToolPart.tool_output_ref。
被 stub 后,消息里保留的是 preview,不再是完整原始 tool result;原始结果仍在外置存储里。ToolPart上还会同步记录一组派生字段:tool_output_truncated、tool_output_original_chars、tool_output_preview_chars、tool_output_sha256、tool_output_storage_uri、tool_output_group_id、tool_output_externalized_reason等,供上层与调试使用。
需要留意的是,失败处理也有分支:如果外置写入抛异常,会依据配置的failure_mode处理——reject直接抛FailedPreconditionError;preview_only则退化为只在本消息内生成 preview(ref 置空);默认preserve_raw则保留原始输出不截断。
3. 生成 synopsis 和 stub
入口在 tool_result_store.py 的make_preview():
- 原生 preview/stub 能力保留,但内容生成逻辑从简单截断演进为typed synopsis;
generate_tool_result_synopsis()负责类型识别和摘要生成;render_tool_result_stub()负责把摘要渲染成最终 stub 文本。
核心实现位于 tool_result_synopsis.py 的generate_tool_result_synopsis()与 tool_result_synopsis.py 的render_tool_result_stub()。
当前原则:
- externalize 触发阈值沿用 OV 原有配置;
- 常见类型使用固定规则上限生成 synopsis,不依赖
preview_chars控制摘要长度; preview_chars只作为无法规则化时的fallback head/tail 采样预算,并作为兼容字段保留在 stub header / metadata 中。
4. 原文回溯
当前回溯能力由 session 暴露三类工具,对应存储层实现在 tool_result_store.py 起的read()/search()/list():
- session.py
read_tool_result():按offset/limit读取原始内容片段; - session.py
search_tool_result():在原始内容中做关键字搜索,返回带 offset 的 snippet; - session.py
list_tool_results():列出当前 session 已 externalize 的结果,支持按tool_name过滤。
此外,从源码中还可以看到一处防抖设计:_rewrite_source_read_tool_output()会把"通过openviking_tool_result_read读回的 tool output"改写为源引用,而不是当作一个新的 tool result 再次进入 externalize 流程,避免读回大文本时产生二次外置(参见 session.py)。
四、支持的数据类型
当前支持的 synopsis kind 定义在 tool_result_synopsis.py:
json / csv / tsv / yaml / xml / code / text / unknown类型对照表
| 类型 | 识别方式 | 处理办法 | stub 中保留内容 |
|---|---|---|---|
json | MIME 含json,或内容以{/[开头且能被 JSON decoder 解析 | 解析 top-level shape,提取 keys、array length、标量示例 | summary + structure + notable_items |
csv | 含逗号,且能按 CSV 读成规则表格 | 统计行列数、列名,保留首条数据样例 | summary + structure |
tsv | 含制表符,且能按 TSV 读成规则表格 | 统计行列数、列名,保留首条数据样例 | summary + structure |
yaml | 满足 YAML 启发式并能yaml.safe_load()成 dict/list | 提取 top-level keys 和 child type | summary + structure |
xml | MIME 含xml,或内容以<开头且可解析 | 提取 root tag、属性数、子标签计数 | summary + structure |
code | 命中代码模式正则 | 提取 imports、symbols、line_count | summary + structure + notable_items |
text | 作为最终 fallback | 规则化文本摘要,不保留全文 sample | summary |
unknown | 空内容、binary-like 内容,或带明确 MIME 但解析失败的结构化内容 | 无法规则化时使用 fallback head/tail sample | summary + sample |
五、类型识别顺序
识别顺序定义在 tool_result_synopsis.py 的generate_tool_result_synopsis()中。当前顺序如下:
- 空内容:直接标为
unknown; - binary-like 内容:出现 NUL 或控制字符比例过高,标为
unknown; jsonxmltsvcsvyamlcode- 最终 fallback 为
text
这个顺序的目的是优先识别结构化格式,再识别代码,最后才把剩余内容视作普通文本。日志样式输出不再作为独立类型处理,会走text,与 lossless-claw 的 large-file exploration 行为保持一致。
从源码看,binary-like 的判定标准是:内容含\x00,或前 1000 个字符中非\n\r\t的控制字符占比超过 5%(_looks_binary(),见 tool_result_synopsis.py)。yaml的启发式(_looks_yaml(),见 tool_result_synopsis.py)要求以---开头,或出现key:后紧跟缩进行;若命中代码模式正则则直接排除。
六、各类型处理策略
JSON
实现位于 tool_result_synopsis.py 的_summarize_json()。输出重点:
- 顶层类型是 object 还是 array;
- top-level keys,最多 10 个;
- 子字段是 object / array / scalar;
- 最多若干条标量示例(
_json_scalar_examples()默认 5 条,单条 80 字符以内); - 若第一个 JSON value 后还有额外字符,会记入
trailing_chars_after_first_json_value; - 不额外保留原始 JSON sample。
JSON shape 的递归深度上限为 2(_JSON_MAX_DEPTH = 2),array 只采样前 3 个元素,object 最多展示 10 个 key。也就是说,一个 10MB 的 JSON 响应最终在上下文里只剩"类型 + 结构 + 少量标量示例",体积可被压缩到极小。
CSV / TSV
实现位于 tool_result_synopsis.py 的_try_table()。输出重点:
- 列数与数据行数;
- 首行列名(空单元格自动命名为
column_N); - 首条数据样例,最多 180 字符(
_TABLE_FIRST_ROW_SAMPLE_CHARS = 180)。
只接受"列数基本一致"的表格:要求至少有 2 行且首行至少 2 列,并且前 10 行的列数必须一致,否则判定为不规则分隔文本,不会被误判成表格。TSV 的识别(制表符分隔)优先于 CSV(逗号分隔)。
YAML
实现位于 tool_result_synopsis.py 的_summarize_yaml()(配合_try_yaml(),见 tool_result_synopsis.py)。输出重点:
- 顶层是 object 还是 array;
- top-level keys,最多 30 个(
_YAML_KEY_LIMIT = 30); - 每个 key 对应的 child type(object / array / scalar 等);
- 不额外保留 YAML sample。
XML
实现位于 tool_result_synopsis.py 的_summarize_xml()。输出重点:
- 根标签名;
- 根节点属性数量;
- 一级子标签频次(按出现次数排序),最多 30 个(
_XML_CHILD_TAG_LIMIT = 30); - 不额外保留 XML sample。
Code
实现位于 tool_result_synopsis.py 的_summarize_code()。输出重点:
- 总行数;
- import 语句,最多 12 条(
_CODE_IMPORT_LIMIT = 12),单条最多 180 字符; - 顶层 symbol,如
class Foo、def bar、fn baz,最多 24 条(_CODE_SYMBOL_LIMIT = 24),单条最多 200 字符; - 不额外保留 head/tail sample。
识别代码的正则(_CODE_PATTERNS,见 tool_result_synopsis.py)覆盖 Python(import/from ... import/class/def/async def)、JS/TS(function/export function/const/let/var)、Rust(package/use/pub fn/fn)等常见语言。当前是轻量规则识别,不做 AST 级代码摘要。
Text
实现位于 tool_result_synopsis.py 的_summarize_text()。text类型明确不接 LLM,只做 deterministic fallback。摘录采用固定上限,不受preview_chars影响。
日志样式输出也归入text。如果需要定位错误/警告行,优先通过 stub 中的 ref 使用openviking_tool_result_search搜索原始 payload,避免仅靠关键字把普通文档误判成日志。
输出重点:
CharactersWordsLinesDetected section headersOpening excerptClosing excerpt
标题提取规则:
- Markdown 标题,如
# Heading; - 全大写风格标题行,如
SYSTEM STATUS(匹配^[A-Z0-9][A-Z0-9\s:_-]{6,}$)。
摘录规则:
- opening excerpt 固定最多取前500 字符(
_TEXT_EXCERPT_CHARS = 500); - closing excerpt 固定最多取后500 字符;
- 先压缩空白(连续空白折叠为单空格),再写入摘要;
- 不额外保留
sample字段,避免把完整原文重新带回上下文。
section headers 最多提取 18 条(_TEXT_HEADER_LIMIT = 18),单条最长 160 字符,且去重。
Unknown
unknown是当前实现里的保守分类,不是富类型支持。当前会落到unknown的场景包括:
- 输出为空(标题为
Empty output); - 文本中存在明显二进制控制字符(标题为
Binary-like output); - MIME 明确标成 JSON/XML,但内容解析失败(标题为
Unparsed JSON-like output/Unparsed XML-like output)。
对这些内容,stub 保留基础说明;非空内容会使用preview_chars生成 head/tail fallback sample(_head_tail_sample(),见 tool_result_synopsis.py,以--- BEGIN SAMPLE HEAD ---/--- BEGIN SAMPLE TAIL ---分段)。原始 payload 仍通过 ref 回溯。
七、Stub 文本结构
渲染逻辑位于 tool_result_synopsis.py 的render_tool_result_stub()。当前 stub 由两部分组成:
Header
首行固定为[OpenViking tool result externalized],随后包含:
tool_namekindoriginal_charspreview_charsrefsha256(仅显示前 16 位,用于安全起见不暴露完整哈希)reason
Body
按 synopsis 内容选择性渲染:
Synopsis(始终存在,逐条-列表)Structure(有 structure 时渲染)Notable items(有 notable_items 时渲染)Sample(有 sample 时渲染)Explore(有 ref 时渲染)
其中Explore会提示模型使用三类回溯工具:
openviking_tool_result_search(配合ref=定位相关原始片段)openviking_tool_result_read(配合ref=与offset/limit逐段检查原始输出)openviking_tool_result_list(发现当前 session 中其他已外置的结果)
这样,即使 stub 被替换后,模型依然被明确引导"去哪里找回原文",形成"摘要可读、原文可溯"的闭环。
八、原始内容存储与回溯
实现位于 tool_result_store.py 的ToolResultStore。每个 externalized tool result 会写入两类文件:
output.txt:原始输出内容;metadata.json:元数据和 synopsis。
存储目录为{session_uri}/tool-results/{tool_result_id}/,其中tool_result_id由tr_前缀 + 安全化后的tool_id+ sha256 前 16 位构成(build_tool_result_id(),见 tool_result_store.py),保证同一内容的幂等去重(写入时会先按 sha256 检查是否已存在)。
metadata 包括:
tool_result_idsession_idmessage_idtool_idtool_namecreated_atoriginal_charspreview_charssha256mime_typesynopsis_kindsynopsisstorage_urioutput_urioffset_unit=unicode_code_point
其中offset_unit=unicode_code_point明确说明所有 offset 均以 Unicode 码点为单位,而非字节。
读取方式
read():按offset/limit读取原始文本片段(默认limit=20_000,limit=-1表示读到底),返回content / offset / limit / total_chars / has_more及可选 metadata,适合长文本逐段展开;search():在原始文本中查关键词(content.find(query)),返回最多limit(默认 20)条带 offset 的 snippet,每个 snippet 在命中位置前后各保留context_chars(默认 300)字符;list():按 session 列出已有外置结果(通过 viking_fs 的ls枚举目录并读取各结果 metadata),支持按tool_name过滤,默认limit=50,便于发现 ref。
当前读取模型是**"面向长文本"**的;它适合回看日志、代码、表格文本、普通文本。
九、配置参数(源码确认)
externalize 触发阈值与 preview 预算均由配置模型ToolOutputExternalizationConfig控制,定义在 server/config.py:
| 参数 | 默认值 | 说明 |
|---|---|---|
enabled | true | 总开关,关闭后不做 externalize |
threshold_chars | 20000 | 单个 tool output 超过该字符数触发外置 |
preview_chars | 2000 | 单个 preview 采样预算;仅作为无法规则化时的 head/tail fallback 预算及兼容字段 |
assistant_turn_inline_budget_chars | 100000 | 同一 assistant turn 内多个 tool output 的总 inline 体积预算 |
assistant_turn_preview_budget_chars | 10000 | 同一 turn 内所有 preview 的合计预算,用于按数量均摊 |
min_preview_chars | 1000 | preview 预算下限,防止均摊后过小 |
aggregate_selection_strategy | largest_first | 聚合超限时的选择策略(当前为按体积从大到小优先选中) |
failure_mode | preserve_raw | 外置失败时的处理:reject/preserve_raw/preview_only |
从 session.py 的_effective_tool_preview_chars()可以看到预算均摊逻辑:当有N个输出被选中外置时,单个 preview 预算取assistant_turn_preview_budget_chars / N与preview_chars的较小值,并受min_preview_chars下限约束。
十、当前边界与取舍
text不接 LLM,原因是我们当前只需要稳定、低成本、可测试的规则化 stub;- 当前回溯接口是
read/search/list,更适合大文本原文回看; offset/limit对"顺序文本展开"很合适,但对图片、二进制、复杂多模态结果并不理想(offset 以 Unicode 码点为单位,无法表达图片/音视频等媒体语义)。
这些取舍决定了当前版本的功能边界:优先解决文本型 tool output 的上下文瘦身问题,媒体解析与 LLM 摘要不在本期范围内。
十一、测试覆盖
当前相关测试包括:
- tests/session/test_tool_result_synopsis.py:覆盖类型识别和 synopsis 生成,包括固定 caps、text 的 500/500 deterministic fallback,以及 unknown 的 head/tail fallback(文件内共 13 个测试函数);
- tests/session/test_tool_result_externalization.py:覆盖 externalization、stub 替换、阈值边界、aggregate budget、ref 回溯等端到端流程;
- tests/server/test_api_sessions.py:覆盖 HTTP API 层的 tool result externalization、stub 文案、
read/list/search回溯,以及synopsis_kind/synopsis.kind元数据透出。
当前相关测试共 29 个用例通过,可作为后续继续补齐真实输出回归用例的基础。
十二、结论
OpenViking 当前的tool stub已经具备一版完整闭环:
- OV 原生的 externalize、preview stub、ref 回溯链路继续保留;
- preview 生成方式已从偏
head + tail的截断,升级为按类型的规则化摘要; text类型采用 deterministic 规则摘要,不接 LLM;- 能通过
read/search/list对外置原文进行回溯。
后续优先级应放在更深的回归测试、更多真实输出样本,以及是否需要继续优化read/search/list的原文回溯体验,而不是先扩展 LLM 摘要或媒体解析。
【免费下载链接】OpenVikingSelf-evolving Context Database for AI Agents. Unify Agent Memory, Knowledge RAG and Skills.项目地址: https://gitcode.com/GitHub_Trending/op/OpenViking
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考