feat: Support max_tokens parameter for Ollama channel #782

2025-02-24 17:31:39 +08:00
parent dc36fdedc2
commit a81a28b7a5
2 changed files with 2 additions and 0 deletions
--- a/relay/channel/ollama/dto.go
+++ b/relay/channel/ollama/dto.go
@@ -11,6 +11,7 @@ type OllamaRequest struct {
 	Topp             float64            `json:"top_p,omitempty"`
 	TopK             int                `json:"top_k,omitempty"`
 	Stop             any                `json:"stop,omitempty"`
+	MaxTokens        uint               `json:"max_tokens,omitempty"`
 	Tools            []dto.ToolCall     `json:"tools,omitempty"`
 	ResponseFormat   any                `json:"response_format,omitempty"`
 	FrequencyPenalty float64            `json:"frequency_penalty,omitempty"`