@axiom-lattice/react-sdk 2.1.101 → 2.1.103

package/dist/index.js

@@ -2031,6 +2031,7 @@ function AgentThreadProvider({
   const sendMessageStreamsRef = (0, import_react7.useRef)(/* @__PURE__ */ new Map());
   const chunkMessageMerger = (0, import_react7.useRef)((0, import_client_sdk3.createSimpleMessageMerger)());
   const lastAgentStateCreatedAtRef = (0, import_react7.useRef)(null);
+  const streamEndedByInterruptRef = (0, import_react7.useRef)(false);
   const messageCountRef = (0, import_react7.useRef)(0);
   const onToolCompletedRef = (0, import_react7.useRef)(onToolCompleted);
   onToolCompletedRef.current = onToolCompleted;
@@ -2116,6 +2117,7 @@ function AgentThreadProvider({
       let interrupt;
       if (chunk.type === "interrupt") {
         const rawChunk = chunk;
+        streamEndedByInterruptRef.current = true;
         interrupt = {
           type: "interrupt",
           id: rawChunk.id,
@@ -2159,6 +2161,7 @@ function AgentThreadProvider({
       if (!threadId) {
         throw new Error("Thread ID is required to send messages");
       }
+      streamEndedByInterruptRef.current = false;
       sendMessageStreamsRef.current.forEach((stopFn) => {
         stopFn();
       });
@@ -2219,9 +2222,10 @@ function AgentThreadProvider({
               },
               // Chunks handled by resumeStream
               async () => {
-                if (options.enableReturnStateWhenStreamCompleted) {
+                if (options.enableReturnStateWhenStreamCompleted && !streamEndedByInterruptRef.current) {
                   await fetchAndUpdateAgentState(threadId);
                 }
+                streamEndedByInterruptRef.current = false;
                 onStreamCompletedRef.current?.();
                 sendMessageStreamsRef.current.delete(userMessage.id);
               },
@@ -2231,6 +2235,7 @@ function AgentThreadProvider({
                   isLoading: false,
                   error
                 }));
+                streamEndedByInterruptRef.current = false;
                 sendMessageStreamsRef.current.delete(userMessage.id);
               }
             );
@@ -2251,9 +2256,10 @@ function AgentThreadProvider({
                   ...prev,
                   isLoading: false
                 }));
-                if (options.enableReturnStateWhenStreamCompleted) {
+                if (options.enableReturnStateWhenStreamCompleted && !streamEndedByInterruptRef.current) {
                   await fetchAndUpdateAgentState(threadId);
                 }
+                streamEndedByInterruptRef.current = false;
                 onStreamCompletedRef.current?.();
                 sendMessageStreamsRef.current.delete(userMessage.id);
               },
@@ -14509,6 +14515,10 @@ var ClarifyFeedback = ({
   const { questions } = data ?? {};
   const { sendMessage, assistantId, threadId, tenantId } = useAgentChat();
   const { config } = useLatticeChatShellContext();
+  const workspaceContext = (0, import_react47.useContext)(WorkspaceContext);
+  const workspaceId = workspaceContext?.workspaceId ?? null;
+  const projectId = workspaceContext?.projectId ?? null;
+  const uploadURL = config?.baseURL ? workspaceId && projectId ? `${config.baseURL}/api/workspaces/${workspaceId}/projects/${projectId}/uploadfile` : `${config.baseURL}/api/assistants/${assistantId}/threads/${threadId}/sandbox/uploadfile` : "";
   const [currentStep, setCurrentStep] = (0, import_react47.useState)(0);
   const [answers, setAnswers] = (0, import_react47.useState)({});
   const [submitted, setSubmitted] = (0, import_react47.useState)(false);
@@ -14532,7 +14542,6 @@ var ClarifyFeedback = ({
       }
     }));
   };
-  const uploadURL = config?.baseURL && assistantId && threadId ? `${config.baseURL}/api/assistants/${assistantId}/threads/${threadId}/sandbox/uploadfile` : "";
   const handleFileUpload = (info) => {
     if (info.file.status === "done") {
       const response = info.file.response;