fumiama/deepinfra
1
0
Fork 0
mirror of https://github.com/fumiama/deepinfra.git synced 2026-06-05 00:32:46 +08:00
Code Activity

feat(model): openai add reasoning effort

Browse Source
This commit is contained in:
源文雨
2026-04-25 16:54:18 +08:00
parent e98ee3ba43
commit d2615e46ea
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
model/openai.go
View File

@@ -32,23 +32,25 @@ type OpenAI struct {
Created int `json:"created,omitempty"` Created int `json:"created,omitempty"`
Choices []OpenAIChoice `json:"choices,omitempty"` Choices []OpenAIChoice `json:"choices,omitempty"`
// callback/request // callback/request
Model string `json:"model"` Model string `json:"model"`
Messages []OpenAIMessage `json:"messages"` Messages []OpenAIMessage `json:"messages"`
Temperature float32 `json:"temperature,omitempty"` // Temperature 0.7 Temperature float32 `json:"temperature,omitempty"` // Temperature 0.7
TopP float32 `json:"top_p,omitempty"` // TopP 0.9 TopP float32 `json:"top_p,omitempty"` // TopP 0.9
MaxTokens int `json:"max_tokens,omitempty"` // MaxTokens 4096 MaxTokens int `json:"max_tokens,omitempty"` // MaxTokens 4096
ReasoningEffort string `json:"reasoning_effort,omitempty"`
// extra body // extra body
EnableThinking bool `json:"enable_thinking"` // EnableThinking is always false in non-stream mode, adapt to 百炼 EnableThinking bool `json:"enable_thinking"` // EnableThinking is always false in non-stream mode, adapt to 百炼
} }
// NewOpenAI use temp 0.7, topp 0.9, maxn 4096 if you don't know the meaning. // NewOpenAI use temp 0.7, topp 0.9, maxn 4096, reasoning low if you don't know the meaning.
func NewOpenAI(model, sep string, temp, topp float32, maxn uint) *OpenAI { func NewOpenAI(model, sep string, temp, topp float32, maxn uint, reasoning string) *OpenAI {
opai := new(OpenAI) opai := new(OpenAI)
opai.sep = sep opai.sep = sep
opai.Model = model opai.Model = model
opai.Temperature = temp opai.Temperature = temp
opai.TopP = topp opai.TopP = topp
opai.MaxTokens = int(maxn) opai.MaxTokens = int(maxn)
opai.ReasoningEffort = reasoning
return opai return opai
} }