@yourgpt/copilot-sdk 2.1.5-alpha.3 → 2.1.5-alpha.4

package/dist/ui/index.d.cts

@@ -7,9 +7,9 @@ import * as class_variance_authority_types from 'class-variance-authority/types'
 import { VariantProps } from 'class-variance-authority';
 import * as use_stick_to_bottom from 'use-stick-to-bottom';
 import { d as WebSearchResult, f as WebSearchResponse } from '../types-ZguuKEs_.cjs';
-import { a as Thread, c as ThreadStorageAdapter, A as AsyncThreadStorageAdapter } from '../types-BeFBBZ5i.cjs';
+import { a as Thread, c as ThreadStorageAdapter, A as AsyncThreadStorageAdapter } from '../types-waEqyE4K.cjs';
 import { T as ToolDefinition } from '../tools-tmksfhUo.cjs';
-import { B as BranchInfo } from '../MessageTree-CoIt_4nB.cjs';
+import { B as BranchInfo } from '../MessageTree-B0JGQOCi.cjs';
 import { d as MCPUIIntent, n as MCPUIFrameProps, M as MCPUIResource } from '../types-B20VCJXL.cjs';
 import { ClassValue } from 'clsx';
 

package/dist/ui/index.d.ts

@@ -7,9 +7,9 @@ import * as class_variance_authority_types from 'class-variance-authority/types'
 import { VariantProps } from 'class-variance-authority';
 import * as use_stick_to_bottom from 'use-stick-to-bottom';
 import { d as WebSearchResult, f as WebSearchResponse } from '../types-ZguuKEs_.js';
-import { a as Thread, c as ThreadStorageAdapter, A as AsyncThreadStorageAdapter } from '../types-BeFBBZ5i.js';
+import { a as Thread, c as ThreadStorageAdapter, A as AsyncThreadStorageAdapter } from '../types-waEqyE4K.js';
 import { T as ToolDefinition } from '../tools-tmksfhUo.js';
-import { B as BranchInfo } from '../MessageTree-CzaN9Eul.js';
+import { B as BranchInfo } from '../MessageTree-CSIHErPK.js';
 import { d as MCPUIIntent, n as MCPUIFrameProps, M as MCPUIResource } from '../types-B20VCJXL.js';
 import { ClassValue } from 'clsx';
 

package/dist/ui/index.js

@@ -1,5 +1,5 @@
-import { useCopilot, useThreadManager } from '../chunk-PT2TOHG5.js';
-import { createServerAdapter } from '../chunk-WZ2TOZ7M.js';
+import { useCopilot, useThreadManager } from '../chunk-FLZO2FO3.js';
+import { createServerAdapter } from '../chunk-EEH3L64W.js';
 import { DEFAULT_MCP_UI_SANDBOX, parseMCPUIMessage } from '../chunk-G4SF2PNQ.js';
 import '../chunk-EWVQWTNV.js';
 import '../chunk-VNLLW3ZI.js';
@@ -6551,7 +6551,7 @@ function useInternalThreadManager(config = {}) {
     clearCurrentThread,
     refreshThreads
   } = threadManager;
-  const { messages, setMessages, status, isLoading, getAllMessages } = useCopilot();
+  const { messages, setMessages, status, isLoading, getAllMessages, switchBranch } = useCopilot();
   const isLoadingMessagesRef = useRef(false);
   const savingToThreadRef = useRef(null);
   const lastSavedSnapshotRef = useRef("");
@@ -6599,6 +6599,9 @@ function useInternalThreadManager(config = {}) {
         lastSavedSnapshotRef.current = getMessageSnapshot(uiMessages);
         savingToThreadRef.current = threadId;
         setMessages(uiMessages);
+        if (thread.activeLeafId) {
+          switchBranch(thread.activeLeafId);
+        }
       } else {
         lastSavedSnapshotRef.current = "";
         savingToThreadRef.current = threadId;
@@ -6609,7 +6612,7 @@ function useInternalThreadManager(config = {}) {
         isLoadingMessagesRef.current = false;
       });
     },
-    [switchThread, setMessages, getMessageSnapshot, onThreadChange]
+    [switchThread, setMessages, switchBranch, getMessageSnapshot, onThreadChange]
   );
   const handleNewThread = useCallback(async () => {
     isLoadingMessagesRef.current = true;
@@ -6647,6 +6650,9 @@ function useInternalThreadManager(config = {}) {
       lastSavedSnapshotRef.current = getMessageSnapshot(uiMessages);
       savingToThreadRef.current = currentThread.id;
       setMessages(uiMessages);
+      if (currentThread.activeLeafId) {
+        switchBranch(currentThread.activeLeafId);
+      }
     } else {
       lastSavedSnapshotRef.current = "";
       savingToThreadRef.current = currentThread.id;
@@ -6655,7 +6661,7 @@ function useInternalThreadManager(config = {}) {
     requestAnimationFrame(() => {
       isLoadingMessagesRef.current = false;
     });
-  }, [enabled, adapter, currentThread, setMessages, getMessageSnapshot, onThreadChange]);
+  }, [enabled, adapter, currentThread, setMessages, switchBranch, getMessageSnapshot, onThreadChange]);
   useEffect(() => {
     if (!enabled) return;
     if (isLoadingMessagesRef.current) {
@@ -6675,9 +6681,10 @@ function useInternalThreadManager(config = {}) {
     const coreMessages = convertToCore(
       allUIMessages.length > 0 ? allUIMessages : messages
     );
+    const activeLeafId = messages[messages.length - 1]?.id;
     if (!currentThreadId && !savingToThreadRef.current) {
       savingToThreadRef.current = "creating";
-      createThread({ messages: coreMessages }).then((thread) => {
+      createThread({ messages: coreMessages, activeLeafId }).then((thread) => {
         lastSavedSnapshotRef.current = currentSnapshot;
         savingToThreadRef.current = thread.id;
         onThreadChange?.(thread.id);
@@ -6687,7 +6694,7 @@ function useInternalThreadManager(config = {}) {
     if (savingToThreadRef.current && savingToThreadRef.current !== currentThreadId) {
       return;
     }
-    updateCurrentThread({ messages: coreMessages });
+    updateCurrentThread({ messages: coreMessages, activeLeafId });
     lastSavedSnapshotRef.current = currentSnapshot;
   }, [
     enabled,