@axiom-lattice/react-sdk 2.1.9 → 2.1.11

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,24 +441,33 @@ 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);
+        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);
@@ -628,6 +637,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) => {
@@ -640,15 +658,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) {
@@ -702,6 +719,7 @@ function AgentThreadProvider({
   );
   const clearMessages = (0, import_react3.useCallback)(() => {
     chunkMessageMerger.current.reset();
+    lastAgentStateCreatedAtRef.current = null;
     setState((prev) => ({
       ...prev,
       messages: [],
@@ -717,6 +735,7 @@ function AgentThreadProvider({
   }, []);
   (0, import_react3.useEffect)(() => {
     if (threadId && clientAssistantId === assistantId) {
+      lastAgentStateCreatedAtRef.current = null;
       loadMessages();
     } else {
       clearMessages();
@@ -3262,17 +3281,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 +3409,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 +3888,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 +3901,7 @@ var Chating = ({
     messages,
     sendMessage,
     stopStreaming,
+    loadMessages,
     isLoading,
     error,
     interrupts,
@@ -4022,6 +4043,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,12 +4061,12 @@ var Chating = ({
         description,
         avatar,
         name,
-        extra,
+        extra: extra ? [...extra, ...headerExtra] : headerExtra,
         extraMeta
       }
     ) }),
     /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(MessageList, { messages, className: styles.messages }),
-    isLoading ? /* @__PURE__ */ (0, import_jsx_runtime28.jsx)("div", { children: /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(import_x4.Bubble, { loading: isLoading, variant: "borderless", content: "" }) }) : /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(import_x4.Prompts, { items: senderPromptsItems, onItemClick: onPromptsItemClick }),
+    isLoading ? /* @__PURE__ */ (0, import_jsx_runtime28.jsx)("div", {}) : /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(import_x4.Prompts, { items: senderPromptsItems, onItemClick: onPromptsItemClick }),
     error && /* @__PURE__ */ (0, import_jsx_runtime28.jsx)("div", { style: { padding: "0 16px 8px" }, children: /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(
       import_antd19.Alert,
       {
@@ -4089,6 +4121,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,
         {
+          showRefreshButton: true,
           name: subagent_type,
           showHeader: true,
           showSender: false,