@liveblocks/core 3.4.0-alpha1 → 3.4.1

package/dist/index.d.cts

@@ -3824,6 +3824,7 @@ type AiGeneratingAssistantMessage = {
     role: "assistant";
     createdAt: ISODateString;
     deletedAt?: ISODateString;
+    copilotId?: CopilotId;
     status: "generating";
     contentSoFar: AiAssistantContentPart[];
 };
@@ -3834,6 +3835,7 @@ type AiAwaitingToolAssistantMessage = {
     role: "assistant";
     createdAt: ISODateString;
     deletedAt?: ISODateString;
+    copilotId?: CopilotId;
     status: "awaiting-tool";
     contentSoFar: AiAssistantContentPart[];
 };
@@ -3845,6 +3847,7 @@ type AiCompletedAssistantMessage = {
     content: AiAssistantContentPart[];
     createdAt: ISODateString;
     deletedAt?: ISODateString;
+    copilotId?: CopilotId;
     status: "completed";
 };
 type AiFailedAssistantMessage = {
@@ -3854,6 +3857,7 @@ type AiFailedAssistantMessage = {
     role: "assistant";
     createdAt: ISODateString;
     deletedAt?: ISODateString;
+    copilotId?: CopilotId;
     status: "failed";
     contentSoFar: AiAssistantContentPart[];
     errorReason: string;
@@ -4020,9 +4024,10 @@ declare function createStore_forTools(): {
 declare function createStore_forChatMessages(toolsStore: ReturnType<typeof createStore_forTools>, setToolResultFn: (chatId: string, messageId: MessageId, invocationId: string, result: ToolResultResponse, options?: SetToolResultOptions) => Promise<void>): {
     getMessageById: (messageId: MessageId) => AiChatMessage | undefined;
     getChatMessagesForBranchΣ: (chatId: string, branch?: MessageId) => DerivedSignal<UiChatMessage[]>;
+    getLastUsedCopilotId: (chatId: string) => CopilotId | undefined;
     createOptimistically: {
         (chatId: string, role: "user", parentId: MessageId | null, content: AiUserContentPart[]): MessageId;
-        (chatId: string, role: "assistant", parentId: MessageId | null): MessageId;
+        (chatId: string, role: "assistant", parentId: MessageId | null, copilotId?: CopilotId): MessageId;
     };
     upsert: (message: AiChatMessage) => void;
     upsertMany: (messages: AiChatMessage[]) => void;
@@ -4081,6 +4086,8 @@ type Ai = {
     /** @private This API will change, and is not considered stable. DO NOT RELY on it. */
     queryChats: (query: AiChatsQuery) => AiChat[];
     /** @private This API will change, and is not considered stable. DO NOT RELY on it. */
+    getLastUsedCopilotId: (chatId: string) => CopilotId | undefined;
+    /** @private This API will change, and is not considered stable. DO NOT RELY on it. */
     registerKnowledgeLayer: (uniqueLayerId: string) => LayerKey;
     /** @private This API will change, and is not considered stable. DO NOT RELY on it. */
     deregisterKnowledgeLayer: (layerKey: LayerKey) => void;

package/dist/index.d.ts

@@ -3824,6 +3824,7 @@ type AiGeneratingAssistantMessage = {
     role: "assistant";
     createdAt: ISODateString;
     deletedAt?: ISODateString;
+    copilotId?: CopilotId;
     status: "generating";
     contentSoFar: AiAssistantContentPart[];
 };
@@ -3834,6 +3835,7 @@ type AiAwaitingToolAssistantMessage = {
     role: "assistant";
     createdAt: ISODateString;
     deletedAt?: ISODateString;
+    copilotId?: CopilotId;
     status: "awaiting-tool";
     contentSoFar: AiAssistantContentPart[];
 };
@@ -3845,6 +3847,7 @@ type AiCompletedAssistantMessage = {
     content: AiAssistantContentPart[];
     createdAt: ISODateString;
     deletedAt?: ISODateString;
+    copilotId?: CopilotId;
     status: "completed";
 };
 type AiFailedAssistantMessage = {
@@ -3854,6 +3857,7 @@ type AiFailedAssistantMessage = {
     role: "assistant";
     createdAt: ISODateString;
     deletedAt?: ISODateString;
+    copilotId?: CopilotId;
     status: "failed";
     contentSoFar: AiAssistantContentPart[];
     errorReason: string;
@@ -4020,9 +4024,10 @@ declare function createStore_forTools(): {
 declare function createStore_forChatMessages(toolsStore: ReturnType<typeof createStore_forTools>, setToolResultFn: (chatId: string, messageId: MessageId, invocationId: string, result: ToolResultResponse, options?: SetToolResultOptions) => Promise<void>): {
     getMessageById: (messageId: MessageId) => AiChatMessage | undefined;
     getChatMessagesForBranchΣ: (chatId: string, branch?: MessageId) => DerivedSignal<UiChatMessage[]>;
+    getLastUsedCopilotId: (chatId: string) => CopilotId | undefined;
     createOptimistically: {
         (chatId: string, role: "user", parentId: MessageId | null, content: AiUserContentPart[]): MessageId;
-        (chatId: string, role: "assistant", parentId: MessageId | null): MessageId;
+        (chatId: string, role: "assistant", parentId: MessageId | null, copilotId?: CopilotId): MessageId;
     };
     upsert: (message: AiChatMessage) => void;
     upsertMany: (messages: AiChatMessage[]) => void;
@@ -4081,6 +4086,8 @@ type Ai = {
     /** @private This API will change, and is not considered stable. DO NOT RELY on it. */
     queryChats: (query: AiChatsQuery) => AiChat[];
     /** @private This API will change, and is not considered stable. DO NOT RELY on it. */
+    getLastUsedCopilotId: (chatId: string) => CopilotId | undefined;
+    /** @private This API will change, and is not considered stable. DO NOT RELY on it. */
     registerKnowledgeLayer: (uniqueLayerId: string) => LayerKey;
     /** @private This API will change, and is not considered stable. DO NOT RELY on it. */
     deregisterKnowledgeLayer: (layerKey: LayerKey) => void;

package/dist/index.js

@@ -6,7 +6,7 @@ var __export = (target, all) => {
 
 // src/version.ts
 var PKG_NAME = "@liveblocks/core";
-var PKG_VERSION = "3.4.0-alpha1";
+var PKG_VERSION = "3.4.1";
 var PKG_FORMAT = "esm";
 
 // src/dupe-detection.ts
@@ -4129,6 +4129,7 @@ function createStore_forChatMessages(toolsStore, setToolResultFn) {
         _optimistic: true
       });
     } else {
+      const copilotId = third;
       upsert({
         id,
         chatId,
@@ -4137,6 +4138,7 @@ function createStore_forChatMessages(toolsStore, setToolResultFn) {
         createdAt,
         status: "generating",
         contentSoFar: [],
+        copilotId,
         _optimistic: true
       });
     }
@@ -4199,8 +4201,9 @@ function createStore_forChatMessages(toolsStore, setToolResultFn) {
                   message.chatId,
                   message.id,
                   toolInvocation.invocationId,
-                  result ?? { data: {} }
-                  // TODO Pass in AiGenerationOptions here, or make the backend use the same options
+                  result ?? { data: {} },
+                  { copilotId: message.copilotId }
+                  // TODO: Should we pass the other generation options (tools, knowledge) as well?
                 );
               })().catch((err) => {
                 error2(
@@ -4348,10 +4351,18 @@ function createStore_forChatMessages(toolsStore, setToolResultFn) {
   function getChatMessagesForBranch\u03A3(chatId, branch) {
     return immutableMessagesByBranch.getOrCreate(chatId).getOrCreate(branch || null);
   }
+  function getLastUsedCopilotId(chatId) {
+    const pool = messagePoolByChatId\u03A3.getOrCreate(chatId).get();
+    const latest = pool.sorted.findRight(
+      (m) => m.role === "assistant" && !m.deletedAt
+    );
+    return latest?.copilotId;
+  }
   return {
     // Readers
     getMessageById,
     getChatMessagesForBranch\u03A3,
+    getLastUsedCopilotId,
     // Mutations
     createOptimistically,
     upsert,
@@ -4453,7 +4464,6 @@ function createAi(config) {
   function onDidConnect() {
   }
   function onDidDisconnect() {
-    warn("onDidDisconnect");
   }
   function handleServerMessage(event) {
     if (typeof event.data !== "string")
@@ -4697,6 +4707,7 @@ function createAi(config) {
       },
       getChatById: context.chatsStore.getChatById,
       queryChats: context.chatsStore.findMany,
+      getLastUsedCopilotId: context.messagesStore.getLastUsedCopilotId,
       registerKnowledgeLayer,
       deregisterKnowledgeLayer,
       updateKnowledge,