This commit is contained in:
c菌
2023-09-16 19:28:08 +08:00
parent 799b177c4f
commit 69cd825180

View File

@@ -243,12 +243,12 @@ func TransReq(chatreq *openai.ChatCompletionRequest) (*bytes.Buffer, error) {
var prompt string
for _, msg := range chatreq.Messages {
switch msg.Role {
case "system":
prompt += fmt.Sprintf("\n\nSystem:%s", msg.Content)
case "user":
prompt += fmt.Sprintf("\n\nUser:%s", msg.Content)
prompt += fmt.Sprintf("\n\nHuman:%s", msg.Content)
case "assistant":
prompt += fmt.Sprintf("\n\nAssistant:%s", msg.Content)
case "system":
prompt += fmt.Sprintf("\n\nSystem:%s", msg.Content)
}
}
transReq.Prompt = prompt + "\n\nAssistant:"