fix(presets): ai message should be encoded (#6707)

message in url should be encoded
toeverything · Apr 9, 2024 · 4585926 · 4585926
1 parent db8b241
commit 4585926
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/packages/presets/src/ai/copilot-client.ts b/packages/presets/src/ai/copilot-client.ts
@@ -186,15 +186,15 @@ export class CopilotClient {
 
   async textToText(text: string, sessionId: string) {
     const res = await fetch(
-      `${this.backendUrl}/api/copilot/chat/${sessionId}?message=${text}`
+      `${this.backendUrl}/api/copilot/chat/${sessionId}?message=${encodeURIComponent(text)}`
     );
     if (!res.ok) return;
     return res.text();
   }
 
   textToTextStream(text: string, sessionId: string) {
     return new EventSource(
-      `${this.backendUrl}/api/copilot/chat/${sessionId}/stream?message=${text}`
+      `${this.backendUrl}/api/copilot/chat/${sessionId}/stream?message=${encodeURIComponent(text)}`
     );
   }
 }
diff --git a/packages/presets/src/ai/provider.ts b/packages/presets/src/ai/provider.ts
@@ -13,7 +13,8 @@ export interface AIUserInfo {
  */
 export class AIProvider {
   private readonly actions: Partial<BlockSuitePresets.AIActions> = {};
-  private userInfoFn: () => AIUserInfo | null = () => null;
+  private userInfoFn: () => AIUserInfo | Promise<AIUserInfo> | null = () =>
+    null;
   private static readonly instance = new AIProvider();
 
   static provideAction<T extends keyof BlockSuitePresets.AIActions>(