mirror of
https://github.com/openclaw/openclaw.git
synced 2026-04-19 13:11:40 +00:00
289 lines
10 KiB
TypeScript
289 lines
10 KiB
TypeScript
import { beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
|
|
import type { createOpenClawCodingTools } from "./pi-tools.js";
|
|
import type { AnyAgentTool } from "./tools/common.js";
|
|
|
|
function mockTool(params: {
|
|
name: string;
|
|
label: string;
|
|
description: string;
|
|
displaySummary?: string;
|
|
}): AnyAgentTool {
|
|
return {
|
|
...params,
|
|
parameters: { type: "object", properties: {} },
|
|
execute: async () => ({ text: params.description }),
|
|
} as unknown as AnyAgentTool;
|
|
}
|
|
|
|
const effectiveInventoryState = vi.hoisted(() => ({
|
|
tools: [
|
|
mockTool({ name: "exec", label: "Exec", description: "Run shell commands" }),
|
|
mockTool({ name: "docs_lookup", label: "Docs Lookup", description: "Search docs" }),
|
|
] as AnyAgentTool[],
|
|
pluginMeta: {} as Record<string, { pluginId: string } | undefined>,
|
|
channelMeta: {} as Record<string, { channelId: string } | undefined>,
|
|
effectivePolicy: {} as { profile?: string; providerProfile?: string },
|
|
resolvedModelCompat: undefined as Record<string, unknown> | undefined,
|
|
createToolsMock: vi.fn<typeof createOpenClawCodingTools>(
|
|
(_options) =>
|
|
[
|
|
mockTool({ name: "exec", label: "Exec", description: "Run shell commands" }),
|
|
mockTool({ name: "docs_lookup", label: "Docs Lookup", description: "Search docs" }),
|
|
] as AnyAgentTool[],
|
|
),
|
|
}));
|
|
|
|
vi.mock("./agent-scope.js", async () => {
|
|
const actual = await vi.importActual<typeof import("./agent-scope.js")>("./agent-scope.js");
|
|
return {
|
|
...actual,
|
|
resolveSessionAgentId: () => "main",
|
|
resolveAgentWorkspaceDir: () => "/tmp/workspace-main",
|
|
resolveAgentDir: () => "/tmp/agents/main/agent",
|
|
};
|
|
});
|
|
|
|
vi.mock("./pi-tools.js", () => ({
|
|
createOpenClawCodingTools: (options?: Parameters<typeof createOpenClawCodingTools>[0]) =>
|
|
effectiveInventoryState.createToolsMock(options),
|
|
}));
|
|
|
|
vi.mock("./pi-embedded-runner/model.js", () => ({
|
|
resolveModel: vi.fn(() => ({
|
|
model: effectiveInventoryState.resolvedModelCompat
|
|
? { compat: effectiveInventoryState.resolvedModelCompat }
|
|
: undefined,
|
|
authStorage: {} as never,
|
|
modelRegistry: {} as never,
|
|
})),
|
|
}));
|
|
|
|
vi.mock("../plugins/tools.js", () => ({
|
|
getPluginToolMeta: (tool: { name: string }) => effectiveInventoryState.pluginMeta[tool.name],
|
|
}));
|
|
|
|
vi.mock("./channel-tools.js", () => ({
|
|
getChannelAgentToolMeta: (tool: { name: string }) =>
|
|
effectiveInventoryState.channelMeta[tool.name],
|
|
}));
|
|
|
|
vi.mock("./pi-tools.policy.js", () => ({
|
|
resolveEffectiveToolPolicy: () => effectiveInventoryState.effectivePolicy,
|
|
}));
|
|
|
|
let resolveEffectiveToolInventory: typeof import("./tools-effective-inventory.js").resolveEffectiveToolInventory;
|
|
|
|
async function loadHarness(options?: {
|
|
tools?: AnyAgentTool[];
|
|
createToolsMock?: typeof effectiveInventoryState.createToolsMock;
|
|
pluginMeta?: Record<string, { pluginId: string } | undefined>;
|
|
channelMeta?: Record<string, { channelId: string } | undefined>;
|
|
effectivePolicy?: { profile?: string; providerProfile?: string };
|
|
resolvedModelCompat?: Record<string, unknown>;
|
|
}) {
|
|
effectiveInventoryState.tools = options?.tools ?? [
|
|
mockTool({ name: "exec", label: "Exec", description: "Run shell commands" }),
|
|
mockTool({ name: "docs_lookup", label: "Docs Lookup", description: "Search docs" }),
|
|
];
|
|
effectiveInventoryState.pluginMeta = options?.pluginMeta ?? {};
|
|
effectiveInventoryState.channelMeta = options?.channelMeta ?? {};
|
|
effectiveInventoryState.effectivePolicy = options?.effectivePolicy ?? {};
|
|
effectiveInventoryState.resolvedModelCompat = options?.resolvedModelCompat;
|
|
effectiveInventoryState.createToolsMock =
|
|
options?.createToolsMock ??
|
|
vi.fn<typeof createOpenClawCodingTools>((_options) => effectiveInventoryState.tools);
|
|
return {
|
|
resolveEffectiveToolInventory,
|
|
createToolsMock: effectiveInventoryState.createToolsMock,
|
|
};
|
|
}
|
|
|
|
describe("resolveEffectiveToolInventory", () => {
|
|
beforeAll(async () => {
|
|
({ resolveEffectiveToolInventory } = await import("./tools-effective-inventory.js"));
|
|
});
|
|
|
|
beforeEach(() => {
|
|
effectiveInventoryState.tools = [
|
|
mockTool({ name: "exec", label: "Exec", description: "Run shell commands" }),
|
|
mockTool({ name: "docs_lookup", label: "Docs Lookup", description: "Search docs" }),
|
|
];
|
|
effectiveInventoryState.pluginMeta = {};
|
|
effectiveInventoryState.channelMeta = {};
|
|
effectiveInventoryState.effectivePolicy = {};
|
|
effectiveInventoryState.resolvedModelCompat = undefined;
|
|
effectiveInventoryState.createToolsMock = vi.fn<typeof createOpenClawCodingTools>(
|
|
(_options) => effectiveInventoryState.tools,
|
|
);
|
|
});
|
|
|
|
it("groups core, plugin, and channel tools from the effective runtime set", async () => {
|
|
const { resolveEffectiveToolInventory } = await loadHarness({
|
|
tools: [
|
|
mockTool({ name: "exec", label: "Exec", description: "Run shell commands" }),
|
|
mockTool({ name: "docs_lookup", label: "Docs Lookup", description: "Search docs" }),
|
|
mockTool({
|
|
name: "message_actions",
|
|
label: "Message Actions",
|
|
description: "Act on messages",
|
|
}),
|
|
],
|
|
pluginMeta: { docs_lookup: { pluginId: "docs" } },
|
|
channelMeta: { message_actions: { channelId: "telegram" } },
|
|
});
|
|
|
|
const result = resolveEffectiveToolInventory({ cfg: {} });
|
|
|
|
expect(result).toEqual({
|
|
agentId: "main",
|
|
profile: "full",
|
|
groups: [
|
|
{
|
|
id: "core",
|
|
label: "Built-in tools",
|
|
source: "core",
|
|
tools: [
|
|
{
|
|
id: "exec",
|
|
label: "Exec",
|
|
description: "Run shell commands",
|
|
rawDescription: "Run shell commands",
|
|
source: "core",
|
|
},
|
|
],
|
|
},
|
|
{
|
|
id: "plugin",
|
|
label: "Connected tools",
|
|
source: "plugin",
|
|
tools: [
|
|
{
|
|
id: "docs_lookup",
|
|
label: "Docs Lookup",
|
|
description: "Search docs",
|
|
rawDescription: "Search docs",
|
|
source: "plugin",
|
|
pluginId: "docs",
|
|
},
|
|
],
|
|
},
|
|
{
|
|
id: "channel",
|
|
label: "Channel tools",
|
|
source: "channel",
|
|
tools: [
|
|
{
|
|
id: "message_actions",
|
|
label: "Message Actions",
|
|
description: "Act on messages",
|
|
rawDescription: "Act on messages",
|
|
source: "channel",
|
|
channelId: "telegram",
|
|
},
|
|
],
|
|
},
|
|
],
|
|
});
|
|
});
|
|
|
|
it("disambiguates duplicate labels with source ids", async () => {
|
|
const { resolveEffectiveToolInventory } = await loadHarness({
|
|
tools: [
|
|
mockTool({ name: "docs_lookup", label: "Lookup", description: "Search docs" }),
|
|
mockTool({ name: "jira_lookup", label: "Lookup", description: "Search Jira" }),
|
|
],
|
|
pluginMeta: {
|
|
docs_lookup: { pluginId: "docs" },
|
|
jira_lookup: { pluginId: "jira" },
|
|
},
|
|
});
|
|
|
|
const result = resolveEffectiveToolInventory({ cfg: {} });
|
|
const labels = result.groups.flatMap((group) => group.tools.map((tool) => tool.label));
|
|
|
|
expect(labels).toEqual(["Lookup (docs)", "Lookup (jira)"]);
|
|
});
|
|
|
|
it("prefers displaySummary over raw description", async () => {
|
|
const { resolveEffectiveToolInventory } = await loadHarness({
|
|
tools: [
|
|
mockTool({
|
|
name: "cron",
|
|
label: "Cron",
|
|
displaySummary: "Schedule and manage cron jobs.",
|
|
description: "Long raw description\n\nACTIONS:\n- status",
|
|
}),
|
|
],
|
|
});
|
|
|
|
const result = resolveEffectiveToolInventory({ cfg: {} });
|
|
|
|
expect(result.groups[0]?.tools[0]).toEqual({
|
|
id: "cron",
|
|
label: "Cron",
|
|
description: "Schedule and manage cron jobs.",
|
|
rawDescription: "Long raw description\n\nACTIONS:\n- status",
|
|
source: "core",
|
|
});
|
|
});
|
|
|
|
it("falls back to a sanitized summary for multi-line raw descriptions", async () => {
|
|
const { resolveEffectiveToolInventory } = await loadHarness({
|
|
tools: [
|
|
mockTool({
|
|
name: "cron",
|
|
label: "Cron",
|
|
description:
|
|
'Manage Gateway cron jobs (status/list/add/update/remove/run/runs) and send wake events. Use this for reminders, "check back later" requests, delayed follow-ups, and recurring tasks. Do not emulate scheduling with exec sleep or process polling.\n\nACTIONS:\n- status: Check cron scheduler status\nJOB SCHEMA:\n{ ... }',
|
|
}),
|
|
],
|
|
});
|
|
|
|
const result = resolveEffectiveToolInventory({ cfg: {} });
|
|
|
|
const description = result.groups[0]?.tools[0]?.description ?? "";
|
|
expect(description).toContain(
|
|
"Manage Gateway cron jobs (status/list/add/update/remove/run/runs) and send wake events.",
|
|
);
|
|
expect(description).toContain("Use this for reminders");
|
|
expect(description.endsWith("...")).toBe(true);
|
|
expect(description.length).toBeLessThanOrEqual(120);
|
|
expect(result.groups[0]?.tools[0]?.rawDescription).toContain("ACTIONS:");
|
|
});
|
|
|
|
it("includes the resolved tool profile", async () => {
|
|
const { resolveEffectiveToolInventory } = await loadHarness({
|
|
tools: [mockTool({ name: "exec", label: "Exec", description: "Run shell commands" })],
|
|
effectivePolicy: { profile: "minimal", providerProfile: "coding" },
|
|
});
|
|
|
|
const result = resolveEffectiveToolInventory({ cfg: {} });
|
|
|
|
expect(result.profile).toBe("coding");
|
|
});
|
|
|
|
it("passes resolved model compat into effective tool creation", async () => {
|
|
const createToolsMock = vi.fn<typeof createOpenClawCodingTools>(() => [
|
|
mockTool({ name: "exec", label: "Exec", description: "Run shell commands" }),
|
|
]);
|
|
const { resolveEffectiveToolInventory } = await loadHarness({
|
|
createToolsMock,
|
|
resolvedModelCompat: { supportsTools: true, supportsNativeWebSearch: true },
|
|
});
|
|
|
|
resolveEffectiveToolInventory({
|
|
cfg: {},
|
|
agentDir: "/tmp/agents/main/agent",
|
|
modelProvider: "xai",
|
|
modelId: "grok-test",
|
|
});
|
|
|
|
expect(createToolsMock).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
allowGatewaySubagentBinding: true,
|
|
modelCompat: { supportsTools: true, supportsNativeWebSearch: true },
|
|
}),
|
|
);
|
|
});
|
|
});
|