Commit b77eb0c6 by JustSong

perf: use max_tokens to reduce token consuming

parent 4a64fede
...@@ -210,11 +210,12 @@ func testChannel(channel *model.Channel, request *ChatRequest) error { ...@@ -210,11 +210,12 @@ func testChannel(channel *model.Channel, request *ChatRequest) error {
func buildTestRequest(c *gin.Context) *ChatRequest { func buildTestRequest(c *gin.Context) *ChatRequest {
model_ := c.Query("model") model_ := c.Query("model")
testRequest := &ChatRequest{ testRequest := &ChatRequest{
Model: model_, Model: model_,
MaxTokens: 1,
} }
testMessage := Message{ testMessage := Message{
Role: "user", Role: "user",
Content: "echo hi", Content: "hi",
} }
testRequest.Messages = append(testRequest.Messages, testMessage) testRequest.Messages = append(testRequest.Messages, testMessage)
return testRequest return testRequest
......
...@@ -21,8 +21,9 @@ type Message struct { ...@@ -21,8 +21,9 @@ type Message struct {
} }
type ChatRequest struct { type ChatRequest struct {
Model string `json:"model"` Model string `json:"model"`
Messages []Message `json:"messages"` Messages []Message `json:"messages"`
MaxTokens int `json:"max_tokens"`
} }
type TextRequest struct { type TextRequest struct {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment