Files
metona-ollama-desktop/tests/tool-parsing.test.ts
T
thzxx b66945c8a7
CI / verify (push) Successful in 1m2s
v0.17.1: 退出释放显存修正 + 备份完整性 + 核心逻辑测试补课 + 上下文逻辑收敛 + 记忆日志可读性
修复:
- main.ts 退出释放模型显存改用 getSetting(serverUrl),不再硬编码 127.0.0.1:11434(避免非默认地址时释放请求打到错误端口)
- 备份导出/导入并入 localStorage 持久化状态(会话摘要、度量历史、轨迹降级缓存、主题),版本升级到 v2,实现完整备份
- 工具数量改为 getEnabledToolDefinitions().length 动态计算,删除写死"32 个"的硬编码
- 记忆日志区分操作来源:memory:write 透传 reason,标注"新增记忆/替换/删除/清空/TTL 衰减清理/访问统计写回(无新条目)",避免"写了但看不到新记忆"的困惑

可维护性:
- 上下文压力逻辑收敛到统一 calculateContextStats,删除 getContextPressureLevel / getTrendAwareCompressThreshold 的重复实现
- 消除 validateToolArgs 同名碰撞(agent-engine 本地版改名 validateToolArgsQuick)
- 子代理工具集改用 getEnabledToolDefinitions() 基线,跟随全局启用开关与 Plan 模式
- 抽取 html-utils.ts 纯函数模块(实体解码/HTML→文本/HTML→Markdown/拦截页检测/相关性评分),tool-handlers-system 净减约 190 行重复代码
- 统一静态导入(savePlanTracker/setPlanModeActive/collectDiagnostics/addWrittenFile)
- console.* 使用处补充豁免说明(启动/退出/刷盘阶段无渲染进程可推送日志)
- run_command 工具描述改为反映可配置执行模式

测试:
- 新增 7 个测试文件 + 扩展 2 个,共 273 个测试(原 34 → 273)
- 覆盖 agent-engine / agent-safety / context-manager / tool-registry / result-formatter / tool-parsing / memory-service / crypto / build-context / html-utils / utils / tool-handlers-fs
- 全部通过 npm run typecheck && npm test && npm run build
2026-08-26 15:02:47 +08:00

61 lines
2.2 KiB
TypeScript

import { describe, it, expect } from 'vitest';
import { parseToolCallsFromText } from '../src/renderer/services/tool-parsing.js';
describe('parseToolCallsFromText — 文本工具调用兜底解析', () => {
it('解析 Action / Action Input 格式', () => {
const content = `
Thought: 我需要读取一个文件
Action: read_file
Action Input: {"path": "src/main.ts"}
`;
const calls = parseToolCallsFromText(content);
expect(calls).toHaveLength(1);
expect(calls[0].function.name).toBe('read_file');
expect(calls[0].function.arguments.path).toBe('src/main.ts');
});
it('解析 <tool_call> XML 格式', () => {
const content = `<tool_call>
{
"name": "web_search",
"arguments": {"query": "rust language"}
}
</tool_call>`;
const calls = parseToolCallsFromText(content);
expect(calls).toHaveLength(1);
expect(calls[0].function.name).toBe('web_search');
expect(calls[0].function.arguments.query).toBe('rust language');
});
it('解析 ```json 代码块中含 name 字段', () => {
const content = '```json\n{"name": "list_directory", "arguments": {"path": "."}}\n```';
const calls = parseToolCallsFromText(content);
expect(calls).toHaveLength(1);
expect(calls[0].function.name).toBe('list_directory');
});
it('解析函数调用语法 func({...}) 且支持嵌套 JSON', () => {
const content = '需要执行 read_file({"path": "a", "opts": {"b": 1}})';
const calls = parseToolCallsFromText(content);
expect(calls).toHaveLength(1);
expect(calls[0].function.name).toBe('read_file');
expect((calls[0].function.arguments as Record<string, unknown>).opts).toEqual({ b: 1 });
});
it('未知工具名被忽略', () => {
const content = 'Action: not_a_real_tool\nAction Input: {"path": "x"}';
expect(parseToolCallsFromText(content)).toHaveLength(0);
});
it('无工具调用返回空数组', () => {
expect(parseToolCallsFromText('这是一个普通回答,没有工具调用。')).toHaveLength(0);
});
it('容忍不带引号的单引号参数', () => {
const content = "Action: read_file\nAction Input: {'path': 'file.txt'}";
const calls = parseToolCallsFromText(content);
expect(calls).toHaveLength(1);
expect(calls[0].function.arguments.path).toBe('file.txt');
});
});