@assistant-ui/react-google-adk 0.0.8 → 0.0.11

package/README.md

@@ -1,36 +1,31 @@
-# @assistant-ui/react-google-adk
+# `@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.
+[Google ADK](https://github.com/google/adk-js) (Agent Development Kit) integration for `@assistant-ui/react`. Connects ADK JS agents to the assistant-ui 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
+```bash
+npm install @assistant-ui/react @assistant-ui/react-google-adk @google/adk
 ```
 
-## Quick Start
-
-### Proxy Mode (with a Next.js API Route)
+## Usage
 
-**Server** (`app/api/adk/route.ts`):
+The recommended setup proxies through your own API route:
 
-```typescript
+```ts
+// app/api/adk/route.ts
 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",
+  sessionId: (req) => new URL(req.url).searchParams.get("sessionId") ?? "default",
 });
 ```
 
-**Client:**
-
 ```tsx
+// client component
 import { useAdkRuntime, createAdkStream } from "@assistant-ui/react-google-adk";
 
 const runtime = useAdkRuntime({
@@ -38,119 +33,11 @@ const runtime = useAdkRuntime({
 });
 ```
 
-### 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
+Or connect directly to an ADK server with `createAdkSessionAdapter`; see the docs for the full setup and option reference.
 
-See the [full documentation](https://www.assistant-ui.com/docs/runtimes/google-adk) for setup guides, advanced APIs, and examples.
+## See also
 
-## License
+- `@assistant-ui/react-langgraph` for LangGraph SDK agents.
+- `@assistant-ui/react-ag-ui` for the AG-UI protocol.
 
-MIT
+Full reference for client hooks (`useAdkAgentInfo`, `useAdkSessionState`, `useAdkToolConfirmations`, `useAdkAuthRequests`, etc.), server exports, and direct mode at [assistant-ui.com/docs/runtimes/google-adk](https://www.assistant-ui.com/docs/runtimes/google-adk).

package/dist/AdkClient.d.ts

@@ -1,28 +1,30 @@
-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;
+import { AdkStreamCallback } from "./types.js";
+
+//#region src/AdkClient.d.ts
+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.
@@ -41,5 +43,7 @@ export type CreateAdkStreamOptions = {
  * });
  * ```
  */
-export declare function createAdkStream(options: CreateAdkStreamOptions): AdkStreamCallback;
+declare function createAdkStream(options: CreateAdkStreamOptions): AdkStreamCallback;
+//#endregion
+export { CreateAdkStreamOptions, createAdkStream };
 //# sourceMappingURL=AdkClient.d.ts.map

package/dist/AdkClient.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AdkClient.d.ts","sourceRoot":"","sources":["../src/AdkClient.ts"],"names":[],"mappings":"AACA,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"}
+{"version":3,"file":"AdkClient.d.ts","names":[],"sources":["../src/AdkClient.ts"],"mappings":";;;KAQY,sBAAA;;AAAZ;;;;;;;EASE,GAAA;EAmB2C;;;;EAb3C,OAAA;EAYI;;;EAPJ,MAAA;EAQkD;AAAA;AAqBpD;;EAvBE,OAAA,GACI,MAAA,0BACO,MAAA,mBAAyB,OAAA,CAAQ,MAAA;AAAA;;;;;AAuB1B;;;;;;;;;;;;;iBAFJ,eAAA,CACd,OAAA,EAAS,sBAAA,GACR,iBAAiB"}

package/dist/AdkClient.js

@@ -1,179 +1,145 @@
 import { contentToParts } from "./contentToParts.js";
+//#region src/AdkClient.ts
 /**
- * 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);
-    };
+* 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",
+* });
+* ```
+*/
+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) {
+			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 {
+			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;
+	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.
- */
+* 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 };
+	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`).
- */
+* 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;
+	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;
+	const toolMsg = messages.find((m) => m.type === "tool");
+	if (toolMsg && toolMsg.type === "tool") {
+		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;
+	}
+	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;
+	}
+	body.parts = messagesToContent(messages).parts;
+	return body;
 }
 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();
-    }
+	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));
+		}
+		if (buffer.trim()) {
+			for (const line of buffer.split("\n")) if (line.startsWith("data: ")) yield JSON.parse(line.slice(6));
+		}
+	} finally {
+		reader.releaseLock();
+	}
 }
+//#endregion
+export { createAdkStream };
+
 //# sourceMappingURL=AdkClient.js.map

package/dist/AdkClient.js.map

@@ -1 +1 @@
-{"version":3,"file":"AdkClient.js","sourceRoot":"","sources":["../src/AdkClient.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,4BAAyB;AAwClD;;;;;;;;;;;;;;;;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,GAAmB,EAAE,CAAC;IAEjC,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,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"}
+{"version":3,"file":"AdkClient.js","names":[],"sources":["../src/AdkClient.ts"],"sourcesContent":["import { contentToParts } from \"./contentToParts\";\nimport type {\n  AdkEvent,\n  AdkEventPart,\n  AdkMessage,\n  AdkStreamCallback,\n} from \"./types\";\n\nexport type CreateAdkStreamOptions = {\n  /**\n   * URL to POST to. Either a proxy route (e.g. \"/api/adk\") or\n   * an ADK server base URL (e.g. \"http://localhost:8000\").\n   *\n   * When `appName` and `userId` are provided, POSTs to `${api}/run_sse`\n   * in ADK-native format. Otherwise POSTs directly to `api` in proxy format\n   * (compatible with `parseAdkRequest`).\n   */\n  api: string;\n\n  /**\n   * ADK application name. When provided along with `userId`,\n   * enables direct connection to an ADK server.\n   */\n  appName?: string | undefined;\n\n  /**\n   * ADK user ID. Required when `appName` is provided.\n   */\n  userId?: string | undefined;\n\n  /**\n   * Extra headers to send with every request.\n   * Can be a static object or an async function for dynamic auth tokens.\n   */\n  headers?:\n    | Record<string, string>\n    | (() => Record<string, string> | Promise<Record<string, string>>)\n    | undefined;\n};\n\n/**\n * Creates an `AdkStreamCallback` that connects to an ADK endpoint.\n *\n * @example Proxy mode (with a Next.js API route)\n * ```ts\n * const stream = createAdkStream({ api: \"/api/adk\" });\n * ```\n *\n * @example Direct mode (connecting to ADK server)\n * ```ts\n * const stream = createAdkStream({\n *   api: \"http://localhost:8000\",\n *   appName: \"my-app\",\n *   userId: \"user-1\",\n * });\n * ```\n */\nexport function createAdkStream(\n  options: CreateAdkStreamOptions,\n): AdkStreamCallback {\n  const isDirect = options.appName != null;\n\n  return async function* (messages, config) {\n    const headers = await resolveHeaders(options.headers);\n\n    let url: string;\n    let body: unknown;\n\n    if (isDirect) {\n      // Direct mode: POST to ADK server's /run_sse\n      url = `${options.api}/run_sse`;\n      const { externalId } = await config.initialize();\n      body = {\n        appName: options.appName,\n        userId: options.userId,\n        sessionId: externalId,\n        newMessage: messagesToContent(messages),\n        streaming: true,\n        ...(config.stateDelta != null && { stateDelta: config.stateDelta }),\n      };\n    } else {\n      // Proxy mode: POST in parseAdkRequest-compatible format\n      url = options.api;\n      body = messagesToProxyBody(messages, config);\n    }\n\n    const response = await fetch(url, {\n      method: \"POST\",\n      headers: { \"Content-Type\": \"application/json\", ...headers },\n      body: JSON.stringify(body),\n      signal: config.abortSignal,\n    });\n\n    if (!response.ok) {\n      throw new Error(\n        `ADK request failed: ${response.status} ${response.statusText}`,\n      );\n    }\n\n    yield* parseSSEResponse(response);\n  };\n}\n\n// ── Internal helpers ──\n\nasync function resolveHeaders(\n  headers:\n    | Record<string, string>\n    | (() => Record<string, string> | Promise<Record<string, string>>)\n    | undefined,\n): Promise<Record<string, string>> {\n  if (!headers) return {};\n  if (typeof headers === \"function\") return await headers();\n  return headers;\n}\n\n/**\n * Converts AdkMessage[] (new messages) into ADK Content format\n * for the direct `/run_sse` endpoint.\n */\nfunction messagesToContent(messages: AdkMessage[]): {\n  role: string;\n  parts: AdkEventPart[];\n} {\n  const parts: AdkEventPart[] = [];\n\n  for (const msg of messages) {\n    if (msg.type === \"human\") {\n      for (const part of contentToParts(msg.content)) {\n        parts.push(part);\n      }\n    } else if (msg.type === \"tool\") {\n      let response: unknown;\n      try {\n        response = JSON.parse(msg.content);\n      } catch {\n        response = msg.content;\n      }\n      parts.push({\n        functionResponse: {\n          name: msg.name,\n          id: msg.tool_call_id,\n          response,\n        },\n      });\n    }\n  }\n\n  if (parts.length === 0) {\n    parts.push({ text: \"\" });\n  }\n\n  return { role: \"user\", parts };\n}\n\n/**\n * Converts AdkMessage[] into the proxy request body format\n * (compatible with `parseAdkRequest`).\n */\nfunction messagesToProxyBody(\n  messages: AdkMessage[],\n  config: {\n    runConfig?: unknown;\n    checkpointId?: string | undefined;\n    stateDelta?: Record<string, unknown> | undefined;\n  },\n): Record<string, unknown> {\n  const body: Record<string, unknown> = {};\n\n  if (config.runConfig != null) body.runConfig = config.runConfig;\n  if (config.checkpointId != null) body.checkpointId = config.checkpointId;\n  if (config.stateDelta != null) body.stateDelta = config.stateDelta;\n\n  // Check if there's a tool result\n  const toolMsg = messages.find((m) => m.type === \"tool\");\n  if (toolMsg && toolMsg.type === \"tool\") {\n    // If there are also other messages (e.g. cancellations), send as parts\n    if (messages.length > 1) {\n      body.parts = messagesToContent(messages).parts;\n      return body;\n    }\n\n    let result: unknown;\n    try {\n      result = JSON.parse(toolMsg.content);\n    } catch {\n      result = toolMsg.content;\n    }\n    body.type = \"tool-result\";\n    body.toolCallId = toolMsg.tool_call_id;\n    body.toolName = toolMsg.name;\n    body.result = result;\n    body.isError = toolMsg.status === \"error\";\n    return body;\n  }\n\n  // Human message(s) - possibly with cancellation tool results prepended\n  if (messages.length === 1 && messages[0]!.type === \"human\") {\n    const msg = messages[0]!;\n    if (typeof msg.content === \"string\") {\n      body.message = msg.content;\n    } else {\n      body.parts = contentToParts(msg.content);\n    }\n    return body;\n  }\n\n  // Multiple messages (e.g. cancellations + human): send as parts array\n  body.parts = messagesToContent(messages).parts;\n  return body;\n}\n\nasync function* parseSSEResponse(response: Response): AsyncGenerator<AdkEvent> {\n  const reader = response.body!.getReader();\n  const decoder = new TextDecoder();\n  let buffer = \"\";\n\n  try {\n    while (true) {\n      const { done, value } = await reader.read();\n      if (done) break;\n\n      buffer += decoder.decode(value, { stream: true });\n      const parts = buffer.split(\"\\n\\n\");\n      buffer = parts.pop() ?? \"\";\n\n      for (const part of parts) {\n        for (const line of part.split(\"\\n\")) {\n          if (line.startsWith(\"data: \")) {\n            yield JSON.parse(line.slice(6)) as AdkEvent;\n          }\n        }\n      }\n    }\n\n    // Handle remaining buffer\n    if (buffer.trim()) {\n      for (const line of buffer.split(\"\\n\")) {\n        if (line.startsWith(\"data: \")) {\n          yield JSON.parse(line.slice(6)) as AdkEvent;\n        }\n      }\n    }\n  } finally {\n    reader.releaseLock();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AAyDA,SAAgB,gBACd,SACmB;CACnB,MAAM,WAAW,QAAQ,WAAW;CAEpC,OAAO,iBAAiB,UAAU,QAAQ;EACxC,MAAM,UAAU,MAAM,eAAe,QAAQ,OAAO;EAEpD,IAAI;EACJ,IAAI;EAEJ,IAAI,UAAU;GAEZ,MAAM,GAAG,QAAQ,IAAI;GACrB,MAAM,EAAE,eAAe,MAAM,OAAO,WAAW;GAC/C,OAAO;IACL,SAAS,QAAQ;IACjB,QAAQ,QAAQ;IAChB,WAAW;IACX,YAAY,kBAAkB,QAAQ;IACtC,WAAW;IACX,GAAI,OAAO,cAAc,QAAQ,EAAE,YAAY,OAAO,WAAW;GACnE;EACF,OAAO;GAEL,MAAM,QAAQ;GACd,OAAO,oBAAoB,UAAU,MAAM;EAC7C;EAEA,MAAM,WAAW,MAAM,MAAM,KAAK;GAChC,QAAQ;GACR,SAAS;IAAE,gBAAgB;IAAoB,GAAG;GAAQ;GAC1D,MAAM,KAAK,UAAU,IAAI;GACzB,QAAQ,OAAO;EACjB,CAAC;EAED,IAAI,CAAC,SAAS,IACZ,MAAM,IAAI,MACR,uBAAuB,SAAS,OAAO,GAAG,SAAS,YACrD;EAGF,OAAO,iBAAiB,QAAQ;CAClC;AACF;AAIA,eAAe,eACb,SAIiC;CACjC,IAAI,CAAC,SAAS,OAAO,CAAC;CACtB,IAAI,OAAO,YAAY,YAAY,OAAO,MAAM,QAAQ;CACxD,OAAO;AACT;;;;;AAMA,SAAS,kBAAkB,UAGzB;CACA,MAAM,QAAwB,CAAC;CAE/B,KAAK,MAAM,OAAO,UAChB,IAAI,IAAI,SAAS,SACf,KAAK,MAAM,QAAQ,eAAe,IAAI,OAAO,GAC3C,MAAM,KAAK,IAAI;MAEZ,IAAI,IAAI,SAAS,QAAQ;EAC9B,IAAI;EACJ,IAAI;GACF,WAAW,KAAK,MAAM,IAAI,OAAO;EACnC,QAAQ;GACN,WAAW,IAAI;EACjB;EACA,MAAM,KAAK,EACT,kBAAkB;GAChB,MAAM,IAAI;GACV,IAAI,IAAI;GACR;EACF,EACF,CAAC;CACH;CAGF,IAAI,MAAM,WAAW,GACnB,MAAM,KAAK,EAAE,MAAM,GAAG,CAAC;CAGzB,OAAO;EAAE,MAAM;EAAQ;CAAM;AAC/B;;;;;AAMA,SAAS,oBACP,UACA,QAKyB;CACzB,MAAM,OAAgC,CAAC;CAEvC,IAAI,OAAO,aAAa,MAAM,KAAK,YAAY,OAAO;CACtD,IAAI,OAAO,gBAAgB,MAAM,KAAK,eAAe,OAAO;CAC5D,IAAI,OAAO,cAAc,MAAM,KAAK,aAAa,OAAO;CAGxD,MAAM,UAAU,SAAS,MAAM,MAAM,EAAE,SAAS,MAAM;CACtD,IAAI,WAAW,QAAQ,SAAS,QAAQ;EAEtC,IAAI,SAAS,SAAS,GAAG;GACvB,KAAK,QAAQ,kBAAkB,QAAQ,EAAE;GACzC,OAAO;EACT;EAEA,IAAI;EACJ,IAAI;GACF,SAAS,KAAK,MAAM,QAAQ,OAAO;EACrC,QAAQ;GACN,SAAS,QAAQ;EACnB;EACA,KAAK,OAAO;EACZ,KAAK,aAAa,QAAQ;EAC1B,KAAK,WAAW,QAAQ;EACxB,KAAK,SAAS;EACd,KAAK,UAAU,QAAQ,WAAW;EAClC,OAAO;CACT;CAGA,IAAI,SAAS,WAAW,KAAK,SAAS,GAAI,SAAS,SAAS;EAC1D,MAAM,MAAM,SAAS;EACrB,IAAI,OAAO,IAAI,YAAY,UACzB,KAAK,UAAU,IAAI;OAEnB,KAAK,QAAQ,eAAe,IAAI,OAAO;EAEzC,OAAO;CACT;CAGA,KAAK,QAAQ,kBAAkB,QAAQ,EAAE;CACzC,OAAO;AACT;AAEA,gBAAgB,iBAAiB,UAA8C;CAC7E,MAAM,SAAS,SAAS,KAAM,UAAU;CACxC,MAAM,UAAU,IAAI,YAAY;CAChC,IAAI,SAAS;CAEb,IAAI;EACF,OAAO,MAAM;GACX,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,KAAK;GAC1C,IAAI,MAAM;GAEV,UAAU,QAAQ,OAAO,OAAO,EAAE,QAAQ,KAAK,CAAC;GAChD,MAAM,QAAQ,OAAO,MAAM,MAAM;GACjC,SAAS,MAAM,IAAI,KAAK;GAExB,KAAK,MAAM,QAAQ,OACjB,KAAK,MAAM,QAAQ,KAAK,MAAM,IAAI,GAChC,IAAI,KAAK,WAAW,QAAQ,GAC1B,MAAM,KAAK,MAAM,KAAK,MAAM,CAAC,CAAC;EAItC;EAGA,IAAI,OAAO,KAAK;QACT,MAAM,QAAQ,OAAO,MAAM,IAAI,GAClC,IAAI,KAAK,WAAW,QAAQ,GAC1B,MAAM,KAAK,MAAM,KAAK,MAAM,CAAC,CAAC;EAAA;CAItC,UAAU;EACR,OAAO,YAAY;CACrB;AACF"}