mirror of
https://github.com/CherryHQ/cherry-studio.git
synced 2025-12-29 05:51:26 +08:00
refactor: renamed generate method to generateText for clarity and consistency
This commit is contained in:
parent
56af85cc3e
commit
563758f69f
@ -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 }> {
|
||||
|
||||
@ -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({
|
||||
|
||||
@ -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[]>
|
||||
}
|
||||
|
||||
@ -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 }
|
||||
|
||||
|
||||
@ -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({
|
||||
|
||||
@ -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 ''
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user