refactor: renamed generate method to generateText for clarity and consistency

This commit is contained in:
kangfenmao 2024-09-13 10:03:30 +08:00
parent 75045a3e6f
commit 7330391813
6 changed files with 7 additions and 7 deletions

View File

@ -30,8 +30,8 @@ export default class AiProvider {
return this.sdk.suggestions(messages, assistant)
}
public async generate({ prompt, content }: { prompt: string; content: string }): Promise<string> {
return this.sdk.generate({ prompt, content })
public async generateText({ prompt, content }: { prompt: string; content: string }): Promise<string> {
return this.sdk.generateText({ prompt, content })
}
public async check(): Promise<{ valid: boolean; error: Error | null }> {

View File

@ -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<string> {
public async generateText({ prompt, content }: { prompt: string; content: string }): Promise<string> {
const model = getDefaultModel()
const message = await this.sdk.messages.create({

View File

@ -28,7 +28,7 @@ export default abstract class BaseProvider {
abstract translate(message: Message, assistant: Assistant): Promise<string>
abstract summaries(messages: Message[], assistant: Assistant): Promise<string>
abstract suggestions(messages: Message[], assistant: Assistant): Promise<Suggestion[]>
abstract generate({ prompt, content }: { prompt: string; content: string }): Promise<string>
abstract generateText({ prompt, content }: { prompt: string; content: string }): Promise<string>
abstract check(): Promise<{ valid: boolean; error: Error | null }>
abstract models(): Promise<OpenAI.Models.Model[]>
}

View File

@ -120,7 +120,7 @@ export default class GeminiProvider extends BaseProvider {
return response.text()
}
public async generate({ prompt, content }: { prompt: string; content: string }): Promise<string> {
public async generateText({ prompt, content }: { prompt: string; content: string }): Promise<string> {
const model = getDefaultModel()
const systemMessage = { role: 'system', content: prompt }

View File

@ -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<string> {
public async generateText({ prompt, content }: { prompt: string; content: string }): Promise<string> {
const model = getDefaultModel()
const response = await this.sdk.chat.completions.create({

View File

@ -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 ''
}