xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-09-18 01:26:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add OpenrouterProviderSort configuration for provider sorting

Browse Source
This commit is contained in:
Laisky.Cai 2025-02-20 01:51:19 +00:00
parent 1a6812182b
commit 7ec33793b7
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
common/config/config.go
View File

@ -164,3 +164,6 @@ var UserContentRequestTimeout = env.Int("USER_CONTENT_REQUEST_TIMEOUT", 30)
var EnforceIncludeUsage = env.Bool("ENFORCE_INCLUDE_USAGE", false) var EnforceIncludeUsage = env.Bool("ENFORCE_INCLUDE_USAGE", false)
var TestPrompt = env.String("TEST_PROMPT", "Output only your specific model name with no additional text.") var TestPrompt = env.String("TEST_PROMPT", "Output only your specific model name with no additional text.")
// OpenrouterProviderSort is used to determine the order of the providers in the openrouter
var OpenrouterProviderSort = env.String("OPENROUTER_PROVIDER_SORT", "")

5
relay/adaptor/openai/adaptor.go
View File

@ -94,12 +94,13 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.G
case channeltype.OpenRouter: case channeltype.OpenRouter:
includeReasoning := true includeReasoning := true
request.IncludeReasoning = &includeReasoning request.IncludeReasoning = &includeReasoning
if request.Provider == nil || request.Provider.Sort == "" { if request.Provider == nil || request.Provider.Sort == "" &&
config.OpenrouterProviderSort != "" {
if request.Provider == nil { if request.Provider == nil {
request.Provider = &openrouter.RequestProvider{} request.Provider = &openrouter.RequestProvider{}
} }
request.Provider.Sort = "throughput" request.Provider.Sort = config.OpenrouterProviderSort
} }
default: default:
} }