cneicy
a5f7b8e586
Publish Docker image (Multi-arch) / Build & push (amd64) (push) Failing after 1m40s
Release (Linux, macOS, Windows) / Linux Release (push) Failing after 42m16s
Publish Docker image (Multi-arch) / Build & push (arm64) (push) Canceled after 0s
Release (Linux, macOS, Windows) / macOS Release (push) Canceled after 0s
Release (Linux, macOS, Windows) / Windows Release (push) Canceled after 0s
Publish Docker image (Multi-arch) / Create multi-arch manifests (push) Canceled after 0s
Sync production source including routing, billing, Astra detection and public uptime. Update custom build instructions and license packaging.
63 lines
2.7 KiB
Go
63 lines
2.7 KiB
Go
package common
|
|
|
|
import "github.com/QuantumNous/new-api/constant"
|
|
|
|
// GetEndpointTypesByChannelType 获取渠道最优先端点类型（所有的渠道都支持 OpenAI 端点）
|
|
func GetEndpointTypesByChannelType(channelType int, modelName string) []constant.EndpointType {
|
|
var endpointTypes []constant.EndpointType
|
|
switch channelType {
|
|
case constant.ChannelTypeJina:
|
|
endpointTypes = []constant.EndpointType{constant.EndpointTypeJinaRerank}
|
|
//case constant.ChannelTypeMidjourney, constant.ChannelTypeMidjourneyPlus:
|
|
// endpointTypes = []constant.EndpointType{constant.EndpointTypeMidjourney}
|
|
//case constant.ChannelTypeSunoAPI:
|
|
// endpointTypes = []constant.EndpointType{constant.EndpointTypeSuno}
|
|
//case constant.ChannelTypeKling:
|
|
// endpointTypes = []constant.EndpointType{constant.EndpointTypeKling}
|
|
//case constant.ChannelTypeJimeng:
|
|
// endpointTypes = []constant.EndpointType{constant.EndpointTypeJimeng}
|
|
case constant.ChannelTypeAws:
|
|
fallthrough
|
|
case constant.ChannelTypeAnthropic:
|
|
endpointTypes = []constant.EndpointType{constant.EndpointTypeAnthropic, constant.EndpointTypeOpenAI}
|
|
case constant.ChannelTypeVertexAi:
|
|
fallthrough
|
|
case constant.ChannelTypeGemini:
|
|
endpointTypes = []constant.EndpointType{constant.EndpointTypeGemini, constant.EndpointTypeOpenAI}
|
|
case constant.ChannelTypeOpenRouter: // OpenRouter 只支持 OpenAI 端点
|
|
endpointTypes = []constant.EndpointType{constant.EndpointTypeOpenAI}
|
|
case constant.ChannelTypeXai:
|
|
endpointTypes = []constant.EndpointType{constant.EndpointTypeOpenAI, constant.EndpointTypeOpenAIResponse}
|
|
case constant.ChannelTypeVLLM, constant.ChannelTypeSGLang:
|
|
endpointTypes = GetAdvancedCustomPreset(channelType).SupportedEndpointTypesForModel(modelName)
|
|
case constant.ChannelTypeSora:
|
|
endpointTypes = []constant.EndpointType{constant.EndpointTypeOpenAIVideo}
|
|
case constant.ChannelTypeSub2API, constant.ChannelTypeNewAPI:
|
|
endpointTypes = []constant.EndpointType{
|
|
constant.EndpointTypeOpenAI,
|
|
constant.EndpointTypeOpenAIResponse,
|
|
constant.EndpointTypeOpenAIResponseCompact,
|
|
constant.EndpointTypeAnthropic,
|
|
constant.EndpointTypeGemini,
|
|
constant.EndpointTypeOpenAIAlphaSearch,
|
|
}
|
|
case constant.ChannelTypeCodex:
|
|
endpointTypes = []constant.EndpointType{
|
|
constant.EndpointTypeOpenAIResponse,
|
|
constant.EndpointTypeOpenAIResponseCompact,
|
|
constant.EndpointTypeOpenAIAlphaSearch,
|
|
}
|
|
default:
|
|
if IsOpenAIResponseOnlyModel(modelName) {
|
|
endpointTypes = []constant.EndpointType{constant.EndpointTypeOpenAIResponse}
|
|
} else {
|
|
endpointTypes = []constant.EndpointType{constant.EndpointTypeOpenAI}
|
|
}
|
|
}
|
|
if IsImageGenerationModel(modelName) {
|
|
// add to first
|
|
endpointTypes = append([]constant.EndpointType{constant.EndpointTypeImageGeneration}, endpointTypes...)
|
|
}
|
|
return endpointTypes
|
|
}
|