feat: add parameter top_k

2025-12-26 08:35:58 +08:00 · 2024-03-12 18:18:47 +08:00
parent 9e5c636490
commit c012306400
2 changed files with 2 additions and 0 deletions
--- a/dto/text_request.go
+++ b/dto/text_request.go
@@ -14,6 +14,7 @@ type GeneralOpenAIRequest struct {
 	MaxTokens        uint            `json:"max_tokens,omitempty"`
 	Temperature      float64         `json:"temperature,omitempty"`
 	TopP             float64         `json:"top_p,omitempty"`
+	TopK             int             `json:"top_k,omitempty"`
 	Stop             any             `json:"stop,omitempty"`
 	N                int             `json:"n,omitempty"`
 	Input            any             `json:"input,omitempty"`