fumiama/deepinfra
1
0
Fork 0
mirror of https://github.com/fumiama/deepinfra.git synced 2026-06-05 00:32:46 +08:00
Code Activity

fix(model): still use string in norm prompt

Browse Source
This commit is contained in:
源文雨
2025-09-24 01:01:56 +08:00
parent e9e3b55ebc
commit 8c92983b13
1 changed files with 17 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
model/openai.go
View File

@@ -128,26 +128,30 @@ func (opai *OpenAI) System(prompt string) Protocol {
return opai
}
func (opai *OpenAI) User(prompt ...Content) Protocol {
raw, err := json.Marshal(&prompt)
func (opai *OpenAI) normal(role string, prompt ...Content) Protocol {
var (
raw json.RawMessage
err error
)
if len(prompt) == 1 && prompt[0].Type == ContentTypeText {
raw, err = json.Marshal(&prompt[0].Text)
} else {
raw, err = json.Marshal(&prompt)
}
if err != nil {
panic(err)
}
opai.Messages = append(opai.Messages, OpenAIMessage{
Role: "user",
Role: role,
Content: raw,
})
return opai
}
func (opai *OpenAI) Assistant(prompt ...Content) Protocol {
raw, err := json.Marshal(&prompt)
if err != nil {
panic(err)
}
opai.Messages = append(opai.Messages, OpenAIMessage{
Role: "assistant",
Content: raw,
})
return opai
func (opai *OpenAI) User(prompt ...Content) Protocol {
return opai.normal("user", prompt...)
}
func (opai *OpenAI) Assistant(prompt ...Content) Protocol {
return opai.normal("assistant", prompt...)
}