@assistant-ui/react-google-adk 0.0.1

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 AgentbaseAI Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,156 @@
+# @assistant-ui/react-google-adk
+
+Google ADK (Agent Development Kit) adapter for [assistant-ui](https://www.assistant-ui.com/).
+
+Connects Google ADK JS agents to assistant-ui's React runtime with streaming, tool calls, multi-agent support, tool confirmations, auth flows, and session state management.
+
+## Installation
+
+```sh
+npm install @assistant-ui/react @assistant-ui/react-google-adk
+```
+
+## Quick Start
+
+### Proxy Mode (with a Next.js API Route)
+
+**Server** (`app/api/adk/route.ts`):
+
+```typescript
+import { createAdkApiRoute } from "@assistant-ui/react-google-adk/server";
+import { runner } from "./agent";
+
+export const POST = createAdkApiRoute({
+  runner,
+  userId: "default-user",
+  sessionId: (req) =>
+    new URL(req.url).searchParams.get("sessionId") ?? "default",
+});
+```
+
+**Client:**
+
+```tsx
+import { useAdkRuntime, createAdkStream } from "@assistant-ui/react-google-adk";
+
+const runtime = useAdkRuntime({
+  stream: createAdkStream({ api: "/api/adk" }),
+});
+```
+
+### Direct Mode (connecting to an ADK server)
+
+```tsx
+import {
+  useAdkRuntime,
+  createAdkStream,
+  createAdkSessionAdapter,
+} from "@assistant-ui/react-google-adk";
+
+const { adapter, load } = createAdkSessionAdapter({
+  apiUrl: "http://localhost:8000",
+  appName: "my-app",
+  userId: "user-1",
+});
+
+const runtime = useAdkRuntime({
+  stream: createAdkStream({
+    api: "http://localhost:8000",
+    appName: "my-app",
+    userId: "user-1",
+  }),
+  sessionAdapter: adapter,
+  load,
+});
+```
+
+## API Reference
+
+### Client Exports
+
+#### `createAdkStream(options)`
+
+Creates an `AdkStreamCallback` that connects to an ADK endpoint via SSE.
+
+| Option | Type | Description |
+|--------|------|-------------|
+| `api` | `string` | URL to POST to (proxy route or ADK server base URL) |
+| `appName` | `string?` | ADK app name (enables direct mode when set) |
+| `userId` | `string?` | ADK user ID (required with `appName`) |
+| `headers` | `Record<string, string> \| (() => ...)` | Static or dynamic request headers |
+
+#### `createAdkSessionAdapter(options)`
+
+Creates a `RemoteThreadListAdapter` backed by ADK's session REST API.
+
+Returns `{ adapter, load }` where `load` reconstructs messages from session events.
+
+| Option | Type | Description |
+|--------|------|-------------|
+| `apiUrl` | `string` | ADK server base URL |
+| `appName` | `string` | ADK app name |
+| `userId` | `string` | ADK user ID |
+| `headers` | `Record<string, string> \| (() => ...)` | Static or dynamic request headers |
+
+#### `useAdkRuntime(options)`
+
+Main hook that creates a full `AssistantRuntime`.
+
+| Option | Type | Description |
+|--------|------|-------------|
+| `stream` | `AdkStreamCallback` | Stream callback (use `createAdkStream`) |
+| `sessionAdapter` | `RemoteThreadListAdapter?` | Alternative to `cloud` for thread persistence |
+| `cloud` | `AssistantCloud?` | Cloud adapter for thread persistence |
+| `load` | `(threadId: string) => Promise<{messages}>` | Load thread history |
+| `autoCancelPendingToolCalls` | `boolean?` | Auto-cancel pending tools on new message (default: true) |
+| `unstable_allowCancellation` | `boolean?` | Enable stream cancellation |
+| `getCheckpointId` | `(threadId, messages) => Promise<string?>` | Enable edit/reload |
+| `adapters` | `{attachments?, speech?, feedback?}` | Optional adapters |
+| `eventHandlers` | `{onError?, onCustomEvent?, onAgentTransfer?}` | Event callbacks |
+
+### Hooks
+
+| Hook | Description |
+|------|-------------|
+| `useAdkAgentInfo()` | Current agent name and branch path |
+| `useAdkSessionState()` | Accumulated session state delta |
+| `useAdkSend()` | Send raw ADK messages |
+| `useAdkLongRunningToolIds()` | Pending tool IDs awaiting input |
+| `useAdkToolConfirmations()` | Tool confirmation requests |
+| `useAdkAuthRequests()` | Auth credential requests |
+| `useAdkArtifacts()` | Artifact delta (filename → version) |
+| `useAdkEscalation()` | Escalation flag |
+| `useAdkMessageMetadata()` | Per-message grounding/citation/usage |
+
+### Server Exports (`/server`)
+
+#### `createAdkApiRoute(options)`
+
+One-liner API route handler combining request parsing and SSE streaming.
+
+| Option | Type | Description |
+|--------|------|-------------|
+| `runner` | `AdkRunner` | ADK Runner instance |
+| `userId` | `string \| (req) => string` | Static or dynamic user ID |
+| `sessionId` | `string \| (req) => string` | Static or dynamic session ID |
+| `onError` | `(error) => void` | Error handler |
+
+#### `adkEventStream(events, options?)`
+
+Converts an ADK event async generator into an SSE `Response`.
+
+#### `parseAdkRequest(request)`
+
+Parses an incoming HTTP request into a structured ADK request.
+
+#### `toAdkContent(parsed)`
+
+Converts a parsed request into a Google GenAI `Content` object.
+
+## Documentation
+
+See the [full documentation](https://www.assistant-ui.com/docs/runtimes/google-adk) for setup guides, advanced APIs, and examples.
+
+## License
+
+MIT

package/dist/AdkClient.d.ts

@@ -0,0 +1,45 @@
+import type { AdkStreamCallback } from "./types.js";
+export type CreateAdkStreamOptions = {
+    /**
+     * URL to POST to. Either a proxy route (e.g. "/api/adk") or
+     * an ADK server base URL (e.g. "http://localhost:8000").
+     *
+     * When `appName` and `userId` are provided, POSTs to `${api}/run_sse`
+     * in ADK-native format. Otherwise POSTs directly to `api` in proxy format
+     * (compatible with `parseAdkRequest`).
+     */
+    api: string;
+    /**
+     * ADK application name. When provided along with `userId`,
+     * enables direct connection to an ADK server.
+     */
+    appName?: string | undefined;
+    /**
+     * ADK user ID. Required when `appName` is provided.
+     */
+    userId?: string | undefined;
+    /**
+     * Extra headers to send with every request.
+     * Can be a static object or an async function for dynamic auth tokens.
+     */
+    headers?: Record<string, string> | (() => Record<string, string> | Promise<Record<string, string>>) | undefined;
+};
+/**
+ * Creates an `AdkStreamCallback` that connects to an ADK endpoint.
+ *
+ * @example Proxy mode (with a Next.js API route)
+ * ```ts
+ * const stream = createAdkStream({ api: "/api/adk" });
+ * ```
+ *
+ * @example Direct mode (connecting to ADK server)
+ * ```ts
+ * const stream = createAdkStream({
+ *   api: "http://localhost:8000",
+ *   appName: "my-app",
+ *   userId: "user-1",
+ * });
+ * ```
+ */
+export declare function createAdkStream(options: CreateAdkStreamOptions): AdkStreamCallback;
+//# sourceMappingURL=AdkClient.d.ts.map

package/dist/AdkClient.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"AdkClient.d.ts","sourceRoot":"","sources":["../src/AdkClient.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAIV,iBAAiB,EAClB,mBAAgB;AAEjB,MAAM,MAAM,sBAAsB,GAAG;IACnC;;;;;;;OAOG;IACH,GAAG,EAAE,MAAM,CAAC;IAEZ;;;OAGG;IACH,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAE7B;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAE5B;;;OAGG;IACH,OAAO,CAAC,EACJ,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GACtB,CAAC,MAAM,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,GAChE,SAAS,CAAC;CACf,CAAC;AAEF;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,eAAe,CAC7B,OAAO,EAAE,sBAAsB,GAC9B,iBAAiB,CA0CnB"}

package/dist/AdkClient.js

@@ -0,0 +1,204 @@
+/**
+ * Creates an `AdkStreamCallback` that connects to an ADK endpoint.
+ *
+ * @example Proxy mode (with a Next.js API route)
+ * ```ts
+ * const stream = createAdkStream({ api: "/api/adk" });
+ * ```
+ *
+ * @example Direct mode (connecting to ADK server)
+ * ```ts
+ * const stream = createAdkStream({
+ *   api: "http://localhost:8000",
+ *   appName: "my-app",
+ *   userId: "user-1",
+ * });
+ * ```
+ */
+export function createAdkStream(options) {
+    const isDirect = options.appName != null;
+    return async function* (messages, config) {
+        const headers = await resolveHeaders(options.headers);
+        let url;
+        let body;
+        if (isDirect) {
+            // Direct mode: POST to ADK server's /run_sse
+            url = `${options.api}/run_sse`;
+            const { externalId } = await config.initialize();
+            body = {
+                appName: options.appName,
+                userId: options.userId,
+                sessionId: externalId,
+                newMessage: messagesToContent(messages),
+                streaming: true,
+                ...(config.stateDelta != null && { stateDelta: config.stateDelta }),
+            };
+        }
+        else {
+            // Proxy mode: POST in parseAdkRequest-compatible format
+            url = options.api;
+            body = messagesToProxyBody(messages, config);
+        }
+        const response = await fetch(url, {
+            method: "POST",
+            headers: { "Content-Type": "application/json", ...headers },
+            body: JSON.stringify(body),
+            signal: config.abortSignal,
+        });
+        if (!response.ok) {
+            throw new Error(`ADK request failed: ${response.status} ${response.statusText}`);
+        }
+        yield* parseSSEResponse(response);
+    };
+}
+// ── Internal helpers ──
+async function resolveHeaders(headers) {
+    if (!headers)
+        return {};
+    if (typeof headers === "function")
+        return await headers();
+    return headers;
+}
+/**
+ * Converts AdkMessage[] (new messages) into ADK Content format
+ * for the direct `/run_sse` endpoint.
+ */
+function messagesToContent(messages) {
+    const parts = [];
+    for (const msg of messages) {
+        if (msg.type === "human") {
+            for (const part of contentToParts(msg.content)) {
+                parts.push(part);
+            }
+        }
+        else if (msg.type === "tool") {
+            let response;
+            try {
+                response = JSON.parse(msg.content);
+            }
+            catch {
+                response = msg.content;
+            }
+            parts.push({
+                functionResponse: {
+                    name: msg.name,
+                    id: msg.tool_call_id,
+                    response,
+                },
+            });
+        }
+    }
+    if (parts.length === 0) {
+        parts.push({ text: "" });
+    }
+    return { role: "user", parts };
+}
+/**
+ * Converts AdkMessage[] into the proxy request body format
+ * (compatible with `parseAdkRequest`).
+ */
+function messagesToProxyBody(messages, config) {
+    const body = {};
+    if (config.runConfig != null)
+        body.runConfig = config.runConfig;
+    if (config.checkpointId != null)
+        body.checkpointId = config.checkpointId;
+    if (config.stateDelta != null)
+        body.stateDelta = config.stateDelta;
+    // Check if there's a tool result
+    const toolMsg = messages.find((m) => m.type === "tool");
+    if (toolMsg && toolMsg.type === "tool") {
+        // If there are also other messages (e.g. cancellations), send as parts
+        if (messages.length > 1) {
+            body.parts = messagesToContent(messages).parts;
+            return body;
+        }
+        let result;
+        try {
+            result = JSON.parse(toolMsg.content);
+        }
+        catch {
+            result = toolMsg.content;
+        }
+        body.type = "tool-result";
+        body.toolCallId = toolMsg.tool_call_id;
+        body.toolName = toolMsg.name;
+        body.result = result;
+        body.isError = toolMsg.status === "error";
+        return body;
+    }
+    // Human message(s) - possibly with cancellation tool results prepended
+    if (messages.length === 1 && messages[0].type === "human") {
+        const msg = messages[0];
+        if (typeof msg.content === "string") {
+            body.message = msg.content;
+        }
+        else {
+            body.parts = contentToParts(msg.content);
+        }
+        return body;
+    }
+    // Multiple messages (e.g. cancellations + human): send as parts array
+    body.parts = messagesToContent(messages).parts;
+    return body;
+}
+function contentToParts(content) {
+    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: "" };
+        }
+    });
+}
+async function* parseSSEResponse(response) {
+    const reader = response.body.getReader();
+    const decoder = new TextDecoder();
+    let buffer = "";
+    try {
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            buffer += decoder.decode(value, { stream: true });
+            const parts = buffer.split("\n\n");
+            buffer = parts.pop() ?? "";
+            for (const part of parts) {
+                for (const line of part.split("\n")) {
+                    if (line.startsWith("data: ")) {
+                        yield JSON.parse(line.slice(6));
+                    }
+                }
+            }
+        }
+        // Handle remaining buffer
+        if (buffer.trim()) {
+            for (const line of buffer.split("\n")) {
+                if (line.startsWith("data: ")) {
+                    yield JSON.parse(line.slice(6));
+                }
+            }
+        }
+    }
+    finally {
+        reader.releaseLock();
+    }
+}
+//# sourceMappingURL=AdkClient.js.map

package/dist/AdkClient.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AdkClient.js","sourceRoot":"","sources":["../src/AdkClient.ts"],"names":[],"mappings":"AAuCA;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,UAAU,eAAe,CAC7B,OAA+B;IAE/B,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC;IAEzC,OAAO,KAAK,SAAS,CAAC,EAAE,QAAQ,EAAE,MAAM;QACtC,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAEtD,IAAI,GAAW,CAAC;QAChB,IAAI,IAAa,CAAC;QAElB,IAAI,QAAQ,EAAE,CAAC;YACb,6CAA6C;YAC7C,GAAG,GAAG,GAAG,OAAO,CAAC,GAAG,UAAU,CAAC;YAC/B,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YACjD,IAAI,GAAG;gBACL,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,SAAS,EAAE,UAAU;gBACrB,UAAU,EAAE,iBAAiB,CAAC,QAAQ,CAAC;gBACvC,SAAS,EAAE,IAAI;gBACf,GAAG,CAAC,MAAM,CAAC,UAAU,IAAI,IAAI,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC;aACpE,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,wDAAwD;YACxD,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;YAClB,IAAI,GAAG,mBAAmB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC/C,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,GAAG,OAAO,EAAE;YAC3D,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YAC1B,MAAM,EAAE,MAAM,CAAC,WAAW;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CACb,uBAAuB,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAChE,CAAC;QACJ,CAAC;QAED,KAAK,CAAC,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC,CAAC;AACJ,CAAC;AAED,yBAAyB;AAEzB,KAAK,UAAU,cAAc,CAC3B,OAGa;IAEb,IAAI,CAAC,OAAO;QAAE,OAAO,EAAE,CAAC;IACxB,IAAI,OAAO,OAAO,KAAK,UAAU;QAAE,OAAO,MAAM,OAAO,EAAE,CAAC;IAC1D,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;GAGG;AACH,SAAS,iBAAiB,CAAC,QAAsB;IAI/C,MAAM,KAAK,GAAmC,EAAE,CAAC;IAEjD,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YACzB,KAAK,MAAM,IAAI,IAAI,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC/C,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnB,CAAC;QACH,CAAC;aAAM,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC/B,IAAI,QAAiB,CAAC;YACtB,IAAI,CAAC;gBACH,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACrC,CAAC;YAAC,MAAM,CAAC;gBACP,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC;YACzB,CAAC;YACD,KAAK,CAAC,IAAI,CAAC;gBACT,gBAAgB,EAAE;oBAChB,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,EAAE,EAAE,GAAG,CAAC,YAAY;oBACpB,QAAQ;iBACT;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;IAC3B,CAAC;IAED,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;AACjC,CAAC;AAED;;;GAGG;AACH,SAAS,mBAAmB,CAC1B,QAAsB,EACtB,MAIC;IAED,MAAM,IAAI,GAA4B,EAAE,CAAC;IAEzC,IAAI,MAAM,CAAC,SAAS,IAAI,IAAI;QAAE,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;IAChE,IAAI,MAAM,CAAC,YAAY,IAAI,IAAI;QAAE,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;IACzE,IAAI,MAAM,CAAC,UAAU,IAAI,IAAI;QAAE,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;IAEnE,iCAAiC;IACjC,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;IACxD,IAAI,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACvC,uEAAuE;QACvE,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,IAAI,CAAC,KAAK,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC;YAC/C,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,MAAe,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;QAC3B,CAAC;QACD,IAAI,CAAC,IAAI,GAAG,aAAa,CAAC;QAC1B,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,YAAY,CAAC;QACvC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,MAAM,KAAK,OAAO,CAAC;QAC1C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,uEAAuE;IACvE,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAE,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAC3D,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAE,CAAC;QACzB,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACpC,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,GAAG,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC3C,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,sEAAsE;IACtE,IAAI,CAAC,KAAK,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC;IAC/C,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,cAAc,CACrB,OAAyC;IAEzC,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IAC5D,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QAC1B,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,MAAM;gBACT,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YAC7B,KAAK,WAAW;gBACd,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAC5C,KAAK,OAAO;gBACV,OAAO,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;YACtE,KAAK,WAAW;gBACd,OAAO,EAAE,QAAQ,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;YAC7C,KAAK,MAAM;gBACT,OAAO;oBACL,cAAc,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE;iBAC7D,CAAC;YACJ,KAAK,aAAa;gBAChB,OAAO;oBACL,mBAAmB,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;iBACpE,CAAC;YACJ;gBACE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;QACxB,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,gBAAgB,CAAC,QAAkB;IACjD,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAK,CAAC,SAAS,EAAE,CAAC;IAC1C,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAClC,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,IAAI,CAAC;QACH,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACnC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;YAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;oBACpC,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;wBAC9B,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAa,CAAC;oBAC9C,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,0BAA0B;QAC1B,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;YAClB,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBACtC,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC9B,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAa,CAAC;gBAC9C,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,WAAW,EAAE,CAAC;IACvB,CAAC;AACH,CAAC"}

package/dist/AdkEventAccumulator.d.ts

@@ -0,0 +1,45 @@
+import type { AdkEvent, AdkMessage, AdkToolConfirmation, AdkAuthRequest, AdkMessageMetadata } from "./types.js";
+/**
+ * Checks if an event is a final response using the same logic as ADK's
+ * `isFinalResponse()`.
+ */
+export declare const isFinalResponse: (event: AdkEvent) => boolean;
+export declare class AdkEventAccumulator {
+    private messagesMap;
+    private currentMessageId;
+    private partialTextBuffer;
+    private partialReasoningBuffer;
+    private accumulatedStateDelta;
+    private accumulatedArtifactDelta;
+    private lastAgentInfo;
+    private lastTransferToAgent;
+    private pendingLongRunningToolIds;
+    private toolConfirmations;
+    private authRequests;
+    private escalated;
+    private messageMetadataMap;
+    constructor(initialMessages?: AdkMessage[]);
+    processEvent(rawEvent: AdkEvent): AdkMessage[];
+    private processPart;
+    private trackMessageMetadata;
+    private getContentArray;
+    private getOrCreateAiMessage;
+    private appendContent;
+    private replaceLastTextContent;
+    private replaceLastReasoningContent;
+    private finalizeCurrentMessage;
+    getMessages(): AdkMessage[];
+    getStateDelta(): Record<string, unknown>;
+    getArtifactDelta(): Record<string, number>;
+    getAgentInfo(): {
+        name?: string | undefined;
+        branch?: string | undefined;
+    };
+    getLastTransferToAgent(): string | undefined;
+    getLongRunningToolIds(): string[];
+    getToolConfirmations(): AdkToolConfirmation[];
+    getAuthRequests(): AdkAuthRequest[];
+    isEscalated(): boolean;
+    getMessageMetadata(): Map<string, AdkMessageMetadata>;
+}
+//# sourceMappingURL=AdkEventAccumulator.d.ts.map

package/dist/AdkEventAccumulator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"AdkEventAccumulator.d.ts","sourceRoot":"","sources":["../src/AdkEventAccumulator.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,QAAQ,EAER,UAAU,EAGV,mBAAmB,EACnB,cAAc,EACd,kBAAkB,EACnB,mBAAgB;AAQjB;;;GAGG;AACH,eAAO,MAAM,eAAe,GAAI,OAAO,QAAQ,KAAG,OAgBjD,CAAC;AAqHF,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,WAAW,CAAiC;IACpD,OAAO,CAAC,gBAAgB,CAAuB;IAC/C,OAAO,CAAC,iBAAiB,CAAM;IAC/B,OAAO,CAAC,sBAAsB,CAAM;IACpC,OAAO,CAAC,qBAAqB,CAA+B;IAC5D,OAAO,CAAC,wBAAwB,CAA8B;IAC9D,OAAO,CAAC,aAAa,CAGd;IACP,OAAO,CAAC,mBAAmB,CAAqB;IAChD,OAAO,CAAC,yBAAyB,CAAgB;IACjD,OAAO,CAAC,iBAAiB,CAA6B;IACtD,OAAO,CAAC,YAAY,CAAwB;IAC5C,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,kBAAkB,CAAyC;gBACvD,eAAe,CAAC,EAAE,UAAU,EAAE;IAQ1C,YAAY,CAAC,QAAQ,EAAE,QAAQ,GAAG,UAAU,EAAE;IA8I9C,OAAO,CAAC,WAAW;IAqJnB,OAAO,CAAC,oBAAoB;IAmB5B,OAAO,CAAC,eAAe;IAMvB,OAAO,CAAC,oBAAoB;IAuB5B,OAAO,CAAC,aAAa;IASrB,OAAO,CAAC,sBAAsB;IAY9B,OAAO,CAAC,2BAA2B;IAenC,OAAO,CAAC,sBAAsB;IAM9B,WAAW,IAAI,UAAU,EAAE;IAI3B,aAAa,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAIxC,gBAAgB,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAI1C,YAAY,IAAI;QAAE,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,GAAG,SAAS,CAAA;KAAE;IAI1E,sBAAsB,IAAI,MAAM,GAAG,SAAS;IAI5C,qBAAqB,IAAI,MAAM,EAAE;IAIjC,oBAAoB,IAAI,mBAAmB,EAAE;IAI7C,eAAe,IAAI,cAAc,EAAE;IAInC,WAAW,IAAI,OAAO;IAItB,kBAAkB,IAAI,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC;CAGtD"}