Something went wrong. Try again.
opencode v2 mirror
Something went wrong. Try again.
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115import { describe, expect, test } from "bun:test"import { Message, ToolResultPart, Media } from "@opencode/ai"import { boundImages, unsupportedParts } from "@opencode/core/session/model-request"
const capabilities = (input: string[]) => ({ tools: true, input, output: ["text"] })
describe("SessionModelRequest.unsupportedParts", () => { test("replaces unsupported user media with a visible error", () => { const messages = unsupportedParts( [ Message.user([ Message.text("Describe these files"), { type: "media", media: Media.base64("aGVsbG8=", "image/png"), filename: "logo.png" }, { type: "media", media: Media.base64("JVBERg==", "application/pdf"), filename: "document.pdf" }, ]), ], capabilities(["text"]), )
expect(messages[0]?.content).toEqual([ Message.text("Describe these files"), Message.text('ERROR: Cannot read "logo.png" (this model does not support image input). Inform the user.'), Message.text('ERROR: Cannot read "document.pdf" (this model does not support pdf input). Inform the user.'), ]) })
test("replaces unsupported media nested in tool results", () => { const messages = unsupportedParts( [ Message.tool( ToolResultPart.make({ id: "call_1", name: "read", result: { type: "content", value: [ { type: "text", text: "Image read successfully" }, { type: "file", uri: "data:image/png;base64,aGVsbG8=", mime: "image/png", name: "logo.png" }, ], }, }), ), ], capabilities(["text"]), )
expect(messages[0]?.content[0]).toMatchObject({ type: "tool-result", result: { type: "content", value: [ { type: "text", text: "Image read successfully" }, { type: "text", text: 'ERROR: Cannot read "logo.png" (this model does not support image input). Inform the user.', }, ], }, }) })
test("preserves supported media", () => { const message = Message.user({ type: "media", media: Media.base64("aGVsbG8=", "image/png") }) expect(unsupportedParts([message], capabilities(["text", "image"]))[0]?.content).toEqual(message.content) })})
describe("SessionModelRequest.boundImages", () => { test("preserves images below the trigger", () => { const messages = [Message.user({ type: "media", media: Media.base64("aGVsbG8=", "image/png") })] expect(boundImages(messages)).toBe(messages) })
test("replaces oldest images until the retained payload reaches the target", () => { const image = "a".repeat(9 * 1024 * 1024) const messages = [ Message.user({ type: "media", media: Media.base64(image, "image/png"), filename: "first.png" }), Message.user({ type: "media", media: Media.base64(image, "image/png"), filename: "second.png" }), Message.user({ type: "media", media: Media.base64(image, "image/png"), filename: "third.png" }), ] const result = boundImages(messages)
expect(result[0]?.content[0]).toMatchObject({ type: "text" }) expect(result[1]?.content[0]).toMatchObject({ type: "text" }) expect(result[2]?.content[0]).toMatchObject({ type: "media", filename: "third.png" }) })
test("replaces images nested in tool results", () => { const image = "a".repeat(13 * 1024 * 1024) const result = boundImages([ Message.tool( ToolResultPart.make({ id: "call_1", name: "read", result: { type: "content", value: [ { type: "file", uri: `data:image/png;base64,${image}`, mime: "image/png", name: "first.png" }, { type: "file", uri: `data:image/png;base64,${image}`, mime: "image/png", name: "second.png" }, ], }, }), ), ])
expect(result[0]?.content[0]).toMatchObject({ type: "tool-result", result: { type: "content", value: [{ type: "text" }, { type: "file", name: "second.png" }], }, }) })})