alexion
6081e3e325
Imported as a linear vendor commit so forge rebases do not replay the upstream root tree. git-subtree-dir: third_party/pi git-subtree-split: 71023aec1546d6d677a65669c5665c0927ce0dd0
40 lines
1.9 KiB
TypeScript
40 lines
1.9 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { type ContentBlock, toLlmContent } from "../src/index.ts";
|
|
|
|
describe("toLlmContent", () => {
|
|
it("passes text and images through and replaces other blocks with text", () => {
|
|
const blocks: ContentBlock[] = [
|
|
{ type: "text", text: "hello", annotations: { priority: 1 } },
|
|
{ type: "image", data: "aW1n", mimeType: "image/png", _meta: { x: 1 } },
|
|
{ type: "audio", data: "YXVk", mimeType: "audio/wav" },
|
|
{ type: "resource_link", uri: "file:///a.txt", name: "a.txt" },
|
|
{ type: "resource", resource: { uri: "file:///b.txt", text: "inline" } },
|
|
{ type: "resource", resource: { uri: "file:///c.png", mimeType: "image/png", blob: "Yw==" } },
|
|
{ type: "resource", resource: { uri: "file:///d.bin", blob: "ZA==" } },
|
|
];
|
|
expect(toLlmContent({ content: blocks })).toEqual([
|
|
{ type: "text", text: "hello" },
|
|
{ type: "image", data: "aW1n", mimeType: "image/png" },
|
|
{ type: "text", text: "[audio audio/wav omitted]" },
|
|
{ type: "text", text: "a.txt: file:///a.txt" },
|
|
{ type: "text", text: "inline" },
|
|
{ type: "image", data: "Yw==", mimeType: "image/png" },
|
|
{ type: "text", text: "[binary resource file:///d.bin (unknown type) omitted]" },
|
|
]);
|
|
});
|
|
|
|
it("falls back to the structured content as JSON when there are no blocks", () => {
|
|
// #10416: structured content may be any JSON value.
|
|
expect(toLlmContent({ content: [], structuredContent: [1, 2] })).toEqual([
|
|
{ type: "text", text: "[\n 1,\n 2\n]" },
|
|
]);
|
|
expect(toLlmContent({ content: [], structuredContent: null })).toEqual([{ type: "text", text: "null" }]);
|
|
expect(toLlmContent({ content: [], structuredContent: { n: 1 } })).toEqual([
|
|
{ type: "text", text: '{\n "n": 1\n}' },
|
|
]);
|
|
expect(toLlmContent({ content: [{ type: "text", text: "n=1" }], structuredContent: { n: 1 } })).toEqual([
|
|
{ type: "text", text: "n=1" },
|
|
]);
|
|
});
|
|
});
|