mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-02-06 20:31:49 +00:00
305ce14fe3
* feat: add Replicate adaptor and integrate into channel and API types * feat: support llm chat on replicate
70 lines
2.2 KiB
Go
70 lines
2.2 KiB
Go
package relay
|
|
|
|
import (
|
|
"github.com/songquanpeng/one-api/relay/adaptor"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/aiproxy"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/ali"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/anthropic"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/aws"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/baidu"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/cloudflare"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/cohere"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/coze"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/deepl"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/gemini"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/ollama"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/openai"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/palm"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/proxy"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/replicate"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/tencent"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/vertexai"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/xunfei"
|
|
"github.com/songquanpeng/one-api/relay/adaptor/zhipu"
|
|
"github.com/songquanpeng/one-api/relay/apitype"
|
|
)
|
|
|
|
func GetAdaptor(apiType int) adaptor.Adaptor {
|
|
switch apiType {
|
|
case apitype.AIProxyLibrary:
|
|
return &aiproxy.Adaptor{}
|
|
case apitype.Ali:
|
|
return &ali.Adaptor{}
|
|
case apitype.Anthropic:
|
|
return &anthropic.Adaptor{}
|
|
case apitype.AwsClaude:
|
|
return &aws.Adaptor{}
|
|
case apitype.Baidu:
|
|
return &baidu.Adaptor{}
|
|
case apitype.Gemini:
|
|
return &gemini.Adaptor{}
|
|
case apitype.OpenAI:
|
|
return &openai.Adaptor{}
|
|
case apitype.PaLM:
|
|
return &palm.Adaptor{}
|
|
case apitype.Tencent:
|
|
return &tencent.Adaptor{}
|
|
case apitype.Xunfei:
|
|
return &xunfei.Adaptor{}
|
|
case apitype.Zhipu:
|
|
return &zhipu.Adaptor{}
|
|
case apitype.Ollama:
|
|
return &ollama.Adaptor{}
|
|
case apitype.Coze:
|
|
return &coze.Adaptor{}
|
|
case apitype.Cohere:
|
|
return &cohere.Adaptor{}
|
|
case apitype.Cloudflare:
|
|
return &cloudflare.Adaptor{}
|
|
case apitype.DeepL:
|
|
return &deepl.Adaptor{}
|
|
case apitype.VertexAI:
|
|
return &vertexai.Adaptor{}
|
|
case apitype.Proxy:
|
|
return &proxy.Adaptor{}
|
|
case apitype.Replicate:
|
|
return &replicate.Adaptor{}
|
|
}
|
|
return nil
|
|
}
|