@assistant-ui/react-google-adk 0.0.31 → 0.0.32

package/dist/AdkClient.d.ts

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

package/dist/AdkClient.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AdkClient.d.ts","names":[],"sources":["../src/AdkClient.ts"],"mappings":";;YAYY;;;;;;;;;EASV;;;;;EAMA;;;;EAKA;;;;;EAMA,UACI,gCACO,yBAAyB,QAAQ;;;;;;;;;;;;;;;;;;;wBAqB9B,gBACd,SAAS,yBACR"}
+{"version":3,"file":"AdkClient.d.ts","sourceRoot":"","sources":["../src/AdkClient.ts"],"names":[],"mappings":"AAMA,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,CAsDnB"}

package/dist/AdkClient.js

@@ -1,5 +1,6 @@
 import { contentToParts } from "./contentToParts.js";
 import { parseAdkEventValue } from "./parseAdkEvent.js";
+import { raceWithAbortSignal } from "./raceWithAbortSignal.js";
 import { toAdkFunctionResponse } from "./toAdkFunctionResponse.js";
 import { trimTrailingSlashes } from "./trimTrailingSlashes.js";
 import { SSEEventDecoder } from "assistant-stream/utils";
@@ -26,7 +27,7 @@ function createAdkStream(options) {
 	const isDirect = options.appName != null;
 	if (isDirect && (options.userId == null || options.userId === "")) throw new Error("createAdkStream direct mode requires \"userId\" when \"appName\" is provided.");
 	return async function* (messages, config) {
-		const headers = await resolveHeaders(options.headers);
+		const headers = await resolveHeaders(options.headers, config.abortSignal);
 		let url;
 		let body;
 		if (isDirect) {
@@ -75,9 +76,9 @@ function parseAdkEvent(data) {
 	}
 	return parseAdkEventValue(value, "Invalid ADK stream event");
 }
-async function resolveHeaders(headers) {
+async function resolveHeaders(headers, signal) {
 	if (!headers) return {};
-	if (typeof headers === "function") return await headers();
+	if (typeof headers === "function") return await raceWithAbortSignal(signal, headers);
 	return headers;
 }
 /**
@@ -144,6 +145,7 @@ function messagesToProxyBody(messages, config) {
 	return body;
 }
 async function* parseSSEResponse(response) {
+	if (!response.body) throw new Error("Expected ADK stream response body, received no body");
 	const reader = response.body.getReader();
 	const decoder = new TextDecoder();
 	const sseDecoder = new SSEEventDecoder({ trailing: "dispatch" });

package/dist/AdkClient.js.map

@@ -1 +1 @@
-{"version":3,"file":"AdkClient.js","names":[],"sources":["../src/AdkClient.ts"],"sourcesContent":["import { SSEEventDecoder } from \"assistant-stream/utils\";\nimport { contentToParts } from \"./contentToParts\";\nimport { parseAdkEventValue } from \"./parseAdkEvent\";\nimport { toAdkFunctionResponse } from \"./toAdkFunctionResponse\";\nimport { trimTrailingSlashes } from \"./trimTrailingSlashes\";\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  if (options.appName === \"\") {\n    throw new Error(\n      'createAdkStream direct mode requires a non-empty \"appName\".',\n    );\n  }\n\n  const isDirect = options.appName != null;\n  if (isDirect && (options.userId == null || options.userId === \"\")) {\n    throw new Error(\n      'createAdkStream direct mode requires \"userId\" when \"appName\" is provided.',\n    );\n  }\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 = `${trimTrailingSlashes(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    validateEventStreamContentType(response);\n    yield* parseSSEResponse(response);\n  };\n}\n\n// ── Internal helpers ──\n\nfunction validateEventStreamContentType(response: Response): void {\n  const contentType = response.headers.get(\"Content-Type\");\n  const mediaType = contentType?.split(\";\", 1)[0]?.trim().toLowerCase();\n  if (mediaType !== \"text/event-stream\") {\n    const received = contentType\n      ? `\"${contentType}\"`\n      : \"no Content-Type header\";\n    void response.body?.cancel().catch(() => undefined);\n    throw new Error(\n      `Expected ADK stream response Content-Type \"text/event-stream\", received ${received}`,\n    );\n  }\n}\n\nfunction parseAdkEvent(data: string): AdkEvent {\n  let value: unknown;\n  try {\n    value = JSON.parse(data);\n  } catch {\n    throw new Error(\"Invalid ADK stream event: expected valid JSON.\");\n  }\n\n  return parseAdkEventValue(value, \"Invalid ADK stream event\");\n}\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: toAdkFunctionResponse(response, msg.status === \"error\"),\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  const sseDecoder = new SSEEventDecoder({ trailing: \"dispatch\" });\n\n  let shouldCancel = true;\n  try {\n    while (true) {\n      let result: ReadableStreamReadResult<Uint8Array>;\n      try {\n        result = await reader.read();\n      } catch (error) {\n        shouldCancel = false;\n        throw error;\n      }\n\n      const { done, value } = result;\n      if (done) {\n        shouldCancel = false;\n        for (const event of sseDecoder.push(decoder.decode())) {\n          yield parseAdkEvent(event.data);\n        }\n        break;\n      }\n\n      for (const event of sseDecoder.push(\n        decoder.decode(value, { stream: true }),\n      )) {\n        yield parseAdkEvent(event.data);\n      }\n    }\n\n    const trailing = sseDecoder.flush();\n    if (trailing !== null) yield parseAdkEvent(trailing.data);\n  } finally {\n    try {\n      if (shouldCancel) await reader.cancel().catch(() => undefined);\n    } finally {\n      reader.releaseLock();\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;AA6DA,SAAgB,gBACd,SACmB;CACnB,IAAI,QAAQ,YAAY,IACtB,MAAM,IAAI,MACR,+DACF;CAGF,MAAM,WAAW,QAAQ,WAAW;CACpC,IAAI,aAAa,QAAQ,UAAU,QAAQ,QAAQ,WAAW,KAC5D,MAAM,IAAI,MACR,+EACF;CAGF,OAAO,iBAAiB,UAAU,QAAQ;EACxC,MAAM,UAAU,MAAM,eAAe,QAAQ,OAAO;EAEpD,IAAI;EACJ,IAAI;EAEJ,IAAI,UAAU;GAEZ,MAAM,GAAG,oBAAoB,QAAQ,GAAG,EAAE;GAC1C,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,+BAA+B,QAAQ;EACvC,OAAO,iBAAiB,QAAQ;CAClC;AACF;AAIA,SAAS,+BAA+B,UAA0B;CAChE,MAAM,cAAc,SAAS,QAAQ,IAAI,cAAc;CAEvD,IADkB,aAAa,MAAM,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,YAAY,MAClD,qBAAqB;EACrC,MAAM,WAAW,cACb,IAAI,YAAY,KAChB;EACJ,SAAc,MAAM,OAAO,CAAC,CAAC,YAAY,KAAA,CAAS;EAClD,MAAM,IAAI,MACR,2EAA2E,UAC7E;CACF;AACF;AAEA,SAAS,cAAc,MAAwB;CAC7C,IAAI;CACJ,IAAI;EACF,QAAQ,KAAK,MAAM,IAAI;CACzB,QAAQ;EACN,MAAM,IAAI,MAAM,gDAAgD;CAClE;CAEA,OAAO,mBAAmB,OAAO,0BAA0B;AAC7D;AAEA,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,UAAU,sBAAsB,UAAU,IAAI,WAAW,OAAO;EAClE,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,CAAC,CAAC;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,EAAE,CAAE,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,CAAC,CAAC;CACzC,OAAO;AACT;AAEA,gBAAgB,iBAAiB,UAA8C;CAC7E,MAAM,SAAS,SAAS,KAAM,UAAU;CACxC,MAAM,UAAU,IAAI,YAAY;CAChC,MAAM,aAAa,IAAI,gBAAgB,EAAE,UAAU,WAAW,CAAC;CAE/D,IAAI,eAAe;CACnB,IAAI;EACF,OAAO,MAAM;GACX,IAAI;GACJ,IAAI;IACF,SAAS,MAAM,OAAO,KAAK;GAC7B,SAAS,OAAO;IACd,eAAe;IACf,MAAM;GACR;GAEA,MAAM,EAAE,MAAM,UAAU;GACxB,IAAI,MAAM;IACR,eAAe;IACf,KAAK,MAAM,SAAS,WAAW,KAAK,QAAQ,OAAO,CAAC,GAClD,MAAM,cAAc,MAAM,IAAI;IAEhC;GACF;GAEA,KAAK,MAAM,SAAS,WAAW,KAC7B,QAAQ,OAAO,OAAO,EAAE,QAAQ,KAAK,CAAC,CACxC,GACE,MAAM,cAAc,MAAM,IAAI;EAElC;EAEA,MAAM,WAAW,WAAW,MAAM;EAClC,IAAI,aAAa,MAAM,MAAM,cAAc,SAAS,IAAI;CAC1D,UAAU;EACR,IAAI;GACF,IAAI,cAAc,MAAM,OAAO,OAAO,CAAC,CAAC,YAAY,KAAA,CAAS;EAC/D,UAAU;GACR,OAAO,YAAY;EACrB;CACF;AACF"}
+{"version":3,"file":"AdkClient.js","names":[],"sources":["../src/AdkClient.ts"],"sourcesContent":["import { SSEEventDecoder } from \"assistant-stream/utils\";\nimport { contentToParts } from \"./contentToParts\";\nimport { parseAdkEventValue } from \"./parseAdkEvent\";\nimport { raceWithAbortSignal } from \"./raceWithAbortSignal\";\nimport { toAdkFunctionResponse } from \"./toAdkFunctionResponse\";\nimport { trimTrailingSlashes } from \"./trimTrailingSlashes\";\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  if (options.appName === \"\") {\n    throw new Error(\n      'createAdkStream direct mode requires a non-empty \"appName\".',\n    );\n  }\n\n  const isDirect = options.appName != null;\n  if (isDirect && (options.userId == null || options.userId === \"\")) {\n    throw new Error(\n      'createAdkStream direct mode requires \"userId\" when \"appName\" is provided.',\n    );\n  }\n\n  return async function* (messages, config) {\n    const headers = await resolveHeaders(options.headers, config.abortSignal);\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 = `${trimTrailingSlashes(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    validateEventStreamContentType(response);\n    yield* parseSSEResponse(response);\n  };\n}\n\n// ── Internal helpers ──\n\nfunction validateEventStreamContentType(response: Response): void {\n  const contentType = response.headers.get(\"Content-Type\");\n  const mediaType = contentType?.split(\";\", 1)[0]?.trim().toLowerCase();\n  if (mediaType !== \"text/event-stream\") {\n    const received = contentType\n      ? `\"${contentType}\"`\n      : \"no Content-Type header\";\n    void response.body?.cancel().catch(() => undefined);\n    throw new Error(\n      `Expected ADK stream response Content-Type \"text/event-stream\", received ${received}`,\n    );\n  }\n}\n\nfunction parseAdkEvent(data: string): AdkEvent {\n  let value: unknown;\n  try {\n    value = JSON.parse(data);\n  } catch {\n    throw new Error(\"Invalid ADK stream event: expected valid JSON.\");\n  }\n\n  return parseAdkEventValue(value, \"Invalid ADK stream event\");\n}\n\nasync function resolveHeaders(\n  headers:\n    | Record<string, string>\n    | (() => Record<string, string> | Promise<Record<string, string>>)\n    | undefined,\n  signal?: AbortSignal,\n): Promise<Record<string, string>> {\n  if (!headers) return {};\n  if (typeof headers === \"function\") {\n    return await raceWithAbortSignal(signal, headers);\n  }\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: toAdkFunctionResponse(response, msg.status === \"error\"),\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  if (!response.body) {\n    throw new Error(\"Expected ADK stream response body, received no body\");\n  }\n  const reader = response.body.getReader();\n  const decoder = new TextDecoder();\n  const sseDecoder = new SSEEventDecoder({ trailing: \"dispatch\" });\n\n  let shouldCancel = true;\n  try {\n    while (true) {\n      let result: ReadableStreamReadResult<Uint8Array>;\n      try {\n        result = await reader.read();\n      } catch (error) {\n        shouldCancel = false;\n        throw error;\n      }\n\n      const { done, value } = result;\n      if (done) {\n        shouldCancel = false;\n        for (const event of sseDecoder.push(decoder.decode())) {\n          yield parseAdkEvent(event.data);\n        }\n        break;\n      }\n\n      for (const event of sseDecoder.push(\n        decoder.decode(value, { stream: true }),\n      )) {\n        yield parseAdkEvent(event.data);\n      }\n    }\n\n    const trailing = sseDecoder.flush();\n    if (trailing !== null) yield parseAdkEvent(trailing.data);\n  } finally {\n    try {\n      if (shouldCancel) await reader.cancel().catch(() => undefined);\n    } finally {\n      reader.releaseLock();\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;AA8DA,SAAgB,gBACd,SACmB;CACnB,IAAI,QAAQ,YAAY,IACtB,MAAM,IAAI,MACR,+DACF;CAGF,MAAM,WAAW,QAAQ,WAAW;CACpC,IAAI,aAAa,QAAQ,UAAU,QAAQ,QAAQ,WAAW,KAC5D,MAAM,IAAI,MACR,+EACF;CAGF,OAAO,iBAAiB,UAAU,QAAQ;EACxC,MAAM,UAAU,MAAM,eAAe,QAAQ,SAAS,OAAO,WAAW;EAExE,IAAI;EACJ,IAAI;EAEJ,IAAI,UAAU;GAEZ,MAAM,GAAG,oBAAoB,QAAQ,GAAG,EAAE;GAC1C,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,+BAA+B,QAAQ;EACvC,OAAO,iBAAiB,QAAQ;CAClC;AACF;AAIA,SAAS,+BAA+B,UAA0B;CAChE,MAAM,cAAc,SAAS,QAAQ,IAAI,cAAc;CAEvD,IADkB,aAAa,MAAM,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,YAAY,MAClD,qBAAqB;EACrC,MAAM,WAAW,cACb,IAAI,YAAY,KAChB;EACJ,SAAc,MAAM,OAAO,CAAC,CAAC,YAAY,KAAA,CAAS;EAClD,MAAM,IAAI,MACR,2EAA2E,UAC7E;CACF;AACF;AAEA,SAAS,cAAc,MAAwB;CAC7C,IAAI;CACJ,IAAI;EACF,QAAQ,KAAK,MAAM,IAAI;CACzB,QAAQ;EACN,MAAM,IAAI,MAAM,gDAAgD;CAClE;CAEA,OAAO,mBAAmB,OAAO,0BAA0B;AAC7D;AAEA,eAAe,eACb,SAIA,QACiC;CACjC,IAAI,CAAC,SAAS,OAAO,CAAC;CACtB,IAAI,OAAO,YAAY,YACrB,OAAO,MAAM,oBAAoB,QAAQ,OAAO;CAElD,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,UAAU,sBAAsB,UAAU,IAAI,WAAW,OAAO;EAClE,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,CAAC,CAAC;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,EAAE,CAAE,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,CAAC,CAAC;CACzC,OAAO;AACT;AAEA,gBAAgB,iBAAiB,UAA8C;CAC7E,IAAI,CAAC,SAAS,MACZ,MAAM,IAAI,MAAM,qDAAqD;CAEvE,MAAM,SAAS,SAAS,KAAK,UAAU;CACvC,MAAM,UAAU,IAAI,YAAY;CAChC,MAAM,aAAa,IAAI,gBAAgB,EAAE,UAAU,WAAW,CAAC;CAE/D,IAAI,eAAe;CACnB,IAAI;EACF,OAAO,MAAM;GACX,IAAI;GACJ,IAAI;IACF,SAAS,MAAM,OAAO,KAAK;GAC7B,SAAS,OAAO;IACd,eAAe;IACf,MAAM;GACR;GAEA,MAAM,EAAE,MAAM,UAAU;GACxB,IAAI,MAAM;IACR,eAAe;IACf,KAAK,MAAM,SAAS,WAAW,KAAK,QAAQ,OAAO,CAAC,GAClD,MAAM,cAAc,MAAM,IAAI;IAEhC;GACF;GAEA,KAAK,MAAM,SAAS,WAAW,KAC7B,QAAQ,OAAO,OAAO,EAAE,QAAQ,KAAK,CAAC,CACxC,GACE,MAAM,cAAc,MAAM,IAAI;EAElC;EAEA,MAAM,WAAW,WAAW,MAAM;EAClC,IAAI,aAAa,MAAM,MAAM,cAAc,SAAS,IAAI;CAC1D,UAAU;EACR,IAAI;GACF,IAAI,cAAc,MAAM,OAAO,OAAO,CAAC,CAAC,YAAY,KAAA,CAAS;EAC/D,UAAU;GACR,OAAO,YAAY;EACrB;CACF;AACF"}

package/dist/AdkEventAccumulator.d.ts

@@ -1,49 +1,47 @@
-import { AdkAuthRequest, AdkEvent, AdkMessage, AdkMessageMetadata, AdkToolConfirmation } from "./types.js";
-//#region src/AdkEventAccumulator.d.ts
+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 finalTextReplacedThisEvent;
-  private finalReasoningReplacedThisEvent;
-  private partialReasoningBuffer;
-  private accumulatedStateDelta;
-  private accumulatedArtifactDelta;
-  private lastAgentInfo;
-  private lastTransferToAgent;
-  private pendingLongRunningToolIds;
-  private escalated;
-  private messageMetadataMap;
-  private aiMessageOrdinals;
-  constructor(initialMessages?: AdkMessage[], initialLongRunningToolIds?: readonly string[]);
-  processEvent(rawEvent: AdkEvent): AdkMessage[];
-  private processPart;
-  private trackMessageMetadata;
-  private getContentArray;
-  private getOrCreateAiMessage;
-  private appendContent;
-  private replaceLastTextContent;
-  private replaceLastReasoningContent;
-  private finalizeCurrentMessage;
-  private getRequestCalls;
-  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>;
+    private messagesMap;
+    private currentMessageId;
+    private partialTextBuffer;
+    private finalTextReplacedThisEvent;
+    private finalReasoningReplacedThisEvent;
+    private partialReasoningBuffer;
+    private accumulatedStateDelta;
+    private accumulatedArtifactDelta;
+    private lastAgentInfo;
+    private lastTransferToAgent;
+    private pendingLongRunningToolIds;
+    private escalated;
+    private messageMetadataMap;
+    private aiMessageOrdinals;
+    constructor(initialMessages?: AdkMessage[], initialLongRunningToolIds?: readonly string[]);
+    processEvent(rawEvent: AdkEvent): AdkMessage[];
+    private processPart;
+    private trackMessageMetadata;
+    private getContentArray;
+    private getOrCreateAiMessage;
+    private appendContent;
+    private replaceLastTextContent;
+    private replaceLastReasoningContent;
+    private finalizeCurrentMessage;
+    private getRequestCalls;
+    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>;
 }
-//#endregion
 //# sourceMappingURL=AdkEventAccumulator.d.ts.map

package/dist/AdkEventAccumulator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AdkEventAccumulator.d.ts","names":[],"sources":["../src/AdkEventAccumulator.ts"],"mappings":";;;;;;qBA4Ca,kBAAmB,OAAO;qBAuI1B;UACH;UACA;UACA;UACA;UACA;UACA;UACA;UACA;UAEA;UAIA;UACA;UACA;UACA;UAGA;EAEN,YAAA,kBAAkB,cAClB;EAYF,aAAa,UAAU,WAAW;UAyL1B;UA4GA;UAmBA;UAMA;UAyBA;UASA;UAYA;UAeA;UAOA;EAWR,eAAe;EAIf,iBAAiB;EAIjB,oBAAoB;EAIpB;IAAkB;IAA2B;;EAI7C;EAIA;EAIA,wBAAwB;EAoBxB,mBAAmB;EAanB;EAIA,sBAAsB,YAAY"}
+{"version":3,"file":"AdkEventAccumulator.d.ts","sourceRoot":"","sources":["../src/AdkEventAccumulator.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,QAAQ,EAER,UAAU,EAGV,mBAAmB,EACnB,cAAc,EACd,kBAAkB,EACnB,mBAAgB;AA4BjB;;;GAGG;AACH,eAAO,MAAM,eAAe,GAAI,OAAO,QAAQ,KAAG,OAgBjD,CAAC;AAuHF,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,WAAW,CAAiC;IACpD,OAAO,CAAC,gBAAgB,CAAuB;IAC/C,OAAO,CAAC,iBAAiB,CAAM;IAC/B,OAAO,CAAC,0BAA0B,CAAS;IAC3C,OAAO,CAAC,+BAA+B,CAAS;IAChD,OAAO,CAAC,sBAAsB,CAAM;IACpC,OAAO,CAAC,qBAAqB,CAAgD;IAC7E,OAAO,CAAC,wBAAwB,CACV;IACtB,OAAO,CAAC,aAAa,CAGd;IACP,OAAO,CAAC,mBAAmB,CAAqB;IAChD,OAAO,CAAC,yBAAyB,CAAqB;IACtD,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,kBAAkB,CAAyC;IAGnE,OAAO,CAAC,iBAAiB,CAA6B;gBAEpD,eAAe,CAAC,EAAE,UAAU,EAAE,EAC9B,yBAAyB,CAAC,EAAE,SAAS,MAAM,EAAE;IAY/C,YAAY,CAAC,QAAQ,EAAE,QAAQ,GAAG,UAAU,EAAE;IAyL9C,OAAO,CAAC,WAAW;IA4GnB,OAAO,CAAC,oBAAoB;IAmB5B,OAAO,CAAC,eAAe;IAMvB,OAAO,CAAC,oBAAoB;IAyB5B,OAAO,CAAC,aAAa;IASrB,OAAO,CAAC,sBAAsB;IAY9B,OAAO,CAAC,2BAA2B;IAenC,OAAO,CAAC,sBAAsB;IAO9B,OAAO,CAAC,eAAe;IAWvB,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;IAoB7C,eAAe,IAAI,cAAc,EAAE;IAanC,WAAW,IAAI,OAAO;IAItB,kBAAkB,IAAI,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC;CAGtD"}

package/dist/AdkSessionAdapter.d.ts

@@ -1,46 +1,45 @@
-import { AdkThreadSnapshot } from "./types.js";
-import { RemoteThreadListAdapter } from "@assistant-ui/core";
-//#region src/AdkSessionAdapter.d.ts
+import type { RemoteThreadListAdapter } from "@assistant-ui/core";
+import type { AdkThreadSnapshot } from "./types.js";
 export type AdkSessionAdapterOptions = {
-  /**
-   * ADK server base URL (e.g. "http://localhost:8000").
-   */
-  apiUrl: string;
-  /**
-   * ADK application name.
-   */
-  appName: string;
-  /**
-   * ADK user ID.
-   */
-  userId: string;
-  /**
-   * Extra headers for API requests.
-   */
-  headers?: Record<string, string> | (() => Record<string, string> | Promise<Record<string, string>>) | undefined;
+    /**
+     * ADK server base URL (e.g. "http://localhost:8000").
+     */
+    apiUrl: string;
+    /**
+     * ADK application name.
+     */
+    appName: string;
+    /**
+     * ADK user ID.
+     */
+    userId: string;
+    /**
+     * Extra headers for API requests.
+     */
+    headers?: Record<string, string> | (() => Record<string, string> | Promise<Record<string, string>>) | undefined;
 };
 export type AdkArtifactData = {
-  inlineData?: {
-    mimeType: string;
-    data: string;
-  } | undefined;
-  fileData?: {
-    fileUri: string;
-    mimeType?: string | undefined;
-  } | undefined;
-  text?: string | undefined;
+    inlineData?: {
+        mimeType: string;
+        data: string;
+    } | undefined;
+    fileData?: {
+        fileUri: string;
+        mimeType?: string | undefined;
+    } | undefined;
+    text?: string | undefined;
 };
 type AdkSessionAdapterResult = {
-  adapter: RemoteThreadListAdapter;
-  load: (sessionId: string, options?: {
-    signal?: AbortSignal | undefined;
-  }) => Promise<AdkThreadSnapshot>;
-  artifacts: {
-    list: (sessionId: string) => Promise<string[]>;
-    load: (sessionId: string, artifactName: string, version?: number) => Promise<AdkArtifactData>;
-    listVersions: (sessionId: string, artifactName: string) => Promise<number[]>;
-    delete: (sessionId: string, artifactName: string) => Promise<void>;
-  };
+    adapter: RemoteThreadListAdapter;
+    load: (sessionId: string, options?: {
+        signal?: AbortSignal | undefined;
+    }) => Promise<AdkThreadSnapshot>;
+    artifacts: {
+        list: (sessionId: string) => Promise<string[]>;
+        load: (sessionId: string, artifactName: string, version?: number) => Promise<AdkArtifactData>;
+        listVersions: (sessionId: string, artifactName: string) => Promise<number[]>;
+        delete: (sessionId: string, artifactName: string) => Promise<void>;
+    };
 };
 /**
  * Creates a `RemoteThreadListAdapter` backed by ADK's session REST API,
@@ -62,5 +61,5 @@ type AdkSessionAdapterResult = {
  * ```
  */
 export declare function createAdkSessionAdapter(options: AdkSessionAdapterOptions): AdkSessionAdapterResult;
-//#endregion
+export {};
 //# sourceMappingURL=AdkSessionAdapter.d.ts.map

package/dist/AdkSessionAdapter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AdkSessionAdapter.d.ts","names":[],"sources":["../src/AdkSessionAdapter.ts"],"mappings":";;;YAaY;;;;EAIV;;;;EAKA;;;;EAKA;;;;EAKA,UACI,gCACO,yBAAyB,QAAQ;;YAIlC;EACV;IAAe;IAAkB;;EACjC;IAAa;IAAiB;;EAC9B;;KAGG;EACH,SAAS;EACT,OACE,mBACA;IAAY,SAAS;QAClB,QAAQ;EACb;IACE,OAAO,sBAAsB;IAC7B,OACE,mBACA,sBACA,qBACG,QAAQ;IACb,eACE,mBACA,yBACG;IACL,SAAS,mBAAmB,yBAAyB;;;;;;;;;;;;;;;;;;;;;;wBAoKzC,wBACd,SAAS,2BACR"}
+{"version":3,"file":"AdkSessionAdapter.d.ts","sourceRoot":"","sources":["../src/AdkSessionAdapter.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAEV,uBAAuB,EAGxB,MAAM,oBAAoB,CAAC;AAI5B,OAAO,KAAK,EAAc,iBAAiB,EAAE,mBAAgB;AAI7D,MAAM,MAAM,wBAAwB,GAAG;IACrC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAEhB;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IAEf;;OAEG;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,MAAM,MAAM,eAAe,GAAG;IAC5B,UAAU,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,GAAG,SAAS,CAAC;IAC5D,QAAQ,CAAC,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAA;KAAE,GAAG,SAAS,CAAC;IAC1E,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC3B,CAAC;AAEF,KAAK,uBAAuB,GAAG;IAC7B,OAAO,EAAE,uBAAuB,CAAC;IACjC,IAAI,EAAE,CACJ,SAAS,EAAE,MAAM,EACjB,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,WAAW,GAAG,SAAS,CAAA;KAAE,KAC3C,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAChC,SAAS,EAAE;QACT,IAAI,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/C,IAAI,EAAE,CACJ,SAAS,EAAE,MAAM,EACjB,YAAY,EAAE,MAAM,EACpB,OAAO,CAAC,EAAE,MAAM,KACb,OAAO,CAAC,eAAe,CAAC,CAAC;QAC9B,YAAY,EAAE,CACZ,SAAS,EAAE,MAAM,EACjB,YAAY,EAAE,MAAM,KACjB,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QACvB,MAAM,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;KACpE,CAAC;CACH,CAAC;AA+IF;;;;;;;;;;;;;;;;;;GAkBG;AACH,wBAAgB,uBAAuB,CACrC,OAAO,EAAE,wBAAwB,GAChC,uBAAuB,CAiMzB"}

package/dist/AdkSessionAdapter.js

@@ -1,4 +1,5 @@
 import { parseAdkEventValue } from "./parseAdkEvent.js";
+import { raceWithAbortSignal } from "./raceWithAbortSignal.js";
 import { trimTrailingSlashes } from "./trimTrailingSlashes.js";
 import { normalizeAdkPart } from "./normalizeAdkPart.js";
 import { AdkEventAccumulator } from "./AdkEventAccumulator.js";
@@ -70,9 +71,9 @@ const parseAdkArtifactVersionsResponse = (value) => {
 function createAdkSessionAdapter(options) {
 	const { apiUrl, appName, userId } = options;
 	const baseUrl = `${trimTrailingSlashes(apiUrl)}/apps/${encodeURIComponent(appName)}/users/${encodeURIComponent(userId)}/sessions`;
-	const getHeaders = async () => {
+	const getHeaders = async (signal) => {
 		if (!options.headers) return {};
-		if (typeof options.headers === "function") return await options.headers();
+		if (typeof options.headers === "function") return await raceWithAbortSignal(signal, options.headers);
 		return options.headers;
 	};
 	const adapter = {
@@ -134,7 +135,7 @@ function createAdkSessionAdapter(options) {
 		}
 	};
 	const load = async (sessionId, options) => {
-		const headers = await getHeaders();
+		const headers = await getHeaders(options?.signal);
 		const res = await fetch(`${baseUrl}/${encodeURIComponent(sessionId)}`, {
 			headers,
 			...options?.signal ? { signal: options.signal } : {}

package/dist/AdkSessionAdapter.js.map

@@ -1 +1 @@
-{"version":3,"file":"AdkSessionAdapter.js","names":[],"sources":["../src/AdkSessionAdapter.ts"],"sourcesContent":["import type { AssistantStream, AssistantStreamChunk } from \"assistant-stream\";\nimport type {\n  RemoteThreadInitializeResponse,\n  RemoteThreadListAdapter,\n  RemoteThreadListResponse,\n  RemoteThreadMetadata,\n} from \"@assistant-ui/core\";\nimport { AdkEventAccumulator } from \"./AdkEventAccumulator\";\nimport { normalizeAdkPart } from \"./normalizeAdkPart\";\nimport { parseAdkEventValue } from \"./parseAdkEvent\";\nimport type { AdkMessage, AdkThreadSnapshot } from \"./types\";\nimport { trimTrailingSlashes } from \"./trimTrailingSlashes\";\n\nexport type AdkSessionAdapterOptions = {\n  /**\n   * ADK server base URL (e.g. \"http://localhost:8000\").\n   */\n  apiUrl: string;\n\n  /**\n   * ADK application name.\n   */\n  appName: string;\n\n  /**\n   * ADK user ID.\n   */\n  userId: string;\n\n  /**\n   * Extra headers for API requests.\n   */\n  headers?:\n    | Record<string, string>\n    | (() => Record<string, string> | Promise<Record<string, string>>)\n    | undefined;\n};\n\nexport type AdkArtifactData = {\n  inlineData?: { mimeType: string; data: string } | undefined;\n  fileData?: { fileUri: string; mimeType?: string | undefined } | undefined;\n  text?: string | undefined;\n};\n\ntype AdkSessionAdapterResult = {\n  adapter: RemoteThreadListAdapter;\n  load: (\n    sessionId: string,\n    options?: { signal?: AbortSignal | undefined },\n  ) => Promise<AdkThreadSnapshot>;\n  artifacts: {\n    list: (sessionId: string) => Promise<string[]>;\n    load: (\n      sessionId: string,\n      artifactName: string,\n      version?: number,\n    ) => Promise<AdkArtifactData>;\n    listVersions: (\n      sessionId: string,\n      artifactName: string,\n    ) => Promise<number[]>;\n    delete: (sessionId: string, artifactName: string) => Promise<void>;\n  };\n};\n\ntype AdkSessionResponse = {\n  id: string;\n  events?: unknown;\n};\n\nconst isRecord = (value: unknown): value is Record<string, unknown> =>\n  typeof value === \"object\" && value !== null && !Array.isArray(value);\n\nconst isAdkSessionResponse = (value: unknown): value is AdkSessionResponse => {\n  if (!isRecord(value)) return false;\n\n  const id = value.id;\n  return typeof id === \"string\" && id.length > 0;\n};\n\nconst parseAdkSessionResponse = (\n  value: unknown,\n  operation: \"create\" | \"fetch\" | \"load\",\n): AdkSessionResponse => {\n  if (!isAdkSessionResponse(value)) {\n    throw new Error(\n      `Invalid ADK session ${operation} response: expected an object with a non-empty string \"id\".`,\n    );\n  }\n  return value;\n};\n\nconst parseAdkSessionListResponse = (value: unknown): AdkSessionResponse[] => {\n  if (!Array.isArray(value)) {\n    throw new Error(\n      \"Invalid ADK session list response: expected an array of sessions.\",\n    );\n  }\n\n  return value.map((session, index) => {\n    if (!isAdkSessionResponse(session)) {\n      throw new Error(\n        `Invalid ADK session list response: session at index ${index} must have a non-empty string \"id\".`,\n      );\n    }\n    return session;\n  });\n};\n\nconst parseAdkArtifactListResponse = (value: unknown): string[] => {\n  if (!Array.isArray(value)) {\n    throw new Error(\n      \"Invalid ADK artifact list response: expected an array of artifact names.\",\n    );\n  }\n\n  return value.map((artifact, index) => {\n    if (typeof artifact === \"string\" && artifact.length > 0) {\n      return artifact;\n    }\n\n    if (isRecord(artifact)) {\n      const filename = artifact.filename;\n      if (typeof filename === \"string\" && filename.length > 0) {\n        return filename;\n      }\n    }\n\n    throw new Error(\n      `Invalid ADK artifact list response: artifact at index ${index} must be a non-empty string or an object with a non-empty string \"filename\".`,\n    );\n  });\n};\n\nconst parseAdkArtifactResponse = (value: unknown): AdkArtifactData => {\n  if (!isRecord(value)) {\n    throw new Error(\n      'Invalid ADK artifact load response: expected an object containing \"text\", \"inlineData\", or \"fileData\".',\n    );\n  }\n\n  const normalizedValue = normalizeAdkPart(value);\n  const { text, inlineData, fileData } = normalizedValue;\n  if (\n    text === undefined &&\n    inlineData === undefined &&\n    fileData === undefined\n  ) {\n    throw new Error(\n      'Invalid ADK artifact load response: expected an object containing \"text\", \"inlineData\", or \"fileData\".',\n    );\n  }\n\n  if (text !== undefined && typeof text !== \"string\") {\n    throw new Error(\n      'Invalid ADK artifact load response: \"text\" must be a string when present.',\n    );\n  }\n\n  if (\n    inlineData !== undefined &&\n    (!isRecord(inlineData) ||\n      typeof inlineData.mimeType !== \"string\" ||\n      typeof inlineData.data !== \"string\")\n  ) {\n    throw new Error(\n      'Invalid ADK artifact load response: \"inlineData\" must contain string \"mimeType\" and \"data\" fields.',\n    );\n  }\n\n  if (\n    fileData !== undefined &&\n    (!isRecord(fileData) ||\n      typeof fileData.fileUri !== \"string\" ||\n      (fileData.mimeType !== undefined &&\n        typeof fileData.mimeType !== \"string\"))\n  ) {\n    throw new Error(\n      'Invalid ADK artifact load response: \"fileData\" must contain a string \"fileUri\" and an optional string \"mimeType\" field.',\n    );\n  }\n\n  return normalizedValue as AdkArtifactData;\n};\n\nconst parseAdkArtifactVersionsResponse = (value: unknown): number[] => {\n  if (!Array.isArray(value)) {\n    throw new Error(\n      \"Invalid ADK artifact versions response: expected an array of version numbers.\",\n    );\n  }\n\n  return value.map((version, index) => {\n    if (\n      typeof version !== \"number\" ||\n      !Number.isInteger(version) ||\n      version < 0\n    ) {\n      throw new Error(\n        `Invalid ADK artifact versions response: version at index ${index} must be a non-negative integer.`,\n      );\n    }\n    return version;\n  });\n};\n\n/**\n * Creates a `RemoteThreadListAdapter` backed by ADK's session REST API,\n * plus a `load` function that reconstructs messages from session events.\n *\n * @example\n * ```ts\n * const { adapter, load } = createAdkSessionAdapter({\n *   apiUrl: \"http://localhost:8000\",\n *   appName: \"my-app\",\n *   userId: \"user-1\",\n * });\n *\n * const runtime = useAdkRuntime({\n *   stream: createAdkStream({ ... }),\n *   sessionAdapter: adapter,\n *   load,\n * });\n * ```\n */\nexport function createAdkSessionAdapter(\n  options: AdkSessionAdapterOptions,\n): AdkSessionAdapterResult {\n  const { apiUrl, appName, userId } = options;\n  const normalizedApiUrl = trimTrailingSlashes(apiUrl);\n  const baseUrl = `${normalizedApiUrl}/apps/${encodeURIComponent(appName)}/users/${encodeURIComponent(userId)}/sessions`;\n\n  const getHeaders = async (): Promise<Record<string, string>> => {\n    if (!options.headers) return {};\n    if (typeof options.headers === \"function\") return await options.headers();\n    return options.headers;\n  };\n\n  const adapter: RemoteThreadListAdapter = {\n    async list(): Promise<RemoteThreadListResponse> {\n      const headers = await getHeaders();\n      const res = await fetch(baseUrl, { headers });\n      if (!res.ok) {\n        throw new Error(`Failed to list sessions: ${res.status}`);\n      }\n      const data = parseAdkSessionListResponse(await res.json());\n\n      const threads: RemoteThreadMetadata[] = data.map((session) => ({\n        status: \"regular\" as const,\n        remoteId: session.id,\n        externalId: session.id,\n        title: undefined,\n      }));\n\n      return { threads };\n    },\n\n    async initialize(\n      _threadId: string,\n    ): Promise<RemoteThreadInitializeResponse> {\n      const headers = await getHeaders();\n      const res = await fetch(baseUrl, {\n        method: \"POST\",\n        headers: { \"Content-Type\": \"application/json\", ...headers },\n        body: JSON.stringify({}),\n      });\n      if (!res.ok) {\n        throw new Error(`Failed to create session: ${res.status}`);\n      }\n      const session = parseAdkSessionResponse(await res.json(), \"create\");\n      return { remoteId: session.id, externalId: session.id };\n    },\n\n    async delete(remoteId: string): Promise<void> {\n      const headers = await getHeaders();\n      const res = await fetch(`${baseUrl}/${encodeURIComponent(remoteId)}`, {\n        method: \"DELETE\",\n        headers,\n      });\n      if (!res.ok && res.status !== 404) {\n        throw new Error(`Failed to delete session: ${res.status}`);\n      }\n    },\n\n    async rename(): Promise<void> {\n      // ADK sessions don't support titles\n    },\n\n    async archive(): Promise<void> {\n      // ADK sessions don't support archiving\n    },\n\n    async unarchive(): Promise<void> {\n      // ADK sessions don't support archiving\n    },\n\n    generateTitle(): Promise<AssistantStream> {\n      // Title generation not supported without assistant-cloud\n      return Promise.resolve(\n        new ReadableStream<AssistantStreamChunk>({\n          start(controller) {\n            controller.close();\n          },\n        }),\n      );\n    },\n\n    async fetch(threadId: string): Promise<RemoteThreadMetadata> {\n      const headers = await getHeaders();\n      const res = await fetch(`${baseUrl}/${encodeURIComponent(threadId)}`, {\n        headers,\n      });\n      if (!res.ok) {\n        throw new Error(`Session not found: ${res.status}`);\n      }\n      const session = parseAdkSessionResponse(await res.json(), \"fetch\");\n      return {\n        status: \"regular\",\n        remoteId: session.id,\n        externalId: session.id,\n        title: undefined,\n      };\n    },\n  };\n\n  const load = async (\n    sessionId: string,\n    options?: { signal?: AbortSignal | undefined },\n  ): Promise<AdkThreadSnapshot> => {\n    const headers = await getHeaders();\n    const res = await fetch(`${baseUrl}/${encodeURIComponent(sessionId)}`, {\n      headers,\n      ...(options?.signal ? { signal: options.signal } : {}),\n    });\n    if (!res.ok) {\n      throw new Error(`Failed to load session: ${res.status}`);\n    }\n    const session = parseAdkSessionResponse(await res.json(), \"load\");\n\n    if (session.events !== undefined && !Array.isArray(session.events)) {\n      throw new Error(\n        'Invalid ADK session load response: expected \"events\" to be an array when present.',\n      );\n    }\n\n    // Events carry ordered state and tool transitions, so partial replay can\n    // produce a plausible but incorrect thread.\n    const events = session.events?.map((event, index) =>\n      parseAdkEventValue(event, `Invalid ADK session event at index ${index}`),\n    );\n\n    if (!events?.length) {\n      return { messages: [] };\n    }\n\n    const accumulator = new AdkEventAccumulator();\n    let messages: AdkMessage[] = [];\n    for (const event of events) {\n      messages = accumulator.processEvent(event);\n    }\n    // The per-turn state rides along, so a refetch can swap the thread over in\n    // one commit instead of reconstructing it from the messages alone.\n    return {\n      messages,\n      longRunningToolIds: accumulator.getLongRunningToolIds(),\n      toolConfirmations: accumulator.getToolConfirmations(),\n      authRequests: accumulator.getAuthRequests(),\n      escalated: accumulator.isEscalated(),\n      messageMetadata: accumulator.getMessageMetadata(),\n      stateDelta: accumulator.getStateDelta(),\n      artifactDelta: accumulator.getArtifactDelta(),\n      agentInfo: accumulator.getAgentInfo(),\n    };\n  };\n\n  const artifactBaseUrl = (sessionId: string) =>\n    `${baseUrl}/${encodeURIComponent(sessionId)}/artifacts`;\n\n  const artifacts: AdkSessionAdapterResult[\"artifacts\"] = {\n    async list(sessionId) {\n      const headers = await getHeaders();\n      const res = await fetch(artifactBaseUrl(sessionId), { headers });\n      if (!res.ok) throw new Error(`Failed to list artifacts: ${res.status}`);\n      return parseAdkArtifactListResponse(await res.json());\n    },\n\n    async load(sessionId, artifactName, version?) {\n      const headers = await getHeaders();\n      let url = `${artifactBaseUrl(sessionId)}/${encodeURIComponent(artifactName)}`;\n      if (version != null) url += `/versions/${version}`;\n      const res = await fetch(url, { headers });\n      if (!res.ok) throw new Error(`Failed to load artifact: ${res.status}`);\n      return parseAdkArtifactResponse(await res.json());\n    },\n\n    async listVersions(sessionId, artifactName) {\n      const headers = await getHeaders();\n      const url = `${artifactBaseUrl(sessionId)}/${encodeURIComponent(artifactName)}/versions`;\n      const res = await fetch(url, { headers });\n      if (!res.ok) {\n        throw new Error(`Failed to list artifact versions: ${res.status}`);\n      }\n      return parseAdkArtifactVersionsResponse(await res.json());\n    },\n\n    async delete(sessionId, artifactName) {\n      const headers = await getHeaders();\n      const url = `${artifactBaseUrl(sessionId)}/${encodeURIComponent(artifactName)}`;\n      const res = await fetch(url, { method: \"DELETE\", headers });\n      if (!res.ok && res.status !== 404) {\n        throw new Error(`Failed to delete artifact: ${res.status}`);\n      }\n    },\n  };\n\n  return { adapter, load, artifacts };\n}\n"],"mappings":";;;;;AAsEA,MAAM,YAAY,UAChB,OAAO,UAAU,YAAY,UAAU,QAAQ,CAAC,MAAM,QAAQ,KAAK;AAErE,MAAM,wBAAwB,UAAgD;CAC5E,IAAI,CAAC,SAAS,KAAK,GAAG,OAAO;CAE7B,MAAM,KAAK,MAAM;CACjB,OAAO,OAAO,OAAO,YAAY,GAAG,SAAS;AAC/C;AAEA,MAAM,2BACJ,OACA,cACuB;CACvB,IAAI,CAAC,qBAAqB,KAAK,GAC7B,MAAM,IAAI,MACR,uBAAuB,UAAU,4DACnC;CAEF,OAAO;AACT;AAEA,MAAM,+BAA+B,UAAyC;CAC5E,IAAI,CAAC,MAAM,QAAQ,KAAK,GACtB,MAAM,IAAI,MACR,mEACF;CAGF,OAAO,MAAM,KAAK,SAAS,UAAU;EACnC,IAAI,CAAC,qBAAqB,OAAO,GAC/B,MAAM,IAAI,MACR,uDAAuD,MAAM,oCAC/D;EAEF,OAAO;CACT,CAAC;AACH;AAEA,MAAM,gCAAgC,UAA6B;CACjE,IAAI,CAAC,MAAM,QAAQ,KAAK,GACtB,MAAM,IAAI,MACR,0EACF;CAGF,OAAO,MAAM,KAAK,UAAU,UAAU;EACpC,IAAI,OAAO,aAAa,YAAY,SAAS,SAAS,GACpD,OAAO;EAGT,IAAI,SAAS,QAAQ,GAAG;GACtB,MAAM,WAAW,SAAS;GAC1B,IAAI,OAAO,aAAa,YAAY,SAAS,SAAS,GACpD,OAAO;EAEX;EAEA,MAAM,IAAI,MACR,yDAAyD,MAAM,6EACjE;CACF,CAAC;AACH;AAEA,MAAM,4BAA4B,UAAoC;CACpE,IAAI,CAAC,SAAS,KAAK,GACjB,MAAM,IAAI,MACR,8GACF;CAGF,MAAM,kBAAkB,iBAAiB,KAAK;CAC9C,MAAM,EAAE,MAAM,YAAY,aAAa;CACvC,IACE,SAAS,KAAA,KACT,eAAe,KAAA,KACf,aAAa,KAAA,GAEb,MAAM,IAAI,MACR,8GACF;CAGF,IAAI,SAAS,KAAA,KAAa,OAAO,SAAS,UACxC,MAAM,IAAI,MACR,6EACF;CAGF,IACE,eAAe,KAAA,MACd,CAAC,SAAS,UAAU,KACnB,OAAO,WAAW,aAAa,YAC/B,OAAO,WAAW,SAAS,WAE7B,MAAM,IAAI,MACR,0GACF;CAGF,IACE,aAAa,KAAA,MACZ,CAAC,SAAS,QAAQ,KACjB,OAAO,SAAS,YAAY,YAC3B,SAAS,aAAa,KAAA,KACrB,OAAO,SAAS,aAAa,WAEjC,MAAM,IAAI,MACR,+HACF;CAGF,OAAO;AACT;AAEA,MAAM,oCAAoC,UAA6B;CACrE,IAAI,CAAC,MAAM,QAAQ,KAAK,GACtB,MAAM,IAAI,MACR,+EACF;CAGF,OAAO,MAAM,KAAK,SAAS,UAAU;EACnC,IACE,OAAO,YAAY,YACnB,CAAC,OAAO,UAAU,OAAO,KACzB,UAAU,GAEV,MAAM,IAAI,MACR,4DAA4D,MAAM,iCACpE;EAEF,OAAO;CACT,CAAC;AACH;;;;;;;;;;;;;;;;;;;;AAqBA,SAAgB,wBACd,SACyB;CACzB,MAAM,EAAE,QAAQ,SAAS,WAAW;CAEpC,MAAM,UAAU,GADS,oBAAoB,MACX,EAAE,QAAQ,mBAAmB,OAAO,EAAE,SAAS,mBAAmB,MAAM,EAAE;CAE5G,MAAM,aAAa,YAA6C;EAC9D,IAAI,CAAC,QAAQ,SAAS,OAAO,CAAC;EAC9B,IAAI,OAAO,QAAQ,YAAY,YAAY,OAAO,MAAM,QAAQ,QAAQ;EACxE,OAAO,QAAQ;CACjB;CAEA,MAAM,UAAmC;EACvC,MAAM,OAA0C;GAC9C,MAAM,UAAU,MAAM,WAAW;GACjC,MAAM,MAAM,MAAM,MAAM,SAAS,EAAE,QAAQ,CAAC;GAC5C,IAAI,CAAC,IAAI,IACP,MAAM,IAAI,MAAM,4BAA4B,IAAI,QAAQ;GAW1D,OAAO,EAAE,SATI,4BAA4B,MAAM,IAAI,KAAK,CAEb,CAAC,CAAC,KAAK,aAAa;IAC7D,QAAQ;IACR,UAAU,QAAQ;IAClB,YAAY,QAAQ;IACpB,OAAO,KAAA;GACT,EAEe,EAAE;EACnB;EAEA,MAAM,WACJ,WACyC;GACzC,MAAM,UAAU,MAAM,WAAW;GACjC,MAAM,MAAM,MAAM,MAAM,SAAS;IAC/B,QAAQ;IACR,SAAS;KAAE,gBAAgB;KAAoB,GAAG;IAAQ;IAC1D,MAAM,KAAK,UAAU,CAAC,CAAC;GACzB,CAAC;GACD,IAAI,CAAC,IAAI,IACP,MAAM,IAAI,MAAM,6BAA6B,IAAI,QAAQ;GAE3D,MAAM,UAAU,wBAAwB,MAAM,IAAI,KAAK,GAAG,QAAQ;GAClE,OAAO;IAAE,UAAU,QAAQ;IAAI,YAAY,QAAQ;GAAG;EACxD;EAEA,MAAM,OAAO,UAAiC;GAC5C,MAAM,UAAU,MAAM,WAAW;GACjC,MAAM,MAAM,MAAM,MAAM,GAAG,QAAQ,GAAG,mBAAmB,QAAQ,KAAK;IACpE,QAAQ;IACR;GACF,CAAC;GACD,IAAI,CAAC,IAAI,MAAM,IAAI,WAAW,KAC5B,MAAM,IAAI,MAAM,6BAA6B,IAAI,QAAQ;EAE7D;EAEA,MAAM,SAAwB,CAE9B;EAEA,MAAM,UAAyB,CAE/B;EAEA,MAAM,YAA2B,CAEjC;EAEA,gBAA0C;GAExC,OAAO,QAAQ,QACb,IAAI,eAAqC,EACvC,MAAM,YAAY;IAChB,WAAW,MAAM;GACnB,EACF,CAAC,CACH;EACF;EAEA,MAAM,MAAM,UAAiD;GAC3D,MAAM,UAAU,MAAM,WAAW;GACjC,MAAM,MAAM,MAAM,MAAM,GAAG,QAAQ,GAAG,mBAAmB,QAAQ,KAAK,EACpE,QACF,CAAC;GACD,IAAI,CAAC,IAAI,IACP,MAAM,IAAI,MAAM,sBAAsB,IAAI,QAAQ;GAEpD,MAAM,UAAU,wBAAwB,MAAM,IAAI,KAAK,GAAG,OAAO;GACjE,OAAO;IACL,QAAQ;IACR,UAAU,QAAQ;IAClB,YAAY,QAAQ;IACpB,OAAO,KAAA;GACT;EACF;CACF;CAEA,MAAM,OAAO,OACX,WACA,YAC+B;EAC/B,MAAM,UAAU,MAAM,WAAW;EACjC,MAAM,MAAM,MAAM,MAAM,GAAG,QAAQ,GAAG,mBAAmB,SAAS,KAAK;GACrE;GACA,GAAI,SAAS,SAAS,EAAE,QAAQ,QAAQ,OAAO,IAAI,CAAC;EACtD,CAAC;EACD,IAAI,CAAC,IAAI,IACP,MAAM,IAAI,MAAM,2BAA2B,IAAI,QAAQ;EAEzD,MAAM,UAAU,wBAAwB,MAAM,IAAI,KAAK,GAAG,MAAM;EAEhE,IAAI,QAAQ,WAAW,KAAA,KAAa,CAAC,MAAM,QAAQ,QAAQ,MAAM,GAC/D,MAAM,IAAI,MACR,qFACF;EAKF,MAAM,SAAS,QAAQ,QAAQ,KAAK,OAAO,UACzC,mBAAmB,OAAO,sCAAsC,OAAO,CACzE;EAEA,IAAI,CAAC,QAAQ,QACX,OAAO,EAAE,UAAU,CAAC,EAAE;EAGxB,MAAM,cAAc,IAAI,oBAAoB;EAC5C,IAAI,WAAyB,CAAC;EAC9B,KAAK,MAAM,SAAS,QAClB,WAAW,YAAY,aAAa,KAAK;EAI3C,OAAO;GACL;GACA,oBAAoB,YAAY,sBAAsB;GACtD,mBAAmB,YAAY,qBAAqB;GACpD,cAAc,YAAY,gBAAgB;GAC1C,WAAW,YAAY,YAAY;GACnC,iBAAiB,YAAY,mBAAmB;GAChD,YAAY,YAAY,cAAc;GACtC,eAAe,YAAY,iBAAiB;GAC5C,WAAW,YAAY,aAAa;EACtC;CACF;CAEA,MAAM,mBAAmB,cACvB,GAAG,QAAQ,GAAG,mBAAmB,SAAS,EAAE;CAuC9C,OAAO;EAAE;EAAS;EAAM,WAAA;GApCtB,MAAM,KAAK,WAAW;IACpB,MAAM,UAAU,MAAM,WAAW;IACjC,MAAM,MAAM,MAAM,MAAM,gBAAgB,SAAS,GAAG,EAAE,QAAQ,CAAC;IAC/D,IAAI,CAAC,IAAI,IAAI,MAAM,IAAI,MAAM,6BAA6B,IAAI,QAAQ;IACtE,OAAO,6BAA6B,MAAM,IAAI,KAAK,CAAC;GACtD;GAEA,MAAM,KAAK,WAAW,cAAc,SAAU;IAC5C,MAAM,UAAU,MAAM,WAAW;IACjC,IAAI,MAAM,GAAG,gBAAgB,SAAS,EAAE,GAAG,mBAAmB,YAAY;IAC1E,IAAI,WAAW,MAAM,OAAO,aAAa;IACzC,MAAM,MAAM,MAAM,MAAM,KAAK,EAAE,QAAQ,CAAC;IACxC,IAAI,CAAC,IAAI,IAAI,MAAM,IAAI,MAAM,4BAA4B,IAAI,QAAQ;IACrE,OAAO,yBAAyB,MAAM,IAAI,KAAK,CAAC;GAClD;GAEA,MAAM,aAAa,WAAW,cAAc;IAC1C,MAAM,UAAU,MAAM,WAAW;IACjC,MAAM,MAAM,GAAG,gBAAgB,SAAS,EAAE,GAAG,mBAAmB,YAAY,EAAE;IAC9E,MAAM,MAAM,MAAM,MAAM,KAAK,EAAE,QAAQ,CAAC;IACxC,IAAI,CAAC,IAAI,IACP,MAAM,IAAI,MAAM,qCAAqC,IAAI,QAAQ;IAEnE,OAAO,iCAAiC,MAAM,IAAI,KAAK,CAAC;GAC1D;GAEA,MAAM,OAAO,WAAW,cAAc;IACpC,MAAM,UAAU,MAAM,WAAW;IACjC,MAAM,MAAM,GAAG,gBAAgB,SAAS,EAAE,GAAG,mBAAmB,YAAY;IAC5E,MAAM,MAAM,MAAM,MAAM,KAAK;KAAE,QAAQ;KAAU;IAAQ,CAAC;IAC1D,IAAI,CAAC,IAAI,MAAM,IAAI,WAAW,KAC5B,MAAM,IAAI,MAAM,8BAA8B,IAAI,QAAQ;GAE9D;EAG8B;CAAE;AACpC"}
+{"version":3,"file":"AdkSessionAdapter.js","names":[],"sources":["../src/AdkSessionAdapter.ts"],"sourcesContent":["import type { AssistantStream, AssistantStreamChunk } from \"assistant-stream\";\nimport type {\n  RemoteThreadInitializeResponse,\n  RemoteThreadListAdapter,\n  RemoteThreadListResponse,\n  RemoteThreadMetadata,\n} from \"@assistant-ui/core\";\nimport { AdkEventAccumulator } from \"./AdkEventAccumulator\";\nimport { normalizeAdkPart } from \"./normalizeAdkPart\";\nimport { parseAdkEventValue } from \"./parseAdkEvent\";\nimport type { AdkMessage, AdkThreadSnapshot } from \"./types\";\nimport { trimTrailingSlashes } from \"./trimTrailingSlashes\";\nimport { raceWithAbortSignal } from \"./raceWithAbortSignal\";\n\nexport type AdkSessionAdapterOptions = {\n  /**\n   * ADK server base URL (e.g. \"http://localhost:8000\").\n   */\n  apiUrl: string;\n\n  /**\n   * ADK application name.\n   */\n  appName: string;\n\n  /**\n   * ADK user ID.\n   */\n  userId: string;\n\n  /**\n   * Extra headers for API requests.\n   */\n  headers?:\n    | Record<string, string>\n    | (() => Record<string, string> | Promise<Record<string, string>>)\n    | undefined;\n};\n\nexport type AdkArtifactData = {\n  inlineData?: { mimeType: string; data: string } | undefined;\n  fileData?: { fileUri: string; mimeType?: string | undefined } | undefined;\n  text?: string | undefined;\n};\n\ntype AdkSessionAdapterResult = {\n  adapter: RemoteThreadListAdapter;\n  load: (\n    sessionId: string,\n    options?: { signal?: AbortSignal | undefined },\n  ) => Promise<AdkThreadSnapshot>;\n  artifacts: {\n    list: (sessionId: string) => Promise<string[]>;\n    load: (\n      sessionId: string,\n      artifactName: string,\n      version?: number,\n    ) => Promise<AdkArtifactData>;\n    listVersions: (\n      sessionId: string,\n      artifactName: string,\n    ) => Promise<number[]>;\n    delete: (sessionId: string, artifactName: string) => Promise<void>;\n  };\n};\n\ntype AdkSessionResponse = {\n  id: string;\n  events?: unknown;\n};\n\nconst isRecord = (value: unknown): value is Record<string, unknown> =>\n  typeof value === \"object\" && value !== null && !Array.isArray(value);\n\nconst isAdkSessionResponse = (value: unknown): value is AdkSessionResponse => {\n  if (!isRecord(value)) return false;\n\n  const id = value.id;\n  return typeof id === \"string\" && id.length > 0;\n};\n\nconst parseAdkSessionResponse = (\n  value: unknown,\n  operation: \"create\" | \"fetch\" | \"load\",\n): AdkSessionResponse => {\n  if (!isAdkSessionResponse(value)) {\n    throw new Error(\n      `Invalid ADK session ${operation} response: expected an object with a non-empty string \"id\".`,\n    );\n  }\n  return value;\n};\n\nconst parseAdkSessionListResponse = (value: unknown): AdkSessionResponse[] => {\n  if (!Array.isArray(value)) {\n    throw new Error(\n      \"Invalid ADK session list response: expected an array of sessions.\",\n    );\n  }\n\n  return value.map((session, index) => {\n    if (!isAdkSessionResponse(session)) {\n      throw new Error(\n        `Invalid ADK session list response: session at index ${index} must have a non-empty string \"id\".`,\n      );\n    }\n    return session;\n  });\n};\n\nconst parseAdkArtifactListResponse = (value: unknown): string[] => {\n  if (!Array.isArray(value)) {\n    throw new Error(\n      \"Invalid ADK artifact list response: expected an array of artifact names.\",\n    );\n  }\n\n  return value.map((artifact, index) => {\n    if (typeof artifact === \"string\" && artifact.length > 0) {\n      return artifact;\n    }\n\n    if (isRecord(artifact)) {\n      const filename = artifact.filename;\n      if (typeof filename === \"string\" && filename.length > 0) {\n        return filename;\n      }\n    }\n\n    throw new Error(\n      `Invalid ADK artifact list response: artifact at index ${index} must be a non-empty string or an object with a non-empty string \"filename\".`,\n    );\n  });\n};\n\nconst parseAdkArtifactResponse = (value: unknown): AdkArtifactData => {\n  if (!isRecord(value)) {\n    throw new Error(\n      'Invalid ADK artifact load response: expected an object containing \"text\", \"inlineData\", or \"fileData\".',\n    );\n  }\n\n  const normalizedValue = normalizeAdkPart(value);\n  const { text, inlineData, fileData } = normalizedValue;\n  if (\n    text === undefined &&\n    inlineData === undefined &&\n    fileData === undefined\n  ) {\n    throw new Error(\n      'Invalid ADK artifact load response: expected an object containing \"text\", \"inlineData\", or \"fileData\".',\n    );\n  }\n\n  if (text !== undefined && typeof text !== \"string\") {\n    throw new Error(\n      'Invalid ADK artifact load response: \"text\" must be a string when present.',\n    );\n  }\n\n  if (\n    inlineData !== undefined &&\n    (!isRecord(inlineData) ||\n      typeof inlineData.mimeType !== \"string\" ||\n      typeof inlineData.data !== \"string\")\n  ) {\n    throw new Error(\n      'Invalid ADK artifact load response: \"inlineData\" must contain string \"mimeType\" and \"data\" fields.',\n    );\n  }\n\n  if (\n    fileData !== undefined &&\n    (!isRecord(fileData) ||\n      typeof fileData.fileUri !== \"string\" ||\n      (fileData.mimeType !== undefined &&\n        typeof fileData.mimeType !== \"string\"))\n  ) {\n    throw new Error(\n      'Invalid ADK artifact load response: \"fileData\" must contain a string \"fileUri\" and an optional string \"mimeType\" field.',\n    );\n  }\n\n  return normalizedValue as AdkArtifactData;\n};\n\nconst parseAdkArtifactVersionsResponse = (value: unknown): number[] => {\n  if (!Array.isArray(value)) {\n    throw new Error(\n      \"Invalid ADK artifact versions response: expected an array of version numbers.\",\n    );\n  }\n\n  return value.map((version, index) => {\n    if (\n      typeof version !== \"number\" ||\n      !Number.isInteger(version) ||\n      version < 0\n    ) {\n      throw new Error(\n        `Invalid ADK artifact versions response: version at index ${index} must be a non-negative integer.`,\n      );\n    }\n    return version;\n  });\n};\n\n/**\n * Creates a `RemoteThreadListAdapter` backed by ADK's session REST API,\n * plus a `load` function that reconstructs messages from session events.\n *\n * @example\n * ```ts\n * const { adapter, load } = createAdkSessionAdapter({\n *   apiUrl: \"http://localhost:8000\",\n *   appName: \"my-app\",\n *   userId: \"user-1\",\n * });\n *\n * const runtime = useAdkRuntime({\n *   stream: createAdkStream({ ... }),\n *   sessionAdapter: adapter,\n *   load,\n * });\n * ```\n */\nexport function createAdkSessionAdapter(\n  options: AdkSessionAdapterOptions,\n): AdkSessionAdapterResult {\n  const { apiUrl, appName, userId } = options;\n  const normalizedApiUrl = trimTrailingSlashes(apiUrl);\n  const baseUrl = `${normalizedApiUrl}/apps/${encodeURIComponent(appName)}/users/${encodeURIComponent(userId)}/sessions`;\n\n  const getHeaders = async (\n    signal?: AbortSignal,\n  ): Promise<Record<string, string>> => {\n    if (!options.headers) return {};\n    if (typeof options.headers === \"function\") {\n      return await raceWithAbortSignal(signal, options.headers);\n    }\n    return options.headers;\n  };\n\n  const adapter: RemoteThreadListAdapter = {\n    async list(): Promise<RemoteThreadListResponse> {\n      const headers = await getHeaders();\n      const res = await fetch(baseUrl, { headers });\n      if (!res.ok) {\n        throw new Error(`Failed to list sessions: ${res.status}`);\n      }\n      const data = parseAdkSessionListResponse(await res.json());\n\n      const threads: RemoteThreadMetadata[] = data.map((session) => ({\n        status: \"regular\" as const,\n        remoteId: session.id,\n        externalId: session.id,\n        title: undefined,\n      }));\n\n      return { threads };\n    },\n\n    async initialize(\n      _threadId: string,\n    ): Promise<RemoteThreadInitializeResponse> {\n      const headers = await getHeaders();\n      const res = await fetch(baseUrl, {\n        method: \"POST\",\n        headers: { \"Content-Type\": \"application/json\", ...headers },\n        body: JSON.stringify({}),\n      });\n      if (!res.ok) {\n        throw new Error(`Failed to create session: ${res.status}`);\n      }\n      const session = parseAdkSessionResponse(await res.json(), \"create\");\n      return { remoteId: session.id, externalId: session.id };\n    },\n\n    async delete(remoteId: string): Promise<void> {\n      const headers = await getHeaders();\n      const res = await fetch(`${baseUrl}/${encodeURIComponent(remoteId)}`, {\n        method: \"DELETE\",\n        headers,\n      });\n      if (!res.ok && res.status !== 404) {\n        throw new Error(`Failed to delete session: ${res.status}`);\n      }\n    },\n\n    async rename(): Promise<void> {\n      // ADK sessions don't support titles\n    },\n\n    async archive(): Promise<void> {\n      // ADK sessions don't support archiving\n    },\n\n    async unarchive(): Promise<void> {\n      // ADK sessions don't support archiving\n    },\n\n    generateTitle(): Promise<AssistantStream> {\n      // Title generation not supported without assistant-cloud\n      return Promise.resolve(\n        new ReadableStream<AssistantStreamChunk>({\n          start(controller) {\n            controller.close();\n          },\n        }),\n      );\n    },\n\n    async fetch(threadId: string): Promise<RemoteThreadMetadata> {\n      const headers = await getHeaders();\n      const res = await fetch(`${baseUrl}/${encodeURIComponent(threadId)}`, {\n        headers,\n      });\n      if (!res.ok) {\n        throw new Error(`Session not found: ${res.status}`);\n      }\n      const session = parseAdkSessionResponse(await res.json(), \"fetch\");\n      return {\n        status: \"regular\",\n        remoteId: session.id,\n        externalId: session.id,\n        title: undefined,\n      };\n    },\n  };\n\n  const load = async (\n    sessionId: string,\n    options?: { signal?: AbortSignal | undefined },\n  ): Promise<AdkThreadSnapshot> => {\n    const headers = await getHeaders(options?.signal);\n    const res = await fetch(`${baseUrl}/${encodeURIComponent(sessionId)}`, {\n      headers,\n      ...(options?.signal ? { signal: options.signal } : {}),\n    });\n    if (!res.ok) {\n      throw new Error(`Failed to load session: ${res.status}`);\n    }\n    const session = parseAdkSessionResponse(await res.json(), \"load\");\n\n    if (session.events !== undefined && !Array.isArray(session.events)) {\n      throw new Error(\n        'Invalid ADK session load response: expected \"events\" to be an array when present.',\n      );\n    }\n\n    // Events carry ordered state and tool transitions, so partial replay can\n    // produce a plausible but incorrect thread.\n    const events = session.events?.map((event, index) =>\n      parseAdkEventValue(event, `Invalid ADK session event at index ${index}`),\n    );\n\n    if (!events?.length) {\n      return { messages: [] };\n    }\n\n    const accumulator = new AdkEventAccumulator();\n    let messages: AdkMessage[] = [];\n    for (const event of events) {\n      messages = accumulator.processEvent(event);\n    }\n    // The per-turn state rides along, so a refetch can swap the thread over in\n    // one commit instead of reconstructing it from the messages alone.\n    return {\n      messages,\n      longRunningToolIds: accumulator.getLongRunningToolIds(),\n      toolConfirmations: accumulator.getToolConfirmations(),\n      authRequests: accumulator.getAuthRequests(),\n      escalated: accumulator.isEscalated(),\n      messageMetadata: accumulator.getMessageMetadata(),\n      stateDelta: accumulator.getStateDelta(),\n      artifactDelta: accumulator.getArtifactDelta(),\n      agentInfo: accumulator.getAgentInfo(),\n    };\n  };\n\n  const artifactBaseUrl = (sessionId: string) =>\n    `${baseUrl}/${encodeURIComponent(sessionId)}/artifacts`;\n\n  const artifacts: AdkSessionAdapterResult[\"artifacts\"] = {\n    async list(sessionId) {\n      const headers = await getHeaders();\n      const res = await fetch(artifactBaseUrl(sessionId), { headers });\n      if (!res.ok) throw new Error(`Failed to list artifacts: ${res.status}`);\n      return parseAdkArtifactListResponse(await res.json());\n    },\n\n    async load(sessionId, artifactName, version?) {\n      const headers = await getHeaders();\n      let url = `${artifactBaseUrl(sessionId)}/${encodeURIComponent(artifactName)}`;\n      if (version != null) url += `/versions/${version}`;\n      const res = await fetch(url, { headers });\n      if (!res.ok) throw new Error(`Failed to load artifact: ${res.status}`);\n      return parseAdkArtifactResponse(await res.json());\n    },\n\n    async listVersions(sessionId, artifactName) {\n      const headers = await getHeaders();\n      const url = `${artifactBaseUrl(sessionId)}/${encodeURIComponent(artifactName)}/versions`;\n      const res = await fetch(url, { headers });\n      if (!res.ok) {\n        throw new Error(`Failed to list artifact versions: ${res.status}`);\n      }\n      return parseAdkArtifactVersionsResponse(await res.json());\n    },\n\n    async delete(sessionId, artifactName) {\n      const headers = await getHeaders();\n      const url = `${artifactBaseUrl(sessionId)}/${encodeURIComponent(artifactName)}`;\n      const res = await fetch(url, { method: \"DELETE\", headers });\n      if (!res.ok && res.status !== 404) {\n        throw new Error(`Failed to delete artifact: ${res.status}`);\n      }\n    },\n  };\n\n  return { adapter, load, artifacts };\n}\n"],"mappings":";;;;;;AAuEA,MAAM,YAAY,UAChB,OAAO,UAAU,YAAY,UAAU,QAAQ,CAAC,MAAM,QAAQ,KAAK;AAErE,MAAM,wBAAwB,UAAgD;CAC5E,IAAI,CAAC,SAAS,KAAK,GAAG,OAAO;CAE7B,MAAM,KAAK,MAAM;CACjB,OAAO,OAAO,OAAO,YAAY,GAAG,SAAS;AAC/C;AAEA,MAAM,2BACJ,OACA,cACuB;CACvB,IAAI,CAAC,qBAAqB,KAAK,GAC7B,MAAM,IAAI,MACR,uBAAuB,UAAU,4DACnC;CAEF,OAAO;AACT;AAEA,MAAM,+BAA+B,UAAyC;CAC5E,IAAI,CAAC,MAAM,QAAQ,KAAK,GACtB,MAAM,IAAI,MACR,mEACF;CAGF,OAAO,MAAM,KAAK,SAAS,UAAU;EACnC,IAAI,CAAC,qBAAqB,OAAO,GAC/B,MAAM,IAAI,MACR,uDAAuD,MAAM,oCAC/D;EAEF,OAAO;CACT,CAAC;AACH;AAEA,MAAM,gCAAgC,UAA6B;CACjE,IAAI,CAAC,MAAM,QAAQ,KAAK,GACtB,MAAM,IAAI,MACR,0EACF;CAGF,OAAO,MAAM,KAAK,UAAU,UAAU;EACpC,IAAI,OAAO,aAAa,YAAY,SAAS,SAAS,GACpD,OAAO;EAGT,IAAI,SAAS,QAAQ,GAAG;GACtB,MAAM,WAAW,SAAS;GAC1B,IAAI,OAAO,aAAa,YAAY,SAAS,SAAS,GACpD,OAAO;EAEX;EAEA,MAAM,IAAI,MACR,yDAAyD,MAAM,6EACjE;CACF,CAAC;AACH;AAEA,MAAM,4BAA4B,UAAoC;CACpE,IAAI,CAAC,SAAS,KAAK,GACjB,MAAM,IAAI,MACR,8GACF;CAGF,MAAM,kBAAkB,iBAAiB,KAAK;CAC9C,MAAM,EAAE,MAAM,YAAY,aAAa;CACvC,IACE,SAAS,KAAA,KACT,eAAe,KAAA,KACf,aAAa,KAAA,GAEb,MAAM,IAAI,MACR,8GACF;CAGF,IAAI,SAAS,KAAA,KAAa,OAAO,SAAS,UACxC,MAAM,IAAI,MACR,6EACF;CAGF,IACE,eAAe,KAAA,MACd,CAAC,SAAS,UAAU,KACnB,OAAO,WAAW,aAAa,YAC/B,OAAO,WAAW,SAAS,WAE7B,MAAM,IAAI,MACR,0GACF;CAGF,IACE,aAAa,KAAA,MACZ,CAAC,SAAS,QAAQ,KACjB,OAAO,SAAS,YAAY,YAC3B,SAAS,aAAa,KAAA,KACrB,OAAO,SAAS,aAAa,WAEjC,MAAM,IAAI,MACR,+HACF;CAGF,OAAO;AACT;AAEA,MAAM,oCAAoC,UAA6B;CACrE,IAAI,CAAC,MAAM,QAAQ,KAAK,GACtB,MAAM,IAAI,MACR,+EACF;CAGF,OAAO,MAAM,KAAK,SAAS,UAAU;EACnC,IACE,OAAO,YAAY,YACnB,CAAC,OAAO,UAAU,OAAO,KACzB,UAAU,GAEV,MAAM,IAAI,MACR,4DAA4D,MAAM,iCACpE;EAEF,OAAO;CACT,CAAC;AACH;;;;;;;;;;;;;;;;;;;;AAqBA,SAAgB,wBACd,SACyB;CACzB,MAAM,EAAE,QAAQ,SAAS,WAAW;CAEpC,MAAM,UAAU,GADS,oBAAoB,MACX,EAAE,QAAQ,mBAAmB,OAAO,EAAE,SAAS,mBAAmB,MAAM,EAAE;CAE5G,MAAM,aAAa,OACjB,WACoC;EACpC,IAAI,CAAC,QAAQ,SAAS,OAAO,CAAC;EAC9B,IAAI,OAAO,QAAQ,YAAY,YAC7B,OAAO,MAAM,oBAAoB,QAAQ,QAAQ,OAAO;EAE1D,OAAO,QAAQ;CACjB;CAEA,MAAM,UAAmC;EACvC,MAAM,OAA0C;GAC9C,MAAM,UAAU,MAAM,WAAW;GACjC,MAAM,MAAM,MAAM,MAAM,SAAS,EAAE,QAAQ,CAAC;GAC5C,IAAI,CAAC,IAAI,IACP,MAAM,IAAI,MAAM,4BAA4B,IAAI,QAAQ;GAW1D,OAAO,EAAE,SATI,4BAA4B,MAAM,IAAI,KAAK,CAEb,CAAC,CAAC,KAAK,aAAa;IAC7D,QAAQ;IACR,UAAU,QAAQ;IAClB,YAAY,QAAQ;IACpB,OAAO,KAAA;GACT,EAEe,EAAE;EACnB;EAEA,MAAM,WACJ,WACyC;GACzC,MAAM,UAAU,MAAM,WAAW;GACjC,MAAM,MAAM,MAAM,MAAM,SAAS;IAC/B,QAAQ;IACR,SAAS;KAAE,gBAAgB;KAAoB,GAAG;IAAQ;IAC1D,MAAM,KAAK,UAAU,CAAC,CAAC;GACzB,CAAC;GACD,IAAI,CAAC,IAAI,IACP,MAAM,IAAI,MAAM,6BAA6B,IAAI,QAAQ;GAE3D,MAAM,UAAU,wBAAwB,MAAM,IAAI,KAAK,GAAG,QAAQ;GAClE,OAAO;IAAE,UAAU,QAAQ;IAAI,YAAY,QAAQ;GAAG;EACxD;EAEA,MAAM,OAAO,UAAiC;GAC5C,MAAM,UAAU,MAAM,WAAW;GACjC,MAAM,MAAM,MAAM,MAAM,GAAG,QAAQ,GAAG,mBAAmB,QAAQ,KAAK;IACpE,QAAQ;IACR;GACF,CAAC;GACD,IAAI,CAAC,IAAI,MAAM,IAAI,WAAW,KAC5B,MAAM,IAAI,MAAM,6BAA6B,IAAI,QAAQ;EAE7D;EAEA,MAAM,SAAwB,CAE9B;EAEA,MAAM,UAAyB,CAE/B;EAEA,MAAM,YAA2B,CAEjC;EAEA,gBAA0C;GAExC,OAAO,QAAQ,QACb,IAAI,eAAqC,EACvC,MAAM,YAAY;IAChB,WAAW,MAAM;GACnB,EACF,CAAC,CACH;EACF;EAEA,MAAM,MAAM,UAAiD;GAC3D,MAAM,UAAU,MAAM,WAAW;GACjC,MAAM,MAAM,MAAM,MAAM,GAAG,QAAQ,GAAG,mBAAmB,QAAQ,KAAK,EACpE,QACF,CAAC;GACD,IAAI,CAAC,IAAI,IACP,MAAM,IAAI,MAAM,sBAAsB,IAAI,QAAQ;GAEpD,MAAM,UAAU,wBAAwB,MAAM,IAAI,KAAK,GAAG,OAAO;GACjE,OAAO;IACL,QAAQ;IACR,UAAU,QAAQ;IAClB,YAAY,QAAQ;IACpB,OAAO,KAAA;GACT;EACF;CACF;CAEA,MAAM,OAAO,OACX,WACA,YAC+B;EAC/B,MAAM,UAAU,MAAM,WAAW,SAAS,MAAM;EAChD,MAAM,MAAM,MAAM,MAAM,GAAG,QAAQ,GAAG,mBAAmB,SAAS,KAAK;GACrE;GACA,GAAI,SAAS,SAAS,EAAE,QAAQ,QAAQ,OAAO,IAAI,CAAC;EACtD,CAAC;EACD,IAAI,CAAC,IAAI,IACP,MAAM,IAAI,MAAM,2BAA2B,IAAI,QAAQ;EAEzD,MAAM,UAAU,wBAAwB,MAAM,IAAI,KAAK,GAAG,MAAM;EAEhE,IAAI,QAAQ,WAAW,KAAA,KAAa,CAAC,MAAM,QAAQ,QAAQ,MAAM,GAC/D,MAAM,IAAI,MACR,qFACF;EAKF,MAAM,SAAS,QAAQ,QAAQ,KAAK,OAAO,UACzC,mBAAmB,OAAO,sCAAsC,OAAO,CACzE;EAEA,IAAI,CAAC,QAAQ,QACX,OAAO,EAAE,UAAU,CAAC,EAAE;EAGxB,MAAM,cAAc,IAAI,oBAAoB;EAC5C,IAAI,WAAyB,CAAC;EAC9B,KAAK,MAAM,SAAS,QAClB,WAAW,YAAY,aAAa,KAAK;EAI3C,OAAO;GACL;GACA,oBAAoB,YAAY,sBAAsB;GACtD,mBAAmB,YAAY,qBAAqB;GACpD,cAAc,YAAY,gBAAgB;GAC1C,WAAW,YAAY,YAAY;GACnC,iBAAiB,YAAY,mBAAmB;GAChD,YAAY,YAAY,cAAc;GACtC,eAAe,YAAY,iBAAiB;GAC5C,WAAW,YAAY,aAAa;EACtC;CACF;CAEA,MAAM,mBAAmB,cACvB,GAAG,QAAQ,GAAG,mBAAmB,SAAS,EAAE;CAuC9C,OAAO;EAAE;EAAS;EAAM,WAAA;GApCtB,MAAM,KAAK,WAAW;IACpB,MAAM,UAAU,MAAM,WAAW;IACjC,MAAM,MAAM,MAAM,MAAM,gBAAgB,SAAS,GAAG,EAAE,QAAQ,CAAC;IAC/D,IAAI,CAAC,IAAI,IAAI,MAAM,IAAI,MAAM,6BAA6B,IAAI,QAAQ;IACtE,OAAO,6BAA6B,MAAM,IAAI,KAAK,CAAC;GACtD;GAEA,MAAM,KAAK,WAAW,cAAc,SAAU;IAC5C,MAAM,UAAU,MAAM,WAAW;IACjC,IAAI,MAAM,GAAG,gBAAgB,SAAS,EAAE,GAAG,mBAAmB,YAAY;IAC1E,IAAI,WAAW,MAAM,OAAO,aAAa;IACzC,MAAM,MAAM,MAAM,MAAM,KAAK,EAAE,QAAQ,CAAC;IACxC,IAAI,CAAC,IAAI,IAAI,MAAM,IAAI,MAAM,4BAA4B,IAAI,QAAQ;IACrE,OAAO,yBAAyB,MAAM,IAAI,KAAK,CAAC;GAClD;GAEA,MAAM,aAAa,WAAW,cAAc;IAC1C,MAAM,UAAU,MAAM,WAAW;IACjC,MAAM,MAAM,GAAG,gBAAgB,SAAS,EAAE,GAAG,mBAAmB,YAAY,EAAE;IAC9E,MAAM,MAAM,MAAM,MAAM,KAAK,EAAE,QAAQ,CAAC;IACxC,IAAI,CAAC,IAAI,IACP,MAAM,IAAI,MAAM,qCAAqC,IAAI,QAAQ;IAEnE,OAAO,iCAAiC,MAAM,IAAI,KAAK,CAAC;GAC1D;GAEA,MAAM,OAAO,WAAW,cAAc;IACpC,MAAM,UAAU,MAAM,WAAW;IACjC,MAAM,MAAM,GAAG,gBAAgB,SAAS,EAAE,GAAG,mBAAmB,YAAY;IAC5E,MAAM,MAAM,MAAM,MAAM,KAAK;KAAE,QAAQ;KAAU;IAAQ,CAAC;IAC1D,IAAI,CAAC,IAAI,MAAM,IAAI,WAAW,KAC5B,MAAM,IAAI,MAAM,8BAA8B,IAAI,QAAQ;GAE9D;EAG8B;CAAE;AACpC"}

package/dist/adkAuthRequest.d.ts

@@ -0,0 +1,17 @@
+import type { AdkAuthCredential, AdkAuthRequest, AdkMessage } from "./types.js";
+export declare const ADK_REQUEST_CREDENTIAL = "adk_request_credential";
+/**
+ * ADK binds a credential reply to the `adk_request_credential` call it issued
+ * and takes only `exchangedAuthCredential` off the response, so the reply is
+ * the request's own auth config with the credential filled in, as ADK's
+ * reference client sends it: adk-python validates the response as a whole auth
+ * config, and it exchanges an OAuth2 code with the client id, redirect uri and
+ * state the request carried, so the caller's fields land over the request's
+ * exchanged credential rather than replacing it. An id that is not a pending
+ * request would send a reply adk-js ignores and adk-python answers by
+ * requesting again, so it throws instead.
+ */
+export declare const toAdkAuthReply: (toolCallId: string, credential: AdkAuthCredential, requests: readonly AdkAuthRequest[]) => AdkMessage & {
+    type: "tool";
+};
+//# sourceMappingURL=adkAuthRequest.d.ts.map

package/dist/adkAuthRequest.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"adkAuthRequest.d.ts","sourceRoot":"","sources":["../src/adkAuthRequest.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,UAAU,EAAE,mBAAgB;AAE7E,eAAO,MAAM,sBAAsB,2BAA2B,CAAC;AAK/D;;;;;;;;;;GAUG;AACH,eAAO,MAAM,cAAc,GACzB,YAAY,MAAM,EAClB,YAAY,iBAAiB,EAC7B,UAAU,SAAS,cAAc,EAAE,KAClC,UAAU,GAAG;IAAE,IAAI,EAAE,MAAM,CAAA;CAyB7B,CAAC"}

package/dist/adkAuthRequest.js

@@ -0,0 +1,45 @@
+import { isRecord } from "@assistant-ui/core/internal";
+import { generateId } from "@assistant-ui/core";
+//#region src/adkAuthRequest.ts
+const ADK_REQUEST_CREDENTIAL = "adk_request_credential";
+const recordOf = (value) => isRecord(value) ? value : {};
+/**
+* ADK binds a credential reply to the `adk_request_credential` call it issued
+* and takes only `exchangedAuthCredential` off the response, so the reply is
+* the request's own auth config with the credential filled in, as ADK's
+* reference client sends it: adk-python validates the response as a whole auth
+* config, and it exchanges an OAuth2 code with the client id, redirect uri and
+* state the request carried, so the caller's fields land over the request's
+* exchanged credential rather than replacing it. An id that is not a pending
+* request would send a reply adk-js ignores and adk-python answers by
+* requesting again, so it throws instead.
+*/
+const toAdkAuthReply = (toolCallId, credential, requests) => {
+	const request = requests.find((r) => r.toolCallId === toolCallId);
+	if (request === void 0) throw new Error(`No pending ADK auth request for tool call id "${toolCallId}"`);
+	const authConfig = recordOf(request.authConfig);
+	const requested = recordOf(authConfig.exchangedAuthCredential);
+	const oauth2 = {
+		...recordOf(requested.oauth2),
+		...credential.oauth2
+	};
+	return {
+		id: generateId(),
+		type: "tool",
+		tool_call_id: toolCallId,
+		name: ADK_REQUEST_CREDENTIAL,
+		content: JSON.stringify({
+			...authConfig,
+			exchangedAuthCredential: {
+				...requested,
+				...credential,
+				...Object.keys(oauth2).length > 0 && { oauth2 }
+			}
+		}),
+		status: "success"
+	};
+};
+//#endregion
+export { ADK_REQUEST_CREDENTIAL, toAdkAuthReply };
+
+//# sourceMappingURL=adkAuthRequest.js.map

package/dist/adkAuthRequest.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"adkAuthRequest.js","names":[],"sources":["../src/adkAuthRequest.ts"],"sourcesContent":["import { generateId } from \"@assistant-ui/core\";\nimport { isRecord } from \"@assistant-ui/core/internal\";\nimport type { AdkAuthCredential, AdkAuthRequest, AdkMessage } from \"./types\";\n\nexport const ADK_REQUEST_CREDENTIAL = \"adk_request_credential\";\n\nconst recordOf = (value: unknown): Record<string, unknown> =>\n  isRecord(value) ? value : {};\n\n/**\n * ADK binds a credential reply to the `adk_request_credential` call it issued\n * and takes only `exchangedAuthCredential` off the response, so the reply is\n * the request's own auth config with the credential filled in, as ADK's\n * reference client sends it: adk-python validates the response as a whole auth\n * config, and it exchanges an OAuth2 code with the client id, redirect uri and\n * state the request carried, so the caller's fields land over the request's\n * exchanged credential rather than replacing it. An id that is not a pending\n * request would send a reply adk-js ignores and adk-python answers by\n * requesting again, so it throws instead.\n */\nexport const toAdkAuthReply = (\n  toolCallId: string,\n  credential: AdkAuthCredential,\n  requests: readonly AdkAuthRequest[],\n): AdkMessage & { type: \"tool\" } => {\n  const request = requests.find((r) => r.toolCallId === toolCallId);\n  if (request === undefined)\n    throw new Error(\n      `No pending ADK auth request for tool call id \"${toolCallId}\"`,\n    );\n\n  const authConfig = recordOf(request.authConfig);\n  const requested = recordOf(authConfig.exchangedAuthCredential);\n  const oauth2 = { ...recordOf(requested.oauth2), ...credential.oauth2 };\n  return {\n    id: generateId(),\n    type: \"tool\",\n    tool_call_id: toolCallId,\n    name: ADK_REQUEST_CREDENTIAL,\n    content: JSON.stringify({\n      ...authConfig,\n      exchangedAuthCredential: {\n        ...requested,\n        ...credential,\n        ...(Object.keys(oauth2).length > 0 && { oauth2 }),\n      },\n    }),\n    status: \"success\",\n  };\n};\n"],"mappings":";;;AAIA,MAAa,yBAAyB;AAEtC,MAAM,YAAY,UAChB,SAAS,KAAK,IAAI,QAAQ,CAAC;;;;;;;;;;;;AAa7B,MAAa,kBACX,YACA,YACA,aACkC;CAClC,MAAM,UAAU,SAAS,MAAM,MAAM,EAAE,eAAe,UAAU;CAChE,IAAI,YAAY,KAAA,GACd,MAAM,IAAI,MACR,iDAAiD,WAAW,EAC9D;CAEF,MAAM,aAAa,SAAS,QAAQ,UAAU;CAC9C,MAAM,YAAY,SAAS,WAAW,uBAAuB;CAC7D,MAAM,SAAS;EAAE,GAAG,SAAS,UAAU,MAAM;EAAG,GAAG,WAAW;CAAO;CACrE,OAAO;EACL,IAAI,WAAW;EACf,MAAM;EACN,cAAc;EACd,MAAM;EACN,SAAS,KAAK,UAAU;GACtB,GAAG;GACH,yBAAyB;IACvB,GAAG;IACH,GAAG;IACH,GAAI,OAAO,KAAK,MAAM,CAAC,CAAC,SAAS,KAAK,EAAE,OAAO;GACjD;EACF,CAAC;EACD,QAAQ;CACV;AACF"}

package/dist/adkExtras.d.ts

@@ -1,5 +1,3 @@
-import { AdkRuntimeExtras } from "./types.js";
-//#region src/adkExtras.d.ts
+import type { AdkRuntimeExtras } from "./types.js";
 export declare const adkExtras: import("@assistant-ui/core/react").RuntimeExtras<AdkRuntimeExtras>;
-//#endregion
 //# sourceMappingURL=adkExtras.d.ts.map

package/dist/adkExtras.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"adkExtras.d.ts","names":[],"sources":["../src/adkExtras.ts"],"mappings":";;qBAGa,8CAAS,cAAA"}
+{"version":3,"file":"adkExtras.d.ts","sourceRoot":"","sources":["../src/adkExtras.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,gBAAgB,EAAE,mBAAgB;AAEhD,eAAO,MAAM,SAAS,oEAAyD,CAAC"}

package/dist/adkToolApproval.d.ts

@@ -1,17 +1,16 @@
-import { AdkMessage } from "./types.js";
-import { ReadonlyJSONValue } from "assistant-stream/utils";
-import { RespondToToolApprovalOptions, ToolCallMessagePart } from "@assistant-ui/core";
-//#region src/adkToolApproval.d.ts
+import type { RespondToToolApprovalOptions, ToolCallMessagePart } from "@assistant-ui/core";
+import type { ReadonlyJSONValue } from "assistant-stream/utils";
+import type { AdkMessage } from "./types.js";
 export type AdkToolApproval = NonNullable<ToolCallMessagePart["approval"]>;
 export declare const ADK_REQUEST_CONFIRMATION = "adk_request_confirmation";
 export type AdkToolApprovalProjection = {
-  approvals: ReadonlyMap<string, AdkToolApproval>;
-  /**
-   * Identity of the decision state. Core re-converts a cached message object
-   * only when the converter callback changes, so the callback is rebuilt when
-   * this key moves rather than on every render.
-   */
-  key: string;
+    approvals: ReadonlyMap<string, AdkToolApproval>;
+    /**
+     * Identity of the decision state. Core re-converts a cached message object
+     * only when the converter callback changes, so the callback is rebuilt when
+     * this key moves rather than on every render.
+     */
+    key: string;
 };
 /**
  * ADK requests a confirmation by emitting a synthetic `adk_request_confirmation`
@@ -25,7 +24,7 @@ export type AdkToolApprovalProjection = {
  */
 export declare const projectAdkToolApprovals: (messages: readonly AdkMessage[]) => AdkToolApprovalProjection;
 export declare const toAdkConfirmationReply: (toolCallId: string, confirmed: boolean, payload?: ReadonlyJSONValue) => AdkMessage & {
-  type: "tool";
+    type: "tool";
 };
 /**
  * ADK confirmations are a plain confirm/reject pair, so no option or reason
@@ -33,7 +32,6 @@ export declare const toAdkConfirmationReply: (toolCallId: string, confirmed: boo
  * reply ADK silently ignores, so it throws instead.
  */
 export declare const toAdkToolConfirmationReply: ({ approvalId, approved }: RespondToToolApprovalOptions, approvals: ReadonlyMap<string, AdkToolApproval>) => AdkMessage & {
-  type: "tool";
+    type: "tool";
 };
-//#endregion
 //# sourceMappingURL=adkToolApproval.d.ts.map