@assistant-ui/react-google-adk 0.0.20 → 0.0.21

package/dist/AdkEventAccumulator.d.ts

@@ -19,6 +19,7 @@ declare class AdkEventAccumulator {
   private authRequests;
   private escalated;
   private messageMetadataMap;
+  private aiMessageOrdinals;
   constructor(initialMessages?: AdkMessage[]);
   processEvent(rawEvent: AdkEvent): AdkMessage[];
   private processPart;

package/dist/AdkEventAccumulator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AdkEventAccumulator.d.ts","names":[],"sources":["../src/AdkEventAccumulator.ts"],"mappings":";;;;;;cAuBa,kBAAmB,OAAO;cAqJ1B;UACH;UACA;UACA;UACA;UACA;UACA;UACA;UAIA;UACA;UACA;UACA;UACA;UACA;EACI,YAAA,kBAAkB;EAQ9B,aAAa,UAAU,WAAW;UAwL1B;UAmJA;UAmBA;UAMA;UAuBA;UASA;UAYA;UAeA;EAMR,eAAe;EAIf,iBAAiB;EAIjB,oBAAoB;EAIpB;IAAkB;IAA2B;;EAI7C;EAIA;EAIA,wBAAwB;EAIxB,mBAAmB;EAInB;EAIA,sBAAsB,YAAY"}
+{"version":3,"file":"AdkEventAccumulator.d.ts","names":[],"sources":["../src/AdkEventAccumulator.ts"],"mappings":";;;;;;cAwCa,kBAAmB,OAAO;cAqJ1B;UACH;UACA;UACA;UACA;UACA;UACA;UACA;UAIA;UACA;UACA;UACA;UACA;UACA;UAGA;EACI,YAAA,kBAAkB;EAQ9B,aAAa,UAAU,WAAW;UAwL1B;UAuJA;UAmBA;UAMA;UAyBA;UASA;UAYA;UAeA;EAMR,eAAe;EAIf,iBAAiB;EAIjB,oBAAoB;EAIpB;IAAkB;IAA2B;;EAI7C;EAIA;EAIA,wBAAwB;EAIxB,mBAAmB;EAInB;EAIA,sBAAsB,YAAY"}

package/dist/AdkEventAccumulator.js

@@ -1,5 +1,18 @@
 import { v4 } from "uuid";
 //#region src/AdkEventAccumulator.ts
+/**
+* A session load replays the stored events through a fresh accumulator, so a
+* message needs an id derived from the event that carries it rather than one
+* minted per replay. An event with no id of its own has never been through the
+* session and has nothing stable to derive from, so it keeps a generated one.
+*
+* A human message keeps the bare event id it has always had. The other kinds
+* take a suffixed namespace, since one event can carry several of them: a tool
+* message by the index of its part, an assistant message by how many this
+* event has already opened.
+*/
+const toolMessageId = (event, partIndex) => event.id ? `${event.id}:${partIndex}` : v4();
+const aiMessageId = (event, ordinal) => event.id ? `${event.id}:ai${ordinal === 0 ? "" : ordinal}` : v4();
 const ADK_REQUEST_CONFIRMATION = "adk_request_confirmation";
 const ADK_REQUEST_CREDENTIAL = "adk_request_credential";
 /**
@@ -120,6 +133,7 @@ var AdkEventAccumulator = class {
 	authRequests = [];
 	escalated = false;
 	messageMetadataMap = /* @__PURE__ */ new Map();
+	aiMessageOrdinals = /* @__PURE__ */ new Map();
 	constructor(initialMessages) {
 		if (initialMessages) for (const msg of initialMessages) this.messagesMap.set(msg.id, msg);
 	}
@@ -217,7 +231,7 @@ var AdkEventAccumulator = class {
 			const current = this.messagesMap.get(this.currentMessageId);
 			if (current && current.type === "ai" && current.author !== event.author) this.finalizeCurrentMessage();
 		}
-		for (const part of parts) this.processPart(part, event);
+		for (const [index, part] of parts.entries()) this.processPart(part, event, index);
 		this.trackMessageMetadata(event);
 		if (isFinalResponse(event) && this.currentMessageId) {
 			const msg = this.messagesMap.get(this.currentMessageId);
@@ -235,7 +249,7 @@ var AdkEventAccumulator = class {
 		if (!event.partial || isFinalResponse(event)) this.finalizeCurrentMessage();
 		return this.getMessages();
 	}
-	processPart(part, event) {
+	processPart(part, event, partIndex) {
 		if (part.functionCall && !event.partial) {
 			const name = part.functionCall.name;
 			if (name === ADK_REQUEST_CONFIRMATION) {
@@ -307,7 +321,7 @@ var AdkEventAccumulator = class {
 		if (part.functionResponse) {
 			this.finalizeCurrentMessage();
 			const toolMsg = {
-				id: v4(),
+				id: toolMessageId(event, partIndex),
 				type: "tool",
 				tool_call_id: part.functionResponse.id ?? "",
 				name: part.functionResponse.name,
@@ -361,7 +375,9 @@ var AdkEventAccumulator = class {
 			const existing = this.messagesMap.get(this.currentMessageId);
 			if (existing && existing.type === "ai") return existing;
 		}
-		const id = v4();
+		const ordinal = this.aiMessageOrdinals.get(event.id ?? "") ?? 0;
+		this.aiMessageOrdinals.set(event.id ?? "", ordinal + 1);
+		const id = aiMessageId(event, ordinal);
 		const msg = {
 			id,
 			type: "ai",

package/dist/AdkEventAccumulator.js.map

@@ -1 +1 @@
-{"version":3,"file":"AdkEventAccumulator.js","names":["uuidv4"],"sources":["../src/AdkEventAccumulator.ts"],"sourcesContent":["import { v4 as uuidv4 } from \"uuid\";\nimport type { MessageStatus } from \"@assistant-ui/core\";\nimport type {\n  AdkEvent,\n  AdkEventPart,\n  AdkMessage,\n  AdkMessageContentPart,\n  AdkToolCall,\n  AdkToolConfirmation,\n  AdkAuthRequest,\n  AdkMessageMetadata,\n} from \"./types\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\n\ntype InProgressMessage = AdkMessage & { type: \"ai\" };\n\nconst ADK_REQUEST_CONFIRMATION = \"adk_request_confirmation\";\nconst ADK_REQUEST_CREDENTIAL = \"adk_request_credential\";\n\n/**\n * Checks if an event is a final response using the same logic as ADK's\n * `isFinalResponse()`.\n */\nexport const isFinalResponse = (event: AdkEvent): boolean => {\n  if (\n    event.actions?.skipSummarization ||\n    (event.longRunningToolIds && event.longRunningToolIds.length > 0)\n  ) {\n    return true;\n  }\n\n  const parts = event.content?.parts;\n  return (\n    !event.partial &&\n    (parts?.length ?? 0) > 0 &&\n    !parts!.some((p) => p.functionCall) &&\n    !parts!.some((p) => p.functionResponse) &&\n    !parts![parts!.length - 1]?.codeExecutionResult\n  );\n};\n\nconst CONTENT_FILTER_REASONS = new Set([\n  \"SAFETY\",\n  \"RECITATION\",\n  \"BLOCKLIST\",\n  \"PROHIBITED_CONTENT\",\n  \"SPII\",\n  \"LANGUAGE\",\n  \"IMAGE_SAFETY\",\n  \"IMAGE_RECITATION\",\n  \"IMAGE_OTHER\",\n  \"IMAGE_PROHIBITED_CONTENT\",\n]);\n\nconst ERROR_REASONS = new Set([\n  \"MALFORMED_FUNCTION_CALL\",\n  \"UNEXPECTED_TOOL_CALL\",\n]);\n\nconst finishReasonToStatus = (\n  finishReason: string | undefined,\n): MessageStatus => {\n  if (finishReason === \"MAX_TOKENS\") {\n    return { type: \"incomplete\", reason: \"length\" };\n  }\n  if (finishReason && CONTENT_FILTER_REASONS.has(finishReason)) {\n    return {\n      type: \"incomplete\",\n      reason: \"content-filter\",\n      error: `Content filtered: ${finishReason}`,\n    };\n  }\n  if (finishReason && ERROR_REASONS.has(finishReason)) {\n    return {\n      type: \"incomplete\",\n      reason: \"error\",\n      error: `LLM error: ${finishReason}`,\n    };\n  }\n  return { type: \"complete\", reason: \"stop\" };\n};\n\nconst inlineDataToPart = (\n  mimeType: string,\n  data: string,\n): AdkMessageContentPart =>\n  mimeType.startsWith(\"image/\")\n    ? { type: \"image\", mimeType, data }\n    : { type: \"file\", mimeType, data };\n\nconst fileDataToPart = (\n  fileUri: string,\n  mimeType: string | undefined,\n): AdkMessageContentPart =>\n  mimeType == null || mimeType.startsWith(\"image/\")\n    ? { type: \"image_url\", url: fileUri }\n    : { type: \"file_url\", url: fileUri, mimeType };\n\n// ── Snake_case normalization ──\n\nconst normalizeEventPart = (part: AdkEventPart): AdkEventPart => {\n  const p = part as Record<string, unknown>;\n  const result: Record<string, unknown> = { ...p };\n  if (\"function_call\" in p && !(\"functionCall\" in p))\n    result.functionCall = p.function_call;\n  if (\"function_response\" in p && !(\"functionResponse\" in p))\n    result.functionResponse = p.function_response;\n  if (\"inline_data\" in p && !(\"inlineData\" in p))\n    result.inlineData = p.inline_data;\n  if (\"file_data\" in p && !(\"fileData\" in p)) result.fileData = p.file_data;\n  if (\"executable_code\" in p && !(\"executableCode\" in p))\n    result.executableCode = p.executable_code;\n  if (\"code_execution_result\" in p && !(\"codeExecutionResult\" in p))\n    result.codeExecutionResult = p.code_execution_result;\n  return result as AdkEventPart;\n};\n\nconst normalizeEvent = (event: AdkEvent): AdkEvent => {\n  const e = event as Record<string, unknown>;\n  const result: Record<string, unknown> = { ...e };\n  if (\"error_code\" in e && !(\"errorCode\" in e)) result.errorCode = e.error_code;\n  if (\"error_message\" in e && !(\"errorMessage\" in e))\n    result.errorMessage = e.error_message;\n  if (\"long_running_tool_ids\" in e && !(\"longRunningToolIds\" in e))\n    result.longRunningToolIds = e.long_running_tool_ids;\n  if (\"turn_complete\" in e && !(\"turnComplete\" in e))\n    result.turnComplete = e.turn_complete;\n  if (\"finish_reason\" in e && !(\"finishReason\" in e))\n    result.finishReason = e.finish_reason;\n  if (\"invocation_id\" in e && !(\"invocationId\" in e))\n    result.invocationId = e.invocation_id;\n  if (\"custom_metadata\" in e && !(\"customMetadata\" in e))\n    result.customMetadata = e.custom_metadata;\n  if (\"grounding_metadata\" in e && !(\"groundingMetadata\" in e))\n    result.groundingMetadata = e.grounding_metadata;\n  if (\"citation_metadata\" in e && !(\"citationMetadata\" in e))\n    result.citationMetadata = e.citation_metadata;\n  if (\"usage_metadata\" in e && !(\"usageMetadata\" in e))\n    result.usageMetadata = e.usage_metadata;\n\n  if (result.actions) {\n    const a = result.actions as Record<string, unknown>;\n    const na: Record<string, unknown> = { ...a };\n    if (\"state_delta\" in a && !(\"stateDelta\" in a))\n      na.stateDelta = a.state_delta;\n    if (\"artifact_delta\" in a && !(\"artifactDelta\" in a))\n      na.artifactDelta = a.artifact_delta;\n    if (\"transfer_to_agent\" in a && !(\"transferToAgent\" in a))\n      na.transferToAgent = a.transfer_to_agent;\n    if (\"skip_summarization\" in a && !(\"skipSummarization\" in a))\n      na.skipSummarization = a.skip_summarization;\n    if (\"requested_auth_configs\" in a && !(\"requestedAuthConfigs\" in a))\n      na.requestedAuthConfigs = a.requested_auth_configs;\n    if (\n      \"requested_tool_confirmations\" in a &&\n      !(\"requestedToolConfirmations\" in a)\n    )\n      na.requestedToolConfirmations = a.requested_tool_confirmations;\n    result.actions = na;\n  }\n\n  if (result.content && (result.content as Record<string, unknown>).parts) {\n    const content = result.content as Record<string, unknown>;\n    const parts = content.parts as AdkEventPart[];\n    result.content = { ...content, parts: parts.map(normalizeEventPart) };\n  }\n\n  return result as AdkEvent;\n};\n\n// ── Accumulator ──\n\nexport class AdkEventAccumulator {\n  private messagesMap = new Map<string, AdkMessage>();\n  private currentMessageId: string | null = null;\n  private partialTextBuffer = \"\";\n  private partialReasoningBuffer = \"\";\n  private accumulatedStateDelta: Record<string, unknown> = {};\n  private accumulatedArtifactDelta: Record<string, number> = {};\n  private lastAgentInfo: {\n    name?: string | undefined;\n    branch?: string | undefined;\n  } = {};\n  private lastTransferToAgent: string | undefined;\n  private pendingLongRunningToolIds = new Set<string>();\n  private toolConfirmations: AdkToolConfirmation[] = [];\n  private authRequests: AdkAuthRequest[] = [];\n  private escalated = false;\n  private messageMetadataMap = new Map<string, AdkMessageMetadata>();\n  constructor(initialMessages?: AdkMessage[]) {\n    if (initialMessages) {\n      for (const msg of initialMessages) {\n        this.messagesMap.set(msg.id, msg);\n      }\n    }\n  }\n\n  processEvent(rawEvent: AdkEvent): AdkMessage[] {\n    const event = normalizeEvent(rawEvent);\n\n    // Accumulate state delta\n    if (event.actions?.stateDelta) {\n      Object.assign(this.accumulatedStateDelta, event.actions.stateDelta);\n    }\n\n    // Accumulate artifact delta\n    if (event.actions?.artifactDelta) {\n      Object.assign(this.accumulatedArtifactDelta, event.actions.artifactDelta);\n    }\n\n    // Track escalation\n    if (event.actions?.escalate) {\n      this.escalated = true;\n    }\n\n    // Track agent transfer\n    if (event.actions?.transferToAgent) {\n      this.lastTransferToAgent = event.actions.transferToAgent;\n    }\n\n    // Track long-running tool IDs. Accumulate across events so that multiple\n    // HITL interrupts emitted in the same turn are all tracked — a\n    // single-event replacement would drop earlier ids.\n    if (event.longRunningToolIds?.length) {\n      for (const id of event.longRunningToolIds) {\n        this.pendingLongRunningToolIds.add(id);\n      }\n    }\n\n    // Track tool confirmations from actions\n    if (event.actions?.requestedToolConfirmations) {\n      for (const [tcId, conf] of Object.entries(\n        event.actions.requestedToolConfirmations,\n      )) {\n        const c = conf as Record<string, unknown>;\n        this.toolConfirmations.push({\n          toolCallId: tcId,\n          toolName: \"\",\n          args: {},\n          hint: (c.hint as string) ?? \"\",\n          confirmed: false,\n          payload: c.payload,\n        });\n      }\n    }\n\n    // Track auth requests from actions\n    if (event.actions?.requestedAuthConfigs) {\n      for (const [tcId, authConf] of Object.entries(\n        event.actions.requestedAuthConfigs,\n      )) {\n        this.authRequests.push({ toolCallId: tcId, authConfig: authConf });\n      }\n    }\n\n    // Track agent info\n    if (event.author && event.author !== \"user\") {\n      this.lastAgentInfo = {\n        name: event.author ?? undefined,\n        branch: event.branch ?? undefined,\n      };\n    }\n\n    // Handle interrupted events\n    if (event.interrupted) {\n      if (this.currentMessageId) {\n        const msg = this.messagesMap.get(this.currentMessageId);\n        if (msg && msg.type === \"ai\" && !msg.status) {\n          const updated: InProgressMessage = {\n            ...msg,\n            content: [...this.getContentArray(msg)],\n            status: { type: \"incomplete\", reason: \"cancelled\" },\n          };\n          this.messagesMap.set(updated.id, updated);\n        }\n      }\n      this.finalizeCurrentMessage();\n      return this.getMessages();\n    }\n\n    // Handle error events\n    if (event.errorCode || event.errorMessage) {\n      this.finalizeCurrentMessage();\n      const errorMsg = this.getOrCreateAiMessage(event);\n      const errorText =\n        event.errorMessage ?? event.errorCode ?? \"Unknown error\";\n      const updated: InProgressMessage = {\n        ...errorMsg,\n        content: [\n          ...this.getContentArray(errorMsg),\n          { type: \"text\", text: errorText },\n        ],\n        status: { type: \"incomplete\", reason: \"error\", error: errorText },\n      };\n      this.messagesMap.set(updated.id, updated);\n      this.currentMessageId = null;\n      return this.getMessages();\n    }\n\n    const parts = event.content?.parts;\n    if (!parts?.length) {\n      // Track metadata even for content-less events (e.g. turnComplete)\n      this.trackMessageMetadata(event);\n      return this.getMessages();\n    }\n\n    // User-authored events → create human message, not AI.\n    // Without this, user events fall through to processPart →\n    // getOrCreateAiMessage, producing type:\"ai\" messages that\n    // convertAdkMessage maps to role:\"assistant\".\n    if (event.author === \"user\") {\n      this.finalizeCurrentMessage();\n      const humanParts: AdkMessageContentPart[] = [];\n      for (const part of parts) {\n        if (part.text != null && !part.thought) {\n          humanParts.push({ type: \"text\", text: part.text });\n        } else if (part.inlineData) {\n          humanParts.push(\n            inlineDataToPart(part.inlineData.mimeType, part.inlineData.data),\n          );\n        } else if (part.fileData) {\n          humanParts.push(\n            fileDataToPart(part.fileData.fileUri, part.fileData.mimeType),\n          );\n        }\n      }\n      if (humanParts.length > 0) {\n        const id = event.id ?? uuidv4();\n        const first = humanParts[0];\n        const content: string | AdkMessageContentPart[] =\n          humanParts.length === 1 && first?.type === \"text\"\n            ? first.text\n            : humanParts;\n        this.messagesMap.set(id, { id, type: \"human\", content });\n      }\n      return this.getMessages();\n    }\n\n    // If author changed, finalize previous message\n    if (this.currentMessageId && event.author && event.author !== \"user\") {\n      const current = this.messagesMap.get(this.currentMessageId);\n      if (current && current.type === \"ai\" && current.author !== event.author) {\n        this.finalizeCurrentMessage();\n      }\n    }\n\n    for (const part of parts) {\n      this.processPart(part, event);\n    }\n\n    // Track per-message metadata (grounding, citation, usage)\n    this.trackMessageMetadata(event);\n\n    // Check isFinalResponse (can be true even for partial events via skipSummarization/longRunningToolIds)\n    if (isFinalResponse(event) && this.currentMessageId) {\n      const msg = this.messagesMap.get(this.currentMessageId);\n      // Skip manual \"complete\" when longRunningToolIds is the sole reason for\n      // isFinalResponse — let auto-status apply requires-action for pending tool calls.\n      const isHitlOnly =\n        event.longRunningToolIds &&\n        event.longRunningToolIds.length > 0 &&\n        !event.actions?.skipSummarization;\n      if (msg && msg.type === \"ai\" && !msg.status && !isHitlOnly) {\n        const status = finishReasonToStatus(event.finishReason);\n        const updated: InProgressMessage = {\n          ...msg,\n          content: [...this.getContentArray(msg)],\n          status,\n        };\n        this.messagesMap.set(updated.id, updated);\n      }\n    }\n\n    // Non-partial event finalizes the current message\n    if (!event.partial || isFinalResponse(event)) {\n      this.finalizeCurrentMessage();\n    }\n\n    return this.getMessages();\n  }\n\n  private processPart(part: AdkEventPart, event: AdkEvent): void {\n    // Detect special ADK function calls\n    if (part.functionCall && !event.partial) {\n      const name = part.functionCall.name;\n\n      // Tool confirmation request\n      if (name === ADK_REQUEST_CONFIRMATION) {\n        const callArgs = part.functionCall.args;\n        // ADK JS: args keys are \"originalFunctionCall\" and \"toolConfirmation\"\n        // ADK Python: args keys are \"original_function_call\" and \"tool_confirmation\"\n        const original =\n          (callArgs.originalFunctionCall as Record<string, unknown>) ??\n          (callArgs.original_function_call as Record<string, unknown>);\n        const conf =\n          (callArgs.toolConfirmation as Record<string, unknown>) ??\n          (callArgs.tool_confirmation as Record<string, unknown>);\n        this.toolConfirmations.push({\n          toolCallId: part.functionCall.id ?? \"\",\n          toolName: (original?.name as string) ?? \"\",\n          args: (original?.args as Record<string, unknown>) ?? {},\n          hint: (conf?.hint as string) ?? \"\",\n          confirmed: false,\n          payload: conf?.payload,\n        });\n      }\n\n      // Auth credential request\n      if (name === ADK_REQUEST_CREDENTIAL) {\n        const credArgs = part.functionCall.args;\n        // ADK JS: args keys are \"function_call_id\" and \"auth_config\"\n        const originalToolCallId =\n          (credArgs.function_call_id as string) ?? part.functionCall.id ?? \"\";\n        const authConfig = credArgs.auth_config ?? credArgs;\n        this.authRequests.push({\n          toolCallId: originalToolCallId,\n          authConfig,\n        });\n      }\n    }\n\n    // Text with thought=true → reasoning (accumulated)\n    if (part.text != null && part.thought) {\n      const msg = this.getOrCreateAiMessage(event);\n      if (event.partial) {\n        this.partialReasoningBuffer += part.text;\n        this.replaceLastReasoningContent(msg, this.partialReasoningBuffer);\n      } else {\n        this.partialReasoningBuffer = \"\";\n        this.replaceLastReasoningContent(msg, part.text);\n      }\n      return;\n    }\n\n    // Regular text\n    if (part.text != null) {\n      const msg = this.getOrCreateAiMessage(event);\n      if (event.partial) {\n        this.partialTextBuffer += part.text;\n        this.replaceLastTextContent(msg, this.partialTextBuffer);\n      } else {\n        this.partialTextBuffer = \"\";\n        this.replaceLastTextContent(msg, part.text);\n      }\n      return;\n    }\n\n    // Function call — skip partial events (args may be incomplete)\n    if (part.functionCall) {\n      if (event.partial) return;\n      const msg = this.getOrCreateAiMessage(event);\n      const toolCall: AdkToolCall = {\n        id: part.functionCall.id ?? uuidv4(),\n        name: part.functionCall.name,\n        args: part.functionCall.args as ReadonlyJSONObject,\n        argsText: JSON.stringify(part.functionCall.args),\n      };\n      const existing = [...(msg.tool_calls ?? [])];\n      const idx = existing.findIndex((tc) => tc.id === toolCall.id);\n      if (idx >= 0) {\n        existing[idx] = toolCall;\n      } else {\n        existing.push(toolCall);\n      }\n      const updated: InProgressMessage = {\n        ...msg,\n        content: [...this.getContentArray(msg)],\n        tool_calls: existing,\n      };\n      this.messagesMap.set(updated.id, updated);\n      return;\n    }\n\n    // Function response → tool message\n    if (part.functionResponse) {\n      this.finalizeCurrentMessage();\n      const toolMsg: AdkMessage = {\n        id: uuidv4(),\n        type: \"tool\",\n        tool_call_id: part.functionResponse.id ?? \"\",\n        name: part.functionResponse.name,\n        content: JSON.stringify(part.functionResponse.response),\n        status: \"success\",\n      };\n      this.messagesMap.set(toolMsg.id, toolMsg);\n      return;\n    }\n\n    // Executable code\n    if (part.executableCode) {\n      const msg = this.getOrCreateAiMessage(event);\n      this.appendContent(msg, {\n        type: \"code\",\n        code: part.executableCode.code,\n        language: part.executableCode.language ?? \"python\",\n      });\n      return;\n    }\n\n    // Code execution result\n    if (part.codeExecutionResult) {\n      const msg = this.getOrCreateAiMessage(event);\n      this.appendContent(msg, {\n        type: \"code_result\",\n        output: part.codeExecutionResult.output,\n        outcome: part.codeExecutionResult.outcome ?? \"OUTCOME_OK\",\n      });\n      return;\n    }\n\n    if (part.inlineData) {\n      const msg = this.getOrCreateAiMessage(event);\n      this.appendContent(\n        msg,\n        inlineDataToPart(part.inlineData.mimeType, part.inlineData.data),\n      );\n      return;\n    }\n\n    if (part.fileData) {\n      const msg = this.getOrCreateAiMessage(event);\n      this.appendContent(\n        msg,\n        fileDataToPart(part.fileData.fileUri, part.fileData.mimeType),\n      );\n    }\n  }\n\n  private trackMessageMetadata(event: AdkEvent): void {\n    if (\n      !this.currentMessageId ||\n      (!event.groundingMetadata &&\n        !event.citationMetadata &&\n        !event.usageMetadata)\n    )\n      return;\n\n    const existing = this.messageMetadataMap.get(this.currentMessageId) ?? {};\n    const updated: AdkMessageMetadata = { ...existing };\n    if (event.groundingMetadata)\n      updated.groundingMetadata = event.groundingMetadata;\n    if (event.citationMetadata)\n      updated.citationMetadata = event.citationMetadata;\n    if (event.usageMetadata) updated.usageMetadata = event.usageMetadata;\n    this.messageMetadataMap.set(this.currentMessageId, updated);\n  }\n\n  private getContentArray(msg: AdkMessage): AdkMessageContentPart[] {\n    return Array.isArray(msg.content)\n      ? (msg.content as AdkMessageContentPart[])\n      : [];\n  }\n\n  private getOrCreateAiMessage(event: AdkEvent): InProgressMessage {\n    if (this.currentMessageId) {\n      const existing = this.messagesMap.get(this.currentMessageId);\n      if (existing && existing.type === \"ai\") {\n        return existing;\n      }\n    }\n\n    const id = uuidv4();\n    const msg: InProgressMessage = {\n      id,\n      type: \"ai\",\n      content: [] as AdkMessageContentPart[],\n      ...(event.author != null && { author: event.author }),\n      ...(event.branch != null && { branch: event.branch }),\n    };\n    this.messagesMap.set(id, msg);\n    this.currentMessageId = id;\n    this.partialTextBuffer = \"\";\n    this.partialReasoningBuffer = \"\";\n    return msg;\n  }\n\n  private appendContent(\n    msg: InProgressMessage,\n    part: AdkMessageContentPart,\n  ): void {\n    const content = [...this.getContentArray(msg), part];\n    const updated: InProgressMessage = { ...msg, content };\n    this.messagesMap.set(updated.id, updated);\n  }\n\n  private replaceLastTextContent(msg: InProgressMessage, text: string): void {\n    const content = [...this.getContentArray(msg)];\n    const lastTextIdx = content.findLastIndex((p) => p.type === \"text\");\n    if (lastTextIdx >= 0) {\n      content[lastTextIdx] = { type: \"text\", text };\n    } else {\n      content.push({ type: \"text\", text });\n    }\n    const updated: InProgressMessage = { ...msg, content };\n    this.messagesMap.set(updated.id, updated);\n  }\n\n  private replaceLastReasoningContent(\n    msg: InProgressMessage,\n    text: string,\n  ): void {\n    const content = [...this.getContentArray(msg)];\n    const lastIdx = content.findLastIndex((p) => p.type === \"reasoning\");\n    if (lastIdx >= 0) {\n      content[lastIdx] = { type: \"reasoning\", text };\n    } else {\n      content.push({ type: \"reasoning\", text });\n    }\n    const updated: InProgressMessage = { ...msg, content };\n    this.messagesMap.set(updated.id, updated);\n  }\n\n  private finalizeCurrentMessage(): void {\n    this.partialTextBuffer = \"\";\n    this.partialReasoningBuffer = \"\";\n    this.currentMessageId = null;\n  }\n\n  getMessages(): AdkMessage[] {\n    return [...this.messagesMap.values()];\n  }\n\n  getStateDelta(): Record<string, unknown> {\n    return { ...this.accumulatedStateDelta };\n  }\n\n  getArtifactDelta(): Record<string, number> {\n    return { ...this.accumulatedArtifactDelta };\n  }\n\n  getAgentInfo(): { name?: string | undefined; branch?: string | undefined } {\n    return { ...this.lastAgentInfo };\n  }\n\n  getLastTransferToAgent(): string | undefined {\n    return this.lastTransferToAgent;\n  }\n\n  getLongRunningToolIds(): string[] {\n    return [...this.pendingLongRunningToolIds];\n  }\n\n  getToolConfirmations(): AdkToolConfirmation[] {\n    return [...this.toolConfirmations];\n  }\n\n  getAuthRequests(): AdkAuthRequest[] {\n    return [...this.authRequests];\n  }\n\n  isEscalated(): boolean {\n    return this.escalated;\n  }\n\n  getMessageMetadata(): Map<string, AdkMessageMetadata> {\n    return new Map(this.messageMetadataMap);\n  }\n}\n"],"mappings":";;AAgBA,MAAM,2BAA2B;AACjC,MAAM,yBAAyB;;;;;AAM/B,MAAa,mBAAmB,UAA6B;CAC3D,IACE,MAAM,SAAS,qBACd,MAAM,sBAAsB,MAAM,mBAAmB,SAAS,GAE/D,OAAO;CAGT,MAAM,QAAQ,MAAM,SAAS;CAC7B,OACE,CAAC,MAAM,YACN,OAAO,UAAU,KAAK,KACvB,CAAC,MAAO,MAAM,MAAM,EAAE,YAAY,KAClC,CAAC,MAAO,MAAM,MAAM,EAAE,gBAAgB,KACtC,CAAC,MAAO,MAAO,SAAS,EAAE,EAAE;AAEhC;AAEA,MAAM,yCAAyB,IAAI,IAAI;CACrC;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;AACF,CAAC;AAED,MAAM,gCAAgB,IAAI,IAAI,CAC5B,2BACA,sBACF,CAAC;AAED,MAAM,wBACJ,iBACkB;CAClB,IAAI,iBAAiB,cACnB,OAAO;EAAE,MAAM;EAAc,QAAQ;CAAS;CAEhD,IAAI,gBAAgB,uBAAuB,IAAI,YAAY,GACzD,OAAO;EACL,MAAM;EACN,QAAQ;EACR,OAAO,qBAAqB;CAC9B;CAEF,IAAI,gBAAgB,cAAc,IAAI,YAAY,GAChD,OAAO;EACL,MAAM;EACN,QAAQ;EACR,OAAO,cAAc;CACvB;CAEF,OAAO;EAAE,MAAM;EAAY,QAAQ;CAAO;AAC5C;AAEA,MAAM,oBACJ,UACA,SAEA,SAAS,WAAW,QAAQ,IACxB;CAAE,MAAM;CAAS;CAAU;AAAK,IAChC;CAAE,MAAM;CAAQ;CAAU;AAAK;AAErC,MAAM,kBACJ,SACA,aAEA,YAAY,QAAQ,SAAS,WAAW,QAAQ,IAC5C;CAAE,MAAM;CAAa,KAAK;AAAQ,IAClC;CAAE,MAAM;CAAY,KAAK;CAAS;AAAS;AAIjD,MAAM,sBAAsB,SAAqC;CAC/D,MAAM,IAAI;CACV,MAAM,SAAkC,EAAE,GAAG,EAAE;CAC/C,IAAI,mBAAmB,KAAK,EAAE,kBAAkB,IAC9C,OAAO,eAAe,EAAE;CAC1B,IAAI,uBAAuB,KAAK,EAAE,sBAAsB,IACtD,OAAO,mBAAmB,EAAE;CAC9B,IAAI,iBAAiB,KAAK,EAAE,gBAAgB,IAC1C,OAAO,aAAa,EAAE;CACxB,IAAI,eAAe,KAAK,EAAE,cAAc,IAAI,OAAO,WAAW,EAAE;CAChE,IAAI,qBAAqB,KAAK,EAAE,oBAAoB,IAClD,OAAO,iBAAiB,EAAE;CAC5B,IAAI,2BAA2B,KAAK,EAAE,yBAAyB,IAC7D,OAAO,sBAAsB,EAAE;CACjC,OAAO;AACT;AAEA,MAAM,kBAAkB,UAA8B;CACpD,MAAM,IAAI;CACV,MAAM,SAAkC,EAAE,GAAG,EAAE;CAC/C,IAAI,gBAAgB,KAAK,EAAE,eAAe,IAAI,OAAO,YAAY,EAAE;CACnE,IAAI,mBAAmB,KAAK,EAAE,kBAAkB,IAC9C,OAAO,eAAe,EAAE;CAC1B,IAAI,2BAA2B,KAAK,EAAE,wBAAwB,IAC5D,OAAO,qBAAqB,EAAE;CAChC,IAAI,mBAAmB,KAAK,EAAE,kBAAkB,IAC9C,OAAO,eAAe,EAAE;CAC1B,IAAI,mBAAmB,KAAK,EAAE,kBAAkB,IAC9C,OAAO,eAAe,EAAE;CAC1B,IAAI,mBAAmB,KAAK,EAAE,kBAAkB,IAC9C,OAAO,eAAe,EAAE;CAC1B,IAAI,qBAAqB,KAAK,EAAE,oBAAoB,IAClD,OAAO,iBAAiB,EAAE;CAC5B,IAAI,wBAAwB,KAAK,EAAE,uBAAuB,IACxD,OAAO,oBAAoB,EAAE;CAC/B,IAAI,uBAAuB,KAAK,EAAE,sBAAsB,IACtD,OAAO,mBAAmB,EAAE;CAC9B,IAAI,oBAAoB,KAAK,EAAE,mBAAmB,IAChD,OAAO,gBAAgB,EAAE;CAE3B,IAAI,OAAO,SAAS;EAClB,MAAM,IAAI,OAAO;EACjB,MAAM,KAA8B,EAAE,GAAG,EAAE;EAC3C,IAAI,iBAAiB,KAAK,EAAE,gBAAgB,IAC1C,GAAG,aAAa,EAAE;EACpB,IAAI,oBAAoB,KAAK,EAAE,mBAAmB,IAChD,GAAG,gBAAgB,EAAE;EACvB,IAAI,uBAAuB,KAAK,EAAE,qBAAqB,IACrD,GAAG,kBAAkB,EAAE;EACzB,IAAI,wBAAwB,KAAK,EAAE,uBAAuB,IACxD,GAAG,oBAAoB,EAAE;EAC3B,IAAI,4BAA4B,KAAK,EAAE,0BAA0B,IAC/D,GAAG,uBAAuB,EAAE;EAC9B,IACE,kCAAkC,KAClC,EAAE,gCAAgC,IAElC,GAAG,6BAA6B,EAAE;EACpC,OAAO,UAAU;CACnB;CAEA,IAAI,OAAO,WAAY,OAAO,QAAoC,OAAO;EACvE,MAAM,UAAU,OAAO;EACvB,MAAM,QAAQ,QAAQ;EACtB,OAAO,UAAU;GAAE,GAAG;GAAS,OAAO,MAAM,IAAI,kBAAkB;EAAE;CACtE;CAEA,OAAO;AACT;AAIA,IAAa,sBAAb,MAAiC;CAC/B,8BAAsB,IAAI,IAAwB;CAClD,mBAA0C;CAC1C,oBAA4B;CAC5B,yBAAiC;CACjC,wBAAyD,CAAC;CAC1D,2BAA2D,CAAC;CAC5D,gBAGI,CAAC;CACL;CACA,4CAAoC,IAAI,IAAY;CACpD,oBAAmD,CAAC;CACpD,eAAyC,CAAC;CAC1C,YAAoB;CACpB,qCAA6B,IAAI,IAAgC;CACjE,YAAY,iBAAgC;EAC1C,IAAI,iBACF,KAAK,MAAM,OAAO,iBAChB,KAAK,YAAY,IAAI,IAAI,IAAI,GAAG;CAGtC;CAEA,aAAa,UAAkC;EAC7C,MAAM,QAAQ,eAAe,QAAQ;EAGrC,IAAI,MAAM,SAAS,YACjB,OAAO,OAAO,KAAK,uBAAuB,MAAM,QAAQ,UAAU;EAIpE,IAAI,MAAM,SAAS,eACjB,OAAO,OAAO,KAAK,0BAA0B,MAAM,QAAQ,aAAa;EAI1E,IAAI,MAAM,SAAS,UACjB,KAAK,YAAY;EAInB,IAAI,MAAM,SAAS,iBACjB,KAAK,sBAAsB,MAAM,QAAQ;EAM3C,IAAI,MAAM,oBAAoB,QAC5B,KAAK,MAAM,MAAM,MAAM,oBACrB,KAAK,0BAA0B,IAAI,EAAE;EAKzC,IAAI,MAAM,SAAS,4BACjB,KAAK,MAAM,CAAC,MAAM,SAAS,OAAO,QAChC,MAAM,QAAQ,0BAChB,GAAG;GACD,MAAM,IAAI;GACV,KAAK,kBAAkB,KAAK;IAC1B,YAAY;IACZ,UAAU;IACV,MAAM,CAAC;IACP,MAAO,EAAE,QAAmB;IAC5B,WAAW;IACX,SAAS,EAAE;GACb,CAAC;EACH;EAIF,IAAI,MAAM,SAAS,sBACjB,KAAK,MAAM,CAAC,MAAM,aAAa,OAAO,QACpC,MAAM,QAAQ,oBAChB,GACE,KAAK,aAAa,KAAK;GAAE,YAAY;GAAM,YAAY;EAAS,CAAC;EAKrE,IAAI,MAAM,UAAU,MAAM,WAAW,QACnC,KAAK,gBAAgB;GACnB,MAAM,MAAM,UAAU,KAAA;GACtB,QAAQ,MAAM,UAAU,KAAA;EAC1B;EAIF,IAAI,MAAM,aAAa;GACrB,IAAI,KAAK,kBAAkB;IACzB,MAAM,MAAM,KAAK,YAAY,IAAI,KAAK,gBAAgB;IACtD,IAAI,OAAO,IAAI,SAAS,QAAQ,CAAC,IAAI,QAAQ;KAC3C,MAAM,UAA6B;MACjC,GAAG;MACH,SAAS,CAAC,GAAG,KAAK,gBAAgB,GAAG,CAAC;MACtC,QAAQ;OAAE,MAAM;OAAc,QAAQ;MAAY;KACpD;KACA,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;IAC1C;GACF;GACA,KAAK,uBAAuB;GAC5B,OAAO,KAAK,YAAY;EAC1B;EAGA,IAAI,MAAM,aAAa,MAAM,cAAc;GACzC,KAAK,uBAAuB;GAC5B,MAAM,WAAW,KAAK,qBAAqB,KAAK;GAChD,MAAM,YACJ,MAAM,gBAAgB,MAAM,aAAa;GAC3C,MAAM,UAA6B;IACjC,GAAG;IACH,SAAS,CACP,GAAG,KAAK,gBAAgB,QAAQ,GAChC;KAAE,MAAM;KAAQ,MAAM;IAAU,CAClC;IACA,QAAQ;KAAE,MAAM;KAAc,QAAQ;KAAS,OAAO;IAAU;GAClE;GACA,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;GACxC,KAAK,mBAAmB;GACxB,OAAO,KAAK,YAAY;EAC1B;EAEA,MAAM,QAAQ,MAAM,SAAS;EAC7B,IAAI,CAAC,OAAO,QAAQ;GAElB,KAAK,qBAAqB,KAAK;GAC/B,OAAO,KAAK,YAAY;EAC1B;EAMA,IAAI,MAAM,WAAW,QAAQ;GAC3B,KAAK,uBAAuB;GAC5B,MAAM,aAAsC,CAAC;GAC7C,KAAK,MAAM,QAAQ,OACjB,IAAI,KAAK,QAAQ,QAAQ,CAAC,KAAK,SAC7B,WAAW,KAAK;IAAE,MAAM;IAAQ,MAAM,KAAK;GAAK,CAAC;QAC5C,IAAI,KAAK,YACd,WAAW,KACT,iBAAiB,KAAK,WAAW,UAAU,KAAK,WAAW,IAAI,CACjE;QACK,IAAI,KAAK,UACd,WAAW,KACT,eAAe,KAAK,SAAS,SAAS,KAAK,SAAS,QAAQ,CAC9D;GAGJ,IAAI,WAAW,SAAS,GAAG;IACzB,MAAM,KAAK,MAAM,MAAMA,GAAO;IAC9B,MAAM,QAAQ,WAAW;IACzB,MAAM,UACJ,WAAW,WAAW,KAAK,OAAO,SAAS,SACvC,MAAM,OACN;IACN,KAAK,YAAY,IAAI,IAAI;KAAE;KAAI,MAAM;KAAS;IAAQ,CAAC;GACzD;GACA,OAAO,KAAK,YAAY;EAC1B;EAGA,IAAI,KAAK,oBAAoB,MAAM,UAAU,MAAM,WAAW,QAAQ;GACpE,MAAM,UAAU,KAAK,YAAY,IAAI,KAAK,gBAAgB;GAC1D,IAAI,WAAW,QAAQ,SAAS,QAAQ,QAAQ,WAAW,MAAM,QAC/D,KAAK,uBAAuB;EAEhC;EAEA,KAAK,MAAM,QAAQ,OACjB,KAAK,YAAY,MAAM,KAAK;EAI9B,KAAK,qBAAqB,KAAK;EAG/B,IAAI,gBAAgB,KAAK,KAAK,KAAK,kBAAkB;GACnD,MAAM,MAAM,KAAK,YAAY,IAAI,KAAK,gBAAgB;GAGtD,MAAM,aACJ,MAAM,sBACN,MAAM,mBAAmB,SAAS,KAClC,CAAC,MAAM,SAAS;GAClB,IAAI,OAAO,IAAI,SAAS,QAAQ,CAAC,IAAI,UAAU,CAAC,YAAY;IAC1D,MAAM,SAAS,qBAAqB,MAAM,YAAY;IACtD,MAAM,UAA6B;KACjC,GAAG;KACH,SAAS,CAAC,GAAG,KAAK,gBAAgB,GAAG,CAAC;KACtC;IACF;IACA,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;GAC1C;EACF;EAGA,IAAI,CAAC,MAAM,WAAW,gBAAgB,KAAK,GACzC,KAAK,uBAAuB;EAG9B,OAAO,KAAK,YAAY;CAC1B;CAEA,YAAoB,MAAoB,OAAuB;EAE7D,IAAI,KAAK,gBAAgB,CAAC,MAAM,SAAS;GACvC,MAAM,OAAO,KAAK,aAAa;GAG/B,IAAI,SAAS,0BAA0B;IACrC,MAAM,WAAW,KAAK,aAAa;IAGnC,MAAM,WACH,SAAS,wBACT,SAAS;IACZ,MAAM,OACH,SAAS,oBACT,SAAS;IACZ,KAAK,kBAAkB,KAAK;KAC1B,YAAY,KAAK,aAAa,MAAM;KACpC,UAAW,UAAU,QAAmB;KACxC,MAAO,UAAU,QAAoC,CAAC;KACtD,MAAO,MAAM,QAAmB;KAChC,WAAW;KACX,SAAS,MAAM;IACjB,CAAC;GACH;GAGA,IAAI,SAAS,wBAAwB;IACnC,MAAM,WAAW,KAAK,aAAa;IAEnC,MAAM,qBACH,SAAS,oBAA+B,KAAK,aAAa,MAAM;IACnE,MAAM,aAAa,SAAS,eAAe;IAC3C,KAAK,aAAa,KAAK;KACrB,YAAY;KACZ;IACF,CAAC;GACH;EACF;EAGA,IAAI,KAAK,QAAQ,QAAQ,KAAK,SAAS;GACrC,MAAM,MAAM,KAAK,qBAAqB,KAAK;GAC3C,IAAI,MAAM,SAAS;IACjB,KAAK,0BAA0B,KAAK;IACpC,KAAK,4BAA4B,KAAK,KAAK,sBAAsB;GACnE,OAAO;IACL,KAAK,yBAAyB;IAC9B,KAAK,4BAA4B,KAAK,KAAK,IAAI;GACjD;GACA;EACF;EAGA,IAAI,KAAK,QAAQ,MAAM;GACrB,MAAM,MAAM,KAAK,qBAAqB,KAAK;GAC3C,IAAI,MAAM,SAAS;IACjB,KAAK,qBAAqB,KAAK;IAC/B,KAAK,uBAAuB,KAAK,KAAK,iBAAiB;GACzD,OAAO;IACL,KAAK,oBAAoB;IACzB,KAAK,uBAAuB,KAAK,KAAK,IAAI;GAC5C;GACA;EACF;EAGA,IAAI,KAAK,cAAc;GACrB,IAAI,MAAM,SAAS;GACnB,MAAM,MAAM,KAAK,qBAAqB,KAAK;GAC3C,MAAM,WAAwB;IAC5B,IAAI,KAAK,aAAa,MAAMA,GAAO;IACnC,MAAM,KAAK,aAAa;IACxB,MAAM,KAAK,aAAa;IACxB,UAAU,KAAK,UAAU,KAAK,aAAa,IAAI;GACjD;GACA,MAAM,WAAW,CAAC,GAAI,IAAI,cAAc,CAAC,CAAE;GAC3C,MAAM,MAAM,SAAS,WAAW,OAAO,GAAG,OAAO,SAAS,EAAE;GAC5D,IAAI,OAAO,GACT,SAAS,OAAO;QAEhB,SAAS,KAAK,QAAQ;GAExB,MAAM,UAA6B;IACjC,GAAG;IACH,SAAS,CAAC,GAAG,KAAK,gBAAgB,GAAG,CAAC;IACtC,YAAY;GACd;GACA,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;GACxC;EACF;EAGA,IAAI,KAAK,kBAAkB;GACzB,KAAK,uBAAuB;GAC5B,MAAM,UAAsB;IAC1B,IAAIA,GAAO;IACX,MAAM;IACN,cAAc,KAAK,iBAAiB,MAAM;IAC1C,MAAM,KAAK,iBAAiB;IAC5B,SAAS,KAAK,UAAU,KAAK,iBAAiB,QAAQ;IACtD,QAAQ;GACV;GACA,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;GACxC;EACF;EAGA,IAAI,KAAK,gBAAgB;GACvB,MAAM,MAAM,KAAK,qBAAqB,KAAK;GAC3C,KAAK,cAAc,KAAK;IACtB,MAAM;IACN,MAAM,KAAK,eAAe;IAC1B,UAAU,KAAK,eAAe,YAAY;GAC5C,CAAC;GACD;EACF;EAGA,IAAI,KAAK,qBAAqB;GAC5B,MAAM,MAAM,KAAK,qBAAqB,KAAK;GAC3C,KAAK,cAAc,KAAK;IACtB,MAAM;IACN,QAAQ,KAAK,oBAAoB;IACjC,SAAS,KAAK,oBAAoB,WAAW;GAC/C,CAAC;GACD;EACF;EAEA,IAAI,KAAK,YAAY;GACnB,MAAM,MAAM,KAAK,qBAAqB,KAAK;GAC3C,KAAK,cACH,KACA,iBAAiB,KAAK,WAAW,UAAU,KAAK,WAAW,IAAI,CACjE;GACA;EACF;EAEA,IAAI,KAAK,UAAU;GACjB,MAAM,MAAM,KAAK,qBAAqB,KAAK;GAC3C,KAAK,cACH,KACA,eAAe,KAAK,SAAS,SAAS,KAAK,SAAS,QAAQ,CAC9D;EACF;CACF;CAEA,qBAA6B,OAAuB;EAClD,IACE,CAAC,KAAK,oBACL,CAAC,MAAM,qBACN,CAAC,MAAM,oBACP,CAAC,MAAM,eAET;EAGF,MAAM,UAA8B,EAAE,GADrB,KAAK,mBAAmB,IAAI,KAAK,gBAAgB,KAAK,CAAC,EACtB;EAClD,IAAI,MAAM,mBACR,QAAQ,oBAAoB,MAAM;EACpC,IAAI,MAAM,kBACR,QAAQ,mBAAmB,MAAM;EACnC,IAAI,MAAM,eAAe,QAAQ,gBAAgB,MAAM;EACvD,KAAK,mBAAmB,IAAI,KAAK,kBAAkB,OAAO;CAC5D;CAEA,gBAAwB,KAA0C;EAChE,OAAO,MAAM,QAAQ,IAAI,OAAO,IAC3B,IAAI,UACL,CAAC;CACP;CAEA,qBAA6B,OAAoC;EAC/D,IAAI,KAAK,kBAAkB;GACzB,MAAM,WAAW,KAAK,YAAY,IAAI,KAAK,gBAAgB;GAC3D,IAAI,YAAY,SAAS,SAAS,MAChC,OAAO;EAEX;EAEA,MAAM,KAAKA,GAAO;EAClB,MAAM,MAAyB;GAC7B;GACA,MAAM;GACN,SAAS,CAAC;GACV,GAAI,MAAM,UAAU,QAAQ,EAAE,QAAQ,MAAM,OAAO;GACnD,GAAI,MAAM,UAAU,QAAQ,EAAE,QAAQ,MAAM,OAAO;EACrD;EACA,KAAK,YAAY,IAAI,IAAI,GAAG;EAC5B,KAAK,mBAAmB;EACxB,KAAK,oBAAoB;EACzB,KAAK,yBAAyB;EAC9B,OAAO;CACT;CAEA,cACE,KACA,MACM;EACN,MAAM,UAAU,CAAC,GAAG,KAAK,gBAAgB,GAAG,GAAG,IAAI;EACnD,MAAM,UAA6B;GAAE,GAAG;GAAK;EAAQ;EACrD,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;CAC1C;CAEA,uBAA+B,KAAwB,MAAoB;EACzE,MAAM,UAAU,CAAC,GAAG,KAAK,gBAAgB,GAAG,CAAC;EAC7C,MAAM,cAAc,QAAQ,eAAe,MAAM,EAAE,SAAS,MAAM;EAClE,IAAI,eAAe,GACjB,QAAQ,eAAe;GAAE,MAAM;GAAQ;EAAK;OAE5C,QAAQ,KAAK;GAAE,MAAM;GAAQ;EAAK,CAAC;EAErC,MAAM,UAA6B;GAAE,GAAG;GAAK;EAAQ;EACrD,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;CAC1C;CAEA,4BACE,KACA,MACM;EACN,MAAM,UAAU,CAAC,GAAG,KAAK,gBAAgB,GAAG,CAAC;EAC7C,MAAM,UAAU,QAAQ,eAAe,MAAM,EAAE,SAAS,WAAW;EACnE,IAAI,WAAW,GACb,QAAQ,WAAW;GAAE,MAAM;GAAa;EAAK;OAE7C,QAAQ,KAAK;GAAE,MAAM;GAAa;EAAK,CAAC;EAE1C,MAAM,UAA6B;GAAE,GAAG;GAAK;EAAQ;EACrD,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;CAC1C;CAEA,yBAAuC;EACrC,KAAK,oBAAoB;EACzB,KAAK,yBAAyB;EAC9B,KAAK,mBAAmB;CAC1B;CAEA,cAA4B;EAC1B,OAAO,CAAC,GAAG,KAAK,YAAY,OAAO,CAAC;CACtC;CAEA,gBAAyC;EACvC,OAAO,EAAE,GAAG,KAAK,sBAAsB;CACzC;CAEA,mBAA2C;EACzC,OAAO,EAAE,GAAG,KAAK,yBAAyB;CAC5C;CAEA,eAA2E;EACzE,OAAO,EAAE,GAAG,KAAK,cAAc;CACjC;CAEA,yBAA6C;EAC3C,OAAO,KAAK;CACd;CAEA,wBAAkC;EAChC,OAAO,CAAC,GAAG,KAAK,yBAAyB;CAC3C;CAEA,uBAA8C;EAC5C,OAAO,CAAC,GAAG,KAAK,iBAAiB;CACnC;CAEA,kBAAoC;EAClC,OAAO,CAAC,GAAG,KAAK,YAAY;CAC9B;CAEA,cAAuB;EACrB,OAAO,KAAK;CACd;CAEA,qBAAsD;EACpD,OAAO,IAAI,IAAI,KAAK,kBAAkB;CACxC;AACF"}
+{"version":3,"file":"AdkEventAccumulator.js","names":["uuidv4"],"sources":["../src/AdkEventAccumulator.ts"],"sourcesContent":["import { v4 as uuidv4 } from \"uuid\";\nimport type { MessageStatus } from \"@assistant-ui/core\";\nimport type {\n  AdkEvent,\n  AdkEventPart,\n  AdkMessage,\n  AdkMessageContentPart,\n  AdkToolCall,\n  AdkToolConfirmation,\n  AdkAuthRequest,\n  AdkMessageMetadata,\n} from \"./types\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\n\ntype InProgressMessage = AdkMessage & { type: \"ai\" };\n\n/**\n * A session load replays the stored events through a fresh accumulator, so a\n * message needs an id derived from the event that carries it rather than one\n * minted per replay. An event with no id of its own has never been through the\n * session and has nothing stable to derive from, so it keeps a generated one.\n *\n * A human message keeps the bare event id it has always had. The other kinds\n * take a suffixed namespace, since one event can carry several of them: a tool\n * message by the index of its part, an assistant message by how many this\n * event has already opened.\n */\nconst toolMessageId = (event: AdkEvent, partIndex: number): string =>\n  event.id ? `${event.id}:${partIndex}` : uuidv4();\n\nconst aiMessageId = (event: AdkEvent, ordinal: number): string =>\n  event.id ? `${event.id}:ai${ordinal === 0 ? \"\" : ordinal}` : uuidv4();\n\nconst ADK_REQUEST_CONFIRMATION = \"adk_request_confirmation\";\nconst ADK_REQUEST_CREDENTIAL = \"adk_request_credential\";\n\n/**\n * Checks if an event is a final response using the same logic as ADK's\n * `isFinalResponse()`.\n */\nexport const isFinalResponse = (event: AdkEvent): boolean => {\n  if (\n    event.actions?.skipSummarization ||\n    (event.longRunningToolIds && event.longRunningToolIds.length > 0)\n  ) {\n    return true;\n  }\n\n  const parts = event.content?.parts;\n  return (\n    !event.partial &&\n    (parts?.length ?? 0) > 0 &&\n    !parts!.some((p) => p.functionCall) &&\n    !parts!.some((p) => p.functionResponse) &&\n    !parts![parts!.length - 1]?.codeExecutionResult\n  );\n};\n\nconst CONTENT_FILTER_REASONS = new Set([\n  \"SAFETY\",\n  \"RECITATION\",\n  \"BLOCKLIST\",\n  \"PROHIBITED_CONTENT\",\n  \"SPII\",\n  \"LANGUAGE\",\n  \"IMAGE_SAFETY\",\n  \"IMAGE_RECITATION\",\n  \"IMAGE_OTHER\",\n  \"IMAGE_PROHIBITED_CONTENT\",\n]);\n\nconst ERROR_REASONS = new Set([\n  \"MALFORMED_FUNCTION_CALL\",\n  \"UNEXPECTED_TOOL_CALL\",\n]);\n\nconst finishReasonToStatus = (\n  finishReason: string | undefined,\n): MessageStatus => {\n  if (finishReason === \"MAX_TOKENS\") {\n    return { type: \"incomplete\", reason: \"length\" };\n  }\n  if (finishReason && CONTENT_FILTER_REASONS.has(finishReason)) {\n    return {\n      type: \"incomplete\",\n      reason: \"content-filter\",\n      error: `Content filtered: ${finishReason}`,\n    };\n  }\n  if (finishReason && ERROR_REASONS.has(finishReason)) {\n    return {\n      type: \"incomplete\",\n      reason: \"error\",\n      error: `LLM error: ${finishReason}`,\n    };\n  }\n  return { type: \"complete\", reason: \"stop\" };\n};\n\nconst inlineDataToPart = (\n  mimeType: string,\n  data: string,\n): AdkMessageContentPart =>\n  mimeType.startsWith(\"image/\")\n    ? { type: \"image\", mimeType, data }\n    : { type: \"file\", mimeType, data };\n\nconst fileDataToPart = (\n  fileUri: string,\n  mimeType: string | undefined,\n): AdkMessageContentPart =>\n  mimeType == null || mimeType.startsWith(\"image/\")\n    ? { type: \"image_url\", url: fileUri }\n    : { type: \"file_url\", url: fileUri, mimeType };\n\n// ── Snake_case normalization ──\n\nconst normalizeEventPart = (part: AdkEventPart): AdkEventPart => {\n  const p = part as Record<string, unknown>;\n  const result: Record<string, unknown> = { ...p };\n  if (\"function_call\" in p && !(\"functionCall\" in p))\n    result.functionCall = p.function_call;\n  if (\"function_response\" in p && !(\"functionResponse\" in p))\n    result.functionResponse = p.function_response;\n  if (\"inline_data\" in p && !(\"inlineData\" in p))\n    result.inlineData = p.inline_data;\n  if (\"file_data\" in p && !(\"fileData\" in p)) result.fileData = p.file_data;\n  if (\"executable_code\" in p && !(\"executableCode\" in p))\n    result.executableCode = p.executable_code;\n  if (\"code_execution_result\" in p && !(\"codeExecutionResult\" in p))\n    result.codeExecutionResult = p.code_execution_result;\n  return result as AdkEventPart;\n};\n\nconst normalizeEvent = (event: AdkEvent): AdkEvent => {\n  const e = event as Record<string, unknown>;\n  const result: Record<string, unknown> = { ...e };\n  if (\"error_code\" in e && !(\"errorCode\" in e)) result.errorCode = e.error_code;\n  if (\"error_message\" in e && !(\"errorMessage\" in e))\n    result.errorMessage = e.error_message;\n  if (\"long_running_tool_ids\" in e && !(\"longRunningToolIds\" in e))\n    result.longRunningToolIds = e.long_running_tool_ids;\n  if (\"turn_complete\" in e && !(\"turnComplete\" in e))\n    result.turnComplete = e.turn_complete;\n  if (\"finish_reason\" in e && !(\"finishReason\" in e))\n    result.finishReason = e.finish_reason;\n  if (\"invocation_id\" in e && !(\"invocationId\" in e))\n    result.invocationId = e.invocation_id;\n  if (\"custom_metadata\" in e && !(\"customMetadata\" in e))\n    result.customMetadata = e.custom_metadata;\n  if (\"grounding_metadata\" in e && !(\"groundingMetadata\" in e))\n    result.groundingMetadata = e.grounding_metadata;\n  if (\"citation_metadata\" in e && !(\"citationMetadata\" in e))\n    result.citationMetadata = e.citation_metadata;\n  if (\"usage_metadata\" in e && !(\"usageMetadata\" in e))\n    result.usageMetadata = e.usage_metadata;\n\n  if (result.actions) {\n    const a = result.actions as Record<string, unknown>;\n    const na: Record<string, unknown> = { ...a };\n    if (\"state_delta\" in a && !(\"stateDelta\" in a))\n      na.stateDelta = a.state_delta;\n    if (\"artifact_delta\" in a && !(\"artifactDelta\" in a))\n      na.artifactDelta = a.artifact_delta;\n    if (\"transfer_to_agent\" in a && !(\"transferToAgent\" in a))\n      na.transferToAgent = a.transfer_to_agent;\n    if (\"skip_summarization\" in a && !(\"skipSummarization\" in a))\n      na.skipSummarization = a.skip_summarization;\n    if (\"requested_auth_configs\" in a && !(\"requestedAuthConfigs\" in a))\n      na.requestedAuthConfigs = a.requested_auth_configs;\n    if (\n      \"requested_tool_confirmations\" in a &&\n      !(\"requestedToolConfirmations\" in a)\n    )\n      na.requestedToolConfirmations = a.requested_tool_confirmations;\n    result.actions = na;\n  }\n\n  if (result.content && (result.content as Record<string, unknown>).parts) {\n    const content = result.content as Record<string, unknown>;\n    const parts = content.parts as AdkEventPart[];\n    result.content = { ...content, parts: parts.map(normalizeEventPart) };\n  }\n\n  return result as AdkEvent;\n};\n\n// ── Accumulator ──\n\nexport class AdkEventAccumulator {\n  private messagesMap = new Map<string, AdkMessage>();\n  private currentMessageId: string | null = null;\n  private partialTextBuffer = \"\";\n  private partialReasoningBuffer = \"\";\n  private accumulatedStateDelta: Record<string, unknown> = {};\n  private accumulatedArtifactDelta: Record<string, number> = {};\n  private lastAgentInfo: {\n    name?: string | undefined;\n    branch?: string | undefined;\n  } = {};\n  private lastTransferToAgent: string | undefined;\n  private pendingLongRunningToolIds = new Set<string>();\n  private toolConfirmations: AdkToolConfirmation[] = [];\n  private authRequests: AdkAuthRequest[] = [];\n  private escalated = false;\n  private messageMetadataMap = new Map<string, AdkMessageMetadata>();\n  // How many assistant messages each event has opened, so a replay of that\n  // event opens them with the same ids.\n  private aiMessageOrdinals = new Map<string, number>();\n  constructor(initialMessages?: AdkMessage[]) {\n    if (initialMessages) {\n      for (const msg of initialMessages) {\n        this.messagesMap.set(msg.id, msg);\n      }\n    }\n  }\n\n  processEvent(rawEvent: AdkEvent): AdkMessage[] {\n    const event = normalizeEvent(rawEvent);\n\n    // Accumulate state delta\n    if (event.actions?.stateDelta) {\n      Object.assign(this.accumulatedStateDelta, event.actions.stateDelta);\n    }\n\n    // Accumulate artifact delta\n    if (event.actions?.artifactDelta) {\n      Object.assign(this.accumulatedArtifactDelta, event.actions.artifactDelta);\n    }\n\n    // Track escalation\n    if (event.actions?.escalate) {\n      this.escalated = true;\n    }\n\n    // Track agent transfer\n    if (event.actions?.transferToAgent) {\n      this.lastTransferToAgent = event.actions.transferToAgent;\n    }\n\n    // Track long-running tool IDs. Accumulate across events so that multiple\n    // HITL interrupts emitted in the same turn are all tracked — a\n    // single-event replacement would drop earlier ids.\n    if (event.longRunningToolIds?.length) {\n      for (const id of event.longRunningToolIds) {\n        this.pendingLongRunningToolIds.add(id);\n      }\n    }\n\n    // Track tool confirmations from actions\n    if (event.actions?.requestedToolConfirmations) {\n      for (const [tcId, conf] of Object.entries(\n        event.actions.requestedToolConfirmations,\n      )) {\n        const c = conf as Record<string, unknown>;\n        this.toolConfirmations.push({\n          toolCallId: tcId,\n          toolName: \"\",\n          args: {},\n          hint: (c.hint as string) ?? \"\",\n          confirmed: false,\n          payload: c.payload,\n        });\n      }\n    }\n\n    // Track auth requests from actions\n    if (event.actions?.requestedAuthConfigs) {\n      for (const [tcId, authConf] of Object.entries(\n        event.actions.requestedAuthConfigs,\n      )) {\n        this.authRequests.push({ toolCallId: tcId, authConfig: authConf });\n      }\n    }\n\n    // Track agent info\n    if (event.author && event.author !== \"user\") {\n      this.lastAgentInfo = {\n        name: event.author ?? undefined,\n        branch: event.branch ?? undefined,\n      };\n    }\n\n    // Handle interrupted events\n    if (event.interrupted) {\n      if (this.currentMessageId) {\n        const msg = this.messagesMap.get(this.currentMessageId);\n        if (msg && msg.type === \"ai\" && !msg.status) {\n          const updated: InProgressMessage = {\n            ...msg,\n            content: [...this.getContentArray(msg)],\n            status: { type: \"incomplete\", reason: \"cancelled\" },\n          };\n          this.messagesMap.set(updated.id, updated);\n        }\n      }\n      this.finalizeCurrentMessage();\n      return this.getMessages();\n    }\n\n    // Handle error events\n    if (event.errorCode || event.errorMessage) {\n      this.finalizeCurrentMessage();\n      const errorMsg = this.getOrCreateAiMessage(event);\n      const errorText =\n        event.errorMessage ?? event.errorCode ?? \"Unknown error\";\n      const updated: InProgressMessage = {\n        ...errorMsg,\n        content: [\n          ...this.getContentArray(errorMsg),\n          { type: \"text\", text: errorText },\n        ],\n        status: { type: \"incomplete\", reason: \"error\", error: errorText },\n      };\n      this.messagesMap.set(updated.id, updated);\n      this.currentMessageId = null;\n      return this.getMessages();\n    }\n\n    const parts = event.content?.parts;\n    if (!parts?.length) {\n      // Track metadata even for content-less events (e.g. turnComplete)\n      this.trackMessageMetadata(event);\n      return this.getMessages();\n    }\n\n    // User-authored events → create human message, not AI.\n    // Without this, user events fall through to processPart →\n    // getOrCreateAiMessage, producing type:\"ai\" messages that\n    // convertAdkMessage maps to role:\"assistant\".\n    if (event.author === \"user\") {\n      this.finalizeCurrentMessage();\n      const humanParts: AdkMessageContentPart[] = [];\n      for (const part of parts) {\n        if (part.text != null && !part.thought) {\n          humanParts.push({ type: \"text\", text: part.text });\n        } else if (part.inlineData) {\n          humanParts.push(\n            inlineDataToPart(part.inlineData.mimeType, part.inlineData.data),\n          );\n        } else if (part.fileData) {\n          humanParts.push(\n            fileDataToPart(part.fileData.fileUri, part.fileData.mimeType),\n          );\n        }\n      }\n      if (humanParts.length > 0) {\n        const id = event.id ?? uuidv4();\n        const first = humanParts[0];\n        const content: string | AdkMessageContentPart[] =\n          humanParts.length === 1 && first?.type === \"text\"\n            ? first.text\n            : humanParts;\n        this.messagesMap.set(id, { id, type: \"human\", content });\n      }\n      return this.getMessages();\n    }\n\n    // If author changed, finalize previous message\n    if (this.currentMessageId && event.author && event.author !== \"user\") {\n      const current = this.messagesMap.get(this.currentMessageId);\n      if (current && current.type === \"ai\" && current.author !== event.author) {\n        this.finalizeCurrentMessage();\n      }\n    }\n\n    for (const [index, part] of parts.entries()) {\n      this.processPart(part, event, index);\n    }\n\n    // Track per-message metadata (grounding, citation, usage)\n    this.trackMessageMetadata(event);\n\n    // Check isFinalResponse (can be true even for partial events via skipSummarization/longRunningToolIds)\n    if (isFinalResponse(event) && this.currentMessageId) {\n      const msg = this.messagesMap.get(this.currentMessageId);\n      // Skip manual \"complete\" when longRunningToolIds is the sole reason for\n      // isFinalResponse — let auto-status apply requires-action for pending tool calls.\n      const isHitlOnly =\n        event.longRunningToolIds &&\n        event.longRunningToolIds.length > 0 &&\n        !event.actions?.skipSummarization;\n      if (msg && msg.type === \"ai\" && !msg.status && !isHitlOnly) {\n        const status = finishReasonToStatus(event.finishReason);\n        const updated: InProgressMessage = {\n          ...msg,\n          content: [...this.getContentArray(msg)],\n          status,\n        };\n        this.messagesMap.set(updated.id, updated);\n      }\n    }\n\n    // Non-partial event finalizes the current message\n    if (!event.partial || isFinalResponse(event)) {\n      this.finalizeCurrentMessage();\n    }\n\n    return this.getMessages();\n  }\n\n  private processPart(\n    part: AdkEventPart,\n    event: AdkEvent,\n    partIndex: number,\n  ): void {\n    // Detect special ADK function calls\n    if (part.functionCall && !event.partial) {\n      const name = part.functionCall.name;\n\n      // Tool confirmation request\n      if (name === ADK_REQUEST_CONFIRMATION) {\n        const callArgs = part.functionCall.args;\n        // ADK JS: args keys are \"originalFunctionCall\" and \"toolConfirmation\"\n        // ADK Python: args keys are \"original_function_call\" and \"tool_confirmation\"\n        const original =\n          (callArgs.originalFunctionCall as Record<string, unknown>) ??\n          (callArgs.original_function_call as Record<string, unknown>);\n        const conf =\n          (callArgs.toolConfirmation as Record<string, unknown>) ??\n          (callArgs.tool_confirmation as Record<string, unknown>);\n        this.toolConfirmations.push({\n          toolCallId: part.functionCall.id ?? \"\",\n          toolName: (original?.name as string) ?? \"\",\n          args: (original?.args as Record<string, unknown>) ?? {},\n          hint: (conf?.hint as string) ?? \"\",\n          confirmed: false,\n          payload: conf?.payload,\n        });\n      }\n\n      // Auth credential request\n      if (name === ADK_REQUEST_CREDENTIAL) {\n        const credArgs = part.functionCall.args;\n        // ADK JS: args keys are \"function_call_id\" and \"auth_config\"\n        const originalToolCallId =\n          (credArgs.function_call_id as string) ?? part.functionCall.id ?? \"\";\n        const authConfig = credArgs.auth_config ?? credArgs;\n        this.authRequests.push({\n          toolCallId: originalToolCallId,\n          authConfig,\n        });\n      }\n    }\n\n    // Text with thought=true → reasoning (accumulated)\n    if (part.text != null && part.thought) {\n      const msg = this.getOrCreateAiMessage(event);\n      if (event.partial) {\n        this.partialReasoningBuffer += part.text;\n        this.replaceLastReasoningContent(msg, this.partialReasoningBuffer);\n      } else {\n        this.partialReasoningBuffer = \"\";\n        this.replaceLastReasoningContent(msg, part.text);\n      }\n      return;\n    }\n\n    // Regular text\n    if (part.text != null) {\n      const msg = this.getOrCreateAiMessage(event);\n      if (event.partial) {\n        this.partialTextBuffer += part.text;\n        this.replaceLastTextContent(msg, this.partialTextBuffer);\n      } else {\n        this.partialTextBuffer = \"\";\n        this.replaceLastTextContent(msg, part.text);\n      }\n      return;\n    }\n\n    // Function call — skip partial events (args may be incomplete)\n    if (part.functionCall) {\n      if (event.partial) return;\n      const msg = this.getOrCreateAiMessage(event);\n      const toolCall: AdkToolCall = {\n        id: part.functionCall.id ?? uuidv4(),\n        name: part.functionCall.name,\n        args: part.functionCall.args as ReadonlyJSONObject,\n        argsText: JSON.stringify(part.functionCall.args),\n      };\n      const existing = [...(msg.tool_calls ?? [])];\n      const idx = existing.findIndex((tc) => tc.id === toolCall.id);\n      if (idx >= 0) {\n        existing[idx] = toolCall;\n      } else {\n        existing.push(toolCall);\n      }\n      const updated: InProgressMessage = {\n        ...msg,\n        content: [...this.getContentArray(msg)],\n        tool_calls: existing,\n      };\n      this.messagesMap.set(updated.id, updated);\n      return;\n    }\n\n    // Function response → tool message\n    if (part.functionResponse) {\n      this.finalizeCurrentMessage();\n      const toolMsg: AdkMessage = {\n        id: toolMessageId(event, partIndex),\n        type: \"tool\",\n        tool_call_id: part.functionResponse.id ?? \"\",\n        name: part.functionResponse.name,\n        content: JSON.stringify(part.functionResponse.response),\n        status: \"success\",\n      };\n      this.messagesMap.set(toolMsg.id, toolMsg);\n      return;\n    }\n\n    // Executable code\n    if (part.executableCode) {\n      const msg = this.getOrCreateAiMessage(event);\n      this.appendContent(msg, {\n        type: \"code\",\n        code: part.executableCode.code,\n        language: part.executableCode.language ?? \"python\",\n      });\n      return;\n    }\n\n    // Code execution result\n    if (part.codeExecutionResult) {\n      const msg = this.getOrCreateAiMessage(event);\n      this.appendContent(msg, {\n        type: \"code_result\",\n        output: part.codeExecutionResult.output,\n        outcome: part.codeExecutionResult.outcome ?? \"OUTCOME_OK\",\n      });\n      return;\n    }\n\n    if (part.inlineData) {\n      const msg = this.getOrCreateAiMessage(event);\n      this.appendContent(\n        msg,\n        inlineDataToPart(part.inlineData.mimeType, part.inlineData.data),\n      );\n      return;\n    }\n\n    if (part.fileData) {\n      const msg = this.getOrCreateAiMessage(event);\n      this.appendContent(\n        msg,\n        fileDataToPart(part.fileData.fileUri, part.fileData.mimeType),\n      );\n    }\n  }\n\n  private trackMessageMetadata(event: AdkEvent): void {\n    if (\n      !this.currentMessageId ||\n      (!event.groundingMetadata &&\n        !event.citationMetadata &&\n        !event.usageMetadata)\n    )\n      return;\n\n    const existing = this.messageMetadataMap.get(this.currentMessageId) ?? {};\n    const updated: AdkMessageMetadata = { ...existing };\n    if (event.groundingMetadata)\n      updated.groundingMetadata = event.groundingMetadata;\n    if (event.citationMetadata)\n      updated.citationMetadata = event.citationMetadata;\n    if (event.usageMetadata) updated.usageMetadata = event.usageMetadata;\n    this.messageMetadataMap.set(this.currentMessageId, updated);\n  }\n\n  private getContentArray(msg: AdkMessage): AdkMessageContentPart[] {\n    return Array.isArray(msg.content)\n      ? (msg.content as AdkMessageContentPart[])\n      : [];\n  }\n\n  private getOrCreateAiMessage(event: AdkEvent): InProgressMessage {\n    if (this.currentMessageId) {\n      const existing = this.messagesMap.get(this.currentMessageId);\n      if (existing && existing.type === \"ai\") {\n        return existing;\n      }\n    }\n\n    const ordinal = this.aiMessageOrdinals.get(event.id ?? \"\") ?? 0;\n    this.aiMessageOrdinals.set(event.id ?? \"\", ordinal + 1);\n    const id = aiMessageId(event, ordinal);\n    const msg: InProgressMessage = {\n      id,\n      type: \"ai\",\n      content: [] as AdkMessageContentPart[],\n      ...(event.author != null && { author: event.author }),\n      ...(event.branch != null && { branch: event.branch }),\n    };\n    this.messagesMap.set(id, msg);\n    this.currentMessageId = id;\n    this.partialTextBuffer = \"\";\n    this.partialReasoningBuffer = \"\";\n    return msg;\n  }\n\n  private appendContent(\n    msg: InProgressMessage,\n    part: AdkMessageContentPart,\n  ): void {\n    const content = [...this.getContentArray(msg), part];\n    const updated: InProgressMessage = { ...msg, content };\n    this.messagesMap.set(updated.id, updated);\n  }\n\n  private replaceLastTextContent(msg: InProgressMessage, text: string): void {\n    const content = [...this.getContentArray(msg)];\n    const lastTextIdx = content.findLastIndex((p) => p.type === \"text\");\n    if (lastTextIdx >= 0) {\n      content[lastTextIdx] = { type: \"text\", text };\n    } else {\n      content.push({ type: \"text\", text });\n    }\n    const updated: InProgressMessage = { ...msg, content };\n    this.messagesMap.set(updated.id, updated);\n  }\n\n  private replaceLastReasoningContent(\n    msg: InProgressMessage,\n    text: string,\n  ): void {\n    const content = [...this.getContentArray(msg)];\n    const lastIdx = content.findLastIndex((p) => p.type === \"reasoning\");\n    if (lastIdx >= 0) {\n      content[lastIdx] = { type: \"reasoning\", text };\n    } else {\n      content.push({ type: \"reasoning\", text });\n    }\n    const updated: InProgressMessage = { ...msg, content };\n    this.messagesMap.set(updated.id, updated);\n  }\n\n  private finalizeCurrentMessage(): void {\n    this.partialTextBuffer = \"\";\n    this.partialReasoningBuffer = \"\";\n    this.currentMessageId = null;\n  }\n\n  getMessages(): AdkMessage[] {\n    return [...this.messagesMap.values()];\n  }\n\n  getStateDelta(): Record<string, unknown> {\n    return { ...this.accumulatedStateDelta };\n  }\n\n  getArtifactDelta(): Record<string, number> {\n    return { ...this.accumulatedArtifactDelta };\n  }\n\n  getAgentInfo(): { name?: string | undefined; branch?: string | undefined } {\n    return { ...this.lastAgentInfo };\n  }\n\n  getLastTransferToAgent(): string | undefined {\n    return this.lastTransferToAgent;\n  }\n\n  getLongRunningToolIds(): string[] {\n    return [...this.pendingLongRunningToolIds];\n  }\n\n  getToolConfirmations(): AdkToolConfirmation[] {\n    return [...this.toolConfirmations];\n  }\n\n  getAuthRequests(): AdkAuthRequest[] {\n    return [...this.authRequests];\n  }\n\n  isEscalated(): boolean {\n    return this.escalated;\n  }\n\n  getMessageMetadata(): Map<string, AdkMessageMetadata> {\n    return new Map(this.messageMetadataMap);\n  }\n}\n"],"mappings":";;;;;;;;;;;;;AA2BA,MAAM,iBAAiB,OAAiB,cACtC,MAAM,KAAK,GAAG,MAAM,GAAG,GAAG,cAAcA,GAAO;AAEjD,MAAM,eAAe,OAAiB,YACpC,MAAM,KAAK,GAAG,MAAM,GAAG,KAAK,YAAY,IAAI,KAAK,YAAYA,GAAO;AAEtE,MAAM,2BAA2B;AACjC,MAAM,yBAAyB;;;;;AAM/B,MAAa,mBAAmB,UAA6B;CAC3D,IACE,MAAM,SAAS,qBACd,MAAM,sBAAsB,MAAM,mBAAmB,SAAS,GAE/D,OAAO;CAGT,MAAM,QAAQ,MAAM,SAAS;CAC7B,OACE,CAAC,MAAM,YACN,OAAO,UAAU,KAAK,KACvB,CAAC,MAAO,MAAM,MAAM,EAAE,YAAY,KAClC,CAAC,MAAO,MAAM,MAAM,EAAE,gBAAgB,KACtC,CAAC,MAAO,MAAO,SAAS,EAAE,EAAE;AAEhC;AAEA,MAAM,yCAAyB,IAAI,IAAI;CACrC;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;AACF,CAAC;AAED,MAAM,gCAAgB,IAAI,IAAI,CAC5B,2BACA,sBACF,CAAC;AAED,MAAM,wBACJ,iBACkB;CAClB,IAAI,iBAAiB,cACnB,OAAO;EAAE,MAAM;EAAc,QAAQ;CAAS;CAEhD,IAAI,gBAAgB,uBAAuB,IAAI,YAAY,GACzD,OAAO;EACL,MAAM;EACN,QAAQ;EACR,OAAO,qBAAqB;CAC9B;CAEF,IAAI,gBAAgB,cAAc,IAAI,YAAY,GAChD,OAAO;EACL,MAAM;EACN,QAAQ;EACR,OAAO,cAAc;CACvB;CAEF,OAAO;EAAE,MAAM;EAAY,QAAQ;CAAO;AAC5C;AAEA,MAAM,oBACJ,UACA,SAEA,SAAS,WAAW,QAAQ,IACxB;CAAE,MAAM;CAAS;CAAU;AAAK,IAChC;CAAE,MAAM;CAAQ;CAAU;AAAK;AAErC,MAAM,kBACJ,SACA,aAEA,YAAY,QAAQ,SAAS,WAAW,QAAQ,IAC5C;CAAE,MAAM;CAAa,KAAK;AAAQ,IAClC;CAAE,MAAM;CAAY,KAAK;CAAS;AAAS;AAIjD,MAAM,sBAAsB,SAAqC;CAC/D,MAAM,IAAI;CACV,MAAM,SAAkC,EAAE,GAAG,EAAE;CAC/C,IAAI,mBAAmB,KAAK,EAAE,kBAAkB,IAC9C,OAAO,eAAe,EAAE;CAC1B,IAAI,uBAAuB,KAAK,EAAE,sBAAsB,IACtD,OAAO,mBAAmB,EAAE;CAC9B,IAAI,iBAAiB,KAAK,EAAE,gBAAgB,IAC1C,OAAO,aAAa,EAAE;CACxB,IAAI,eAAe,KAAK,EAAE,cAAc,IAAI,OAAO,WAAW,EAAE;CAChE,IAAI,qBAAqB,KAAK,EAAE,oBAAoB,IAClD,OAAO,iBAAiB,EAAE;CAC5B,IAAI,2BAA2B,KAAK,EAAE,yBAAyB,IAC7D,OAAO,sBAAsB,EAAE;CACjC,OAAO;AACT;AAEA,MAAM,kBAAkB,UAA8B;CACpD,MAAM,IAAI;CACV,MAAM,SAAkC,EAAE,GAAG,EAAE;CAC/C,IAAI,gBAAgB,KAAK,EAAE,eAAe,IAAI,OAAO,YAAY,EAAE;CACnE,IAAI,mBAAmB,KAAK,EAAE,kBAAkB,IAC9C,OAAO,eAAe,EAAE;CAC1B,IAAI,2BAA2B,KAAK,EAAE,wBAAwB,IAC5D,OAAO,qBAAqB,EAAE;CAChC,IAAI,mBAAmB,KAAK,EAAE,kBAAkB,IAC9C,OAAO,eAAe,EAAE;CAC1B,IAAI,mBAAmB,KAAK,EAAE,kBAAkB,IAC9C,OAAO,eAAe,EAAE;CAC1B,IAAI,mBAAmB,KAAK,EAAE,kBAAkB,IAC9C,OAAO,eAAe,EAAE;CAC1B,IAAI,qBAAqB,KAAK,EAAE,oBAAoB,IAClD,OAAO,iBAAiB,EAAE;CAC5B,IAAI,wBAAwB,KAAK,EAAE,uBAAuB,IACxD,OAAO,oBAAoB,EAAE;CAC/B,IAAI,uBAAuB,KAAK,EAAE,sBAAsB,IACtD,OAAO,mBAAmB,EAAE;CAC9B,IAAI,oBAAoB,KAAK,EAAE,mBAAmB,IAChD,OAAO,gBAAgB,EAAE;CAE3B,IAAI,OAAO,SAAS;EAClB,MAAM,IAAI,OAAO;EACjB,MAAM,KAA8B,EAAE,GAAG,EAAE;EAC3C,IAAI,iBAAiB,KAAK,EAAE,gBAAgB,IAC1C,GAAG,aAAa,EAAE;EACpB,IAAI,oBAAoB,KAAK,EAAE,mBAAmB,IAChD,GAAG,gBAAgB,EAAE;EACvB,IAAI,uBAAuB,KAAK,EAAE,qBAAqB,IACrD,GAAG,kBAAkB,EAAE;EACzB,IAAI,wBAAwB,KAAK,EAAE,uBAAuB,IACxD,GAAG,oBAAoB,EAAE;EAC3B,IAAI,4BAA4B,KAAK,EAAE,0BAA0B,IAC/D,GAAG,uBAAuB,EAAE;EAC9B,IACE,kCAAkC,KAClC,EAAE,gCAAgC,IAElC,GAAG,6BAA6B,EAAE;EACpC,OAAO,UAAU;CACnB;CAEA,IAAI,OAAO,WAAY,OAAO,QAAoC,OAAO;EACvE,MAAM,UAAU,OAAO;EACvB,MAAM,QAAQ,QAAQ;EACtB,OAAO,UAAU;GAAE,GAAG;GAAS,OAAO,MAAM,IAAI,kBAAkB;EAAE;CACtE;CAEA,OAAO;AACT;AAIA,IAAa,sBAAb,MAAiC;CAC/B,8BAAsB,IAAI,IAAwB;CAClD,mBAA0C;CAC1C,oBAA4B;CAC5B,yBAAiC;CACjC,wBAAyD,CAAC;CAC1D,2BAA2D,CAAC;CAC5D,gBAGI,CAAC;CACL;CACA,4CAAoC,IAAI,IAAY;CACpD,oBAAmD,CAAC;CACpD,eAAyC,CAAC;CAC1C,YAAoB;CACpB,qCAA6B,IAAI,IAAgC;CAGjE,oCAA4B,IAAI,IAAoB;CACpD,YAAY,iBAAgC;EAC1C,IAAI,iBACF,KAAK,MAAM,OAAO,iBAChB,KAAK,YAAY,IAAI,IAAI,IAAI,GAAG;CAGtC;CAEA,aAAa,UAAkC;EAC7C,MAAM,QAAQ,eAAe,QAAQ;EAGrC,IAAI,MAAM,SAAS,YACjB,OAAO,OAAO,KAAK,uBAAuB,MAAM,QAAQ,UAAU;EAIpE,IAAI,MAAM,SAAS,eACjB,OAAO,OAAO,KAAK,0BAA0B,MAAM,QAAQ,aAAa;EAI1E,IAAI,MAAM,SAAS,UACjB,KAAK,YAAY;EAInB,IAAI,MAAM,SAAS,iBACjB,KAAK,sBAAsB,MAAM,QAAQ;EAM3C,IAAI,MAAM,oBAAoB,QAC5B,KAAK,MAAM,MAAM,MAAM,oBACrB,KAAK,0BAA0B,IAAI,EAAE;EAKzC,IAAI,MAAM,SAAS,4BACjB,KAAK,MAAM,CAAC,MAAM,SAAS,OAAO,QAChC,MAAM,QAAQ,0BAChB,GAAG;GACD,MAAM,IAAI;GACV,KAAK,kBAAkB,KAAK;IAC1B,YAAY;IACZ,UAAU;IACV,MAAM,CAAC;IACP,MAAO,EAAE,QAAmB;IAC5B,WAAW;IACX,SAAS,EAAE;GACb,CAAC;EACH;EAIF,IAAI,MAAM,SAAS,sBACjB,KAAK,MAAM,CAAC,MAAM,aAAa,OAAO,QACpC,MAAM,QAAQ,oBAChB,GACE,KAAK,aAAa,KAAK;GAAE,YAAY;GAAM,YAAY;EAAS,CAAC;EAKrE,IAAI,MAAM,UAAU,MAAM,WAAW,QACnC,KAAK,gBAAgB;GACnB,MAAM,MAAM,UAAU,KAAA;GACtB,QAAQ,MAAM,UAAU,KAAA;EAC1B;EAIF,IAAI,MAAM,aAAa;GACrB,IAAI,KAAK,kBAAkB;IACzB,MAAM,MAAM,KAAK,YAAY,IAAI,KAAK,gBAAgB;IACtD,IAAI,OAAO,IAAI,SAAS,QAAQ,CAAC,IAAI,QAAQ;KAC3C,MAAM,UAA6B;MACjC,GAAG;MACH,SAAS,CAAC,GAAG,KAAK,gBAAgB,GAAG,CAAC;MACtC,QAAQ;OAAE,MAAM;OAAc,QAAQ;MAAY;KACpD;KACA,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;IAC1C;GACF;GACA,KAAK,uBAAuB;GAC5B,OAAO,KAAK,YAAY;EAC1B;EAGA,IAAI,MAAM,aAAa,MAAM,cAAc;GACzC,KAAK,uBAAuB;GAC5B,MAAM,WAAW,KAAK,qBAAqB,KAAK;GAChD,MAAM,YACJ,MAAM,gBAAgB,MAAM,aAAa;GAC3C,MAAM,UAA6B;IACjC,GAAG;IACH,SAAS,CACP,GAAG,KAAK,gBAAgB,QAAQ,GAChC;KAAE,MAAM;KAAQ,MAAM;IAAU,CAClC;IACA,QAAQ;KAAE,MAAM;KAAc,QAAQ;KAAS,OAAO;IAAU;GAClE;GACA,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;GACxC,KAAK,mBAAmB;GACxB,OAAO,KAAK,YAAY;EAC1B;EAEA,MAAM,QAAQ,MAAM,SAAS;EAC7B,IAAI,CAAC,OAAO,QAAQ;GAElB,KAAK,qBAAqB,KAAK;GAC/B,OAAO,KAAK,YAAY;EAC1B;EAMA,IAAI,MAAM,WAAW,QAAQ;GAC3B,KAAK,uBAAuB;GAC5B,MAAM,aAAsC,CAAC;GAC7C,KAAK,MAAM,QAAQ,OACjB,IAAI,KAAK,QAAQ,QAAQ,CAAC,KAAK,SAC7B,WAAW,KAAK;IAAE,MAAM;IAAQ,MAAM,KAAK;GAAK,CAAC;QAC5C,IAAI,KAAK,YACd,WAAW,KACT,iBAAiB,KAAK,WAAW,UAAU,KAAK,WAAW,IAAI,CACjE;QACK,IAAI,KAAK,UACd,WAAW,KACT,eAAe,KAAK,SAAS,SAAS,KAAK,SAAS,QAAQ,CAC9D;GAGJ,IAAI,WAAW,SAAS,GAAG;IACzB,MAAM,KAAK,MAAM,MAAMA,GAAO;IAC9B,MAAM,QAAQ,WAAW;IACzB,MAAM,UACJ,WAAW,WAAW,KAAK,OAAO,SAAS,SACvC,MAAM,OACN;IACN,KAAK,YAAY,IAAI,IAAI;KAAE;KAAI,MAAM;KAAS;IAAQ,CAAC;GACzD;GACA,OAAO,KAAK,YAAY;EAC1B;EAGA,IAAI,KAAK,oBAAoB,MAAM,UAAU,MAAM,WAAW,QAAQ;GACpE,MAAM,UAAU,KAAK,YAAY,IAAI,KAAK,gBAAgB;GAC1D,IAAI,WAAW,QAAQ,SAAS,QAAQ,QAAQ,WAAW,MAAM,QAC/D,KAAK,uBAAuB;EAEhC;EAEA,KAAK,MAAM,CAAC,OAAO,SAAS,MAAM,QAAQ,GACxC,KAAK,YAAY,MAAM,OAAO,KAAK;EAIrC,KAAK,qBAAqB,KAAK;EAG/B,IAAI,gBAAgB,KAAK,KAAK,KAAK,kBAAkB;GACnD,MAAM,MAAM,KAAK,YAAY,IAAI,KAAK,gBAAgB;GAGtD,MAAM,aACJ,MAAM,sBACN,MAAM,mBAAmB,SAAS,KAClC,CAAC,MAAM,SAAS;GAClB,IAAI,OAAO,IAAI,SAAS,QAAQ,CAAC,IAAI,UAAU,CAAC,YAAY;IAC1D,MAAM,SAAS,qBAAqB,MAAM,YAAY;IACtD,MAAM,UAA6B;KACjC,GAAG;KACH,SAAS,CAAC,GAAG,KAAK,gBAAgB,GAAG,CAAC;KACtC;IACF;IACA,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;GAC1C;EACF;EAGA,IAAI,CAAC,MAAM,WAAW,gBAAgB,KAAK,GACzC,KAAK,uBAAuB;EAG9B,OAAO,KAAK,YAAY;CAC1B;CAEA,YACE,MACA,OACA,WACM;EAEN,IAAI,KAAK,gBAAgB,CAAC,MAAM,SAAS;GACvC,MAAM,OAAO,KAAK,aAAa;GAG/B,IAAI,SAAS,0BAA0B;IACrC,MAAM,WAAW,KAAK,aAAa;IAGnC,MAAM,WACH,SAAS,wBACT,SAAS;IACZ,MAAM,OACH,SAAS,oBACT,SAAS;IACZ,KAAK,kBAAkB,KAAK;KAC1B,YAAY,KAAK,aAAa,MAAM;KACpC,UAAW,UAAU,QAAmB;KACxC,MAAO,UAAU,QAAoC,CAAC;KACtD,MAAO,MAAM,QAAmB;KAChC,WAAW;KACX,SAAS,MAAM;IACjB,CAAC;GACH;GAGA,IAAI,SAAS,wBAAwB;IACnC,MAAM,WAAW,KAAK,aAAa;IAEnC,MAAM,qBACH,SAAS,oBAA+B,KAAK,aAAa,MAAM;IACnE,MAAM,aAAa,SAAS,eAAe;IAC3C,KAAK,aAAa,KAAK;KACrB,YAAY;KACZ;IACF,CAAC;GACH;EACF;EAGA,IAAI,KAAK,QAAQ,QAAQ,KAAK,SAAS;GACrC,MAAM,MAAM,KAAK,qBAAqB,KAAK;GAC3C,IAAI,MAAM,SAAS;IACjB,KAAK,0BAA0B,KAAK;IACpC,KAAK,4BAA4B,KAAK,KAAK,sBAAsB;GACnE,OAAO;IACL,KAAK,yBAAyB;IAC9B,KAAK,4BAA4B,KAAK,KAAK,IAAI;GACjD;GACA;EACF;EAGA,IAAI,KAAK,QAAQ,MAAM;GACrB,MAAM,MAAM,KAAK,qBAAqB,KAAK;GAC3C,IAAI,MAAM,SAAS;IACjB,KAAK,qBAAqB,KAAK;IAC/B,KAAK,uBAAuB,KAAK,KAAK,iBAAiB;GACzD,OAAO;IACL,KAAK,oBAAoB;IACzB,KAAK,uBAAuB,KAAK,KAAK,IAAI;GAC5C;GACA;EACF;EAGA,IAAI,KAAK,cAAc;GACrB,IAAI,MAAM,SAAS;GACnB,MAAM,MAAM,KAAK,qBAAqB,KAAK;GAC3C,MAAM,WAAwB;IAC5B,IAAI,KAAK,aAAa,MAAMA,GAAO;IACnC,MAAM,KAAK,aAAa;IACxB,MAAM,KAAK,aAAa;IACxB,UAAU,KAAK,UAAU,KAAK,aAAa,IAAI;GACjD;GACA,MAAM,WAAW,CAAC,GAAI,IAAI,cAAc,CAAC,CAAE;GAC3C,MAAM,MAAM,SAAS,WAAW,OAAO,GAAG,OAAO,SAAS,EAAE;GAC5D,IAAI,OAAO,GACT,SAAS,OAAO;QAEhB,SAAS,KAAK,QAAQ;GAExB,MAAM,UAA6B;IACjC,GAAG;IACH,SAAS,CAAC,GAAG,KAAK,gBAAgB,GAAG,CAAC;IACtC,YAAY;GACd;GACA,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;GACxC;EACF;EAGA,IAAI,KAAK,kBAAkB;GACzB,KAAK,uBAAuB;GAC5B,MAAM,UAAsB;IAC1B,IAAI,cAAc,OAAO,SAAS;IAClC,MAAM;IACN,cAAc,KAAK,iBAAiB,MAAM;IAC1C,MAAM,KAAK,iBAAiB;IAC5B,SAAS,KAAK,UAAU,KAAK,iBAAiB,QAAQ;IACtD,QAAQ;GACV;GACA,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;GACxC;EACF;EAGA,IAAI,KAAK,gBAAgB;GACvB,MAAM,MAAM,KAAK,qBAAqB,KAAK;GAC3C,KAAK,cAAc,KAAK;IACtB,MAAM;IACN,MAAM,KAAK,eAAe;IAC1B,UAAU,KAAK,eAAe,YAAY;GAC5C,CAAC;GACD;EACF;EAGA,IAAI,KAAK,qBAAqB;GAC5B,MAAM,MAAM,KAAK,qBAAqB,KAAK;GAC3C,KAAK,cAAc,KAAK;IACtB,MAAM;IACN,QAAQ,KAAK,oBAAoB;IACjC,SAAS,KAAK,oBAAoB,WAAW;GAC/C,CAAC;GACD;EACF;EAEA,IAAI,KAAK,YAAY;GACnB,MAAM,MAAM,KAAK,qBAAqB,KAAK;GAC3C,KAAK,cACH,KACA,iBAAiB,KAAK,WAAW,UAAU,KAAK,WAAW,IAAI,CACjE;GACA;EACF;EAEA,IAAI,KAAK,UAAU;GACjB,MAAM,MAAM,KAAK,qBAAqB,KAAK;GAC3C,KAAK,cACH,KACA,eAAe,KAAK,SAAS,SAAS,KAAK,SAAS,QAAQ,CAC9D;EACF;CACF;CAEA,qBAA6B,OAAuB;EAClD,IACE,CAAC,KAAK,oBACL,CAAC,MAAM,qBACN,CAAC,MAAM,oBACP,CAAC,MAAM,eAET;EAGF,MAAM,UAA8B,EAAE,GADrB,KAAK,mBAAmB,IAAI,KAAK,gBAAgB,KAAK,CAAC,EACtB;EAClD,IAAI,MAAM,mBACR,QAAQ,oBAAoB,MAAM;EACpC,IAAI,MAAM,kBACR,QAAQ,mBAAmB,MAAM;EACnC,IAAI,MAAM,eAAe,QAAQ,gBAAgB,MAAM;EACvD,KAAK,mBAAmB,IAAI,KAAK,kBAAkB,OAAO;CAC5D;CAEA,gBAAwB,KAA0C;EAChE,OAAO,MAAM,QAAQ,IAAI,OAAO,IAC3B,IAAI,UACL,CAAC;CACP;CAEA,qBAA6B,OAAoC;EAC/D,IAAI,KAAK,kBAAkB;GACzB,MAAM,WAAW,KAAK,YAAY,IAAI,KAAK,gBAAgB;GAC3D,IAAI,YAAY,SAAS,SAAS,MAChC,OAAO;EAEX;EAEA,MAAM,UAAU,KAAK,kBAAkB,IAAI,MAAM,MAAM,EAAE,KAAK;EAC9D,KAAK,kBAAkB,IAAI,MAAM,MAAM,IAAI,UAAU,CAAC;EACtD,MAAM,KAAK,YAAY,OAAO,OAAO;EACrC,MAAM,MAAyB;GAC7B;GACA,MAAM;GACN,SAAS,CAAC;GACV,GAAI,MAAM,UAAU,QAAQ,EAAE,QAAQ,MAAM,OAAO;GACnD,GAAI,MAAM,UAAU,QAAQ,EAAE,QAAQ,MAAM,OAAO;EACrD;EACA,KAAK,YAAY,IAAI,IAAI,GAAG;EAC5B,KAAK,mBAAmB;EACxB,KAAK,oBAAoB;EACzB,KAAK,yBAAyB;EAC9B,OAAO;CACT;CAEA,cACE,KACA,MACM;EACN,MAAM,UAAU,CAAC,GAAG,KAAK,gBAAgB,GAAG,GAAG,IAAI;EACnD,MAAM,UAA6B;GAAE,GAAG;GAAK;EAAQ;EACrD,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;CAC1C;CAEA,uBAA+B,KAAwB,MAAoB;EACzE,MAAM,UAAU,CAAC,GAAG,KAAK,gBAAgB,GAAG,CAAC;EAC7C,MAAM,cAAc,QAAQ,eAAe,MAAM,EAAE,SAAS,MAAM;EAClE,IAAI,eAAe,GACjB,QAAQ,eAAe;GAAE,MAAM;GAAQ;EAAK;OAE5C,QAAQ,KAAK;GAAE,MAAM;GAAQ;EAAK,CAAC;EAErC,MAAM,UAA6B;GAAE,GAAG;GAAK;EAAQ;EACrD,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;CAC1C;CAEA,4BACE,KACA,MACM;EACN,MAAM,UAAU,CAAC,GAAG,KAAK,gBAAgB,GAAG,CAAC;EAC7C,MAAM,UAAU,QAAQ,eAAe,MAAM,EAAE,SAAS,WAAW;EACnE,IAAI,WAAW,GACb,QAAQ,WAAW;GAAE,MAAM;GAAa;EAAK;OAE7C,QAAQ,KAAK;GAAE,MAAM;GAAa;EAAK,CAAC;EAE1C,MAAM,UAA6B;GAAE,GAAG;GAAK;EAAQ;EACrD,KAAK,YAAY,IAAI,QAAQ,IAAI,OAAO;CAC1C;CAEA,yBAAuC;EACrC,KAAK,oBAAoB;EACzB,KAAK,yBAAyB;EAC9B,KAAK,mBAAmB;CAC1B;CAEA,cAA4B;EAC1B,OAAO,CAAC,GAAG,KAAK,YAAY,OAAO,CAAC;CACtC;CAEA,gBAAyC;EACvC,OAAO,EAAE,GAAG,KAAK,sBAAsB;CACzC;CAEA,mBAA2C;EACzC,OAAO,EAAE,GAAG,KAAK,yBAAyB;CAC5C;CAEA,eAA2E;EACzE,OAAO,EAAE,GAAG,KAAK,cAAc;CACjC;CAEA,yBAA6C;EAC3C,OAAO,KAAK;CACd;CAEA,wBAAkC;EAChC,OAAO,CAAC,GAAG,KAAK,yBAAyB;CAC3C;CAEA,uBAA8C;EAC5C,OAAO,CAAC,GAAG,KAAK,iBAAiB;CACnC;CAEA,kBAAoC;EAClC,OAAO,CAAC,GAAG,KAAK,YAAY;CAC9B;CAEA,cAAuB;EACrB,OAAO,KAAK;CACd;CAEA,qBAAsD;EACpD,OAAO,IAAI,IAAI,KAAK,kBAAkB;CACxC;AACF"}

package/dist/AdkSessionAdapter.d.ts

@@ -1,4 +1,4 @@
-import { AdkMessage } from "./types.js";
+import { AdkThreadSnapshot } from "./types.js";
 import { RemoteThreadListAdapter } from "@assistant-ui/core";
 //#region src/AdkSessionAdapter.d.ts
 type AdkSessionAdapterOptions = {
@@ -28,9 +28,9 @@ type AdkArtifactData = {
 };
 type AdkSessionAdapterResult = {
   adapter: RemoteThreadListAdapter;
-  load: (sessionId: string) => Promise<{
-    messages: AdkMessage[];
-  }>;
+  load: (sessionId: string, options?: {
+    signal?: AbortSignal | undefined;
+  }) => Promise<AdkThreadSnapshot>;
   artifacts: {
     list: (sessionId: string) => Promise<string[]>;
     load: (sessionId: string, artifactName: string, version?: number) => Promise<AdkArtifactData>;

package/dist/AdkSessionAdapter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AdkSessionAdapter.d.ts","names":[],"sources":["../src/AdkSessionAdapter.ts"],"mappings":";;;KAWY;;;;EAIV;;;;EAKA;;;;EAKA;;;;EAKA,UACI,gCACO,yBAAyB,QAAQ;;KAIlC;EACV;IAAe;IAAkB;;EACjC;;KAGG;EACH,SAAS;EACT,OAAO,sBAAsB;IAAU,UAAU;;EACjD;IACE,OAAO,sBAAsB;IAC7B,OACE,mBACA,sBACA,qBACG,QAAQ;IACb,eACE,mBACA,yBACG;IACL,SAAS,mBAAmB,yBAAyB;;;;;;;;;;;;;;;;;;;;;;iBAmJzC,wBACd,SAAS,2BACR"}
+{"version":3,"file":"AdkSessionAdapter.d.ts","names":[],"sources":["../src/AdkSessionAdapter.ts"],"mappings":";;;KAWY;;;;EAIV;;;;EAKA;;;;EAKA;;;;EAKA,UACI,gCACO,yBAAyB,QAAQ;;KAIlC;EACV;IAAe;IAAkB;;EACjC;;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;;;;;;;;;;;;;;;;;;;;;;iBAmJzC,wBACd,SAAS,2BACR"}

package/dist/AdkSessionAdapter.js

@@ -129,9 +129,12 @@ function createAdkSessionAdapter(options) {
 			};
 		}
 	};
-	const load = async (sessionId) => {
+	const load = async (sessionId, options) => {
 		const headers = await getHeaders();
-		const res = await fetch(`${baseUrl}/${encodeURIComponent(sessionId)}`, { headers });
+		const res = await fetch(`${baseUrl}/${encodeURIComponent(sessionId)}`, {
+			headers,
+			...options?.signal ? { signal: options.signal } : {}
+		});
 		if (!res.ok) throw new Error(`Failed to load session: ${res.status}`);
 		const session = parseAdkSessionResponse(await res.json(), "load");
 		if (session.events !== void 0 && !Array.isArray(session.events)) throw new Error("Invalid ADK session load response: expected \"events\" to be an array when present.");
@@ -140,7 +143,17 @@ function createAdkSessionAdapter(options) {
 		const accumulator = new AdkEventAccumulator();
 		let messages = [];
 		for (const event of events) messages = accumulator.processEvent(event);
-		return { messages };
+		return {
+			messages,
+			longRunningToolIds: accumulator.getLongRunningToolIds(),
+			toolConfirmations: accumulator.getToolConfirmations(),
+			authRequests: accumulator.getAuthRequests(),
+			escalated: accumulator.isEscalated(),
+			messageMetadata: accumulator.getMessageMetadata(),
+			stateDelta: accumulator.getStateDelta(),
+			artifactDelta: accumulator.getArtifactDelta(),
+			agentInfo: accumulator.getAgentInfo()
+		};
 	};
 	const artifactBaseUrl = (sessionId) => `${baseUrl}/${encodeURIComponent(sessionId)}/artifacts`;
 	return {

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 type { AdkEvent, AdkMessage } 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  text?: string | undefined;\n};\n\ntype AdkSessionAdapterResult = {\n  adapter: RemoteThreadListAdapter;\n  load: (sessionId: string) => Promise<{ messages: AdkMessage[] }>;\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\" or \"inlineData\".',\n    );\n  }\n\n  const { text, inlineData } = value;\n  if (text === undefined && inlineData === undefined) {\n    throw new Error(\n      'Invalid ADK artifact load response: expected an object containing \"text\" or \"inlineData\".',\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  return value 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  ): Promise<{ messages: AdkMessage[] }> => {\n    const headers = await getHeaders();\n    const res = await fetch(`${baseUrl}/${encodeURIComponent(sessionId)}`, {\n      headers,\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    const events = session.events as AdkEvent[] | undefined;\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    return { messages };\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":";;;AAgEA,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,+FACF;CAGF,MAAM,EAAE,MAAM,eAAe;CAC7B,IAAI,SAAS,KAAA,KAAa,eAAe,KAAA,GACvC,MAAM,IAAI,MACR,+FACF;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,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,cACwC;EACxC,MAAM,UAAU,MAAM,WAAW;EACjC,MAAM,MAAM,MAAM,MAAM,GAAG,QAAQ,GAAG,mBAAmB,SAAS,KAAK,EACrE,QACF,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;EAGF,MAAM,SAAS,QAAQ;EAEvB,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;EAE3C,OAAO,EAAE,SAAS;CACpB;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 type { AdkEvent, 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  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\" or \"inlineData\".',\n    );\n  }\n\n  const { text, inlineData } = value;\n  if (text === undefined && inlineData === undefined) {\n    throw new Error(\n      'Invalid ADK artifact load response: expected an object containing \"text\" or \"inlineData\".',\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  return value 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    const events = session.events as AdkEvent[] | undefined;\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":";;;AAmEA,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,+FACF;CAGF,MAAM,EAAE,MAAM,eAAe;CAC7B,IAAI,SAAS,KAAA,KAAa,eAAe,KAAA,GACvC,MAAM,IAAI,MACR,+FACF;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,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;EAGF,MAAM,SAAS,QAAQ;EAEvB,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/convertAdkMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"convertAdkMessages.d.ts","names":[],"sources":["../src/convertAdkMessages.ts"],"mappings":";;;cAkFa,mBAAmB,4BAA4B,SAC1D"}
+{"version":3,"file":"convertAdkMessages.d.ts","names":[],"sources":["../src/convertAdkMessages.ts"],"mappings":";;;cAmFa,mBAAmB,4BAA4B,SAC1D"}

package/dist/convertAdkMessages.js

@@ -23,30 +23,27 @@ const contentToParts = (content, role) => {
 				type: "image",
 				image: part.url
 			};
-			case "file": {
-				const format = role === "user" && part.filename == null ? part.mimeType === "audio/wav" ? "wav" : part.mimeType === "audio/mp3" ? "mp3" : null : null;
-				if (format) return {
-					type: "audio",
-					audio: {
-						data: part.data,
-						format
-					}
+			case "file": return {
+				type: "file",
+				data: part.data,
+				mimeType: part.mimeType,
+				...part.filename != null && { filename: part.filename }
+			};
+			case "file_url":
+				if (role === "user") return {
+					type: "file",
+					data: part.url,
+					mimeType: part.mimeType ?? "application/octet-stream",
+					sourceType: "url"
 				};
 				return {
-					type: "file",
-					data: part.data,
-					mimeType: part.mimeType,
-					...part.filename != null && { filename: part.filename }
+					type: "data",
+					name: "file_url",
+					data: {
+						url: part.url,
+						...part.mimeType != null && { mimeType: part.mimeType }
+					}
 				};
-			}
-			case "file_url": return {
-				type: "data",
-				name: "file_url",
-				data: {
-					url: part.url,
-					...part.mimeType != null && { mimeType: part.mimeType }
-				}
-			};
 			case "code": return {
 				type: "data",
 				name: "executable_code",

package/dist/convertAdkMessages.js.map

@@ -1 +1 @@
-{"version":3,"file":"convertAdkMessages.js","names":[],"sources":["../src/convertAdkMessages.ts"],"sourcesContent":["\"use client\";\n\nimport type { ToolCallMessagePart } from \"@assistant-ui/core\";\nimport type { useExternalMessageConverter } from \"@assistant-ui/core/react\";\nimport type { AdkMessage, AdkMessageContentPart } from \"./types\";\n\ntype ContentPart =\n  | { type: \"text\"; text: string }\n  | { type: \"reasoning\"; text: string }\n  | { type: \"image\"; image: string }\n  | { type: \"file\"; data: string; mimeType: string; filename?: string }\n  | { type: \"audio\"; audio: { data: string; format: \"mp3\" | \"wav\" } }\n  | { type: \"data\"; name: string; data: unknown };\n\nconst contentToParts = (\n  content: AdkMessage[\"content\"],\n  role: \"user\" | \"assistant\",\n): ContentPart[] => {\n  if (typeof content === \"string\")\n    return [{ type: \"text\" as const, text: content }];\n\n  return (content as AdkMessageContentPart[])\n    .map((part): ContentPart | null => {\n      switch (part.type) {\n        case \"text\":\n          return { type: \"text\", text: part.text };\n        case \"reasoning\":\n          return { type: \"reasoning\", text: part.text };\n        case \"image\":\n          return {\n            type: \"image\",\n            image: `data:${part.mimeType};base64,${part.data}`,\n          };\n        case \"image_url\":\n          return { type: \"image\", image: part.url };\n        case \"file\": {\n          const format =\n            role === \"user\" && part.filename == null\n              ? part.mimeType === \"audio/wav\"\n                ? (\"wav\" as const)\n                : part.mimeType === \"audio/mp3\"\n                  ? (\"mp3\" as const)\n                  : null\n              : null;\n          if (format) {\n            return { type: \"audio\", audio: { data: part.data, format } };\n          }\n          return {\n            type: \"file\",\n            data: part.data,\n            mimeType: part.mimeType,\n            ...(part.filename != null && { filename: part.filename }),\n          };\n        }\n        case \"file_url\":\n          return {\n            type: \"data\",\n            name: \"file_url\",\n            data: {\n              url: part.url,\n              ...(part.mimeType != null && { mimeType: part.mimeType }),\n            },\n          };\n        case \"code\":\n          return {\n            type: \"data\",\n            name: \"executable_code\",\n            data: { code: part.code, language: part.language },\n          };\n        case \"code_result\":\n          return {\n            type: \"data\",\n            name: \"code_execution_result\",\n            data: { output: part.output, outcome: part.outcome },\n          };\n        default:\n          return null;\n      }\n    })\n    .filter((p): p is NonNullable<typeof p> => p !== null);\n};\n\nexport const convertAdkMessage: useExternalMessageConverter.Callback<\n  AdkMessage\n> = (message) => {\n  switch (message.type) {\n    case \"human\":\n      return {\n        role: \"user\",\n        id: message.id,\n        content: contentToParts(message.content, \"user\"),\n      };\n\n    case \"ai\": {\n      const toolCallParts: ToolCallMessagePart[] =\n        message.tool_calls?.map((tc) => ({\n          type: \"tool-call\",\n          toolCallId: tc.id,\n          toolName: tc.name,\n          args: tc.args,\n          argsText: tc.argsText ?? JSON.stringify(tc.args),\n        })) ?? [];\n\n      return {\n        role: \"assistant\",\n        id: message.id,\n        content: [\n          ...contentToParts(message.content, \"assistant\"),\n          ...toolCallParts,\n        ],\n        ...(message.status && { status: message.status }),\n        ...(message.author && {\n          metadata: {\n            custom: { author: message.author, branch: message.branch },\n          },\n        }),\n      };\n    }\n\n    case \"tool\":\n      return {\n        role: \"tool\",\n        toolCallId: message.tool_call_id,\n        toolName: message.name,\n        result: message.content,\n        isError: message.status === \"error\",\n      };\n  }\n};\n"],"mappings":";;AAcA,MAAM,kBACJ,SACA,SACkB;CAClB,IAAI,OAAO,YAAY,UACrB,OAAO,CAAC;EAAE,MAAM;EAAiB,MAAM;CAAQ,CAAC;CAElD,OAAQ,QACL,KAAK,SAA6B;EACjC,QAAQ,KAAK,MAAb;GACE,KAAK,QACH,OAAO;IAAE,MAAM;IAAQ,MAAM,KAAK;GAAK;GACzC,KAAK,aACH,OAAO;IAAE,MAAM;IAAa,MAAM,KAAK;GAAK;GAC9C,KAAK,SACH,OAAO;IACL,MAAM;IACN,OAAO,QAAQ,KAAK,SAAS,UAAU,KAAK;GAC9C;GACF,KAAK,aACH,OAAO;IAAE,MAAM;IAAS,OAAO,KAAK;GAAI;GAC1C,KAAK,QAAQ;IACX,MAAM,SACJ,SAAS,UAAU,KAAK,YAAY,OAChC,KAAK,aAAa,cACf,QACD,KAAK,aAAa,cACf,QACD,OACJ;IACN,IAAI,QACF,OAAO;KAAE,MAAM;KAAS,OAAO;MAAE,MAAM,KAAK;MAAM;KAAO;IAAE;IAE7D,OAAO;KACL,MAAM;KACN,MAAM,KAAK;KACX,UAAU,KAAK;KACf,GAAI,KAAK,YAAY,QAAQ,EAAE,UAAU,KAAK,SAAS;IACzD;GACF;GACA,KAAK,YACH,OAAO;IACL,MAAM;IACN,MAAM;IACN,MAAM;KACJ,KAAK,KAAK;KACV,GAAI,KAAK,YAAY,QAAQ,EAAE,UAAU,KAAK,SAAS;IACzD;GACF;GACF,KAAK,QACH,OAAO;IACL,MAAM;IACN,MAAM;IACN,MAAM;KAAE,MAAM,KAAK;KAAM,UAAU,KAAK;IAAS;GACnD;GACF,KAAK,eACH,OAAO;IACL,MAAM;IACN,MAAM;IACN,MAAM;KAAE,QAAQ,KAAK;KAAQ,SAAS,KAAK;IAAQ;GACrD;GACF,SACE,OAAO;EACX;CACF,CAAC,CAAC,CACD,QAAQ,MAAkC,MAAM,IAAI;AACzD;AAEA,MAAa,qBAER,YAAY;CACf,QAAQ,QAAQ,MAAhB;EACE,KAAK,SACH,OAAO;GACL,MAAM;GACN,IAAI,QAAQ;GACZ,SAAS,eAAe,QAAQ,SAAS,MAAM;EACjD;EAEF,KAAK,MAAM;GACT,MAAM,gBACJ,QAAQ,YAAY,KAAK,QAAQ;IAC/B,MAAM;IACN,YAAY,GAAG;IACf,UAAU,GAAG;IACb,MAAM,GAAG;IACT,UAAU,GAAG,YAAY,KAAK,UAAU,GAAG,IAAI;GACjD,EAAE,KAAK,CAAC;GAEV,OAAO;IACL,MAAM;IACN,IAAI,QAAQ;IACZ,SAAS,CACP,GAAG,eAAe,QAAQ,SAAS,WAAW,GAC9C,GAAG,aACL;IACA,GAAI,QAAQ,UAAU,EAAE,QAAQ,QAAQ,OAAO;IAC/C,GAAI,QAAQ,UAAU,EACpB,UAAU,EACR,QAAQ;KAAE,QAAQ,QAAQ;KAAQ,QAAQ,QAAQ;IAAO,EAC3D,EACF;GACF;EACF;EAEA,KAAK,QACH,OAAO;GACL,MAAM;GACN,YAAY,QAAQ;GACpB,UAAU,QAAQ;GAClB,QAAQ,QAAQ;GAChB,SAAS,QAAQ,WAAW;EAC9B;CACJ;AACF"}
+{"version":3,"file":"convertAdkMessages.js","names":[],"sources":["../src/convertAdkMessages.ts"],"sourcesContent":["\"use client\";\n\nimport type { ToolCallMessagePart } from \"@assistant-ui/core\";\nimport type { useExternalMessageConverter } from \"@assistant-ui/core/react\";\nimport type { AdkMessage, AdkMessageContentPart } from \"./types\";\n\ntype ContentPart =\n  | { type: \"text\"; text: string }\n  | { type: \"reasoning\"; text: string }\n  | { type: \"image\"; image: string }\n  | {\n      type: \"file\";\n      data: string;\n      mimeType: string;\n      filename?: string;\n      sourceType?: \"url\";\n    }\n  | { type: \"data\"; name: string; data: unknown };\n\nconst contentToParts = (\n  content: AdkMessage[\"content\"],\n  role: \"user\" | \"assistant\",\n): ContentPart[] => {\n  if (typeof content === \"string\")\n    return [{ type: \"text\" as const, text: content }];\n\n  return (content as AdkMessageContentPart[])\n    .map((part): ContentPart | null => {\n      switch (part.type) {\n        case \"text\":\n          return { type: \"text\", text: part.text };\n        case \"reasoning\":\n          return { type: \"reasoning\", text: part.text };\n        case \"image\":\n          return {\n            type: \"image\",\n            image: `data:${part.mimeType};base64,${part.data}`,\n          };\n        case \"image_url\":\n          return { type: \"image\", image: part.url };\n        case \"file\":\n          return {\n            type: \"file\",\n            data: part.data,\n            mimeType: part.mimeType,\n            ...(part.filename != null && { filename: part.filename }),\n          };\n        case \"file_url\":\n          if (role === \"user\") {\n            return {\n              type: \"file\",\n              data: part.url,\n              mimeType: part.mimeType ?? \"application/octet-stream\",\n              sourceType: \"url\",\n            };\n          }\n          return {\n            type: \"data\",\n            name: \"file_url\",\n            data: {\n              url: part.url,\n              ...(part.mimeType != null && { mimeType: part.mimeType }),\n            },\n          };\n        case \"code\":\n          return {\n            type: \"data\",\n            name: \"executable_code\",\n            data: { code: part.code, language: part.language },\n          };\n        case \"code_result\":\n          return {\n            type: \"data\",\n            name: \"code_execution_result\",\n            data: { output: part.output, outcome: part.outcome },\n          };\n        default:\n          return null;\n      }\n    })\n    .filter((p): p is NonNullable<typeof p> => p !== null);\n};\n\nexport const convertAdkMessage: useExternalMessageConverter.Callback<\n  AdkMessage\n> = (message) => {\n  switch (message.type) {\n    case \"human\":\n      return {\n        role: \"user\",\n        id: message.id,\n        content: contentToParts(message.content, \"user\"),\n      };\n\n    case \"ai\": {\n      const toolCallParts: ToolCallMessagePart[] =\n        message.tool_calls?.map((tc) => ({\n          type: \"tool-call\",\n          toolCallId: tc.id,\n          toolName: tc.name,\n          args: tc.args,\n          argsText: tc.argsText ?? JSON.stringify(tc.args),\n        })) ?? [];\n\n      return {\n        role: \"assistant\",\n        id: message.id,\n        content: [\n          ...contentToParts(message.content, \"assistant\"),\n          ...toolCallParts,\n        ],\n        ...(message.status && { status: message.status }),\n        ...(message.author && {\n          metadata: {\n            custom: { author: message.author, branch: message.branch },\n          },\n        }),\n      };\n    }\n\n    case \"tool\":\n      return {\n        role: \"tool\",\n        toolCallId: message.tool_call_id,\n        toolName: message.name,\n        result: message.content,\n        isError: message.status === \"error\",\n      };\n  }\n};\n"],"mappings":";;AAmBA,MAAM,kBACJ,SACA,SACkB;CAClB,IAAI,OAAO,YAAY,UACrB,OAAO,CAAC;EAAE,MAAM;EAAiB,MAAM;CAAQ,CAAC;CAElD,OAAQ,QACL,KAAK,SAA6B;EACjC,QAAQ,KAAK,MAAb;GACE,KAAK,QACH,OAAO;IAAE,MAAM;IAAQ,MAAM,KAAK;GAAK;GACzC,KAAK,aACH,OAAO;IAAE,MAAM;IAAa,MAAM,KAAK;GAAK;GAC9C,KAAK,SACH,OAAO;IACL,MAAM;IACN,OAAO,QAAQ,KAAK,SAAS,UAAU,KAAK;GAC9C;GACF,KAAK,aACH,OAAO;IAAE,MAAM;IAAS,OAAO,KAAK;GAAI;GAC1C,KAAK,QACH,OAAO;IACL,MAAM;IACN,MAAM,KAAK;IACX,UAAU,KAAK;IACf,GAAI,KAAK,YAAY,QAAQ,EAAE,UAAU,KAAK,SAAS;GACzD;GACF,KAAK;IACH,IAAI,SAAS,QACX,OAAO;KACL,MAAM;KACN,MAAM,KAAK;KACX,UAAU,KAAK,YAAY;KAC3B,YAAY;IACd;IAEF,OAAO;KACL,MAAM;KACN,MAAM;KACN,MAAM;MACJ,KAAK,KAAK;MACV,GAAI,KAAK,YAAY,QAAQ,EAAE,UAAU,KAAK,SAAS;KACzD;IACF;GACF,KAAK,QACH,OAAO;IACL,MAAM;IACN,MAAM;IACN,MAAM;KAAE,MAAM,KAAK;KAAM,UAAU,KAAK;IAAS;GACnD;GACF,KAAK,eACH,OAAO;IACL,MAAM;IACN,MAAM;IACN,MAAM;KAAE,QAAQ,KAAK;KAAQ,SAAS,KAAK;IAAQ;GACrD;GACF,SACE,OAAO;EACX;CACF,CAAC,CAAC,CACD,QAAQ,MAAkC,MAAM,IAAI;AACzD;AAEA,MAAa,qBAER,YAAY;CACf,QAAQ,QAAQ,MAAhB;EACE,KAAK,SACH,OAAO;GACL,MAAM;GACN,IAAI,QAAQ;GACZ,SAAS,eAAe,QAAQ,SAAS,MAAM;EACjD;EAEF,KAAK,MAAM;GACT,MAAM,gBACJ,QAAQ,YAAY,KAAK,QAAQ;IAC/B,MAAM;IACN,YAAY,GAAG;IACf,UAAU,GAAG;IACb,MAAM,GAAG;IACT,UAAU,GAAG,YAAY,KAAK,UAAU,GAAG,IAAI;GACjD,EAAE,KAAK,CAAC;GAEV,OAAO;IACL,MAAM;IACN,IAAI,QAAQ;IACZ,SAAS,CACP,GAAG,eAAe,QAAQ,SAAS,WAAW,GAC9C,GAAG,aACL;IACA,GAAI,QAAQ,UAAU,EAAE,QAAQ,QAAQ,OAAO;IAC/C,GAAI,QAAQ,UAAU,EACpB,UAAU,EACR,QAAQ;KAAE,QAAQ,QAAQ;KAAQ,QAAQ,QAAQ;IAAO,EAC3D,EACF;GACF;EACF;EAEA,KAAK,QACH,OAAO;GACL,MAAM;GACN,YAAY,QAAQ;GACpB,UAAU,QAAQ;GAClB,QAAQ,QAAQ;GAChB,SAAS,QAAQ,WAAW;EAC9B;CACJ;AACF"}

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { AdkAuthCredential, AdkAuthCredentialType, AdkAuthRequest, AdkEvent, AdkEventActions, AdkEventPart, AdkEventType, AdkMessage, AdkMessageContentPart, AdkMessageMetadata, AdkRunConfig, AdkSendMessageConfig, AdkStreamCallback, AdkStructuredEvent, AdkToolCall, AdkToolConfirmation, OnAdkAgentTransferCallback, OnAdkCustomEventCallback, OnAdkErrorCallback } from "./types.js";
+import { AdkAuthCredential, AdkAuthCredentialType, AdkAuthRequest, AdkEvent, AdkEventActions, AdkEventPart, AdkEventType, AdkMessage, AdkMessageContentPart, AdkMessageMetadata, AdkRunConfig, AdkSendMessageConfig, AdkStreamCallback, AdkStructuredEvent, AdkThreadSnapshot, AdkToolCall, AdkToolConfirmation, OnAdkAgentTransferCallback, OnAdkCustomEventCallback, OnAdkErrorCallback } from "./types.js";
 import { CreateAdkStreamOptions, createAdkStream } from "./AdkClient.js";
 import { AdkEventAccumulator } from "./AdkEventAccumulator.js";
 import { AdkArtifactData, AdkSessionAdapterOptions, createAdkSessionAdapter } from "./AdkSessionAdapter.js";
@@ -7,4 +7,4 @@ import { useAdkAgentInfo, useAdkAppState, useAdkArtifacts, useAdkAuthRequests, u
 import { UseAdkRuntimeOptions, useAdkRuntime } from "./useAdkRuntime.js";
 import { UseAdkMessagesOptions, useAdkMessages } from "./useAdkMessages.js";
 import { toAdkStructuredEvents } from "./structuredEvents.js";
-export { type AdkArtifactData, type AdkAuthCredential, type AdkAuthCredentialType, type AdkAuthRequest, type AdkEvent, AdkEventAccumulator, type AdkEventActions, type AdkEventPart, AdkEventType, type AdkMessage, type AdkMessageContentPart, type AdkMessageMetadata, type AdkRunConfig, type AdkSendMessageConfig, type AdkSessionAdapterOptions, type AdkStreamCallback, type AdkStructuredEvent, type AdkToolCall, type AdkToolConfirmation, type CreateAdkStreamOptions, type OnAdkAgentTransferCallback, type OnAdkCustomEventCallback, type OnAdkErrorCallback, type UseAdkMessagesOptions, type UseAdkRuntimeOptions, convertAdkMessage, createAdkSessionAdapter, createAdkStream, toAdkStructuredEvents, useAdkAgentInfo, useAdkAppState, useAdkArtifacts, useAdkAuthRequests, useAdkConfirmTool, useAdkEscalation, useAdkLongRunningToolIds, useAdkMessageMetadata, useAdkMessages, useAdkRuntime, useAdkSend, useAdkSessionState, useAdkSubmitAuth, useAdkSubmitInput, useAdkTempState, useAdkToolConfirmations, useAdkUserState };
+export { type AdkArtifactData, type AdkAuthCredential, type AdkAuthCredentialType, type AdkAuthRequest, type AdkEvent, AdkEventAccumulator, type AdkEventActions, type AdkEventPart, AdkEventType, type AdkMessage, type AdkMessageContentPart, type AdkMessageMetadata, type AdkRunConfig, type AdkSendMessageConfig, type AdkSessionAdapterOptions, type AdkStreamCallback, type AdkStructuredEvent, type AdkThreadSnapshot, type AdkToolCall, type AdkToolConfirmation, type CreateAdkStreamOptions, type OnAdkAgentTransferCallback, type OnAdkCustomEventCallback, type OnAdkErrorCallback, type UseAdkMessagesOptions, type UseAdkRuntimeOptions, convertAdkMessage, createAdkSessionAdapter, createAdkStream, toAdkStructuredEvents, useAdkAgentInfo, useAdkAppState, useAdkArtifacts, useAdkAuthRequests, useAdkConfirmTool, useAdkEscalation, useAdkLongRunningToolIds, useAdkMessageMetadata, useAdkMessages, useAdkRuntime, useAdkSend, useAdkSessionState, useAdkSubmitAuth, useAdkSubmitInput, useAdkTempState, useAdkToolConfirmations, useAdkUserState };

package/dist/types.d.ts

@@ -275,6 +275,27 @@ type AdkRuntimeExtras = {
   escalated: boolean;
   messageMetadata: Map<string, AdkMessageMetadata>;
 };
+/**
+ * What a session load reconstructs. The messages are the whole thread; the
+ * rest is the per-turn state the events imply, which a caller that replays
+ * them through {@link AdkEventAccumulator} already has. Omitting a field
+ * clears it, since a snapshot that cannot describe the state is not evidence
+ * that the state survived.
+ */
+type AdkThreadSnapshot = {
+  messages: AdkMessage[];
+  longRunningToolIds?: string[] | undefined;
+  toolConfirmations?: AdkToolConfirmation[] | undefined;
+  authRequests?: AdkAuthRequest[] | undefined;
+  escalated?: boolean | undefined;
+  messageMetadata?: Map<string, AdkMessageMetadata> | undefined;
+  stateDelta?: Record<string, unknown> | undefined;
+  artifactDelta?: Record<string, number> | undefined;
+  agentInfo?: {
+    name?: string | undefined;
+    branch?: string | undefined;
+  } | undefined;
+};
 //#endregion
-export { AdkAuthCredential, AdkAuthCredentialType, AdkAuthRequest, AdkEvent, AdkEventActions, AdkEventPart, AdkEventType, AdkMessage, AdkMessageContentPart, AdkMessageMetadata, AdkRunConfig, AdkRuntimeExtras, AdkSendMessageConfig, AdkStreamCallback, AdkStructuredEvent, AdkToolCall, AdkToolConfirmation, OnAdkAgentTransferCallback, OnAdkCustomEventCallback, OnAdkErrorCallback };
+export { AdkAuthCredential, AdkAuthCredentialType, AdkAuthRequest, AdkEvent, AdkEventActions, AdkEventPart, AdkEventType, AdkMessage, AdkMessageContentPart, AdkMessageMetadata, AdkRunConfig, AdkRuntimeExtras, AdkSendMessageConfig, AdkStreamCallback, AdkStructuredEvent, AdkThreadSnapshot, AdkToolCall, AdkToolConfirmation, OnAdkAgentTransferCallback, OnAdkCustomEventCallback, OnAdkErrorCallback };
 //# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","names":[],"sources":["../src/types.ts"],"mappings":";;;KAKY;EACV;EACA;EACA;IACE;IACA;IACA,MAAM;;EAER;IACE;IACA;IACA;;EAEF;IACE;IACA;;EAEF;IACE;IACA;;EAEF;IACE;IACA;;EAEF;IACE;IACA;;;KAIQ;EACV,aAAa;EACb,gBAAgB;EAChB;EACA;EACA;EACA,uBAAuB;EACvB,6BAA6B;;KAGnB;EACV;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;IAEM;IACA,QAAQ;;EAGd,UAAU;EACV;EACA;EACA;EACA;EACA;EACA;EACA,iBAAiB;;KAKP;EACV;EACA;EACA,MAAM;EACN;;KAGU;EAEN;EACA;EACA,kBAAkB;;EAGlB;EACA;EACA,kBAAkB;EAClB,aAAa;EACb;EACA;EACA,SAAS;;EAGT;EACA;EACA;EACA;EACA;EACA;EACA;;KAGM;EACN;EAAc;;EACd;EAAmB;;EACnB;EAAe;EAAkB;;EACjC;EAAmB;;EAEnB;EACA;EACA;EACA;;EAGA;EACA;EACA;;EAEA;EAAc;EAAc;;EAC5B;EAAqB;EAAgB;;EACrC;EAAkB;;KAIZ;EACV;EACA;EACA,MAAM;EACN;EACA;EACA;;KAGU;KAOA;EACV,UAAU;EACV;EACA;EACA;IAEM;IACA;MACE;MACA;MACA;;;EAIR;IAEM;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;;EAGN;;KAGU;EACV;EACA;;KAGU;EACV;EACA;EACA;;KAKU;EACV;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;;KAKU;;;;;;EAMV;EACA;EACA,aAAa;;cAKF;;;;;;;;;;;;KAaD;EACN;EAAiB;;EACjB;EAAiB;;EAEjB;EACA;IAAQ;IAAc;IAAa,MAAM;;;EAGzC;EACA;IAAU;IAAc;IAAa;;;EAErC;EAAmB;IAAQ;IAAc;;;EACzC;EAAqB;IAAU;IAAgB;;;EAC/C;EAAe;EAAoB;;EACnC;EAAkB;;EAElB;EACA,eAAe;;EAEf;;KAEM,qBACV,UAAU,cACV,QAAQ;EACN,aAAa;EACb,kBAAkB;IAChB;IACA;;MAGD,QAAQ,eAAe,aAAa,eAAe;KAI5C,sBAAsB,0BAA0B;KAEhD,4BACV,cACA,yBACU;KAEA,8BACV,2BACU;;KAGA;EACV,OAAO,UAAU,cAAc,QAAQ,yBAAyB;EAChE;IAAa;IAA2B;;EACxC,YAAY;EACZ,eAAe;EACf;EACA,mBAAmB;EACnB,cAAc;EACd;EACA,iBAAiB,YAAY"}
+{"version":3,"file":"types.d.ts","names":[],"sources":["../src/types.ts"],"mappings":";;;KAKY;EACV;EACA;EACA;IACE;IACA;IACA,MAAM;;EAER;IACE;IACA;IACA;;EAEF;IACE;IACA;;EAEF;IACE;IACA;;EAEF;IACE;IACA;;EAEF;IACE;IACA;;;KAIQ;EACV,aAAa;EACb,gBAAgB;EAChB;EACA;EACA;EACA,uBAAuB;EACvB,6BAA6B;;KAGnB;EACV;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;IAEM;IACA,QAAQ;;EAGd,UAAU;EACV;EACA;EACA;EACA;EACA;EACA;EACA,iBAAiB;;KAKP;EACV;EACA;EACA,MAAM;EACN;;KAGU;EAEN;EACA;EACA,kBAAkB;;EAGlB;EACA;EACA,kBAAkB;EAClB,aAAa;EACb;EACA;EACA,SAAS;;EAGT;EACA;EACA;EACA;EACA;EACA;EACA;;KAGM;EACN;EAAc;;EACd;EAAmB;;EACnB;EAAe;EAAkB;;EACjC;EAAmB;;EAEnB;EACA;EACA;EACA;;EAGA;EACA;EACA;;EAEA;EAAc;EAAc;;EAC5B;EAAqB;EAAgB;;EACrC;EAAkB;;KAIZ;EACV;EACA;EACA,MAAM;EACN;EACA;EACA;;KAGU;KAOA;EACV,UAAU;EACV;EACA;EACA;IAEM;IACA;MACE;MACA;MACA;;;EAIR;IAEM;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;;EAGN;;KAGU;EACV;EACA;;KAGU;EACV;EACA;EACA;;KAKU;EACV;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;;KAKU;;;;;;EAMV;EACA;EACA,aAAa;;cAKF;;;;;;;;;;;;KAaD;EACN;EAAiB;;EACjB;EAAiB;;EAEjB;EACA;IAAQ;IAAc;IAAa,MAAM;;;EAGzC;EACA;IAAU;IAAc;IAAa;;;EAErC;EAAmB;IAAQ;IAAc;;;EACzC;EAAqB;IAAU;IAAgB;;;EAC/C;EAAe;EAAoB;;EACnC;EAAkB;;EAElB;EACA,eAAe;;EAEf;;KAEM,qBACV,UAAU,cACV,QAAQ;EACN,aAAa;EACb,kBAAkB;IAChB;IACA;;MAGD,QAAQ,eAAe,aAAa,eAAe;KAI5C,sBAAsB,0BAA0B;KAEhD,4BACV,cACA,yBACU;KAEA,8BACV,2BACU;;KAGA;EACV,OAAO,UAAU,cAAc,QAAQ,yBAAyB;EAChE;IAAa;IAA2B;;EACxC,YAAY;EACZ,eAAe;EACf;EACA,mBAAmB;EACnB,cAAc;EACd;EACA,iBAAiB,YAAY;;;;;;;;;KAUnB;EACV,UAAU;EACV;EACA,oBAAoB;EACpB,eAAe;EACf;EACA,kBAAkB,YAAY;EAC9B,aAAa;EACb,gBAAgB;EAChB;IACM;IAA2B"}

package/dist/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","names":[],"sources":["../src/types.ts"],"sourcesContent":["import type { MessageStatus } from \"@assistant-ui/core\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\n\n// ── ADK Event wire types (lightweight, no @google/adk dependency) ──\n\nexport type AdkEventPart = {\n  text?: string;\n  thought?: boolean;\n  functionCall?: {\n    name: string;\n    id?: string;\n    args: Record<string, unknown>;\n  };\n  functionResponse?: {\n    name: string;\n    id?: string;\n    response: unknown;\n  };\n  executableCode?: {\n    code: string;\n    language?: string;\n  };\n  codeExecutionResult?: {\n    output: string;\n    outcome?: string;\n  };\n  inlineData?: {\n    mimeType: string;\n    data: string;\n  };\n  fileData?: {\n    fileUri: string;\n    mimeType?: string;\n  };\n};\n\nexport type AdkEventActions = {\n  stateDelta?: Record<string, unknown> | undefined;\n  artifactDelta?: Record<string, number> | undefined;\n  transferToAgent?: string | undefined;\n  escalate?: boolean | undefined;\n  skipSummarization?: boolean | undefined;\n  requestedAuthConfigs?: Record<string, unknown> | undefined;\n  requestedToolConfirmations?: Record<string, unknown> | undefined;\n};\n\nexport type AdkEvent = {\n  id: string;\n  invocationId?: string | undefined;\n  author?: string | undefined;\n  branch?: string | undefined;\n  partial?: boolean | undefined;\n  turnComplete?: boolean | undefined;\n  interrupted?: boolean | undefined;\n  finishReason?: string | undefined;\n  timestamp?: number | undefined;\n  content?:\n    | {\n        role?: string | undefined;\n        parts?: AdkEventPart[] | undefined;\n      }\n    | undefined;\n  actions?: AdkEventActions | undefined;\n  longRunningToolIds?: string[] | undefined;\n  errorCode?: string | undefined;\n  errorMessage?: string | undefined;\n  groundingMetadata?: unknown;\n  citationMetadata?: unknown;\n  usageMetadata?: unknown;\n  customMetadata?: Record<string, unknown> | undefined;\n};\n\n// ── ADK Message types (accumulated from events) ──\n\nexport type AdkToolCall = {\n  id: string;\n  name: string;\n  args: ReadonlyJSONObject;\n  argsText?: string;\n};\n\nexport type AdkMessage =\n  | {\n      id: string;\n      type: \"human\";\n      content: string | AdkMessageContentPart[];\n    }\n  | {\n      id: string;\n      type: \"ai\";\n      content: string | AdkMessageContentPart[];\n      tool_calls?: AdkToolCall[] | undefined;\n      author?: string | undefined;\n      branch?: string | undefined;\n      status?: MessageStatus | undefined;\n    }\n  | {\n      id: string;\n      type: \"tool\";\n      content: string;\n      tool_call_id: string;\n      name: string;\n      status?: \"success\" | \"error\" | undefined;\n      artifact?: unknown;\n    };\n\nexport type AdkMessageContentPart =\n  | { type: \"text\"; text: string }\n  | { type: \"reasoning\"; text: string }\n  | { type: \"image\"; mimeType: string; data: string }\n  | { type: \"image_url\"; url: string }\n  | {\n      type: \"file\";\n      mimeType: string;\n      data: string;\n      filename?: string | undefined;\n    }\n  | {\n      type: \"file_url\";\n      url: string;\n      mimeType?: string | undefined;\n    }\n  | { type: \"code\"; code: string; language: string }\n  | { type: \"code_result\"; output: string; outcome: string }\n  | { type: \"activity\"; message: string };\n\n// ── ADK-specific state types ──\n\nexport type AdkToolConfirmation = {\n  toolCallId: string;\n  toolName: string;\n  args: Record<string, unknown>;\n  hint: string;\n  confirmed: boolean;\n  payload?: unknown;\n};\n\nexport type AdkAuthCredentialType =\n  | \"apiKey\"\n  | \"http\"\n  | \"oauth2\"\n  | \"openIdConnect\"\n  | \"serviceAccount\";\n\nexport type AdkAuthCredential = {\n  authType: AdkAuthCredentialType;\n  resourceRef?: string | undefined;\n  apiKey?: string | undefined;\n  http?:\n    | {\n        scheme: string;\n        credentials: {\n          username?: string;\n          password?: string;\n          token?: string;\n        };\n      }\n    | undefined;\n  oauth2?:\n    | {\n        clientId?: string;\n        clientSecret?: string;\n        authUri?: string;\n        state?: string;\n        redirectUri?: string;\n        authResponseUri?: string;\n        authCode?: string;\n        accessToken?: string;\n        refreshToken?: string;\n        expiresAt?: number;\n        expiresIn?: number;\n      }\n    | undefined;\n  serviceAccount?: unknown;\n};\n\nexport type AdkAuthRequest = {\n  toolCallId: string;\n  authConfig: unknown;\n};\n\nexport type AdkMessageMetadata = {\n  groundingMetadata?: unknown;\n  citationMetadata?: unknown;\n  usageMetadata?: unknown;\n};\n\n// ── RunConfig ──\n\nexport type AdkRunConfig = {\n  streamingMode?: \"none\" | \"sse\" | \"bidi\" | undefined;\n  pauseOnToolCalls?: boolean | undefined;\n  maxLlmCalls?: number | undefined;\n  saveInputBlobsAsArtifacts?: boolean | undefined;\n  supportCfc?: boolean | undefined;\n  speechConfig?: unknown;\n  responseModalities?: string[] | undefined;\n  outputAudioTranscription?: unknown;\n  inputAudioTranscription?: unknown;\n  enableAffectiveDialog?: boolean | undefined;\n  proactivity?: unknown;\n  realtimeInputConfig?: unknown;\n};\n\n// ── Stream callback types ──\n\nexport type AdkSendMessageConfig = {\n  /**\n   * ADK RunConfig. Typed as `unknown` for compatibility with\n   * assistant-ui core's RunConfig type. Use `AdkRunConfig` when\n   * constructing configs manually for type safety.\n   */\n  runConfig?: unknown;\n  checkpointId?: string | undefined;\n  stateDelta?: Record<string, unknown> | undefined;\n};\n\n// ── Structured events ──\n\nexport const AdkEventType = {\n  THOUGHT: \"thought\",\n  CONTENT: \"content\",\n  TOOL_CALL: \"tool_call\",\n  TOOL_RESULT: \"tool_result\",\n  CALL_CODE: \"call_code\",\n  CODE_RESULT: \"code_result\",\n  ERROR: \"error\",\n  ACTIVITY: \"activity\",\n  TOOL_CONFIRMATION: \"tool_confirmation\",\n  FINISHED: \"finished\",\n} as const;\n\nexport type AdkStructuredEvent =\n  | { type: \"thought\"; content: string }\n  | { type: \"content\"; content: string }\n  | {\n      type: \"tool_call\";\n      call: { name: string; id?: string; args: Record<string, unknown> };\n    }\n  | {\n      type: \"tool_result\";\n      result: { name: string; id?: string; response: unknown };\n    }\n  | { type: \"call_code\"; code: { code: string; language?: string } }\n  | { type: \"code_result\"; result: { output: string; outcome?: string } }\n  | { type: \"error\"; errorCode?: string; errorMessage?: string }\n  | { type: \"activity\"; message: string }\n  | {\n      type: \"tool_confirmation\";\n      confirmations: Record<string, unknown>;\n    }\n  | { type: \"finished\" };\n\nexport type AdkStreamCallback = (\n  messages: AdkMessage[],\n  config: AdkSendMessageConfig & {\n    abortSignal: AbortSignal;\n    initialize: () => Promise<{\n      remoteId: string;\n      externalId: string | undefined;\n    }>;\n  },\n) => Promise<AsyncGenerator<AdkEvent>> | AsyncGenerator<AdkEvent>;\n\n// ── Event handler callbacks ──\n\nexport type OnAdkErrorCallback = (error: unknown) => void | Promise<void>;\n\nexport type OnAdkCustomEventCallback = (\n  type: string,\n  data: unknown,\n) => void | Promise<void>;\n\nexport type OnAdkAgentTransferCallback = (\n  toAgent: string,\n) => void | Promise<void>;\n\n/** Private state and actions `useAdkRuntime` exposes through `thread.extras`. */\nexport type AdkRuntimeExtras = {\n  send: (messages: AdkMessage[], config: AdkSendMessageConfig) => Promise<void>;\n  agentInfo: { name?: string | undefined; branch?: string | undefined };\n  stateDelta: Record<string, unknown>;\n  artifactDelta: Record<string, number>;\n  longRunningToolIds: string[];\n  toolConfirmations: AdkToolConfirmation[];\n  authRequests: AdkAuthRequest[];\n  escalated: boolean;\n  messageMetadata: Map<string, AdkMessageMetadata>;\n};\n"],"mappings":";AA2NA,MAAa,eAAe;CAC1B,SAAS;CACT,SAAS;CACT,WAAW;CACX,aAAa;CACb,WAAW;CACX,aAAa;CACb,OAAO;CACP,UAAU;CACV,mBAAmB;CACnB,UAAU;AACZ"}
+{"version":3,"file":"types.js","names":[],"sources":["../src/types.ts"],"sourcesContent":["import type { MessageStatus } from \"@assistant-ui/core\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\n\n// ── ADK Event wire types (lightweight, no @google/adk dependency) ──\n\nexport type AdkEventPart = {\n  text?: string;\n  thought?: boolean;\n  functionCall?: {\n    name: string;\n    id?: string;\n    args: Record<string, unknown>;\n  };\n  functionResponse?: {\n    name: string;\n    id?: string;\n    response: unknown;\n  };\n  executableCode?: {\n    code: string;\n    language?: string;\n  };\n  codeExecutionResult?: {\n    output: string;\n    outcome?: string;\n  };\n  inlineData?: {\n    mimeType: string;\n    data: string;\n  };\n  fileData?: {\n    fileUri: string;\n    mimeType?: string;\n  };\n};\n\nexport type AdkEventActions = {\n  stateDelta?: Record<string, unknown> | undefined;\n  artifactDelta?: Record<string, number> | undefined;\n  transferToAgent?: string | undefined;\n  escalate?: boolean | undefined;\n  skipSummarization?: boolean | undefined;\n  requestedAuthConfigs?: Record<string, unknown> | undefined;\n  requestedToolConfirmations?: Record<string, unknown> | undefined;\n};\n\nexport type AdkEvent = {\n  id: string;\n  invocationId?: string | undefined;\n  author?: string | undefined;\n  branch?: string | undefined;\n  partial?: boolean | undefined;\n  turnComplete?: boolean | undefined;\n  interrupted?: boolean | undefined;\n  finishReason?: string | undefined;\n  timestamp?: number | undefined;\n  content?:\n    | {\n        role?: string | undefined;\n        parts?: AdkEventPart[] | undefined;\n      }\n    | undefined;\n  actions?: AdkEventActions | undefined;\n  longRunningToolIds?: string[] | undefined;\n  errorCode?: string | undefined;\n  errorMessage?: string | undefined;\n  groundingMetadata?: unknown;\n  citationMetadata?: unknown;\n  usageMetadata?: unknown;\n  customMetadata?: Record<string, unknown> | undefined;\n};\n\n// ── ADK Message types (accumulated from events) ──\n\nexport type AdkToolCall = {\n  id: string;\n  name: string;\n  args: ReadonlyJSONObject;\n  argsText?: string;\n};\n\nexport type AdkMessage =\n  | {\n      id: string;\n      type: \"human\";\n      content: string | AdkMessageContentPart[];\n    }\n  | {\n      id: string;\n      type: \"ai\";\n      content: string | AdkMessageContentPart[];\n      tool_calls?: AdkToolCall[] | undefined;\n      author?: string | undefined;\n      branch?: string | undefined;\n      status?: MessageStatus | undefined;\n    }\n  | {\n      id: string;\n      type: \"tool\";\n      content: string;\n      tool_call_id: string;\n      name: string;\n      status?: \"success\" | \"error\" | undefined;\n      artifact?: unknown;\n    };\n\nexport type AdkMessageContentPart =\n  | { type: \"text\"; text: string }\n  | { type: \"reasoning\"; text: string }\n  | { type: \"image\"; mimeType: string; data: string }\n  | { type: \"image_url\"; url: string }\n  | {\n      type: \"file\";\n      mimeType: string;\n      data: string;\n      filename?: string | undefined;\n    }\n  | {\n      type: \"file_url\";\n      url: string;\n      mimeType?: string | undefined;\n    }\n  | { type: \"code\"; code: string; language: string }\n  | { type: \"code_result\"; output: string; outcome: string }\n  | { type: \"activity\"; message: string };\n\n// ── ADK-specific state types ──\n\nexport type AdkToolConfirmation = {\n  toolCallId: string;\n  toolName: string;\n  args: Record<string, unknown>;\n  hint: string;\n  confirmed: boolean;\n  payload?: unknown;\n};\n\nexport type AdkAuthCredentialType =\n  | \"apiKey\"\n  | \"http\"\n  | \"oauth2\"\n  | \"openIdConnect\"\n  | \"serviceAccount\";\n\nexport type AdkAuthCredential = {\n  authType: AdkAuthCredentialType;\n  resourceRef?: string | undefined;\n  apiKey?: string | undefined;\n  http?:\n    | {\n        scheme: string;\n        credentials: {\n          username?: string;\n          password?: string;\n          token?: string;\n        };\n      }\n    | undefined;\n  oauth2?:\n    | {\n        clientId?: string;\n        clientSecret?: string;\n        authUri?: string;\n        state?: string;\n        redirectUri?: string;\n        authResponseUri?: string;\n        authCode?: string;\n        accessToken?: string;\n        refreshToken?: string;\n        expiresAt?: number;\n        expiresIn?: number;\n      }\n    | undefined;\n  serviceAccount?: unknown;\n};\n\nexport type AdkAuthRequest = {\n  toolCallId: string;\n  authConfig: unknown;\n};\n\nexport type AdkMessageMetadata = {\n  groundingMetadata?: unknown;\n  citationMetadata?: unknown;\n  usageMetadata?: unknown;\n};\n\n// ── RunConfig ──\n\nexport type AdkRunConfig = {\n  streamingMode?: \"none\" | \"sse\" | \"bidi\" | undefined;\n  pauseOnToolCalls?: boolean | undefined;\n  maxLlmCalls?: number | undefined;\n  saveInputBlobsAsArtifacts?: boolean | undefined;\n  supportCfc?: boolean | undefined;\n  speechConfig?: unknown;\n  responseModalities?: string[] | undefined;\n  outputAudioTranscription?: unknown;\n  inputAudioTranscription?: unknown;\n  enableAffectiveDialog?: boolean | undefined;\n  proactivity?: unknown;\n  realtimeInputConfig?: unknown;\n};\n\n// ── Stream callback types ──\n\nexport type AdkSendMessageConfig = {\n  /**\n   * ADK RunConfig. Typed as `unknown` for compatibility with\n   * assistant-ui core's RunConfig type. Use `AdkRunConfig` when\n   * constructing configs manually for type safety.\n   */\n  runConfig?: unknown;\n  checkpointId?: string | undefined;\n  stateDelta?: Record<string, unknown> | undefined;\n};\n\n// ── Structured events ──\n\nexport const AdkEventType = {\n  THOUGHT: \"thought\",\n  CONTENT: \"content\",\n  TOOL_CALL: \"tool_call\",\n  TOOL_RESULT: \"tool_result\",\n  CALL_CODE: \"call_code\",\n  CODE_RESULT: \"code_result\",\n  ERROR: \"error\",\n  ACTIVITY: \"activity\",\n  TOOL_CONFIRMATION: \"tool_confirmation\",\n  FINISHED: \"finished\",\n} as const;\n\nexport type AdkStructuredEvent =\n  | { type: \"thought\"; content: string }\n  | { type: \"content\"; content: string }\n  | {\n      type: \"tool_call\";\n      call: { name: string; id?: string; args: Record<string, unknown> };\n    }\n  | {\n      type: \"tool_result\";\n      result: { name: string; id?: string; response: unknown };\n    }\n  | { type: \"call_code\"; code: { code: string; language?: string } }\n  | { type: \"code_result\"; result: { output: string; outcome?: string } }\n  | { type: \"error\"; errorCode?: string; errorMessage?: string }\n  | { type: \"activity\"; message: string }\n  | {\n      type: \"tool_confirmation\";\n      confirmations: Record<string, unknown>;\n    }\n  | { type: \"finished\" };\n\nexport type AdkStreamCallback = (\n  messages: AdkMessage[],\n  config: AdkSendMessageConfig & {\n    abortSignal: AbortSignal;\n    initialize: () => Promise<{\n      remoteId: string;\n      externalId: string | undefined;\n    }>;\n  },\n) => Promise<AsyncGenerator<AdkEvent>> | AsyncGenerator<AdkEvent>;\n\n// ── Event handler callbacks ──\n\nexport type OnAdkErrorCallback = (error: unknown) => void | Promise<void>;\n\nexport type OnAdkCustomEventCallback = (\n  type: string,\n  data: unknown,\n) => void | Promise<void>;\n\nexport type OnAdkAgentTransferCallback = (\n  toAgent: string,\n) => void | Promise<void>;\n\n/** Private state and actions `useAdkRuntime` exposes through `thread.extras`. */\nexport type AdkRuntimeExtras = {\n  send: (messages: AdkMessage[], config: AdkSendMessageConfig) => Promise<void>;\n  agentInfo: { name?: string | undefined; branch?: string | undefined };\n  stateDelta: Record<string, unknown>;\n  artifactDelta: Record<string, number>;\n  longRunningToolIds: string[];\n  toolConfirmations: AdkToolConfirmation[];\n  authRequests: AdkAuthRequest[];\n  escalated: boolean;\n  messageMetadata: Map<string, AdkMessageMetadata>;\n};\n\n/**\n * What a session load reconstructs. The messages are the whole thread; the\n * rest is the per-turn state the events imply, which a caller that replays\n * them through {@link AdkEventAccumulator} already has. Omitting a field\n * clears it, since a snapshot that cannot describe the state is not evidence\n * that the state survived.\n */\nexport type AdkThreadSnapshot = {\n  messages: AdkMessage[];\n  longRunningToolIds?: string[] | undefined;\n  toolConfirmations?: AdkToolConfirmation[] | undefined;\n  authRequests?: AdkAuthRequest[] | undefined;\n  escalated?: boolean | undefined;\n  messageMetadata?: Map<string, AdkMessageMetadata> | undefined;\n  stateDelta?: Record<string, unknown> | undefined;\n  artifactDelta?: Record<string, number> | undefined;\n  agentInfo?:\n    | { name?: string | undefined; branch?: string | undefined }\n    | undefined;\n};\n"],"mappings":";AA2NA,MAAa,eAAe;CAC1B,SAAS;CACT,SAAS;CACT,WAAW;CACX,aAAa;CACb,WAAW;CACX,aAAa;CACb,OAAO;CACP,UAAU;CACV,mBAAmB;CACnB,UAAU;AACZ"}

package/dist/useAdkMessages.d.ts

@@ -1,4 +1,4 @@
-import { AdkAuthRequest, AdkMessage, AdkMessageMetadata, AdkSendMessageConfig, AdkStreamCallback, AdkToolConfirmation, OnAdkAgentTransferCallback, OnAdkCustomEventCallback, OnAdkErrorCallback } from "./types.js";
+import { AdkAuthRequest, AdkMessage, AdkMessageMetadata, AdkSendMessageConfig, AdkStreamCallback, AdkThreadSnapshot, AdkToolConfirmation, OnAdkAgentTransferCallback, OnAdkCustomEventCallback, OnAdkErrorCallback } from "./types.js";
 //#region src/useAdkMessages.d.ts
 type UseAdkMessagesOptions = {
   stream: AdkStreamCallback;
@@ -25,6 +25,7 @@ declare const useAdkMessages: ({ stream, eventHandlers }: UseAdkMessagesOptions)
   cancel: () => void;
   setMessages: (msgs: AdkMessage[]) => void;
   replaceMessages: (msgs: AdkMessage[]) => void;
+  applySnapshot: (snapshot: AdkThreadSnapshot) => void;
 };
 //#endregion
 export { UseAdkMessagesOptions, useAdkMessages };

package/dist/useAdkMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useAdkMessages.d.ts","names":[],"sources":["../src/useAdkMessages.ts"],"mappings":";;KAkBY;EACV,QAAQ;EACR;IACE,UAAU;IACV,gBAAgB;IAChB,kBAAkB;;;cAIT,mBAAkB,QAAA,iBAG5B;;;;IAIQ;IACE;;;;;;;;EAsDW,cAAA,aAAA,cAAY,QAAU,yBAAoB;;EA9BhB,cAAA,MAAA;EAUvC,kBAAA,MAAA"}
+{"version":3,"file":"useAdkMessages.d.ts","names":[],"sources":["../src/useAdkMessages.ts"],"mappings":";;KAmBY;EACV,QAAQ;EACR;IACE,UAAU;IACV,gBAAgB;IAChB,kBAAkB;;;cAIT,mBAAkB,QAAA,iBAG5B;;;;IAIQ;IACE;;;;;;;;EA2EW,cAAA,aAAA,cAAY,QAAU,yBAAoB;;EAnDhB,cAAA,MAAA;EA+BvC,kBAAA,MAAA;EAnBI,gBAAA,UAAA"}