fix: support chat in different doc (#7693)

fix: [BS-990](https://linear.app/affine-design/issue/BS-990/避免centerpeek中发起的会话，等待ai返回时，页面失去响应) [BS-1005](https://linear.app/affine-design/issue/BS-1005/chat-block无法被copy-paste到别的文档中，duplicate全篇文档后，可以center-peek)
2024-12-03 06:03:21 +03:00 · 2024-08-02 03:53:17 +00:00 · 2024-08-02 03:53:17 +00:00 · 62fc7e2f4d
commit 62fc7e2f4d
parent f7798a00c1
2 changed files with 103 additions and 66 deletions
--- a/packages/frontend/core/src/blocksuite/presets/ai/_common/chat-actions-handle.ts
+++ b/packages/frontend/core/src/blocksuite/presets/ai/_common/chat-actions-handle.ts
@ -48,16 +48,16 @@ export type ChatAction = {
  ) => Promise<boolean>;
 };

-export async function queryHistoryMessages(doc: Doc, forkSessionId: string) {
+export async function queryHistoryMessages(
+  workspaceId: string,
+  docId: string,
+  forkSessionId: string
+) {
  // Get fork session messages
-  const histories = await AIProvider.histories?.chats(
-    doc.collection.id,
-    doc.id,
-    {
-      sessionId: forkSessionId,
-      messageOrder: ChatHistoryOrder.asc,
-    }
-  );
+  const histories = await AIProvider.histories?.chats(workspaceId, docId, {
+    sessionId: forkSessionId,
+    messageOrder: ChatHistoryOrder.asc,
+  });

  if (!histories || !histories.length) {
    return [];
@ -98,7 +98,11 @@ export async function constructRootChatBlockMessages(
 ) {
  // Convert chat messages to AI chat block messages
  const userInfo = await AIProvider.userInfo;
-  const forkMessages = await queryHistoryMessages(doc, forkSessionId);
+  const forkMessages = await queryHistoryMessages(
+    doc.collection.id,
+    doc.id,
+    forkSessionId
+  );
  return constructUserInfoWithMessages(forkMessages, userInfo);
 }

@ -163,6 +167,8 @@ function addAIChatBlock(
      messages: JSON.stringify(messages),
      index: layer.generateIndex('affine:embed-ai-chat'),
      sessionId,
+      rootWorkspaceId: doc.collection.id,
+      rootDocId: doc.id,
    },
    surfaceBlock.id
  );
--- a/packages/frontend/core/src/blocksuite/presets/ai/peek-view/chat-block-peek-view.ts
+++ b/packages/frontend/core/src/blocksuite/presets/ai/peek-view/chat-block-peek-view.ts
@ -15,7 +15,6 @@ import {
  type ChatMessage,
  ChatMessagesSchema,
 } from '@blocksuite/presets';
-import type { Doc } from '@blocksuite/store';
 import { html, LitElement, nothing } from 'lit';
 import { customElement, property, query, state } from 'lit/decorators.js';
 import { classMap } from 'lit/directives/class-map.js';
@ -57,6 +56,14 @@ export class AIChatBlockPeekView extends LitElement {
    return this.parentModel.id;
  }

+  private get parentRootDocId() {
+    return this.parentModel.rootDocId;
+  }
+
+  private get parentRootWorkspaceId() {
+    return this.parentModel.rootWorkspaceId;
+  }
+
  private readonly _deserializeHistoryChatMessages = (
    historyMessagesString: string
  ) => {
@ -75,11 +82,16 @@ export class AIChatBlockPeekView extends LitElement {
  };

  private readonly _constructBranchChatBlockMessages = async (
-    doc: Doc,
+    rootWorkspaceId: string,
+    rootDocId: string,
    forkSessionId: string
  ) => {
    const currentUserInfo = await AIProvider.userInfo;
-    const forkMessages = await queryHistoryMessages(doc, forkSessionId);
+    const forkMessages = await queryHistoryMessages(
+      rootWorkspaceId,
+      rootDocId,
+      forkSessionId
+    );
    const forkLength = forkMessages.length;
    const historyLength = this._historyMessages.length;

@ -153,8 +165,10 @@ export class AIChatBlockPeekView extends LitElement {
    }

    // Get fork session messages
+    const { parentRootWorkspaceId, parentRootDocId } = this;
    const messages = await this._constructBranchChatBlockMessages(
-      doc,
+      parentRootWorkspaceId,
+      parentRootDocId,
      this.chatContext.currentSessionId
    );
    if (!messages.length) {
@ -169,6 +183,8 @@ export class AIChatBlockPeekView extends LitElement {
        xywh: bound.serialize(),
        messages: JSON.stringify(messages),
        sessionId: this.chatContext.currentSessionId,
+        rootWorkspaceId: parentRootWorkspaceId,
+        rootDocId: parentRootDocId,
      },
      surfaceBlock.id
    );
@ -212,8 +228,10 @@ export class AIChatBlockPeekView extends LitElement {
    const chatBlock = doc.getBlock(this.chatContext.currentChatBlockId);

    // Get fork session messages
+    const { parentRootWorkspaceId, parentRootDocId } = this;
    const messages = await this._constructBranchChatBlockMessages(
-      doc,
+      parentRootWorkspaceId,
+      parentRootDocId,
      this.chatContext.currentSessionId
    );
    if (!messages.length) {
@ -346,58 +364,66 @@ export class AIChatBlockPeekView extends LitElement {
    const { host } = this;
    const actions = ChatBlockPeekViewActions;

-    return html`${repeat(
-      currentMessages,
-      message => message.createdAt + message.content,
-      (message, idx) => {
-        const { status, error } = this.chatContext;
-        const isAssistantMessage = message.role === 'assistant';
-        const isLastReply =
-          idx === currentMessages.length - 1 && isAssistantMessage;
-        const messageState =
-          isLastReply && status === 'transmitting' ? 'generating' : 'finished';
-        const shouldRenderError = isLastReply && status === 'error' && !!error;
-        const isNotReady = status === 'transmitting' || status === 'loading';
-        const shouldRenderCopyMore =
-          isAssistantMessage && !(isLastReply && isNotReady);
-        const shouldRenderActions =
-          isLastReply && !!message.content && !isNotReady;
+    return html`${repeat(currentMessages, (message, idx) => {
+      const { status, error } = this.chatContext;
+      const isAssistantMessage = message.role === 'assistant';
+      const isLastReply =
+        idx === currentMessages.length - 1 && isAssistantMessage;
+      const messageState =
+        isLastReply && (status === 'transmitting' || status === 'loading')
+          ? 'generating'
+          : 'finished';
+      const shouldRenderError = isLastReply && status === 'error' && !!error;
+      const isNotReady = status === 'transmitting' || status === 'loading';
+      const shouldRenderCopyMore =
+        isAssistantMessage && !(isLastReply && isNotReady);
+      const shouldRenderActions =
+        isLastReply && !!message.content && !isNotReady;

-        const messageClasses = classMap({
-          'assistant-message-container': isAssistantMessage,
-        });
+      const messageClasses = classMap({
+        'assistant-message-container': isAssistantMessage,
+      });

-        return html`<div class=${messageClasses}>
-          <ai-chat-message
-            .host=${host}
-            .message=${message}
-            .state=${messageState}
-          ></ai-chat-message>
-          ${shouldRenderError ? AIChatErrorRenderer(host, error) : nothing}
-          ${shouldRenderCopyMore
-            ? html` <chat-copy-more
-                .host=${host}
-                .actions=${actions}
-                .content=${message.content}
-                .isLast=${isLastReply}
-                .chatSessionId=${this.chatContext.currentSessionId ?? undefined}
-                .messageId=${message.id ?? undefined}
-                .retry=${() => this.retry()}
-              ></chat-copy-more>`
-            : nothing}
-          ${shouldRenderActions
-            ? html`<chat-action-list
-                .host=${host}
-                .actions=${actions}
-                .content=${message.content}
-                .chatSessionId=${this.chatContext.currentSessionId ?? undefined}
-                .messageId=${message.id ?? undefined}
-                .layoutDirection=${'horizontal'}
-              ></chat-action-list>`
-            : nothing}
-        </div>`;
-      }
-    )}`;
+      const { attachments, role, content } = message;
+      const userInfo = {
+        userId: message.userId,
+        userName: message.userName,
+        avatarUrl: message.avatarUrl,
+      };
+
+      return html`<div class=${messageClasses}>
+        <ai-chat-message
+          .host=${host}
+          .state=${messageState}
+          .content=${content}
+          .attachments=${attachments}
+          .role=${role}
+          .userInfo=${userInfo}
+        ></ai-chat-message>
+        ${shouldRenderError ? AIChatErrorRenderer(host, error) : nothing}
+        ${shouldRenderCopyMore
+          ? html` <chat-copy-more
+              .host=${host}
+              .actions=${actions}
+              .content=${message.content}
+              .isLast=${isLastReply}
+              .chatSessionId=${this.chatContext.currentSessionId ?? undefined}
+              .messageId=${message.id ?? undefined}
+              .retry=${() => this.retry()}
+            ></chat-copy-more>`
+          : nothing}
+        ${shouldRenderActions
+          ? html`<chat-action-list
+              .host=${host}
+              .actions=${actions}
+              .content=${message.content}
+              .chatSessionId=${this.chatContext.currentSessionId ?? undefined}
+              .messageId=${message.id ?? undefined}
+              .layoutDirection=${'horizontal'}
+            ></chat-action-list>`
+          : nothing}
+      </div>`;
+    })}`;
  };

  override connectedCallback() {
@ -405,7 +431,12 @@ export class AIChatBlockPeekView extends LitElement {
    this._historyMessages = this._deserializeHistoryChatMessages(
      this.historyMessagesString
    );
-    queryHistoryMessages(this.host.doc, this.parentSessionId)
+    const { parentRootWorkspaceId, parentRootDocId, parentSessionId } = this;
+    queryHistoryMessages(
+      parentRootWorkspaceId,
+      parentRootDocId,
+      parentSessionId
+    )
      .then(messages => {
        this._historyMessages = this._historyMessages.map((message, idx) => {
          return {