@axiom-lattice/react-sdk 2.1.9 → 2.1.11

package/dist/index.mjs

@@ -396,24 +396,33 @@ 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);
+        const currentCreatedAt = agentState?.createdAt;
+        const needsUpdate = !lastAgentStateCreatedAtRef.current || currentCreatedAt !== lastAgentStateCreatedAtRef.current;
+        if (currentCreatedAt) {
+          lastAgentStateCreatedAtRef.current = currentCreatedAt;
+        }
+        let needUpdateFields = {};
+        needUpdateFields.agentState = agentState;
+        needUpdateFields.interrupts = agentState?.tasks?.flatMap((task) => {
+          return task.interrupts.map((interrupt) => {
+            return {
+              id: interrupt.id,
+              value: interrupt.value,
+              role: "ai",
+              type: "interrupt"
+            };
+          });
+        });
+        needUpdateFields.todos = agentState?.values?.todos;
         setState((prev) => ({
           ...prev,
-          agentState,
-          interrupts: agentState?.tasks?.flatMap((task) => {
-            return task.interrupts.map((interrupt) => {
-              return {
-                id: interrupt.id,
-                value: interrupt.value,
-                role: "ai",
-                type: "interrupt"
-              };
-            });
-          })
+          ...needUpdateFields
         }));
       } catch (error) {
         console.warn("Failed to fetch agent state:", error);
@@ -583,6 +592,15 @@ function AgentThreadProvider({
       setState((prev) => ({ ...prev, isLoading: true, error: null }));
       try {
         const agentState = await client.getAgentState(threadId);
+        const currentCreatedAt = agentState?.createdAt;
+        const needsUpdate = !lastAgentStateCreatedAtRef.current || currentCreatedAt !== lastAgentStateCreatedAtRef.current;
+        if (currentCreatedAt) {
+          lastAgentStateCreatedAtRef.current = currentCreatedAt;
+        }
+        let needUpdateFields = {};
+        let fetchedMessages = [];
+        needUpdateFields.agentState = agentState;
+        needUpdateFields.todos = agentState?.values?.todos;
         const interrupts = agentState?.tasks?.flatMap(
           (task) => {
             return task.interrupts.map((interrupt) => {
@@ -595,15 +613,14 @@ function AgentThreadProvider({
             });
           }
         );
-        const fetchedMessages = await client.getMessages({ threadId });
+        needUpdateFields.interrupts = interrupts;
+        fetchedMessages = await client.getMessages({ threadId });
+        needUpdateFields.messages = fetchedMessages;
         chunkMessageMerger.current.reset();
         chunkMessageMerger.current.initialMessages(fetchedMessages);
         setState((prev) => ({
           ...prev,
-          agentState,
-          interrupts,
-          todos: agentState?.values?.todos,
-          messages: chunkMessageMerger.current.getMessages(),
+          ...needUpdateFields,
           isLoading: false
         }));
         if (options.enableResumeStream && fetchedMessages.length > 0) {
@@ -657,6 +674,7 @@ function AgentThreadProvider({
   );
   const clearMessages = useCallback2(() => {
     chunkMessageMerger.current.reset();
+    lastAgentStateCreatedAtRef.current = null;
     setState((prev) => ({
       ...prev,
       messages: [],
@@ -672,6 +690,7 @@ function AgentThreadProvider({
   }, []);
   useEffect3(() => {
     if (threadId && clientAssistantId === assistantId) {
+      lastAgentStateCreatedAtRef.current = null;
       loadMessages();
     } else {
       clearMessages();
@@ -3259,17 +3278,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 +3406,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,10 +3504,13 @@ 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,
   Prompts,
   Sender
 } from "@ant-design/x";
@@ -3888,7 +3910,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 +3923,7 @@ var Chating = ({
     messages,
     sendMessage,
     stopStreaming,
+    loadMessages,
     isLoading,
     error,
     interrupts,
@@ -4041,6 +4065,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,12 +4083,12 @@ var Chating = ({
         description,
         avatar,
         name,
-        extra,
+        extra: extra ? [...extra, ...headerExtra] : headerExtra,
         extraMeta
       }
     ) }),
     /* @__PURE__ */ jsx28(MessageList, { messages, className: styles.messages }),
-    isLoading ? /* @__PURE__ */ jsx28("div", { children: /* @__PURE__ */ jsx28(Bubble2, { loading: isLoading, variant: "borderless", content: "" }) }) : /* @__PURE__ */ jsx28(Prompts, { items: senderPromptsItems, onItemClick: onPromptsItemClick }),
+    isLoading ? /* @__PURE__ */ jsx28("div", {}) : /* @__PURE__ */ jsx28(Prompts, { items: senderPromptsItems, onItemClick: onPromptsItemClick }),
     error && /* @__PURE__ */ jsx28("div", { style: { padding: "0 16px 8px" }, children: /* @__PURE__ */ jsx28(
       Alert2,
       {
@@ -4108,6 +4143,7 @@ var TaskDetail = ({ data, component_key, interactive = true }) => {
       children: /* @__PURE__ */ jsx29("div", { style: { overflow: "hidden" }, children: /* @__PURE__ */ jsx29(
         Chating,
         {
+          showRefreshButton: true,
           name: subagent_type,
           showHeader: true,
           showSender: false,