File size: 1,612 Bytes
0dbc9de | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | import { describe, expect, test } from "bun:test"
import { toolDisplayMetadata, webSearchProviderLabel } from "../../src/util/tool-display"
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 structured metadata for non-pending states", () => {
const structured = { provider: "parallel", numResults: 3 }
expect(toolDisplayMetadata({ status: "running", structured })).toBe(structured)
expect(toolDisplayMetadata({ status: "completed", structured })).toBe(structured)
expect(toolDisplayMetadata({ status: "error", structured })).toBe(structured)
})
test("does not expose pending or malformed metadata", () => {
expect(toolDisplayMetadata({ status: "pending", structured: { provider: "exa" } })).toEqual({})
expect(toolDisplayMetadata({ status: "completed" })).toEqual({})
expect(toolDisplayMetadata({ status: "completed", structured: null })).toEqual({})
expect(toolDisplayMetadata({ status: "completed", structured: [] })).toEqual({})
expect(toolDisplayMetadata(undefined)).toEqual({})
})
})
|