diff --git a/src/renderer/src/providers/AiProvider.ts b/src/renderer/src/providers/AiProvider.ts index f2ea32486c..2ee7a29552 100644 --- a/src/renderer/src/providers/AiProvider.ts +++ b/src/renderer/src/providers/AiProvider.ts @@ -30,8 +30,8 @@ export default class AiProvider { return this.sdk.suggestions(messages, assistant) } - public async generate({ prompt, content }: { prompt: string; content: string }): Promise { - return this.sdk.generate({ prompt, content }) + public async generateText({ prompt, content }: { prompt: string; content: string }): Promise { + return this.sdk.generateText({ prompt, content }) } public async check(): Promise<{ valid: boolean; error: Error | null }> { diff --git a/src/renderer/src/providers/AnthropicProvider.ts b/src/renderer/src/providers/AnthropicProvider.ts index 4bc4ef6312..c8f3bb059e 100644 --- a/src/renderer/src/providers/AnthropicProvider.ts +++ b/src/renderer/src/providers/AnthropicProvider.ts @@ -118,7 +118,7 @@ export default class AnthropicProvider extends BaseProvider { return message.content[0].type === 'text' ? message.content[0].text : '' } - public async generate({ prompt, content }: { prompt: string; content: string }): Promise { + public async generateText({ prompt, content }: { prompt: string; content: string }): Promise { const model = getDefaultModel() const message = await this.sdk.messages.create({ diff --git a/src/renderer/src/providers/BaseProvider.ts b/src/renderer/src/providers/BaseProvider.ts index ec1e4db1b1..7d36bd68fb 100644 --- a/src/renderer/src/providers/BaseProvider.ts +++ b/src/renderer/src/providers/BaseProvider.ts @@ -28,7 +28,7 @@ export default abstract class BaseProvider { abstract translate(message: Message, assistant: Assistant): Promise abstract summaries(messages: Message[], assistant: Assistant): Promise abstract suggestions(messages: Message[], assistant: Assistant): Promise - abstract generate({ prompt, content }: { prompt: string; content: string }): Promise + abstract generateText({ prompt, content }: { prompt: string; content: string }): Promise abstract check(): Promise<{ valid: boolean; error: Error | null }> abstract models(): Promise } diff --git a/src/renderer/src/providers/GeminiProvider.ts b/src/renderer/src/providers/GeminiProvider.ts index 4b09d45cb7..7360674530 100644 --- a/src/renderer/src/providers/GeminiProvider.ts +++ b/src/renderer/src/providers/GeminiProvider.ts @@ -120,7 +120,7 @@ export default class GeminiProvider extends BaseProvider { return response.text() } - public async generate({ prompt, content }: { prompt: string; content: string }): Promise { + public async generateText({ prompt, content }: { prompt: string; content: string }): Promise { const model = getDefaultModel() const systemMessage = { role: 'system', content: prompt } diff --git a/src/renderer/src/providers/OpenAIProvider.ts b/src/renderer/src/providers/OpenAIProvider.ts index 1f229d48e9..755d56a100 100644 --- a/src/renderer/src/providers/OpenAIProvider.ts +++ b/src/renderer/src/providers/OpenAIProvider.ts @@ -128,7 +128,7 @@ export default class OpenAIProvider extends BaseProvider { return removeQuotes(response.choices[0].message?.content?.substring(0, 50) || '') } - public async generate({ prompt, content }: { prompt: string; content: string }): Promise { + public async generateText({ prompt, content }: { prompt: string; content: string }): Promise { const model = getDefaultModel() const response = await this.sdk.chat.completions.create({ diff --git a/src/renderer/src/services/api.ts b/src/renderer/src/services/api.ts index dacffe97b9..450b9032f0 100644 --- a/src/renderer/src/services/api.ts +++ b/src/renderer/src/services/api.ts @@ -140,7 +140,7 @@ export async function fetchGenerate({ prompt, content }: { prompt: string; conte const AI = new AiProvider(provider) try { - return await AI.generate({ prompt, content }) + return await AI.generateText({ prompt, content }) } catch (error: any) { return '' }