From 27a4e697106dc5549774a0f7920a08eceb29fff0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?= <41315874+fumiama@users.noreply.github.com> Date: Wed, 10 Sep 2025 22:48:55 +0800 Subject: [PATCH] feat(model): support extra_body for openai --- model/openai.go | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) diff --git a/model/openai.go b/model/openai.go index 1f08ef0..e63c944 100644 --- a/model/openai.go +++ b/model/openai.go @@ -34,10 +34,11 @@ type OpenAI struct { // callback/request Model string `json:"model"` Messages []Message `json:"messages"` - Temperature float32 `json:"temperature"` // Temperature 0.7 - TopP float32 `json:"top_p"` // TopP 0.9 - MaxTokens int `json:"max_tokens"` // MaxTokens 4096 - ExtraBody any `json:"extra_body"` + Temperature float32 `json:"temperature,omitempty"` // Temperature 0.7 + TopP float32 `json:"top_p,omitempty"` // TopP 0.9 + MaxTokens int `json:"max_tokens,omitempty"` // MaxTokens 4096 + // extra body + EnableThinking bool `json:"enable_thinking"` } // NewOpenAI use temp 0.7, topp 0.9, maxn 4096 if you don't know the meaning. @@ -60,11 +61,6 @@ func (*OpenAI) Header(key string, h http.Header) { h.Add("Authorization", "Bearer "+key) } -func (opai *OpenAI) SetExtra(body any) *OpenAI { - opai.ExtraBody = body - return opai -} - func (opai *OpenAI) Body() *bytes.Buffer { w := bytes.NewBuffer(make([]byte, 0, 8192)) err := json.NewEncoder(w).Encode(opai)