1m timeout for non stream request

This commit is contained in:
2023-11-01 15:41:49 +08:00
parent 5a78c61e5f
commit d9a42842b2
2 changed files with 44 additions and 7 deletions

32
request_body.go Normal file
View File

@@ -0,0 +1,32 @@
package main
import (
"encoding/json"
)
type Message struct {
Content string `json:"content"`
Role string `json:"role"`
}
type RequestBody struct {
Model string `json:"model"`
Messages []Message `json:"messages"`
Stream bool `json:"stream"`
Temperature float64 `json:"temperature"`
TopP int64 `json:"top_p"`
PresencePenalty float64 `json:"presence_penalty"`
FrequencyPenalty float64 `json:"frequency_penalty"`
}
func ParseRequestBody(data []byte) (RequestBody, error) {
ret := RequestBody{}
var requestBody RequestBody
err := json.Unmarshal(data, &requestBody)
if err != nil {
return ret, err
}
return requestBody, nil
}