diff --git a/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/provider.ts b/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/provider.ts index 0f545550de0a8..12d0b132a6b44 100644 --- a/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/provider.ts +++ b/packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/provider.ts @@ -121,14 +121,14 @@ export function setupAIProvider() { AIProvider.provideAction('makeItReal', options => { assertExists(options.stream); - const prompt = - 'Here are the latest wireframes. Could you make a new website based on these wireframes and notes and send back just the html file?'; + const prompt = 'Make it real'; return imageToTextStream({ + prompt, docId: options.docId, workspaceId: options.workspaceId, + content: options.content, params: options.params, - attachments: options.atachments, - prompt, + attachments: options.attachments, }); }); } 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 31c8df7c416e6..7223289643f60 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 @@ -42,13 +42,16 @@ export function imageToTextStream({ const client = new CopilotClient(getBaseUrl()); return { [Symbol.asyncIterator]: async function* () { - const session = await client.createSession({ + const sessionId = await client.createSession({ workspaceId, docId, promptName, }); - const message = await client.createMessage(options); - const eventSource = client.textStream(message, session); + const messageId = await client.createMessage({ + sessionId, + ...options, + }); + const eventSource = client.textStream(messageId, sessionId); yield* toTextStream(eventSource, { timeout: TIMEOUT }); }, }; @@ -58,24 +61,29 @@ export function imageToTextStream({ export function imageToImagesStream({ docId, workspaceId, + prompt: promptName, ...options }: { docId: string; workspaceId: string; prompt: string; + content: string; params?: string; attachments?: string[]; }) { const client = new CopilotClient(getBaseUrl()); return { [Symbol.asyncIterator]: async function* () { - const session = await client.createSession({ + const sessionId = await client.createSession({ workspaceId, docId, - promptName: 'Summary', // placeholder + promptName, + }); + const messageId = await client.createMessage({ + sessionId, + ...options, }); - const message = await client.createMessage(options); - const eventSource = client.imagesStream(message, session); + const eventSource = client.imagesStream(messageId, sessionId); yield* toTextStream(eventSource, { timeout: TIMEOUT }); }, };