fix: missing sessionId

toeverything · Apr 11, 2024 · a349670 · a349670
1 parent b274db3
commit a349670
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 11 deletions.
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
@@ -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
@@ -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 });
     },
   };