@axiom-lattice/react-sdk 2.1.0 → 2.1.2

package/dist/index.d.mts

@@ -1,4 +1,4 @@
-import { Message, MessageChunk } from '@axiom-lattice/protocols';
+import { Message, MessageChunk, InterruptMessage } from '@axiom-lattice/protocols';
 export * from '@axiom-lattice/protocols';
 import * as React$1 from 'react';
 import React__default, { ReactNode } from 'react';
@@ -92,6 +92,10 @@ interface ChatState {
      * Current streaming message (if any)
      */
     streamingMessage: MessageChunk | null;
+    /**
+     * Interrupt data when agent requires user intervention
+     */
+    interrupts: InterruptMessage[] | undefined;
 }
 /**
  * Extended chat state with agent state
@@ -386,6 +390,7 @@ interface ChatingProps {
     onOpenSidePanel: (data: any) => void;
     onReminderClick: () => void;
     onClearError?: () => void;
+    interrupts: InterruptMessage[] | undefined;
     styles: any;
     reminderCount: number;
     handleMDResponseEvent: (action: string, data?: any, message?: string, agent?: string) => void;

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { Message, MessageChunk } from '@axiom-lattice/protocols';
+import { Message, MessageChunk, InterruptMessage } from '@axiom-lattice/protocols';
 export * from '@axiom-lattice/protocols';
 import * as React$1 from 'react';
 import React__default, { ReactNode } from 'react';
@@ -92,6 +92,10 @@ interface ChatState {
      * Current streaming message (if any)
      */
     streamingMessage: MessageChunk | null;
+    /**
+     * Interrupt data when agent requires user intervention
+     */
+    interrupts: InterruptMessage[] | undefined;
 }
 /**
  * Extended chat state with agent state
@@ -386,6 +390,7 @@ interface ChatingProps {
     onOpenSidePanel: (data: any) => void;
     onReminderClick: () => void;
     onClearError?: () => void;
+    interrupts: InterruptMessage[] | undefined;
     styles: any;
     reminderCount: number;
     handleMDResponseEvent: (action: string, data?: any, message?: string, agent?: string) => void;

package/dist/index.js

@@ -96,6 +96,7 @@ function useChat(threadId, options = {}) {
     isLoading: false,
     error: null,
     streamingMessage: null,
+    interrupts: void 0,
     ...options.enableReturnStateWhenStreamCompleted ? { agentState: null } : {}
   });
   const stopStreamingRef = (0, import_react2.useRef)(null);
@@ -108,7 +109,15 @@ function useChat(threadId, options = {}) {
         setState(
           (prev) => ({
             ...prev,
-            agentState
+            agentState,
+            interrupts: agentState?.tasks?.flatMap((task) => {
+              return task.interrupts.map((interrupt) => {
+                return {
+                  id: interrupt.id,
+                  value: interrupt.value
+                };
+              });
+            })
           })
         );
       } catch (error) {
@@ -122,29 +131,38 @@ function useChat(threadId, options = {}) {
       chunkMessageMerger.current.initialMessages(options.initialMessages);
     }
   }, [options.initialMessages]);
-  const handleStreamEvent = (0, import_react2.useCallback)((chunk) => {
-    chunkMessageMerger.current.push(chunk);
-    let todos;
-    if (chunk.type === "tool" && chunk.data && typeof chunk.data.content === "string" && chunk.data.content.startsWith("```todo_list")) {
-      try {
-        const content = chunk.data.content;
-        const match = content.match(/```todo_list\s*([\s\S]*?)\s*```/);
-        if (match && match[1]) {
-          todos = JSON.parse(match[1]);
+  const handleStreamEvent = (0, import_react2.useCallback)(
+    (chunk) => {
+      let interrupt;
+      if (chunk.type === "interrupt") {
+        interrupt = chunk;
+      } else {
+        chunkMessageMerger.current.push(chunk);
+      }
+      let todos;
+      if (chunk.type === "tool" && chunk.data && typeof chunk.data.content === "string" && chunk.data.content.startsWith("```todo_list")) {
+        try {
+          const content = chunk.data.content;
+          const match = content.match(/```todo_list\s*([\s\S]*?)\s*```/);
+          if (match && match[1]) {
+            todos = JSON.parse(match[1]);
+          }
+        } catch (e) {
+          console.error("Failed to parse todo list from chunk", e);
         }
-      } catch (e) {
-        console.error("Failed to parse todo list from chunk", e);
       }
-    }
-    const updatedMessages = chunkMessageMerger.current.getMessages();
-    setState((prev) => ({
-      ...prev,
-      todos: todos || prev.todos,
-      messages: updatedMessages,
-      isLoading: true,
-      streamingMessage: null
-    }));
-  }, []);
+      const updatedMessages = chunkMessageMerger.current.getMessages();
+      setState((prev) => ({
+        ...prev,
+        interrupts: interrupt ? [interrupt] : void 0,
+        todos: todos || prev.todos,
+        messages: updatedMessages,
+        isLoading: true,
+        streamingMessage: null
+      }));
+    },
+    []
+  );
   const sendMessage = (0, import_react2.useCallback)(
     async (data) => {
       if (!threadId) {
@@ -271,12 +289,21 @@ function useChat(threadId, options = {}) {
       setState((prev) => ({ ...prev, isLoading: true, error: null }));
       try {
         const agentState = await client.getAgentState(threadId);
+        const interrupts = agentState?.tasks?.flatMap((task) => {
+          return task.interrupts.map((interrupt) => {
+            return {
+              id: interrupt.id,
+              value: interrupt.value
+            };
+          });
+        });
         const fetchedMessages = await client.getMessages({ threadId });
         chunkMessageMerger.current.reset();
         chunkMessageMerger.current.initialMessages(fetchedMessages);
         setState((prev) => ({
           ...prev,
           agentState,
+          interrupts,
           todos: agentState?.values?.todos,
           messages: chunkMessageMerger.current.getMessages(),
           isLoading: false
@@ -331,6 +358,7 @@ function useChat(threadId, options = {}) {
     setState((prev) => ({
       ...prev,
       messages: [],
+      interrupts: void 0,
       streamingMessage: null
     }));
   }, []);
@@ -1202,7 +1230,7 @@ var Todo = ({
       case "completed":
         return /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(import_icons4.CheckCircleOutlined, { style: { color: "#52c41a" } });
       case "in_progress":
-        return /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(import_icons4.LoadingOutlined, { style: { color: "#1890ff" } });
+        return /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(import_icons4.ArrowRightOutlined, { style: { fontWeight: "500" } });
       case "pending":
         return /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(import_icons4.ClockCircleOutlined, { style: { color: "gray" } });
       default:
@@ -1327,7 +1355,7 @@ var WriteTodos = ({
     {
       size: "small",
       bordered: false,
-      defaultActiveKey: [],
+      defaultActiveKey: [toolCallData.id],
       expandIcon,
       children: /* @__PURE__ */ (0, import_jsx_runtime8.jsx)(
         import_CollapsePanel2.default,
@@ -2680,6 +2708,7 @@ MemoizedBubbleList.displayName = "MemoizedBubbleList";
 var Chating = ({
   avatar,
   name,
+  interrupts,
   description,
   default_submit_message,
   tenant_id,
@@ -3091,9 +3120,18 @@ ${JSON.stringify(tool_call)}
         message: `${error.message}`
       }
     ) }),
+    interrupts && interrupts.length > 0 && /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(import_antd13.Space, { direction: "vertical", style: { width: "100%" }, children: interrupts.map((interrupt) => /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(
+      MDResponse,
+      {
+        content: interrupt.value,
+        eventHandler: handleMDResponseEvent
+      },
+      interrupt.id
+    )) }),
     /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(
       import_x2.Sender,
       {
+        disabled: interrupts && interrupts.length > 0,
         allowSpeech: true,
         ref: senderRef,
         value: content,