ChatGPT/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-10-16 16:13:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: reorganize constant related package

Browse Source
This commit is contained in:
JustSong
2024-04-06 00:44:33 +08:00
parent 880e12c855
commit f9d914873f
30 changed files with 269 additions and 257 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
relay/helper/main.go
View File

@@ -1,6 +1,7 @@
package helper
import (
"github.com/songquanpeng/one-api/relay/apitype"
"github.com/songquanpeng/one-api/relay/channel"
"github.com/songquanpeng/one-api/relay/channel/aiproxy"
"github.com/songquanpeng/one-api/relay/channel/ali"
@@ -13,32 +14,31 @@ import (
"github.com/songquanpeng/one-api/relay/channel/tencent"
"github.com/songquanpeng/one-api/relay/channel/xunfei"
"github.com/songquanpeng/one-api/relay/channel/zhipu"
"github.com/songquanpeng/one-api/relay/constant"
)
func GetAdaptor(apiType int) channel.Adaptor {
switch apiType {
case constant.APITypeAIProxyLibrary:
case apitype.AIProxyLibrary:
return &aiproxy.Adaptor{}
case constant.APITypeAli:
case apitype.Ali:
return &ali.Adaptor{}
case constant.APITypeAnthropic:
case apitype.Anthropic:
return &anthropic.Adaptor{}
case constant.APITypeBaidu:
case apitype.Baidu:
return &baidu.Adaptor{}
case constant.APITypeGemini:
case apitype.Gemini:
return &gemini.Adaptor{}
case constant.APITypeOpenAI:
case apitype.OpenAI:
return &openai.Adaptor{}
case constant.APITypePaLM:
case apitype.PaLM:
return &palm.Adaptor{}
case constant.APITypeTencent:
case apitype.Tencent:
return &tencent.Adaptor{}
case constant.APITypeXunfei:
case apitype.Xunfei:
return &xunfei.Adaptor{}
case constant.APITypeZhipu:
case apitype.Zhipu:
return &zhipu.Adaptor{}
case constant.APITypeOllama:
case apitype.Ollama:
return &ollama.Adaptor{}
}
return nil