feitianbubu
08f88d25e5
* feat: support tencent tokenhub api key via openai-compatible protocol * fix: use tokenhub base url for tencent api key channels * test: cover tencent key-format dispatch and add TokenHub key prompt locales
38 lines
971 B
Go
38 lines
971 B
Go
package tencent
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/QuantumNous/new-api/constant"
|
|
"github.com/QuantumNous/new-api/relay/channel"
|
|
"github.com/QuantumNous/new-api/relay/channel/openai"
|
|
relaycommon "github.com/QuantumNous/new-api/relay/common"
|
|
)
|
|
|
|
const tokenHubBaseURL = "https://tokenhub.tencentmaas.com"
|
|
|
|
// DispatchAdaptor 按密钥格式分流:三段式 ak/sk 走原生 TC3,单段 TokenHub key 走 OpenAI 兼容。
|
|
type DispatchAdaptor struct {
|
|
channel.Adaptor
|
|
}
|
|
|
|
func (a *DispatchAdaptor) Init(info *relaycommon.RelayInfo) {
|
|
if strings.Contains(info.ApiKey, "|") {
|
|
a.Adaptor = &Adaptor{}
|
|
} else {
|
|
a.Adaptor = &openai.Adaptor{}
|
|
if info.ChannelBaseUrl == "" || info.ChannelBaseUrl == constant.ChannelBaseURLs[constant.ChannelTypeTencent] {
|
|
info.ChannelBaseUrl = tokenHubBaseURL
|
|
}
|
|
}
|
|
a.Adaptor.Init(info)
|
|
}
|
|
|
|
func (a *DispatchAdaptor) GetModelList() []string {
|
|
return ModelList
|
|
}
|
|
|
|
func (a *DispatchAdaptor) GetChannelName() string {
|
|
return ChannelName
|
|
}
|