@assistant-ui/react-google-adk 0.0.7 → 0.0.10

package/src/useAdkMessages.ts

@@ -3,11 +3,10 @@ import { useState, useCallback, useRef, useMemo } from "react";
 import { v4 as uuidv4 } from "uuid";
 import { useAui } from "@assistant-ui/store";
 import { AdkEventAccumulator } from "./AdkEventAccumulator";
+import { contentToParts } from "./contentToParts";
 import type {
   AdkEvent,
-  AdkEventPart,
   AdkMessage,
-  AdkMessageContentPart,
   AdkMessageMetadata,
   AdkSendMessageConfig,
   AdkStreamCallback,
@@ -200,7 +199,8 @@ export const useAdkMessages = ({
   };
 };
 
-const messageToEvent = (msg: AdkMessage): AdkEvent => {
+/** @internal — exported for unit tests. */
+export const messageToEvent = (msg: AdkMessage): AdkEvent => {
   if (msg.type === "human") {
     return {
       id: msg.id ?? uuidv4(),
@@ -246,31 +246,3 @@ const messageToEvent = (msg: AdkMessage): AdkEvent => {
   };
   return result;
 };
-
-const contentToParts = (
-  content: string | AdkMessageContentPart[],
-): AdkEventPart[] => {
-  if (typeof content === "string") return [{ text: content }];
-  return content.map((part) => {
-    switch (part.type) {
-      case "text":
-        return { text: part.text };
-      case "reasoning":
-        return { text: part.text, thought: true };
-      case "image":
-        return { inlineData: { mimeType: part.mimeType, data: part.data } };
-      case "image_url":
-        return { fileData: { fileUri: part.url } };
-      case "code":
-        return {
-          executableCode: { code: part.code, language: part.language },
-        };
-      case "code_result":
-        return {
-          codeExecutionResult: { output: part.output, outcome: part.outcome },
-        };
-      default:
-        return { text: "" };
-    }
-  });
-};

package/src/useAdkRuntime.test.ts

@@ -1,7 +1,23 @@
 import { describe, it, expect } from "vitest";
-import { getPendingCancellations, getPendingToolCalls } from "./useAdkRuntime";
+import {
+  getMessageContent,
+  getPendingCancellations,
+  getPendingToolCalls,
+} from "./useAdkRuntime";
+import type { AppendMessage } from "@assistant-ui/core";
 import type { AdkMessage } from "./types";
 
+const makeAppendMessage = (content: AppendMessage["content"]): AppendMessage =>
+  ({
+    role: "user",
+    content,
+    attachments: [],
+    parentId: null,
+    sourceId: null,
+    runConfig: undefined,
+    metadata: { custom: {} },
+  }) as unknown as AppendMessage;
+
 const aiWithToolCalls = (
   id: string,
   toolCalls: Array<{ id: string; name: string }>,
@@ -136,3 +152,60 @@ describe("getPendingCancellations", () => {
     });
   });
 });
+
+describe("getMessageContent", () => {
+  it("preserves file part data and mimeType end-to-end", () => {
+    const result = getMessageContent(
+      makeAppendMessage([
+        { type: "text", text: "see attached" },
+        {
+          type: "file",
+          mimeType: "application/pdf",
+          data: "JVBERi0xLjQK",
+          filename: "report.pdf",
+        },
+      ]),
+    );
+    expect(result).toEqual([
+      { type: "text", text: "see attached" },
+      {
+        type: "file",
+        mimeType: "application/pdf",
+        data: "JVBERi0xLjQK",
+        filename: "report.pdf",
+      },
+    ]);
+  });
+
+  it("keeps file-only content as an array (does not collapse to a text marker)", () => {
+    const result = getMessageContent(
+      makeAppendMessage([
+        {
+          type: "file",
+          mimeType: "image/png",
+          data: "AAAA",
+          filename: "x.png",
+        },
+      ]),
+    );
+    expect(result).toEqual([
+      {
+        type: "file",
+        mimeType: "image/png",
+        data: "AAAA",
+        filename: "x.png",
+      },
+    ]);
+  });
+
+  it("omits filename when not provided", () => {
+    const result = getMessageContent(
+      makeAppendMessage([
+        { type: "file", mimeType: "application/pdf", data: "AAAA" },
+      ]),
+    );
+    expect(result).toEqual([
+      { type: "file", mimeType: "application/pdf", data: "AAAA" },
+    ]);
+  });
+});

package/src/useAdkRuntime.ts

@@ -3,6 +3,7 @@ import { useEffect, useRef, useState } from "react";
 import {
   getExternalStoreMessages,
   type AttachmentAdapter,
+  type DictationAdapter,
   type FeedbackAdapter,
   type SpeechSynthesisAdapter,
   type AppendMessage,
@@ -32,7 +33,8 @@ import { convertAdkMessage } from "./convertAdkMessages";
 import { symbolAdkRuntimeExtras, type AdkRuntimeExtras } from "./hooks";
 import { v4 as uuidv4 } from "uuid";
 
-const getMessageContent = (msg: AppendMessage) => {
+/** @internal — exported for unit tests. */
+export const getMessageContent = (msg: AppendMessage) => {
   const allContent = [
     ...msg.content,
     ...(msg.attachments?.flatMap((a) => a.content) ?? []),
@@ -46,8 +48,10 @@ const getMessageContent = (msg: AppendMessage) => {
         return { type: "image_url" as const, url: part.image };
       case "file":
         return {
-          type: "text" as const,
-          text: `[File: ${part.filename ?? "file"}]`,
+          type: "file" as const,
+          mimeType: part.mimeType,
+          data: part.data,
+          ...(part.filename != null && { filename: part.filename }),
         };
 
       case "tool-call":
@@ -144,6 +148,7 @@ export type UseAdkRuntimeOptions = {
     | {
         attachments?: AttachmentAdapter;
         speech?: SpeechSynthesisAdapter;
+        dictation?: DictationAdapter;
         feedback?: FeedbackAdapter;
       }
     | undefined;
@@ -164,7 +169,7 @@ export type UseAdkRuntimeOptions = {
 
 const useAdkRuntimeImpl = ({
   autoCancelPendingToolCalls,
-  adapters: { attachments, feedback, speech } = {},
+  adapters: { attachments, dictation, feedback, speech } = {},
   unstable_allowCancellation,
   stream,
   load,
@@ -261,7 +266,7 @@ const useAdkRuntimeImpl = ({
   const runtime = useExternalStoreRuntime({
     isRunning: effectiveIsRunning,
     messages: threadMessages,
-    adapters: { attachments, feedback, speech },
+    adapters: { attachments, dictation, feedback, speech },
     extras: {
       [symbolAdkRuntimeExtras]: true,
       agentInfo,