mirror of
https://github.com/RooVetGit/Roo-Code.git
synced 2026-08-28 05:27:24 +00:00
feat: Add completePrompt method to MistralHandler for chat completions
This commit is contained in:
parent
8842497f20
commit
b4cf69793e
1 changed files with 21 additions and 0 deletions
|
|
@ -86,4 +86,25 @@ export class MistralHandler implements ApiHandler {
|
|||
info: mistralModels[mistralDefaultModelId],
|
||||
}
|
||||
}
|
||||
|
||||
async completePrompt(prompt: string): Promise<string> {
|
||||
try {
|
||||
const response = await this.client.chat.complete({
|
||||
model: this.options.apiModelId || mistralDefaultModelId,
|
||||
messages: [{ role: "user", content: prompt }],
|
||||
temperature: this.options.modelTemperature ?? MISTRAL_DEFAULT_TEMPERATURE,
|
||||
})
|
||||
|
||||
const content = response.choices?.[0]?.message.content
|
||||
if (Array.isArray(content)) {
|
||||
return content.map((c) => (c.type === "text" ? c.text : "")).join("")
|
||||
}
|
||||
return content || ""
|
||||
} catch (error) {
|
||||
if (error instanceof Error) {
|
||||
throw new Error(`Mistral completion error: ${error.message}`)
|
||||
}
|
||||
throw error
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue