1
0
mirror of https://github.com/fumiama/deepinfra.git synced 2026-06-21 12:21:37 +08:00

fix(model): still use string in system prompt

This commit is contained in:
源文雨
2025-09-24 00:43:09 +08:00
parent 8757564fe7
commit e9e3b55ebc
5 changed files with 7 additions and 7 deletions

View File

@@ -77,11 +77,11 @@ func (ollm *OLLaMA) OutputRaw() Contents {
return Contents{NewContentText(ollm.Message.Content)}
}
func (ollm *OLLaMA) System(prompt ...Content) Protocol {
func (ollm *OLLaMA) System(prompt string) Protocol {
ollm.Messages = make([]OLLaMAMessage, 1, 8)
ollm.Messages[0] = OLLaMAMessage{
Role: "system",
Content: prompt[0].Text,
Content: prompt,
}
return ollm
}