mirror of
https://bitbucket.org/siakitem/my-pi.git
synced 2026-10-26 10:45:22 +00:00
云服务部-叶林立andDevin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
ee1e6999d7
7 个按厂商命名的搜索工具收敛为 2 个统一工具，schema 与提示词不再暴露 provider 概念：web_search 由当前主模型按内置规则在 Tavily/Exa/Keenable 间路由（会话内 LRU 缓存，失败按可用性降级），web_fetch 固定走 Exa 并 静默降级。三家均改为纯 fetch 直连——Exa 直接对 mcp.exa.ai 发 JSON-RPC 并缓存会话，彻底移除 pi-mcp-adapter、@tavily/pi-extension 与 @keenable/pi-search 依赖。 Generated with [Devin](https://devin.ai) Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
62 lines
4.0 KiB
TypeScript
62 lines
4.0 KiB
TypeScript
import assert from "node:assert/strict";
|
|
import test from "node:test";
|
|
|
|
import type { ToolInfo } from "@earendil-works/pi-coding-agent";
|
|
import { BUNDLE_GROUP_DEFINITIONS, createBundleCatalog } from "../extensions/bundle-groups.ts";
|
|
import { buildManifestHash, rankGroups, selectConfidentGroup } from "../extensions/catalog.ts";
|
|
|
|
function tool(name: string): ToolInfo {
|
|
return {
|
|
name,
|
|
description: `Full description for ${name}`,
|
|
parameters: { type: "object", properties: {} } as ToolInfo["parameters"],
|
|
sourceInfo: { source: "extension", scope: "user", path: `/test/${name}.ts` } as unknown as ToolInfo["sourceInfo"],
|
|
};
|
|
}
|
|
|
|
const constraints = { maxToolsPerGroup: 8, groupOverrides: {} };
|
|
|
|
test("bundle seed names are unique and cover the declared my-pi tools", () => {
|
|
const names = BUNDLE_GROUP_DEFINITIONS.flatMap((group) => group.tools);
|
|
assert.equal(new Set(names).size, names.length);
|
|
const tools = names.map(tool);
|
|
const hash = buildManifestHash(tools, constraints);
|
|
const result = createBundleCatalog(tools, hash, constraints);
|
|
assert.deepEqual(result.unknownTools, []);
|
|
assert.equal(result.coveredNames.size, names.length);
|
|
assert.equal(result.catalog.tools.length, names.length);
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "ctx_purge")?.primaryGroup, "context-administration");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "web_fetch")?.primaryGroup, "web-search");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "memory_search")?.primaryGroup, "memory-recall");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "session_search")?.primaryGroup, "memory-recall");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "memory_add")?.primaryGroup, "memory-management");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "skill_manage")?.primaryGroup, "memory-management");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "Agent")?.primaryGroup, "subagents");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "get_subagent_result")?.primaryGroup, "subagents");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "steer_subagent")?.primaryGroup, "subagents");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "ssh_find")?.primaryGroup, "ssh-remote-search");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "ssh_grep")?.primaryGroup, "ssh-remote-search");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "ask_user_question")?.primaryGroup, "user-interaction");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "chrome_snapshot")?.primaryGroup, "chrome-navigation");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "chrome_click")?.primaryGroup, "chrome-interaction");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "chrome_list_network_requests")?.primaryGroup, "chrome-debugging");
|
|
for (const query of ["read", "read banana", "search", "web search", "查找", "运行", "查找香蕉"]) {
|
|
assert.equal(selectConfidentGroup(rankGroups(result.catalog, query), query), undefined);
|
|
}
|
|
assert.equal(selectConfidentGroup(rankGroups(result.catalog, "ssh_read"), "ssh_read")?.id, "ssh-remote-files");
|
|
assert.equal(
|
|
selectConfidentGroup(rankGroups(result.catalog, "remote find grep"), "remote find grep")?.id,
|
|
"ssh-remote-search",
|
|
);
|
|
});
|
|
|
|
test("user overrides take priority and unknown tools retain deterministic fallback groups", () => {
|
|
const tools = [tool("web_search"), tool("custom_analyze")];
|
|
const configured = { maxToolsPerGroup: 8, groupOverrides: { preferred: ["web_search"] } };
|
|
const hash = buildManifestHash(tools, configured);
|
|
const result = createBundleCatalog(tools, hash, configured);
|
|
assert.deepEqual(result.unknownTools.map((item) => item.name), ["custom_analyze"]);
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "web_search")?.primaryGroup, "preferred");
|
|
assert.equal(result.catalog.tools.find((card) => card.name === "custom_analyze")?.primaryGroup, "custom");
|
|
});
|