@axiom-lattice/protocols 4.1.3 → 4.1.4

package/dist/index.mjs

@@ -451,6 +451,85 @@ function hasOnlyKeys(value, required, optional) {
   return required.every((key) => key in value) && keys.every((key) => required.includes(key) || optional.includes(key));
 }
 
+// src/AgentWebAppStreamProjection.ts
+function createAgentWebAppStreamProjectionContext(options = {}) {
+  return { startedToolIds: /* @__PURE__ */ new Set(), allowInterrupts: options.allowInterrupts ?? true, allowGenUI: options.allowGenUI ?? false };
+}
+function projectAgentWebAppChunk(chunk, context) {
+  if (chunk.type === MessageChunkTypes.AI) {
+    const events = [];
+    const startedToolIds = context?.startedToolIds ?? /* @__PURE__ */ new Set();
+    if (typeof chunk.data.content === "string" && chunk.data.content) events.push({ type: "message.delta", text: chunk.data.content });
+    if (context?.allowGenUI && Array.isArray(chunk.data.content)) {
+      for (const value of chunk.data.content) {
+        const block = parseAgentWebAppGenUIBlock(value);
+        if (block) events.push({ type: "genui.render", block });
+      }
+    }
+    for (const call of chunk.data.tool_call_chunks ?? []) {
+      if (call.id && call.name && !startedToolIds.has(call.id)) {
+        startedToolIds.add(call.id);
+        events.push({ type: "tool.started", id: call.id, name: call.name });
+      }
+    }
+    for (const call of chunk.data.tool_calls ?? []) {
+      if (!startedToolIds.has(call.id)) {
+        startedToolIds.add(call.id);
+        events.push({ type: "tool.started", id: call.id, name: call.name });
+      }
+    }
+    return events;
+  }
+  if (chunk.type === MessageChunkTypes.TOOL && chunk.data.tool_call_id) {
+    return [{ type: "tool.completed", id: chunk.data.tool_call_id }];
+  }
+  if (chunk.type === MessageChunkTypes.INTERRUPT) {
+    if (context && !context.allowInterrupts) return [{ type: "stream.completed" }];
+    const interrupt = publicInterrupt(chunk);
+    return interrupt ? [{ type: "interrupt.created", interrupt }, { type: "stream.completed" }] : [{ type: "stream.completed" }];
+  }
+  if (chunk.type === MessageChunkTypes.MESSAGE_COMPLETED) {
+    return [
+      { type: "message.completed", messageId: chunk.data.id },
+      { type: "stream.completed" }
+    ];
+  }
+  if (chunk.type === MessageChunkTypes.MESSAGE_FAILED) {
+    return [
+      { type: "error", error: { code: "STREAM_FAILED", message: "Stream failed", retryable: true } },
+      { type: "stream.completed" }
+    ];
+  }
+  return [];
+}
+function publicInterrupt(chunk) {
+  const value = parseInterruptContent(chunk.data.content);
+  if (!value) return void 0;
+  const topLevelId = chunk.id;
+  const id = typeof topLevelId === "string" ? topLevelId : chunk.data.id;
+  if (!id) return void 0;
+  return {
+    id,
+    type: typeof value.type === "string" ? value.type : "input",
+    prompt: typeof value.prompt === "string" ? value.prompt : typeof value.message === "string" ? value.message : "Input required",
+    ...isRecord2(value.data) ? { data: value.data } : {}
+  };
+}
+function parseInterruptContent(content) {
+  if (!content) return {};
+  if (isRecord2(content)) return content;
+  if (typeof content !== "string") return void 0;
+  try {
+    const parsed = JSON.parse(content);
+    return isRecord2(parsed) ? parsed : { prompt: content };
+  } catch {
+    return { prompt: content };
+  }
+}
+function isRecord2(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+
 // src/ExactDataSnapshot.ts
 function ownDescriptorField(descriptor, key) {
   const field = Object.getOwnPropertyDescriptor(descriptor, key);
@@ -796,6 +875,7 @@ export {
   ScheduledTaskStatus,
   UIComponentType,
   assertGenericProjectConfig,
+  createAgentWebAppStreamProjectionContext,
   descriptorDataValue,
   getSubAgentsFromConfig,
   getToolsFromConfig,
@@ -812,6 +892,7 @@ export {
   parseQueuedExecutionMode,
   parseTaskBeliefState,
   parseTrustedRunContext,
+  projectAgentWebAppChunk,
   replaceTaskBeliefState,
   requireProjectTaskWorkItemStore,
   snapshotExactArray,