luulam/AFFiNE
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix(core): should choose different models based on content or attachments (#6567)

Browse Source
This commit is contained in:
fundon
2024-04-15 13:12:57 +00:00
parent 4b24722f3d
commit b2ca8249c1
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/provider.ts
View File

@@ -1,6 +1,7 @@
import { assertExists } from '@blocksuite/global/utils'; import { assertExists } from '@blocksuite/global/utils';
import { AIProvider } from '@blocksuite/presets'; import { AIProvider } from '@blocksuite/presets';
import type { PromptKey } from './prompt';
import { import {
createChatSession, createChatSession,
listHistories, listHistories,
@@ -250,8 +251,12 @@ export function setupAIProvider() {
}); });
AIProvider.provide('createImage', options => { AIProvider.provide('createImage', options => {
// const promptName = 'debug:action:fal-sd15'; // test to image
const promptName = 'debug:action:dalle3'; let promptName: PromptKey = 'debug:action:dalle3';
// image to image
if (options.attachments?.length) {
promptName = 'debug:action:fal-sd15';
}
return toImage({ return toImage({
...options, ...options,
promptName, promptName,