mirror of
https://github.com/fumiama/deepinfra.git
synced 2026-06-19 11:01:54 +08:00
feat(model): openai add reasoning effort
This commit is contained in:
@@ -37,18 +37,20 @@ type OpenAI struct {
|
|||||||
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
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user