@axiom-lattice/react-sdk 2.1.3 → 2.1.5

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/index.mjs CHANGED
@@ -257,20 +257,23 @@ function useChat(threadId, options = {}) {
257
257
  }));
258
258
  if (options.enableResumeStream && fetchedMessages.length > 0) {
259
259
  const lastMessage = fetchedMessages[fetchedMessages.length - 1];
260
- const lastContent = typeof lastMessage.content === "string" ? lastMessage.content : JSON.stringify(lastMessage.content || "");
260
+ chunkMessageMerger.current.removeMessageById(lastMessage.id);
261
261
  const stopResumeStream = client.resumeStream(
262
262
  {
263
263
  threadId,
264
264
  messageId: lastMessage.id,
265
- knownContent: lastContent,
265
+ knownContent: "",
266
266
  pollInterval: options.resumeStreamPollInterval || 100
267
267
  },
268
268
  handleStreamEvent,
269
- () => {
269
+ async () => {
270
270
  setState((prev) => ({
271
271
  ...prev,
272
272
  isLoading: false
273
273
  }));
274
+ if (options.enableReturnStateWhenStreamCompleted) {
275
+ await fetchAndUpdateAgentState(threadId);
276
+ }
274
277
  stopStreamingRef.current = null;
275
278
  },
276
279
  (error) => {