8b7dd10a8a
- Added `jsonSchemaToTypeScript` and `renderToolInventory` to generate tool blocks with TypeScript signatures. - Added `examples` and `TSchema` fields to dump-tool metadata and passed them through prompt rendering. - Changed Harmony invocation rendering to omit `<|constrain|>json` markers in tool call payloads. - Added compact native tool list-mode inventory rendering with full `# Tool:` output elsewhere.
81 lines
2.5 KiB
TypeScript
81 lines
2.5 KiB
TypeScript
/**
|
|
* Contract: /dump renders the tool catalog through the shared AI inventory
|
|
* renderer — a simplified TypeScript signature (derived from the wire JSON
|
|
* Schema) plus each tool's examples in the model's native tool-call syntax.
|
|
*
|
|
* Tools carry live Zod v4 schemas; the dump must surface a readable signature
|
|
* (not the schema instance's internals) and must include examples, which the
|
|
* previous `<parameter>`-per-key JSON Schema dump dropped entirely.
|
|
*/
|
|
import { describe, expect, it } from "bun:test";
|
|
import { formatSessionDumpText } from "@oh-my-pi/pi-coding-agent/session/session-dump-format";
|
|
import { z } from "zod/v4";
|
|
|
|
describe("formatSessionDumpText tool parameters", () => {
|
|
it("renders Zod schemas as a TypeScript signature, not schema internals", () => {
|
|
const out = formatSessionDumpText({
|
|
messages: [],
|
|
tools: [
|
|
{
|
|
name: "web_search",
|
|
description: "Searches the web.",
|
|
parameters: z.object({
|
|
query: z.string().describe("search query"),
|
|
recency: z.enum(["day", "week"]).optional(),
|
|
}),
|
|
},
|
|
],
|
|
});
|
|
|
|
expect(out).toContain("# Tool: web_search");
|
|
expect(out).toContain("Parameters: {");
|
|
expect(out).toContain("/** search query */");
|
|
expect(out).toContain("query: string;");
|
|
expect(out).toContain('recency?: "day" | "week";');
|
|
// Live Zod instance internals must never leak into the dump.
|
|
expect(out).not.toContain("_zod");
|
|
expect(out).not.toContain("ZodObject");
|
|
// Tool params are no longer emitted as XML <parameter> elements.
|
|
expect(out).not.toContain('<parameter name="type">');
|
|
});
|
|
|
|
it("passes plain JSON-Schema parameters through to a TypeScript signature", () => {
|
|
const out = formatSessionDumpText({
|
|
messages: [],
|
|
tools: [
|
|
{
|
|
name: "legacy",
|
|
description: "Legacy tool.",
|
|
parameters: {
|
|
type: "object",
|
|
properties: { path: { type: "string", description: "a path" } },
|
|
required: ["path"],
|
|
},
|
|
},
|
|
],
|
|
});
|
|
|
|
expect(out).toContain("# Tool: legacy");
|
|
expect(out).toContain("/** a path */");
|
|
expect(out).toContain("path: string;");
|
|
});
|
|
|
|
it("includes tool examples in the model's native syntax", () => {
|
|
const out = formatSessionDumpText({
|
|
messages: [],
|
|
tools: [
|
|
{
|
|
name: "find",
|
|
description: "Finds files.",
|
|
parameters: z.object({ paths: z.array(z.string()) }),
|
|
examples: [{ call: { paths: ["src/**/*.ts"] } }],
|
|
},
|
|
],
|
|
});
|
|
|
|
expect(out).toContain("## Available Tools");
|
|
expect(out).toContain("<examples>");
|
|
expect(out).toContain('<invoke name="find">');
|
|
});
|
|
});
|