58 lines
2.1 KiB
TypeScript
58 lines
2.1 KiB
TypeScript
import { describe, expect, test } from "bun:test"
|
|
import {
|
|
canonicalToolName,
|
|
finiteNumber,
|
|
primitiveInputSummary,
|
|
toolDisplayMetadata,
|
|
webSearchProviderLabel,
|
|
} from "../../src/util/tool-display"
|
|
|
|
test("normalizes shared tool primitives", () => {
|
|
expect(["bash", "task", "apply_patch", "plugin_tool"].map(canonicalToolName)).toEqual([
|
|
"shell",
|
|
"subagent",
|
|
"patch",
|
|
"plugin_tool",
|
|
])
|
|
expect([finiteNumber(-1.5), finiteNumber(Number.NaN), finiteNumber("1")]).toEqual([-1.5, undefined, undefined])
|
|
expect(primitiveInputSummary({ command: "pwd", count: 2, nested: {} })).toBe("[command=pwd, count=2]")
|
|
expect(primitiveInputSummary({ path: "src/a.ts", line: 2 }, ["path"])).toBe("[line=2]")
|
|
})
|
|
|
|
describe("webSearchProviderLabel", () => {
|
|
test("labels known providers", () => {
|
|
expect(webSearchProviderLabel("parallel")).toBe("Parallel Web Search")
|
|
expect(webSearchProviderLabel("exa")).toBe("Exa Web Search")
|
|
})
|
|
|
|
for (const [name, provider] of [
|
|
["undefined", undefined],
|
|
["null", null],
|
|
["an object", {}],
|
|
["an array", []],
|
|
["a number", 1],
|
|
["an unexpected string", "other"],
|
|
] as const) {
|
|
test(`uses the generic label for ${name}`, () => {
|
|
expect(webSearchProviderLabel(provider)).toBe("Web Search")
|
|
})
|
|
}
|
|
})
|
|
|
|
describe("toolDisplayMetadata", () => {
|
|
test("returns tool metadata for non-pending states", () => {
|
|
const metadata = { provider: "parallel", numResults: 3 }
|
|
|
|
expect(toolDisplayMetadata({ status: "running", metadata })).toBe(metadata)
|
|
expect(toolDisplayMetadata({ status: "completed", metadata })).toBe(metadata)
|
|
expect(toolDisplayMetadata({ status: "error", metadata })).toBe(metadata)
|
|
})
|
|
|
|
test("does not expose pending or malformed metadata", () => {
|
|
expect(toolDisplayMetadata({ status: "streaming", metadata: { provider: "exa" } })).toEqual({})
|
|
expect(toolDisplayMetadata({ status: "completed" })).toEqual({})
|
|
expect(toolDisplayMetadata({ status: "completed", metadata: null })).toEqual({})
|
|
expect(toolDisplayMetadata({ status: "completed", metadata: [] })).toEqual({})
|
|
expect(toolDisplayMetadata(undefined)).toEqual({})
|
|
})
|
|
})
|