From e806169f6067e89430bb46500183d0deacdc45ae Mon Sep 17 00:00:00 2001
From: pengx17 <pengxiao@outlook.com>
Date: Wed, 17 Apr 2024 13:27:15 +0000
Subject: [PATCH] fix(core): missing copilot param (#6602)

---
 .../block-suite-editor/ai/copilot-client.ts        | 14 ++++++++++++++
 .../blocksuite/block-suite-editor/ai/request.ts    |  2 ++
 2 files changed, 16 insertions(+)

diff --git a/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/copilot-client.ts b/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/copilot-client.ts
index 5b2691162..a6fa5ad93 100644
--- a/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/copilot-client.ts
+++ b/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/copilot-client.ts
@@ -88,10 +88,12 @@ export class CopilotClient {
     sessionId,
     messageId,
     message,
+    params,
   }: {
     sessionId: string;
     messageId?: string;
     message?: string;
+    params?: Record<string, string>;
   }) {
     if (messageId && message) {
       throw new Error('Only one of messageId or message can be provided');
@@ -105,6 +107,11 @@ export class CopilotClient {
     if (message) {
       url.searchParams.set('message', message);
     }
+    if (!messageId && params) {
+      Object.entries(params).forEach(([key, value]) => {
+        url.searchParams.set(key, value);
+      });
+    }
 
     return new Promise(resolve => {
       fetch(url.toString())
@@ -120,10 +127,12 @@ export class CopilotClient {
     sessionId,
     messageId,
     message,
+    params,
   }: {
     sessionId: string;
     messageId?: string;
     message?: string;
+    params?: Record<string, string>;
   }) {
     if (messageId && message) {
       throw new Error('Only one of messageId or message can be provided');
@@ -139,6 +148,11 @@ export class CopilotClient {
     if (message) {
       url.searchParams.set('message', message);
     }
+    if (!messageId && params) {
+      Object.entries(params).forEach(([key, value]) => {
+        url.searchParams.set(key, value);
+      });
+    }
     return new EventSource(url.toString());
   }
 
diff --git a/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/request.ts b/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/request.ts
index aaee0c7f5..28dc005ca 100644
--- a/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/request.ts
+++ b/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/request.ts
@@ -136,6 +136,7 @@ export function textToText({
           sessionId: message.sessionId,
           messageId: message.messageId,
           message: message.message,
+          params,
         });
         yield* toTextStream(eventSource, { timeout });
       },
@@ -162,6 +163,7 @@ export function textToText({
           sessionId: message.sessionId,
           messageId: message.messageId,
           message: message.message,
+          params,
         });
       }),
     ]);