@agent-native/toolkit 0.225.2-nightly-20261106152255 → 0.226.0

package/src/app/chat/prepare-prompt-composer-agent-handoff.spec.ts

@@ -0,0 +1,162 @@
+// @vitest-environment happy-dom
+import { describe, expect, it, vi } from "vitest";
+
+import { preparePromptComposerAgentHandoff } from "./prepare-prompt-composer-agent-handoff.js";
+
+describe("prompt composer agent handoff", () => {
+  it("counts serialized metadata only once near the body budget", async () => {
+    const name = '"\\é'.repeat(450_000);
+    const result = await preparePromptComposerAgentHandoff(
+      [],
+      {
+        attachments: [
+          {
+            name,
+            content: [{ type: "file", url: "https://example.test/file" }],
+          },
+        ],
+      },
+      "Upload failed",
+    );
+    expect(result.attachments?.[0]?.name).toBe(name);
+  });
+  it.each([[3 * 1024 * 1024], [2 * 1024 * 1024, 2 * 1024 * 1024]])(
+    "rejects oversized raw files before reading any",
+    async (...sizes) => {
+      const files = sizes.map((size, index) => {
+        const file = new File(["small"], `${index}.txt`);
+        Object.defineProperty(file, "size", { value: size });
+        return file;
+      });
+      const read = vi.spyOn(FileReader.prototype, "readAsDataURL");
+      try {
+        await expect(
+          preparePromptComposerAgentHandoff(files, {}, "Upload failed"),
+        ).rejects.toMatchObject({
+          code: "upload_too_large",
+          cause: { code: "upload_too_large" },
+        });
+        expect(read).not.toHaveBeenCalled();
+      } finally {
+        read.mockRestore();
+      }
+    },
+  );
+  it("preserves raw file bytes once and durable attachment URLs", async () => {
+    const file = new File(["review content"], "review.txt", {
+      type: "text/plain",
+    });
+    const result = await preparePromptComposerAgentHandoff(
+      [file],
+      {
+        attachments: [
+          { file, name: file.name },
+          {
+            name: "saved.pdf",
+            content: [
+              {
+                type: "file",
+                url: "https://example.test/saved.pdf",
+                mimeType: "application/pdf",
+              },
+            ],
+          },
+        ],
+      },
+      "Localized upload failure",
+    );
+    expect(result.attachments).toEqual([
+      {
+        type: "file",
+        name: "saved.pdf",
+        url: "https://example.test/saved.pdf",
+        contentType: "application/pdf",
+      },
+      {
+        type: "file",
+        name: "review.txt",
+        contentType: "text/plain",
+        data: "data:text/plain;base64,cmV2aWV3IGNvbnRlbnQ=",
+      },
+    ]);
+  });
+  it("preserves orphan files and prefers uploaded content over retained raw files", async () => {
+    const file = new File(["context"], "context.txt", { type: "text/plain" });
+    const orphan = await preparePromptComposerAgentHandoff(
+      [],
+      { attachments: [{ file, name: file.name }] },
+      "Upload failed",
+    );
+    expect(orphan.attachments?.[0]).toMatchObject({
+      name: "context.txt",
+      data: "data:text/plain;base64,Y29udGV4dA==",
+    });
+    const uploaded = await preparePromptComposerAgentHandoff(
+      [file],
+      {
+        attachments: [
+          {
+            file,
+            name: file.name,
+            content: [
+              { type: "file", url: "https://example.test/context.txt" },
+            ],
+          },
+        ],
+      },
+      "Upload failed",
+    );
+    expect(uploaded.attachments).toEqual([
+      {
+        type: "file",
+        name: "context.txt",
+        url: "https://example.test/context.txt",
+        contentType: undefined,
+      },
+    ]);
+  });
+  it("preserves trusted HTTP storage references without reading the raw file", async () => {
+    const file = new File(["small"], "large.txt");
+    Object.defineProperty(file, "size", { value: 100 * 1024 * 1024 });
+    const read = vi.spyOn(FileReader.prototype, "readAsDataURL");
+    try {
+      const result = await preparePromptComposerAgentHandoff(
+        [file],
+        {
+          attachments: [
+            {
+              file,
+              name: file.name,
+              content: [
+                { type: "file", url: "http://127.0.0.1:5203/bucket/file.txt" },
+              ],
+            },
+          ],
+        },
+        "Upload failed",
+      );
+      expect(result.attachments?.[0]?.url).toBe(
+        "http://127.0.0.1:5203/bucket/file.txt",
+      );
+      expect(read).not.toHaveBeenCalled();
+    } finally {
+      read.mockRestore();
+    }
+  });
+  it("rejects unreadable descriptors rather than submitting without selected context", async () => {
+    await expect(
+      preparePromptComposerAgentHandoff(
+        [],
+        {
+          attachments: [
+            {
+              name: "missing",
+              content: [{ type: "file", url: "blob:unavailable" }],
+            },
+          ],
+        },
+        "Localized upload failure",
+      ),
+    ).rejects.toThrow("Localized upload failure");
+  });
+});

package/src/app/chat/prepare-prompt-composer-agent-handoff.ts

@@ -0,0 +1,134 @@
+import { isPersistableAttachmentReferenceUrl } from "@agent-native/agentkit/protocol";
+import type { AgentChatMessage } from "@agent-native/core/client/agent-chat";
+
+import { MAX_ESTIMATED_BODY_BYTES } from "../../composer/attachment-accept.js";
+import { isPastedTextAttachmentName } from "../../composer/pasted-text.js";
+
+function readFile(file: File): Promise<string> {
+  return new Promise((resolve, reject) => {
+    const reader = new FileReader();
+    reader.onload = () =>
+      typeof reader.result === "string"
+        ? resolve(reader.result)
+        : reject(new TypeError("Invalid file reader result"));
+    reader.onerror = () =>
+      reject(reader.error ?? new TypeError("File read failed"));
+    reader.readAsDataURL(file);
+  });
+}
+
+async function prepareHandoff(
+  files: readonly File[],
+  options: { attachments?: readonly unknown[] },
+): Promise<Pick<AgentChatMessage, "attachments">> {
+  const attachments: NonNullable<AgentChatMessage["attachments"]> = [];
+  const pending = new Set(files);
+  for (const value of options.attachments ?? []) {
+    if (!value || typeof value !== "object")
+      throw new TypeError("Invalid composer attachment");
+    if ("file" in value && value.file instanceof File) {
+      const hasDurableContent =
+        "content" in value &&
+        Array.isArray(value.content) &&
+        value.content.some(
+          (part: unknown) =>
+            part &&
+            typeof part === "object" &&
+            (("url" in part && isPersistableAttachmentReferenceUrl(part.url)) ||
+              ("image" in part &&
+                isPersistableAttachmentReferenceUrl(part.image))),
+        );
+      if (hasDurableContent) pending.delete(value.file);
+      else {
+        if (
+          !isPastedTextAttachmentName(value.file.name) ||
+          pending.has(value.file)
+        )
+          pending.add(value.file);
+        continue;
+      }
+    }
+    if (
+      !("name" in value) ||
+      typeof value.name !== "string" ||
+      !("content" in value) ||
+      !Array.isArray(value.content)
+    ) {
+      throw new TypeError("Invalid composer attachment");
+    }
+    for (const part of value.content) {
+      if (!part || typeof part !== "object" || !("type" in part))
+        throw new TypeError("Invalid attachment content");
+      if (
+        part.type === "text" &&
+        "text" in part &&
+        typeof part.text === "string"
+      ) {
+        attachments.push({ type: "file", name: value.name, text: part.text });
+      } else if (
+        part.type === "file" &&
+        "url" in part &&
+        isPersistableAttachmentReferenceUrl(part.url)
+      ) {
+        attachments.push({
+          type: "file",
+          name: value.name,
+          url: part.url,
+          contentType:
+            "mimeType" in part && typeof part.mimeType === "string"
+              ? part.mimeType
+              : undefined,
+        });
+      } else if (
+        part.type === "image" &&
+        "image" in part &&
+        typeof part.image === "string" &&
+        isPersistableAttachmentReferenceUrl(part.image)
+      ) {
+        attachments.push({ type: "image", name: value.name, url: part.image });
+      } else throw new TypeError("Unsupported attachment content");
+    }
+  }
+  const metadata = [
+    ...attachments,
+    ...Array.from(pending, (file) => ({ name: file.name, type: file.type })),
+  ];
+  const estimatedBytes =
+    new TextEncoder().encode(JSON.stringify(metadata)).byteLength * 1.15 +
+    Array.from(pending).reduce(
+      (sum, file) => sum + (4 * Math.ceil(file.size / 3) + 64) * 1.15,
+      0,
+    );
+  if (estimatedBytes > MAX_ESTIMATED_BODY_BYTES)
+    throw Object.assign(new Error("upload_too_large"), {
+      code: "upload_too_large",
+    });
+  for (const file of pending)
+    attachments.push({
+      type: file.type.startsWith("image/") ? "image" : "file",
+      name: file.name,
+      contentType: file.type || "application/octet-stream",
+      data: await readFile(file),
+    });
+  return attachments.length ? { attachments } : {};
+}
+
+export async function preparePromptComposerAgentHandoff(
+  files: readonly File[],
+  options: { attachments?: readonly unknown[] },
+  failureMessage: string,
+): Promise<Pick<AgentChatMessage, "attachments">> {
+  try {
+    return await prepareHandoff(files, options);
+  } catch (cause) {
+    throw Object.assign(new Error(failureMessage), {
+      cause,
+      ...(cause &&
+      typeof cause === "object" &&
+      "code" in cause &&
+      typeof cause.code === "string"
+        ? { code: cause.code }
+        : {}),
+    });
+  }
+}

package/src/app/i18n/catalogs/ar-SA.ts

@@ -497,7 +497,12 @@ const messages: ToolkitAgentChatTranslation = {
   "composer.useIntegrationTools": "استخدم أدوات {{integration}}",
   "composer.menu.scheduleTask": "جدولة مهمة",
   "composer.menu.scheduleTaskDescription": "تشغيل مهمة وفق جدول زمني",
-  "composer.menu.uploadFile": "رفع ملف",
+  "composer.menu.uploadFile": "الملفات والصور",
+  "composer.menu.searchWorkspace": "البحث في مساحة العمل…",
+  "composer.menu.references": "المراجع",
+  "composer.menu.referenceFile": "الإشارة إلى ملف",
+  "composer.menu.attachToMessage": "إضافة سياق",
+  "composer.menu.commands": "الأوامر",
   "composer.menu.uploadFileDescription":
     "صور وPDFs ونصوص/تعليمات برمجية وJSON وCSV",
   "composer.messageAgent": "مراسلة الوكيل...",

package/src/app/i18n/catalogs/de-DE.ts

@@ -1217,7 +1217,12 @@ const messages: ToolkitAgentChatTranslation = {
   "composer.menu.scheduleTask": "Aufgabe planen",
   "composer.menu.scheduleTaskDescription":
     "Etwas nach einem Zeitplan ausführen",
-  "composer.menu.uploadFile": "Datei hochladen",
+  "composer.menu.uploadFile": "Dateien und Bilder",
+  "composer.menu.searchWorkspace": "Arbeitsbereich durchsuchen…",
+  "composer.menu.references": "Referenzen",
+  "composer.menu.referenceFile": "Datei referenzieren",
+  "composer.menu.attachToMessage": "Kontext hinzufügen",
+  "composer.menu.commands": "Befehle",
   "composer.menu.uploadFileDescription": "Bilder, PDFs, Text/Code, JSON, CSV",
   "composer.messageAgent": "Nachricht an den Agenten...",
   "composer.model": "Modell",

package/src/app/i18n/catalogs/en-US.ts

@@ -506,7 +506,12 @@ const messages = {
   "composer.useIntegrationTools": "Use {{integration}} tools",
   "composer.menu.scheduleTask": "Schedule Task",
   "composer.menu.scheduleTaskDescription": "Run something on a schedule",
-  "composer.menu.uploadFile": "Upload File",
+  "composer.menu.uploadFile": "Files and images",
+  "composer.menu.searchWorkspace": "Search workspace…",
+  "composer.menu.references": "References",
+  "composer.menu.referenceFile": "Reference a file",
+  "composer.menu.attachToMessage": "Add context",
+  "composer.menu.commands": "Commands",
   "composer.menu.uploadFileDescription": "Images, PDFs, text/code, JSON, CSV",
   "composer.messageAgent": "Message agent...",
   "composer.model": "Model",

package/src/app/i18n/catalogs/es-ES.ts

@@ -1211,7 +1211,12 @@ const messages: ToolkitAgentChatTranslation = {
   "composer.useIntegrationTools": "Usar las herramientas de {{integration}}",
   "composer.menu.scheduleTask": "Programar tarea",
   "composer.menu.scheduleTaskDescription": "Ejecutar algo según un horario",
-  "composer.menu.uploadFile": "Cargar archivo",
+  "composer.menu.uploadFile": "Archivos e imágenes",
+  "composer.menu.searchWorkspace": "Buscar en el espacio de trabajo…",
+  "composer.menu.references": "Referencias",
+  "composer.menu.referenceFile": "Referenciar un archivo",
+  "composer.menu.attachToMessage": "Añadir contexto",
+  "composer.menu.commands": "Comandos",
   "composer.menu.uploadFileDescription":
     "Imágenes, PDFs, texto/código, JSON, CSV",
   "composer.messageAgent": "Escribe al agente...",

package/src/app/i18n/catalogs/fr-FR.ts

@@ -1219,7 +1219,12 @@ const messages: ToolkitAgentChatTranslation = {
   "composer.menu.scheduleTask": "Planifier une tâche",
   "composer.menu.scheduleTaskDescription":
     "Exécuter quelque chose selon un calendrier",
-  "composer.menu.uploadFile": "Télécharger le fichier",
+  "composer.menu.uploadFile": "Fichiers et images",
+  "composer.menu.searchWorkspace": "Rechercher dans l’espace de travail…",
+  "composer.menu.references": "Références",
+  "composer.menu.referenceFile": "Référencer un fichier",
+  "composer.menu.attachToMessage": "Ajouter du contexte",
+  "composer.menu.commands": "Commandes",
   "composer.menu.uploadFileDescription": "Images, PDFs, texte/code, JSON, CSV",
   "composer.messageAgent": "Écrire à l’agent...",
   "composer.model": "Modèle",

package/src/app/i18n/catalogs/hi-IN.ts

@@ -498,7 +498,12 @@ const messages: ToolkitAgentChatTranslation = {
   "composer.useIntegrationTools": "{{integration}} टूल का उपयोग करें",
   "composer.menu.scheduleTask": "कार्य शेड्यूल करें",
   "composer.menu.scheduleTaskDescription": "किसी कार्य को शेड्यूल के अनुसार चलाएँ",
-  "composer.menu.uploadFile": "फ़ाइल अपलोड करें",
+  "composer.menu.uploadFile": "फ़ाइलें और छवियाँ",
+  "composer.menu.searchWorkspace": "वर्कस्पेस में खोजें…",
+  "composer.menu.references": "संदर्भ",
+  "composer.menu.referenceFile": "फ़ाइल का संदर्भ दें",
+  "composer.menu.attachToMessage": "संदर्भ जोड़ें",
+  "composer.menu.commands": "कमांड",
   "composer.menu.uploadFileDescription": "इमेज, PDFs, टेक्स्ट/कोड, JSON, CSV",
   "composer.messageAgent": "एजेंट को संदेश भेजें...",
   "composer.model": "मॉडल",

package/src/app/i18n/catalogs/ja-JP.ts

@@ -518,7 +518,12 @@ const messages: ToolkitAgentChatTranslation = {
   "composer.useIntegrationTools": "{{integration}}のツールを使用",
   "composer.menu.scheduleTask": "タスクをスケジュール",
   "composer.menu.scheduleTaskDescription": "スケジュールに沿って実行",
-  "composer.menu.uploadFile": "ファイルをアップロード",
+  "composer.menu.uploadFile": "ファイルと画像",
+  "composer.menu.searchWorkspace": "ワークスペースを検索…",
+  "composer.menu.references": "参照",
+  "composer.menu.referenceFile": "ファイルを参照",
+  "composer.menu.attachToMessage": "コンテキストを追加",
+  "composer.menu.commands": "コマンド",
   "composer.menu.uploadFileDescription":
     "画像、PDF、テキスト／コード、JSON、CSV",
   "composer.messageAgent": "エージェントにメッセージを送信...",

package/src/app/i18n/catalogs/ko-KR.ts

@@ -502,7 +502,12 @@ const messages: ToolkitAgentChatTranslation = {
   "composer.useIntegrationTools": "{{integration}} 도구 사용",
   "composer.menu.scheduleTask": "작업 예약",
   "composer.menu.scheduleTaskDescription": "일정에 따라 실행",
-  "composer.menu.uploadFile": "파일 업로드",
+  "composer.menu.uploadFile": "파일 및 이미지",
+  "composer.menu.searchWorkspace": "워크스페이스 검색…",
+  "composer.menu.references": "참조",
+  "composer.menu.referenceFile": "파일 참조",
+  "composer.menu.attachToMessage": "컨텍스트 추가",
+  "composer.menu.commands": "명령",
   "composer.menu.uploadFileDescription": "이미지, PDF, 텍스트/코드, JSON, CSV",
   "composer.messageAgent": "에이전트에게 메시지 보내기...",
   "composer.model": "모델",

package/src/app/i18n/catalogs/pt-BR.ts

@@ -1202,7 +1202,12 @@ const messages: ToolkitAgentChatTranslation = {
   "composer.menu.scheduleTask": "Agendar tarefa",
   "composer.menu.scheduleTaskDescription":
     "Executar algo conforme uma programação",
-  "composer.menu.uploadFile": "Carregar arquivo",
+  "composer.menu.uploadFile": "Arquivos e imagens",
+  "composer.menu.searchWorkspace": "Pesquisar no espaço de trabalho…",
+  "composer.menu.references": "Referências",
+  "composer.menu.referenceFile": "Referenciar um arquivo",
+  "composer.menu.attachToMessage": "Adicionar contexto",
+  "composer.menu.commands": "Comandos",
   "composer.menu.uploadFileDescription":
     "Imagens, PDFs, texto/código, JSON, CSV",
   "composer.messageAgent": "Enviar mensagem ao agente...",

package/src/app/i18n/catalogs/zh-CN.ts

@@ -483,7 +483,12 @@ const messages: ToolkitAgentChatTranslation = {
   "composer.useIntegrationTools": "使用 {{integration}} 工具",
   "composer.menu.scheduleTask": "安排任务",
   "composer.menu.scheduleTaskDescription": "按计划运行任务",
-  "composer.menu.uploadFile": "上传文件",
+  "composer.menu.uploadFile": "文件和图片",
+  "composer.menu.searchWorkspace": "搜索工作区…",
+  "composer.menu.references": "引用",
+  "composer.menu.referenceFile": "引用文件",
+  "composer.menu.attachToMessage": "添加上下文",
+  "composer.menu.commands": "命令",
   "composer.menu.uploadFileDescription": "图片、PDF、文本/代码、JSON、CSV",
   "composer.messageAgent": "向智能体发送消息...",
   "composer.model": "模型",

package/src/app/i18n/catalogs/zh-TW.ts

@@ -484,7 +484,12 @@ const messages: ToolkitAgentChatTranslation = {
   "composer.useIntegrationTools": "使用 {{integration}} 工具",
   "composer.menu.scheduleTask": "排程工作",
   "composer.menu.scheduleTaskDescription": "依排程執行工作",
-  "composer.menu.uploadFile": "上傳檔案",
+  "composer.menu.uploadFile": "檔案和圖片",
+  "composer.menu.searchWorkspace": "搜尋工作區…",
+  "composer.menu.references": "參照",
+  "composer.menu.referenceFile": "引用檔案",
+  "composer.menu.attachToMessage": "新增內容脈絡",
+  "composer.menu.commands": "命令",
   "composer.menu.uploadFileDescription": "圖片、PDF、文字/程式碼、JSON、CSV",
   "composer.messageAgent": "傳送訊息給代理...",
   "composer.model": "模型",

package/src/app/resources/McpIntegrationDialog.tsx

@@ -80,6 +80,9 @@ type DialogMode = "catalog" | "choice" | "form";
 export interface McpIntegrationDialogProps {
   open: boolean;
   onOpenChange: (open: boolean) => void;
+  onCloseAutoFocus?: React.ComponentProps<
+    typeof DialogContent
+  >["onCloseAutoFocus"];
   initialIntegrationId?: string | null;
   connectIntegrationId?: string | null;
   quickConnectIntegrationId?: string | null;
@@ -157,6 +160,7 @@ function resolveIntegrationScope(
 export function McpIntegrationDialog({
   open,
   onOpenChange,
+  onCloseAutoFocus,
   initialIntegrationId = null,
   connectIntegrationId = null,
   quickConnectIntegrationId = null,
@@ -889,6 +893,7 @@ export function McpIntegrationDialog({
   return (
     <Dialog open={open} onOpenChange={onOpenChange}>
       <DialogContent
+        onCloseAutoFocus={onCloseAutoFocus}
         aria-describedby={undefined}
         className={cn(
           "flex flex-col gap-0 overflow-hidden p-0",