@axiom-lattice/react-sdk 2.1.8 → 2.1.10

package/dist/index.mjs

@@ -396,15 +396,21 @@ function AgentThreadProvider({
   });
   const stopStreamingRef = useRef2(null);
   const chunkMessageMerger = useRef2(createSimpleMessageMerger2());
+  const lastAgentStateCreatedAtRef = useRef2(null);
   const fetchAndUpdateAgentState = useCallback2(
     async (threadId2) => {
       if (!options.enableReturnStateWhenStreamCompleted) return;
       try {
         const agentState = await client.getAgentState(threadId2);
-        setState((prev) => ({
-          ...prev,
-          agentState,
-          interrupts: agentState?.tasks?.flatMap((task) => {
+        const currentCreatedAt = agentState?.createdAt;
+        const needsUpdate = !lastAgentStateCreatedAtRef.current || currentCreatedAt !== lastAgentStateCreatedAtRef.current;
+        if (currentCreatedAt) {
+          lastAgentStateCreatedAtRef.current = currentCreatedAt;
+        }
+        let needUpdateFields = {};
+        if (needsUpdate) {
+          needUpdateFields.agentState = agentState;
+          needUpdateFields.interrupts = agentState?.tasks?.flatMap((task) => {
             return task.interrupts.map((interrupt) => {
               return {
                 id: interrupt.id,
@@ -413,7 +419,12 @@ function AgentThreadProvider({
                 type: "interrupt"
               };
             });
-          })
+          });
+          needUpdateFields.todos = agentState?.values?.todos;
+        }
+        setState((prev) => ({
+          ...prev,
+          ...needUpdateFields
         }));
       } catch (error) {
         console.warn("Failed to fetch agent state:", error);
@@ -583,27 +594,39 @@ function AgentThreadProvider({
       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,
-                role: "ai",
-                type: "interrupt"
-              };
-            });
-          }
-        );
-        const fetchedMessages = await client.getMessages({ threadId });
-        chunkMessageMerger.current.reset();
-        chunkMessageMerger.current.initialMessages(fetchedMessages);
+        const currentCreatedAt = agentState?.createdAt;
+        const needsUpdate = !lastAgentStateCreatedAtRef.current || currentCreatedAt !== lastAgentStateCreatedAtRef.current;
+        if (currentCreatedAt) {
+          lastAgentStateCreatedAtRef.current = currentCreatedAt;
+        }
+        let needUpdateFields = {};
+        let fetchedMessages = [];
+        if (needsUpdate) {
+          needUpdateFields.agentState = agentState;
+          needUpdateFields.todos = agentState?.values?.todos;
+          const interrupts = agentState?.tasks?.flatMap(
+            (task) => {
+              return task.interrupts.map((interrupt) => {
+                return {
+                  id: interrupt.id,
+                  value: interrupt.value,
+                  role: "ai",
+                  type: "interrupt"
+                };
+              });
+            }
+          );
+          needUpdateFields.interrupts = interrupts;
+          fetchedMessages = await client.getMessages({ threadId });
+          needUpdateFields.messages = fetchedMessages;
+          chunkMessageMerger.current.reset();
+          chunkMessageMerger.current.initialMessages(fetchedMessages);
+        } else {
+          fetchedMessages = chunkMessageMerger.current.getMessages();
+        }
         setState((prev) => ({
           ...prev,
-          agentState,
-          interrupts,
-          todos: agentState?.values?.todos,
-          messages: chunkMessageMerger.current.getMessages(),
+          ...needUpdateFields,
           isLoading: false
         }));
         if (options.enableResumeStream && fetchedMessages.length > 0) {
@@ -657,6 +680,7 @@ function AgentThreadProvider({
   );
   const clearMessages = useCallback2(() => {
     chunkMessageMerger.current.reset();
+    lastAgentStateCreatedAtRef.current = null;
     setState((prev) => ({
       ...prev,
       messages: [],
@@ -671,7 +695,8 @@ function AgentThreadProvider({
     }));
   }, []);
   useEffect3(() => {
-    if (threadId) {
+    if (threadId && clientAssistantId === assistantId) {
+      lastAgentStateCreatedAtRef.current = null;
       loadMessages();
     } else {
       clearMessages();
@@ -682,7 +707,7 @@ function AgentThreadProvider({
         stopStreamingRef.current = null;
       }
     };
-  }, [threadId, loadMessages, clearMessages]);
+  }, [threadId, loadMessages, clearMessages, clientAssistantId, assistantId]);
   const value = {
     state,
     sendMessage,
@@ -3259,17 +3284,17 @@ var useStyle5 = createStyles7(({ token, css }) => ({
     &:hover {
       border-color: ${token.colorPrimary};
       box-shadow: 0 8px 24px rgba(24, 144, 255, 0.12);
-      transform: translateY(-4px);
+      transform: translateX(4px);
     }
     &::before {
       content: "";
       position: absolute;
       top: 0;
       left: 0;
-      right: 0;
-      height: 4px;
+      bottom: 0;
+      width: 4px;
       background: linear-gradient(
-        90deg,
+        180deg,
         ${token.colorPrimary} 0%,
         ${token.colorPrimaryHover} 100%
       );
@@ -3387,7 +3412,7 @@ var TaskCard = ({
   const { description, subagent_type, assignee } = toolCallData?.args || {};
   const status = toolCallData.status || "pending";
   const { threadId } = useAgentChat();
-  const subagent_thread_id = (threadId || "") + "_" + subagent_type + "_" + toolCallData.id;
+  const subagent_thread_id = (threadId || "") + "____" + subagent_type + "_" + toolCallData.id;
   const getStatusConfig = (status2) => {
     switch (status2) {
       case "success":
@@ -3485,7 +3510,11 @@ var TaskCard = ({
 import { Typography as Typography12 } from "antd";
 
 // src/components/Chat/Chating.tsx
-import { CloudUploadOutlined, PaperClipOutlined } from "@ant-design/icons";
+import {
+  CloudUploadOutlined,
+  PaperClipOutlined,
+  ReloadOutlined
+} from "@ant-design/icons";
 import {
   Attachments,
   Bubble as Bubble2,
@@ -3888,7 +3917,8 @@ var Chating = ({
   uploadAction = "/api/file_storage/upload?path=temp",
   showHeader = true,
   showSender = true,
-  showHITL = true
+  showHITL = true,
+  showRefreshButton = false
 }) => {
   const [content, setContent] = useState16("");
   const [attachedFiles, setAttachedFiles] = useState16([]);
@@ -3900,6 +3930,7 @@ var Chating = ({
     messages,
     sendMessage,
     stopStreaming,
+    loadMessages,
     isLoading,
     error,
     interrupts,
@@ -4041,6 +4072,17 @@ var Chating = ({
       )
     }
   );
+  const refreshButton = /* @__PURE__ */ jsx28(
+    Button10,
+    {
+      type: "text",
+      icon: /* @__PURE__ */ jsx28(ReloadOutlined, {}),
+      onClick: () => {
+        loadMessages();
+      }
+    }
+  );
+  const headerExtra = showRefreshButton ? [refreshButton] : [];
   return /* @__PURE__ */ jsxs17(Fragment4, { children: [
     /* @__PURE__ */ jsx28("div", { children: showHeader && /* @__PURE__ */ jsx28(
       AgentHeader,
@@ -4048,7 +4090,7 @@ var Chating = ({
         description,
         avatar,
         name,
-        extra,
+        extra: extra ? [...extra, ...headerExtra] : headerExtra,
         extraMeta
       }
     ) }),
@@ -4108,7 +4150,7 @@ var TaskDetail = ({ data, component_key, interactive = true }) => {
       children: /* @__PURE__ */ jsx29("div", { style: { overflow: "hidden" }, children: /* @__PURE__ */ jsx29(
         Chating,
         {
-          description,
+          showRefreshButton: true,
           name: subagent_type,
           showHeader: true,
           showSender: false,