assistant-ui--assistant-ui/packages/react-ag-ui/test/conversions.spec.ts

"use client";

import { describe, it, expect } from "vitest";
import { z } from "zod";
import { UserMessageSchema } from "@ag-ui/client";
import { ExportedMessageRepository } from "@assistant-ui/core";
import { fromAgUiMessages as publicFromAgUiMessages } from "../src";
import {
  fromAgUiMessages,
  toAgUiMessages,
  toAgUiTools,
} from "../src/runtime/adapter/conversions";

describe("adapter conversions", () => {
  it("converts thread messages to AG-UI format", () => {
    const result = toAgUiMessages([
      {
        id: "1",
        role: "user",
        content: [{ type: "text", text: "Hello" }],
      },
      {
        id: "2",
        role: "assistant",
        content: [
          { type: "text", text: "Hi" },
          {
            type: "tool-call",
            toolCallId: "call-1",
            toolName: "search",
            argsText: '{"query":"x"}',
          },
        ],
      },
    ] as any);

    expect(result).toHaveLength(2);
    expect(result[0]).toMatchObject({ role: "user", content: "Hello" });
    const toolCall = (result[1] as any).toolCalls?.[0];
    expect(toolCall).toMatchObject({
      id: "call-1",
      function: { name: "search", arguments: '{"query":"x"}' },
    });
  });

  it("marks errored tool call results with error content", () => {
    const result = toAgUiMessages([
      {
        id: "assistant-1",
        role: "assistant",
        content: [
          {
            type: "tool-call",
            toolCallId: "call-99",
            toolName: "do-it",
            argsText: "{}",
            result: { error: "nope" },
            isError: true,
          },
        ],
      },
    ] as any);

    expect(result).toHaveLength(2);
    expect(result[1]).toMatchObject({
      role: "tool",
      toolCallId: "call-99",
      content: '{"error":"nope"}',
      error: '{"error":"nope"}',
    });
  });

  it("includes tool messages for completed tool calls", () => {
    const result = toAgUiMessages([
      {
        id: "assistant-1",
        role: "assistant",
        content: [
          { type: "text", text: "Working..." },
          {
            type: "tool-call",
            toolCallId: "call-42",
            toolName: "search",
            argsText: '{"query":"x"}',
            result: { ok: true },
          },
        ],
      },
    ] as any);

    expect(result).toHaveLength(2);
    expect(result[0]).toMatchObject({
      role: "assistant",
      content: "Working...",
      toolCalls: [
        {
          id: "call-42",
          function: { name: "search", arguments: '{"query":"x"}' },
        },
      ],
    });
    expect(result[1]).toMatchObject({
      role: "tool",
      toolCallId: "call-42",
      content: '{"ok":true}',
    });
  });

  it("merges tool role snapshot messages back into assistant tool-call parts", () => {
    const result = fromAgUiMessages([
      {
        id: "msg-1",
        role: "user",
        content: "What's the weather?",
      },
      {
        id: "msg-2",
        role: "assistant",
        content: "",
        tool_calls: [
          {
            id: "call-1",
            type: "function",
            function: {
              name: "get_weather",
              arguments: '{"city":"Paris"}',
            },
          },
        ],
      },
      {
        id: "msg-3",
        role: "tool",
        tool_call_id: "call-1",
        content: '{"temperature":"22C"}',
      },
    ] as any);

    expect(result).toHaveLength(2);
    const assistantMessage = result[1] as any;
    expect(assistantMessage.role).toBe("assistant");
    const toolPart = assistantMessage.content.find(
      (part: { type: string }) => part.type === "tool-call",
    );
    expect(toolPart).toMatchObject({
      toolCallId: "call-1",
      toolName: "get_weather",
      argsText: '{"city":"Paris"}',
      result: { temperature: "22C" },
    });
  });

  it("restores requires-action status for an assistant message with an unresolved tool call", () => {
    const result = fromAgUiMessages([
      {
        id: "msg-1",
        role: "assistant",
        content: "",
        tool_calls: [
          {
            id: "call-1",
            type: "function",
            function: { name: "ask_user", arguments: "{}" },
          },
        ],
      },
    ] as any);

    expect(result).toHaveLength(1);
    expect((result[0] as any).status).toMatchObject({
      type: "requires-action",
      reason: "tool-calls",
    });
  });

  it("leaves a resolved tool call without a requires-action status", () => {
    const result = fromAgUiMessages([
      {
        id: "msg-1",
        role: "assistant",
        content: "",
        tool_calls: [
          {
            id: "call-1",
            type: "function",
            function: { name: "get_weather", arguments: "{}" },
          },
        ],
      },
      {
        id: "msg-2",
        role: "tool",
        tool_call_id: "call-1",
        content: '{"temperature":"22C"}',
      },
    ] as any);

    expect(result).toHaveLength(1);
    expect((result[0] as any).status).toBeUndefined();
  });

  it("does not add status to an assistant message without tool calls", () => {
    const result = fromAgUiMessages([
      { id: "msg-1", role: "assistant", content: "done" },
    ] as any);

    expect(result).toHaveLength(1);
    expect((result[0] as any).status).toBeUndefined();
  });

  it("restores interrupt status and metadata from a persisted interrupts blob", () => {
    const interrupts = [
      { id: "int-1", reason: "confirmation", message: "Proceed?" },
    ];
    const result = fromAgUiMessages([
      {
        id: "msg-1",
        role: "assistant",
        content: "waiting",
        metadata: { custom: { agui: { interrupts } } },
      },
    ] as any);

    expect(result).toHaveLength(1);
    expect((result[0] as any).status).toMatchObject({
      type: "requires-action",
      reason: "interrupt",
    });
    expect((result[0] as any).metadata.custom.agui.interrupts).toEqual(
      interrupts,
    );
  });

  it("prefers interrupt status over tool-calls when both are present", () => {
    const result = fromAgUiMessages([
      {
        id: "msg-1",
        role: "assistant",
        content: "",
        tool_calls: [
          {
            id: "call-1",
            type: "function",
            function: { name: "ask_user", arguments: "{}" },
          },
        ],
        metadata: {
          custom: {
            agui: { interrupts: [{ id: "int-1", reason: "tool_call" }] },
          },
        },
      },
    ] as any);

    expect(result).toHaveLength(1);
    expect((result[0] as any).status).toMatchObject({
      type: "requires-action",
      reason: "interrupt",
    });
  });

  it("ignores a malformed interrupts blob without throwing", () => {
    const result = fromAgUiMessages([
      {
        id: "msg-1",
        role: "assistant",
        content: "done",
        metadata: {
          custom: {
            agui: {
              interrupts: [{ reason: "no-id" }, { id: "no-reason" }, 42],
            },
          },
        },
      },
    ] as any);

    expect(result).toHaveLength(1);
    expect((result[0] as any).status).toBeUndefined();
    expect(
      (result[0] as any).metadata?.custom?.agui?.interrupts,
    ).toBeUndefined();
  });

  it("creates a synthetic assistant tool-call when snapshot has an orphan tool message", () => {
    const result = fromAgUiMessages([
      {
        id: "tool-only",
        role: "tool",
        tool_call_id: "call-9",
        name: "lookup",
        content: '{"ok":true}',
      },
    ] as any);

    expect(result).toHaveLength(1);
    expect(result[0]).toMatchObject({
      role: "assistant",
      id: "tool-only:assistant",
    });
    const toolPart = (result[0] as any).content[0];
    expect(toolPart).toMatchObject({
      type: "tool-call",
      toolCallId: "call-9",
      toolName: "lookup",
      result: { ok: true },
    });
  });

  it("imports a reasoning snapshot message as a reasoning assistant part", () => {
    const result = fromAgUiMessages([
      {
        id: "reason-1",
        role: "reasoning",
        content: "Let me think about this step by step.",
      },
    ] as any);

    expect(result).toHaveLength(1);
    expect(result[0]).toMatchObject({
      id: "reason-1",
      role: "assistant",
      content: [
        { type: "reasoning", text: "Let me think about this step by step." },
      ],
    });
  });

  it("preserves cross-message order around reasoning messages", () => {
    const result = fromAgUiMessages([
      { id: "u-1", role: "user", content: "hi" },
      { id: "r-1", role: "reasoning", content: "thinking" },
      { id: "a-1", role: "assistant", content: "done" },
    ] as any);

    expect(result.map((m) => m.role)).toEqual([
      "user",
      "assistant",
      "assistant",
    ]);
    expect((result[1] as any).content).toEqual([
      { type: "reasoning", text: "thinking" },
    ]);
    expect((result[2] as any).content).toEqual([
      { type: "text", text: "done" },
    ]);
  });

  it("skips empty reasoning messages", () => {
    const result = fromAgUiMessages([
      { id: "r-1", role: "reasoning", content: "" },
    ] as any);

    expect(result).toHaveLength(0);
  });

  it("drops reasoning messages when showThinking is false", () => {
    const result = fromAgUiMessages(
      [
        { id: "u-1", role: "user", content: "hi" },
        { id: "r-1", role: "reasoning", content: "thinking" },
        { id: "a-1", role: "assistant", content: "done" },
      ] as any,
      { showThinking: false },
    );

    expect(result.map((m) => m.role)).toEqual(["user", "assistant"]);
    expect((result[1] as any).content).toEqual([
      { type: "text", text: "done" },
    ]);
  });

  it("drops activity messages (no assistant-part equivalent)", () => {
    const result = fromAgUiMessages([
      { id: "u-1", role: "user", content: "hi" },
      {
        id: "act-1",
        role: "activity",
        activityType: "search",
        content: { query: "weather" },
      },
    ] as any);

    expect(result).toHaveLength(1);
    expect(result[0]).toMatchObject({ role: "user" });
  });

  it("skips whitespace-only reasoning messages", () => {
    const result = fromAgUiMessages([
      { id: "r-1", role: "reasoning", content: "   " },
    ] as any);

    expect(result).toHaveLength(0);
  });

  it("does not re-emit an imported reasoning message as an empty assistant message", () => {
    const imported = fromAgUiMessages([
      { id: "u-1", role: "user", content: "hi" },
      { id: "r-1", role: "reasoning", content: "thinking" },
      { id: "a-1", role: "assistant", content: "done" },
    ] as any);

    const roundTripped = toAgUiMessages(imported);

    expect(roundTripped.map((m) => m.role)).toEqual(["user", "assistant"]);
    expect(
      roundTripped.filter((m) => m.role === "assistant" && m.content === ""),
    ).toHaveLength(0);
    expect(roundTripped[1]).toMatchObject({
      role: "assistant",
      content: "done",
    });
  });

  it("filters disabled/back-end tools", () => {
    const tools = toAgUiTools({
      search: { description: "Search", parameters: { type: "object" } },
      disabled: { disabled: true },
      backend: { type: "backend" },
    });

    expect(tools).toHaveLength(1);
    expect(tools[0]).toMatchObject({ name: "search" });
  });

  it("prefers available schema conversion helpers for tools", () => {
    const tools = toAgUiTools({
      jsonTool: { parameters: { toJSON: () => ({ type: "object" }) } },
      schemaTool: { parameters: { toJSONSchema: () => ({ type: "string" }) } },
      plain: { parameters: { type: "boolean" } },
    });

    expect(tools).toHaveLength(3);
    expect(tools).toEqual(
      expect.arrayContaining([
        {
          name: "jsonTool",
          description: undefined,
          parameters: { type: "object" },
        },
        {
          name: "schemaTool",
          description: undefined,
          parameters: { type: "string" },
        },
        {
          name: "plain",
          description: undefined,
          parameters: { type: "boolean" },
        },
      ]),
    );
  });

  it("preserves tool message ID through round-trip conversion", () => {
    const agUiMessages = [
      {
        id: "msg-1",
        role: "user",
        content: "What's the weather?",
      },
      {
        id: "msg-2",
        role: "assistant",
        content: "",
        tool_calls: [
          {
            id: "call-1",
            type: "function",
            function: {
              name: "get_weather",
              arguments: '{"city":"Paris"}',
            },
          },
        ],
      },
      {
        id: "tool-msg-original-id",
        role: "tool",
        tool_call_id: "call-1",
        content: '{"temperature":"22C"}',
      },
    ] as any;

    const threadMessages = fromAgUiMessages(agUiMessages);
    const roundTripped = toAgUiMessages(threadMessages);

    const toolMessage = roundTripped.find((m) => m.role === "tool");
    expect(toolMessage).toBeDefined();
    expect(toolMessage!.id).toBe("tool-msg-original-id");
  });

  it("converts Zod schemas to JSON Schema format", () => {
    const zodSchema = z.object({
      message: z.string().describe("Text to log to the console."),
    });

    const tools = toAgUiTools({
      console_log: {
        description: "Log a message to the console.",
        parameters: zodSchema,
      },
    });

    expect(tools).toHaveLength(1);
    expect(tools[0]).toMatchObject({
      name: "console_log",
      description: "Log a message to the console.",
    });
    // Verify parameters is a plain JSON Schema object, not a Zod instance
    expect(tools[0]!.parameters).toMatchObject({
      type: "object",
      properties: {
        message: {
          type: "string",
          description: "Text to log to the console.",
        },
      },
      required: ["message"],
    });
    // Ensure it's not a Zod instance (no Zod methods)
    expect(tools[0]!.parameters).not.toHaveProperty("parse");
    expect(tools[0]!.parameters).not.toHaveProperty("_def");
  });

  it("returns plain string content when user message has no attachments", () => {
    const result = toAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [{ type: "text", text: "Hello" }],
      },
    ] as any);

    expect(result).toHaveLength(1);
    expect(result[0]).toMatchObject({ role: "user", content: "Hello" });
    expect(() => UserMessageSchema.parse(result[0])).not.toThrow();
  });

  it("converts image attachment with data URL to AG-UI image source", () => {
    const result = toAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [{ type: "text", text: "What is this?" }],
        attachments: [
          {
            id: "a-1",
            type: "image",
            name: "photo.png",
            content: [
              {
                type: "image",
                image: "data:image/png;base64,iVBORw0KGgoAAAA=",
              },
            ],
          },
        ],
      },
    ] as any);

    expect(result).toHaveLength(1);
    expect(result[0]).toMatchObject({
      role: "user",
      content: [
        { type: "text", text: "What is this?" },
        {
          type: "image",
          source: {
            type: "data",
            value: "iVBORw0KGgoAAAA=",
            mimeType: "image/png",
          },
        },
      ],
    });
    expect(() => UserMessageSchema.parse(result[0])).not.toThrow();
  });

  it("converts image attachment with http URL to AG-UI image url source", () => {
    const result = toAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [{ type: "text", text: "hi" }],
        attachments: [
          {
            id: "a-1",
            type: "image",
            name: "remote.jpg",
            contentType: "image/jpeg",
            content: [
              { type: "image", image: "https://example.com/remote.jpg" },
            ],
          },
        ],
      },
    ] as any);

    expect(result[0]).toMatchObject({
      role: "user",
      content: [
        { type: "text", text: "hi" },
        {
          type: "image",
          source: {
            type: "url",
            value: "https://example.com/remote.jpg",
            mimeType: "image/jpeg",
          },
        },
      ],
    });
    expect(() => UserMessageSchema.parse(result[0])).not.toThrow();
  });

  it("converts a file attachment to an AG-UI document part with filename in metadata", () => {
    const result = toAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [{ type: "text", text: "review this" }],
        attachments: [
          {
            id: "a-1",
            type: "document",
            name: "spec.pdf",
            contentType: "application/pdf",
            content: [
              {
                type: "file",
                data: "JVBERi0xLjQK",
                mimeType: "application/pdf",
                filename: "spec.pdf",
              },
            ],
          },
        ],
      },
    ] as any);

    expect(result[0]).toMatchObject({
      role: "user",
      content: [
        { type: "text", text: "review this" },
        {
          type: "document",
          source: {
            type: "data",
            value: "JVBERi0xLjQK",
            mimeType: "application/pdf",
          },
          metadata: { filename: "spec.pdf" },
        },
      ],
    });
    expect((result[0] as any).content[1]).not.toHaveProperty("filename");
    expect(() => UserMessageSchema.parse(result[0])).not.toThrow();
  });

  it("classifies audio/video files into their own multimodal parts", () => {
    const result = toAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [],
        attachments: [
          {
            id: "a-1",
            type: "file",
            name: "clip.mp3",
            content: [
              {
                type: "file",
                data: "data:audio/mpeg;base64,QUJD",
                mimeType: "audio/mpeg",
              },
            ],
          },
          {
            id: "a-2",
            type: "file",
            name: "clip.mp4",
            content: [
              {
                type: "file",
                data: "ZmFrZQ==",
                mimeType: "video/mp4",
              },
            ],
          },
        ],
      },
    ] as any);

    expect(result[0]).toMatchObject({
      role: "user",
      content: [
        {
          type: "audio",
          source: { type: "data", value: "QUJD", mimeType: "audio/mpeg" },
        },
        {
          type: "video",
          source: { type: "data", value: "ZmFrZQ==", mimeType: "video/mp4" },
        },
      ],
    });
    expect(() => UserMessageSchema.parse(result[0])).not.toThrow();
  });

  it("converts an http(s) file attachment to a url source document part", () => {
    const result = toAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [],
        attachments: [
          {
            id: "a-1",
            type: "document",
            name: "spec.pdf",
            contentType: "application/pdf",
            content: [
              {
                type: "file",
                data: "https://example.com/spec.pdf",
                mimeType: "application/pdf",
                filename: "spec.pdf",
              },
            ],
          },
        ],
      },
    ] as any);

    expect(result[0]).toMatchObject({
      role: "user",
      content: [
        {
          type: "document",
          source: {
            type: "url",
            value: "https://example.com/spec.pdf",
            mimeType: "application/pdf",
          },
          metadata: { filename: "spec.pdf" },
        },
      ],
    });
    expect(() => UserMessageSchema.parse(result[0])).not.toThrow();
  });

  it("handles user message with only attachment and empty text", () => {
    const result = toAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [],
        attachments: [
          {
            id: "a-1",
            type: "image",
            name: "pic.png",
            content: [
              {
                type: "image",
                image: "data:image/png;base64,AAAA",
              },
            ],
          },
        ],
      },
    ] as any);

    expect(result[0]).toMatchObject({
      role: "user",
      content: [
        {
          type: "image",
          source: {
            type: "data",
            value: "AAAA",
            mimeType: "image/png",
          },
        },
      ],
    });
    expect(() => UserMessageSchema.parse(result[0])).not.toThrow();
  });

  it("preserves string-form content when non-text attachments are present", () => {
    const result = toAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: "What is in this image?",
        attachments: [
          {
            id: "a-1",
            type: "image",
            name: "photo.png",
            content: [
              {
                type: "image",
                image: "data:image/png;base64,AAAA",
              },
            ],
          },
        ],
      },
    ] as any);

    expect(result[0]).toMatchObject({
      role: "user",
      content: [
        { type: "text", text: "What is in this image?" },
        {
          type: "image",
          source: {
            type: "data",
            value: "AAAA",
            mimeType: "image/png",
          },
        },
      ],
    });
    expect(() => UserMessageSchema.parse(result[0])).not.toThrow();
  });

  it("preserves text parts from attachments in the string fallback", () => {
    const result = toAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [{ type: "text", text: "hi" }],
        attachments: [
          {
            id: "a-1",
            type: "document",
            name: "notes.txt",
            content: [
              {
                type: "text",
                text: "<attachment name=notes.txt>extracted content</attachment>",
              },
            ],
          },
        ],
      },
    ] as any);

    expect(result[0]!.content).toBe(
      "hi\n<attachment name=notes.txt>extracted content</attachment>",
    );
    expect(() => UserMessageSchema.parse(result[0])).not.toThrow();
  });

  it("parses data URL with charset parameter (e.g. SVG)", () => {
    const result = toAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [],
        attachments: [
          {
            id: "a-1",
            type: "image",
            name: "icon.svg",
            content: [
              {
                type: "image",
                image:
                  "data:image/svg+xml;charset=utf-8;base64,PHN2ZyB4bWxucz0=",
              },
            ],
          },
        ],
      },
    ] as any);

    expect(result[0]).toMatchObject({
      role: "user",
      content: [
        {
          type: "image",
          source: {
            type: "data",
            value: "PHN2ZyB4bWxucz0=",
            mimeType: "image/svg+xml",
          },
        },
      ],
    });
    expect(() => UserMessageSchema.parse(result[0])).not.toThrow();
  });

  it("routes http file data to a url source, not an inline data source", () => {
    const result = toAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [],
        attachments: [
          {
            id: "a-1",
            type: "file",
            name: "report.pdf",
            content: [
              {
                type: "file",
                data: "https://cdn.example.com/report.pdf",
                mimeType: "application/pdf",
                filename: "report.pdf",
              },
            ],
          },
        ],
      },
    ] as any);

    expect(result[0]).toMatchObject({
      role: "user",
      content: [
        {
          type: "document",
          source: {
            type: "url",
            value: "https://cdn.example.com/report.pdf",
            mimeType: "application/pdf",
          },
          metadata: { filename: "report.pdf" },
        },
      ],
    });
    expect((result[0] as any).content[0].source).not.toHaveProperty("data");
    expect(() => UserMessageSchema.parse(result[0])).not.toThrow();
  });

  it("restores a document input part as a user attachment", () => {
    const result = fromAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [
          { type: "text", text: "review this" },
          {
            type: "document",
            source: {
              type: "url",
              value: "https://example.com/spec.pdf",
              mimeType: "application/pdf",
            },
            metadata: { filename: "spec.pdf" },
          },
        ],
      },
    ]);

    expect(result[0]).toMatchObject({
      role: "user",
      content: "review this",
      attachments: [
        {
          type: "document",
          name: "spec.pdf",
          contentType: "application/pdf",
          status: { type: "complete" },
          content: [
            {
              type: "file",
              data: "https://example.com/spec.pdf",
              mimeType: "application/pdf",
              filename: "spec.pdf",
            },
          ],
        },
      ],
    });
  });

  it("restores an image input part with a data source as an image attachment", () => {
    const result = fromAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [
          {
            type: "image",
            source: {
              type: "data",
              value: "iVBORw0KGgo=",
              mimeType: "image/png",
            },
          },
        ],
      },
    ]);

    expect(result[0]).toMatchObject({
      role: "user",
      content: "",
      attachments: [
        {
          type: "image",
          name: "image",
          contentType: "image/png",
          status: { type: "complete" },
          content: [
            { type: "image", image: "data:image/png;base64,iVBORw0KGgo=" },
          ],
        },
      ],
    });
  });

  it("restores audio and video input parts as file attachments", () => {
    const result = fromAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [
          {
            type: "audio",
            source: { type: "data", value: "QUJD", mimeType: "audio/mpeg" },
            metadata: { filename: "memo.mp3" },
          },
          {
            type: "video",
            source: {
              type: "url",
              value: "https://example.com/clip.mp4",
              mimeType: "video/mp4",
            },
          },
        ],
      },
    ]);

    expect(result[0]!.attachments).toMatchObject([
      {
        type: "file",
        name: "memo.mp3",
        contentType: "audio/mpeg",
        content: [
          {
            type: "file",
            data: "data:audio/mpeg;base64,QUJD",
            mimeType: "audio/mpeg",
            filename: "memo.mp3",
          },
        ],
      },
      {
        type: "file",
        name: "file",
        contentType: "video/mp4",
        content: [
          {
            type: "file",
            data: "https://example.com/clip.mp4",
            mimeType: "video/mp4",
          },
        ],
      },
    ]);
  });

  it("does not add attachments to text-only user messages", () => {
    const result = fromAgUiMessages([
      { id: "u-1", role: "user", content: "Hi" },
      {
        id: "u-2",
        role: "user",
        content: [{ type: "text", text: "Hello" }],
      },
    ]);

    expect(result[0]).not.toHaveProperty("attachments");
    expect(result[1]).not.toHaveProperty("attachments");
  });

  it("round-trips multimodal user input through fromAgUiMessages and back", () => {
    const original = {
      id: "u-1",
      role: "user",
      content: [
        { type: "text", text: "see attached" },
        {
          type: "document",
          source: {
            type: "data",
            value: "JVBERi0xLjQK",
            mimeType: "application/pdf",
          },
          metadata: { filename: "spec.pdf" },
        },
      ],
    };

    const restored = fromAgUiMessages([original]);
    const roundTripped = toAgUiMessages(
      restored as Parameters<typeof toAgUiMessages>[0],
    );

    expect(roundTripped[0]).toMatchObject({
      role: "user",
      content: [
        { type: "text", text: "see attached" },
        {
          type: "document",
          source: {
            type: "data",
            value: "JVBERi0xLjQK",
            mimeType: "application/pdf",
          },
          metadata: { filename: "spec.pdf" },
        },
      ],
    });
    expect(() => UserMessageSchema.parse(roundTripped[0])).not.toThrow();
  });

  it("restores a legacy binary input part and re-sends it as a modern part", () => {
    const restored = fromAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [
          { type: "text", text: "old history" },
          {
            type: "binary",
            mimeType: "application/pdf",
            data: "JVBERi0xLjQK",
            filename: "legacy.pdf",
          },
        ],
      },
    ]);

    expect(restored[0]).toMatchObject({
      role: "user",
      content: "old history",
      attachments: [
        {
          type: "document",
          name: "legacy.pdf",
          contentType: "application/pdf",
          status: { type: "complete" },
          content: [
            {
              type: "file",
              data: "data:application/pdf;base64,JVBERi0xLjQK",
              mimeType: "application/pdf",
              filename: "legacy.pdf",
            },
          ],
        },
      ],
    });

    const roundTripped = toAgUiMessages(
      restored as Parameters<typeof toAgUiMessages>[0],
    );
    expect(roundTripped[0]).toMatchObject({
      role: "user",
      content: [
        { type: "text", text: "old history" },
        {
          type: "document",
          source: {
            type: "data",
            value: "JVBERi0xLjQK",
            mimeType: "application/pdf",
          },
          metadata: { filename: "legacy.pdf" },
        },
      ],
    });
    expect(() => UserMessageSchema.parse(roundTripped[0])).not.toThrow();
  });

  it("restores a legacy binary image part with a url as an image attachment", () => {
    const result = fromAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [
          {
            type: "binary",
            mimeType: "image/png",
            url: "https://example.com/cat.png",
          },
        ],
      },
    ]);

    expect(result[0]).toMatchObject({
      role: "user",
      content: "",
      attachments: [
        {
          type: "image",
          name: "image",
          contentType: "image/png",
          status: { type: "complete" },
          content: [{ type: "image", image: "https://example.com/cat.png" }],
        },
      ],
    });
  });

  it("prefers data over url in legacy binary parts and routes them by mime type", () => {
    const result = fromAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [
          {
            type: "binary",
            mimeType: "audio/mpeg",
            data: "QUJD",
            url: "https://example.com/memo.mp3",
          },
          {
            type: "binary",
            mimeType: "video/mp4",
            data: "",
            url: "https://example.com/clip.mp4",
          },
        ],
      },
    ]);

    expect(result[0]!.attachments).toMatchObject([
      {
        type: "file",
        contentType: "audio/mpeg",
        content: [
          {
            type: "file",
            data: "data:audio/mpeg;base64,QUJD",
            mimeType: "audio/mpeg",
          },
        ],
      },
      {
        type: "file",
        contentType: "video/mp4",
        content: [
          {
            type: "file",
            data: "https://example.com/clip.mp4",
            mimeType: "video/mp4",
          },
        ],
      },
    ]);
  });

  it("skips legacy binary parts that only carry a file id", () => {
    const result = fromAgUiMessages([
      {
        id: "u-1",
        role: "user",
        content: [
          { type: "text", text: "see file" },
          { type: "binary", mimeType: "application/pdf", id: "file-1" },
        ],
      },
    ]);

    expect(result[0]).toMatchObject({ role: "user", content: "see file" });
    expect(result[0]).not.toHaveProperty("attachments");
  });
});

describe("package exports", () => {
  it("exposes fromAgUiMessages from the package root", () => {
    expect(publicFromAgUiMessages).toBe(fromAgUiMessages);
  });

  it("composes with ExportedMessageRepository.fromArray for history loading", () => {
    const repo = ExportedMessageRepository.fromArray(
      publicFromAgUiMessages([
        { id: "u-1", role: "user", content: "Hi" },
        { id: "a-1", role: "assistant", content: "Hello!" },
      ]),
    );

    expect(repo.messages).toHaveLength(2);
    expect(repo.messages[0]).toMatchObject({
      parentId: null,
      message: { role: "user" },
    });
    expect(repo.messages[1]!.parentId).toBe(repo.messages[0]!.message.id);
  });
});