diff --git a/pkg/claude/claude.go b/pkg/claude/claude.go index 8e99647..5036eab 100644 --- a/pkg/claude/claude.go +++ b/pkg/claude/claude.go @@ -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\nHuman:%s", msg.Content) case "user": 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:" @@ -294,6 +294,7 @@ func TransRsp(c *gin.Context, isStream bool, reader *bufio.Reader) { dataChan = make(chan string) stopChan = make(chan bool) ) + wg.Add(2) go func() { defer wg.Done() for {