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; }) { 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; }) { 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, }); }), ]);