@axiom-lattice/react-sdk 2.1.8 → 2.1.10

package/dist/index.d.mts

@@ -58,6 +58,7 @@ interface ChatResponse {
  * Agent state interface
  */
 interface AgentState {
+    createdAt?: string;
     values: {
         messages: Message[];
         [key: string]: any;
@@ -446,6 +447,7 @@ interface ChatingProps {
     showHeader?: boolean;
     showSender?: boolean;
     showHITL?: boolean;
+    showRefreshButton?: boolean;
 }
 declare const Chating: React__default.FC<ChatingProps>;
 

package/dist/index.d.ts

@@ -58,6 +58,7 @@ interface ChatResponse {
  * Agent state interface
  */
 interface AgentState {
+    createdAt?: string;
     values: {
         messages: Message[];
         [key: string]: any;
@@ -446,6 +447,7 @@ interface ChatingProps {
     showHeader?: boolean;
     showSender?: boolean;
     showHITL?: boolean;
+    showRefreshButton?: boolean;
 }
 declare const Chating: React__default.FC<ChatingProps>;
 

package/dist/index.js

@@ -441,15 +441,21 @@ function AgentThreadProvider({
   });
   const stopStreamingRef = (0, import_react3.useRef)(null);
   const chunkMessageMerger = (0, import_react3.useRef)((0, import_client_sdk3.createSimpleMessageMerger)());
+  const lastAgentStateCreatedAtRef = (0, import_react3.useRef)(null);
   const fetchAndUpdateAgentState = (0, import_react3.useCallback)(
     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,
@@ -458,7 +464,12 @@ function AgentThreadProvider({
                 type: "interrupt"
               };
             });
-          })
+          });
+          needUpdateFields.todos = agentState?.values?.todos;
+        }
+        setState((prev) => ({
+          ...prev,
+          ...needUpdateFields
         }));
       } catch (error) {
         console.warn("Failed to fetch agent state:", error);
@@ -628,27 +639,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) {
@@ -702,6 +725,7 @@ function AgentThreadProvider({
   );
   const clearMessages = (0, import_react3.useCallback)(() => {
     chunkMessageMerger.current.reset();
+    lastAgentStateCreatedAtRef.current = null;
     setState((prev) => ({
       ...prev,
       messages: [],
@@ -716,7 +740,8 @@ function AgentThreadProvider({
     }));
   }, []);
   (0, import_react3.useEffect)(() => {
-    if (threadId) {
+    if (threadId && clientAssistantId === assistantId) {
+      lastAgentStateCreatedAtRef.current = null;
       loadMessages();
     } else {
       clearMessages();
@@ -727,7 +752,7 @@ function AgentThreadProvider({
         stopStreamingRef.current = null;
       }
     };
-  }, [threadId, loadMessages, clearMessages]);
+  }, [threadId, loadMessages, clearMessages, clientAssistantId, assistantId]);
   const value = {
     state,
     sendMessage,
@@ -3262,17 +3287,17 @@ var useStyle5 = (0, import_antd_style7.createStyles)(({ 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%
       );
@@ -3390,7 +3415,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":
@@ -3869,7 +3894,8 @@ var Chating = ({
   uploadAction = "/api/file_storage/upload?path=temp",
   showHeader = true,
   showSender = true,
-  showHITL = true
+  showHITL = true,
+  showRefreshButton = false
 }) => {
   const [content, setContent] = (0, import_react18.useState)("");
   const [attachedFiles, setAttachedFiles] = (0, import_react18.useState)([]);
@@ -3881,6 +3907,7 @@ var Chating = ({
     messages,
     sendMessage,
     stopStreaming,
+    loadMessages,
     isLoading,
     error,
     interrupts,
@@ -4022,6 +4049,17 @@ var Chating = ({
       )
     }
   );
+  const refreshButton = /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(
+    import_antd19.Button,
+    {
+      type: "text",
+      icon: /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(import_icons11.ReloadOutlined, {}),
+      onClick: () => {
+        loadMessages();
+      }
+    }
+  );
+  const headerExtra = showRefreshButton ? [refreshButton] : [];
   return /* @__PURE__ */ (0, import_jsx_runtime28.jsxs)(import_jsx_runtime28.Fragment, { children: [
     /* @__PURE__ */ (0, import_jsx_runtime28.jsx)("div", { children: showHeader && /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(
       AgentHeader,
@@ -4029,7 +4067,7 @@ var Chating = ({
         description,
         avatar,
         name,
-        extra,
+        extra: extra ? [...extra, ...headerExtra] : headerExtra,
         extraMeta
       }
     ) }),
@@ -4089,7 +4127,7 @@ var TaskDetail = ({ data, component_key, interactive = true }) => {
       children: /* @__PURE__ */ (0, import_jsx_runtime29.jsx)("div", { style: { overflow: "hidden" }, children: /* @__PURE__ */ (0, import_jsx_runtime29.jsx)(
         Chating,
         {
-          description,
+          showRefreshButton: true,
           name: subagent_type,
           showHeader: true,
           showSender: false,