From b2ca8249c1187702fee654840a901d8637354420 Mon Sep 17 00:00:00 2001
From: fundon <fundon@pindash.io>
Date: Mon, 15 Apr 2024 13:12:57 +0000
Subject: [PATCH] fix(core): should choose different models based on content or
 attachments (#6567)

---
 .../blocksuite/block-suite-editor/ai/provider.ts         | 9 +++++++--
 1 file changed, 7 insertions(+), 2 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
index 5aea5c9424..e81fb5c2c8 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
@@ -1,6 +1,7 @@
 import { assertExists } from '@blocksuite/global/utils';
 import { AIProvider } from '@blocksuite/presets';
 
+import type { PromptKey } from './prompt';
 import {
   createChatSession,
   listHistories,
@@ -250,8 +251,12 @@ export function setupAIProvider() {
   });
 
   AIProvider.provide('createImage', options => {
-    // const promptName = 'debug:action:fal-sd15';
-    const promptName = 'debug:action:dalle3';
+    // test to image
+    let promptName: PromptKey = 'debug:action:dalle3';
+    // image to image
+    if (options.attachments?.length) {
+      promptName = 'debug:action:fal-sd15';
+    }
     return toImage({
       ...options,
       promptName,