Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🎨 添加工厂方法

Browse Source
This commit is contained in:
MartialBE 2023-12-02 18:14:48 +08:00
parent 5e08cc8719
commit da87fca2a2
14 changed files with 125 additions and 84 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
providers/aiproxy/base.go
View File

@ -1,18 +1,20 @@
package aiproxy
import (
"one-api/providers/base"
"one-api/providers/openai"
"github.com/gin-gonic/gin"
)
type AIProxyProvider struct {
*openai.OpenAIProvider
}
type AIProxyProviderFactory struct{}
// 创建 CreateAIProxyProvider
func CreateAIProxyProvider(c *gin.Context) *AIProxyProvider {
func (f AIProxyProviderFactory) Create(c *gin.Context) base.ProviderInterface {
return &AIProxyProvider{
OpenAIProvider: openai.CreateOpenAIProvider(c, "https://api.aiproxy.io"),
}
}
type AIProxyProvider struct {
*openai.OpenAIProvider
}

13
providers/ali/base.go
View File

@ -8,13 +8,12 @@ import (
"github.com/gin-gonic/gin"
)
type AliProvider struct {
base.BaseProvider
}
// 定义供应商工厂
type AliProviderFactory struct{}
// 创建 AliProvider
// https://dashscope.aliyuncs.com/api/v1/services/aigc/text-generation/generation
// 创建 AliAIProvider
func CreateAliAIProvider(c *gin.Context) *AliProvider {
func (f AliProviderFactory) Create(c *gin.Context) base.ProviderInterface {
return &AliProvider{
BaseProvider: base.BaseProvider{
BaseURL: "https://dashscope.aliyuncs.com",
@ -25,6 +24,10 @@ func CreateAliAIProvider(c *gin.Context) *AliProvider {
}
}
type AliProvider struct {
base.BaseProvider
}
// 获取请求头
func (p *AliProvider) GetRequestHeaders() (headers map[string]string) {
headers = make(map[string]string)

11
providers/api2d/base.go
View File

@ -1,18 +1,21 @@
package api2d
import (
"one-api/providers/base"
"one-api/providers/openai"
"github.com/gin-gonic/gin"
)
type Api2dProvider struct {
*openai.OpenAIProvider
}
type Api2dProviderFactory struct{}
// 创建 Api2dProvider
func CreateApi2dProvider(c *gin.Context) *Api2dProvider {
func (f Api2dProviderFactory) Create(c *gin.Context) base.ProviderInterface {
return &Api2dProvider{
OpenAIProvider: openai.CreateOpenAIProvider(c, "https://oa.api2d.net"),
}
}
type Api2dProvider struct {
*openai.OpenAIProvider
}

12
providers/azure/base.go
View File

@ -7,12 +7,10 @@ import (
"github.com/gin-gonic/gin"
)
type AzureProvider struct {
openai.OpenAIProvider
}
type AzureProviderFactory struct{}
// 创建 OpenAIProvider
func CreateAzureProvider(c *gin.Context) *AzureProvider {
// 创建 AzureProvider
func (f AzureProviderFactory) Create(c *gin.Context) base.ProviderInterface {
return &AzureProvider{
OpenAIProvider: openai.OpenAIProvider{
BaseProvider: base.BaseProvider{
@ -32,3 +30,7 @@ func CreateAzureProvider(c *gin.Context) *AzureProvider {
},
}
}
type AzureProvider struct {
openai.OpenAIProvider
}

15
providers/baidu/base.go
View File

@ -13,13 +13,12 @@ import (
"github.com/gin-gonic/gin"
)
var baiduTokenStore sync.Map
// 定义供应商工厂
type BaiduProviderFactory struct{}
type BaiduProvider struct {
base.BaseProvider
}
// 创建 BaiduProvider
func CreateBaiduProvider(c *gin.Context) *BaiduProvider {
func (f BaiduProviderFactory) Create(c *gin.Context) base.ProviderInterface {
return &BaiduProvider{
BaseProvider: base.BaseProvider{
BaseURL: "https://aip.baidubce.com",
@ -30,6 +29,12 @@ func CreateBaiduProvider(c *gin.Context) *BaiduProvider {
}
}
var baiduTokenStore sync.Map
type BaiduProvider struct {
base.BaseProvider
}
// 获取完整请求 URL
func (p *BaiduProvider) GetFullRequestURL(requestURL string, modelName string) string {
var modelNameMap = map[string]string{

11
providers/claude/base.go
View File

@ -6,11 +6,10 @@ import (
"github.com/gin-gonic/gin"
)
type ClaudeProvider struct {
base.BaseProvider
}
type ClaudeProviderFactory struct{}
func CreateClaudeProvider(c *gin.Context) *ClaudeProvider {
// 创建 ClaudeProvider
func (f ClaudeProviderFactory) Create(c *gin.Context) base.ProviderInterface {
return &ClaudeProvider{
BaseProvider: base.BaseProvider{
BaseURL: "https://api.anthropic.com",
@ -20,6 +19,10 @@ func CreateClaudeProvider(c *gin.Context) *ClaudeProvider {
}
}
type ClaudeProvider struct {
base.BaseProvider
}
// 获取请求头
func (p *ClaudeProvider) GetRequestHeaders() (headers map[string]string) {
headers = make(map[string]string)

13
providers/closeai/base.go
View File

@ -1,18 +1,21 @@
package closeai
import (
"one-api/providers/base"
"one-api/providers/openai"
"github.com/gin-gonic/gin"
)
type CloseaiProxyProvider struct {
*openai.OpenAIProvider
}
type CloseaiProviderFactory struct{}
// 创建 CloseaiProxyProvider
func CreateCloseaiProxyProvider(c *gin.Context) *CloseaiProxyProvider {
// 创建 CloseaiProvider
func (f CloseaiProviderFactory) Create(c *gin.Context) base.ProviderInterface {
return &CloseaiProxyProvider{
OpenAIProvider: openai.CreateOpenAIProvider(c, "https://api.closeai-proxy.xyz"),
}
}
type CloseaiProxyProvider struct {
*openai.OpenAIProvider
}

7
providers/openai/base.go
View File

@ -16,6 +16,13 @@ import (
"github.com/gin-gonic/gin"
)
type OpenAIProviderFactory struct{}
// 创建 OpenAIProvider
func (f OpenAIProviderFactory) Create(c *gin.Context) base.ProviderInterface {
return CreateOpenAIProvider(c, "")
}
type OpenAIProvider struct {
base.BaseProvider
IsAzure bool

11
providers/openaisb/base.go
View File

@ -1,18 +1,21 @@
package openaisb
import (
"one-api/providers/base"
"one-api/providers/openai"
"github.com/gin-gonic/gin"
)
type OpenaiSBProvider struct {
*openai.OpenAIProvider
}
type OpenaiSBProviderFactory struct{}
// 创建 OpenaiSBProvider
func CreateOpenaiSBProvider(c *gin.Context) *OpenaiSBProvider {
func (f OpenaiSBProviderFactory) Create(c *gin.Context) base.ProviderInterface {
return &OpenaiSBProvider{
OpenAIProvider: openai.CreateOpenAIProvider(c, "https://api.openai-sb.com"),
}
}
type OpenaiSBProvider struct {
*openai.OpenAIProvider
}

10
providers/palm/base.go
View File

@ -8,12 +8,10 @@ import (
"github.com/gin-gonic/gin"
)
type PalmProvider struct {
base.BaseProvider
}
type PalmProviderFactory struct{}
// 创建 PalmProvider
func CreatePalmProvider(c *gin.Context) *PalmProvider {
func (f PalmProviderFactory) Create(c *gin.Context) base.ProviderInterface {
return &PalmProvider{
BaseProvider: base.BaseProvider{
BaseURL: "https://generativelanguage.googleapis.com",
@ -23,6 +21,10 @@ func CreatePalmProvider(c *gin.Context) *PalmProvider {
}
}
type PalmProvider struct {
base.BaseProvider
}
// 获取请求头
func (p *PalmProvider) GetRequestHeaders() (headers map[string]string) {
headers = make(map[string]string)

58
providers/providers.go
View File

@ -20,35 +20,36 @@ import (
"github.com/gin-gonic/gin"
)
// 定义供应商工厂接口
type ProviderFactory interface {
Create(c *gin.Context) base.ProviderInterface
}
// 创建全局的供应商工厂映射
var providerFactories = make(map[int]ProviderFactory)
// 在程序启动时，添加所有的供应商工厂
func init() {
providerFactories[common.ChannelTypeOpenAI] = openai.OpenAIProviderFactory{}
providerFactories[common.ChannelTypeAzure] = azure.AzureProviderFactory{}
providerFactories[common.ChannelTypeAli] = ali.AliProviderFactory{}
providerFactories[common.ChannelTypeTencent] = tencent.TencentProviderFactory{}
providerFactories[common.ChannelTypeBaidu] = baidu.BaiduProviderFactory{}
providerFactories[common.ChannelTypeAnthropic] = claude.ClaudeProviderFactory{}
providerFactories[common.ChannelTypePaLM] = palm.PalmProviderFactory{}
providerFactories[common.ChannelTypeZhipu] = zhipu.ZhipuProviderFactory{}
providerFactories[common.ChannelTypeXunfei] = xunfei.XunfeiProviderFactory{}
providerFactories[common.ChannelTypeAIProxy] = aiproxy.AIProxyProviderFactory{}
providerFactories[common.ChannelTypeAPI2D] = api2d.Api2dProviderFactory{}
providerFactories[common.ChannelTypeCloseAI] = closeai.CloseaiProviderFactory{}
providerFactories[common.ChannelTypeOpenAISB] = openaisb.OpenaiSBProviderFactory{}
}
// 获取供应商
func GetProvider(channelType int, c *gin.Context) base.ProviderInterface {
switch channelType {
case common.ChannelTypeOpenAI:
return openai.CreateOpenAIProvider(c, "")
case common.ChannelTypeAzure:
return azure.CreateAzureProvider(c)
case common.ChannelTypeAli:
return ali.CreateAliAIProvider(c)
case common.ChannelTypeTencent:
return tencent.CreateTencentProvider(c)
case common.ChannelTypeBaidu:
return baidu.CreateBaiduProvider(c)
case common.ChannelTypeAnthropic:
return claude.CreateClaudeProvider(c)
case common.ChannelTypePaLM:
return palm.CreatePalmProvider(c)
case common.ChannelTypeZhipu:
return zhipu.CreateZhipuProvider(c)
case common.ChannelTypeXunfei:
return xunfei.CreateXunfeiProvider(c)
case common.ChannelTypeAIProxy:
return aiproxy.CreateAIProxyProvider(c)
case common.ChannelTypeAPI2D:
return api2d.CreateApi2dProvider(c)
case common.ChannelTypeCloseAI:
return closeai.CreateCloseaiProxyProvider(c)
case common.ChannelTypeOpenAISB:
return openaisb.CreateOpenaiSBProvider(c)
default:
factory, ok := providerFactories[channelType]
if !ok {
// 处理未找到的供应商工厂
baseURL := common.ChannelBaseURLs[channelType]
if c.GetString("base_url") != "" {
baseURL = c.GetString("base_url")
@ -59,4 +60,5 @@ func GetProvider(channelType int, c *gin.Context) base.ProviderInterface {
return nil
}
return factory.Create(c)
}

10
providers/tencent/base.go
View File

@ -14,12 +14,10 @@ import (
"github.com/gin-gonic/gin"
)
type TencentProvider struct {
base.BaseProvider
}
type TencentProviderFactory struct{}
// 创建 TencentProvider
func CreateTencentProvider(c *gin.Context) *TencentProvider {
func (f TencentProviderFactory) Create(c *gin.Context) base.ProviderInterface {
return &TencentProvider{
BaseProvider: base.BaseProvider{
BaseURL: "https://hunyuan.cloud.tencent.com",
@ -29,6 +27,10 @@ func CreateTencentProvider(c *gin.Context) *TencentProvider {
}
}
type TencentProvider struct {
base.BaseProvider
}
// 获取请求头
func (p *TencentProvider) GetRequestHeaders() (headers map[string]string) {
headers = make(map[string]string)

16
providers/xunfei/base.go
View File

@ -14,15 +14,10 @@ import (
"github.com/gin-gonic/gin"
)
// https://www.xfyun.cn/doc/spark/Web.html
type XunfeiProvider struct {
base.BaseProvider
domain string
apiId string
}
type XunfeiProviderFactory struct{}
// 创建 XunfeiProvider
func CreateXunfeiProvider(c *gin.Context) *XunfeiProvider {
func (f XunfeiProviderFactory) Create(c *gin.Context) base.ProviderInterface {
return &XunfeiProvider{
BaseProvider: base.BaseProvider{
BaseURL: "wss://spark-api.xf-yun.com",
@ -32,6 +27,13 @@ func CreateXunfeiProvider(c *gin.Context) *XunfeiProvider {
}
}
// https://www.xfyun.cn/doc/spark/Web.html
type XunfeiProvider struct {
base.BaseProvider
domain string
apiId string
}
// 获取请求头
func (p *XunfeiProvider) GetRequestHeaders() (headers map[string]string) {
headers = make(map[string]string)

10
providers/zhipu/base.go
View File

@ -15,12 +15,10 @@ import (
var zhipuTokens sync.Map
var expSeconds int64 = 24 * 3600
type ZhipuProvider struct {
base.BaseProvider
}
type ZhipuProviderFactory struct{}
// 创建 ZhipuProvider
func CreateZhipuProvider(c *gin.Context) *ZhipuProvider {
func (f ZhipuProviderFactory) Create(c *gin.Context) base.ProviderInterface {
return &ZhipuProvider{
BaseProvider: base.BaseProvider{
BaseURL: "https://open.bigmodel.cn",
@ -30,6 +28,10 @@ func CreateZhipuProvider(c *gin.Context) *ZhipuProvider {
}
}
type ZhipuProvider struct {
base.BaseProvider
}
// 获取请求头
func (p *ZhipuProvider) GetRequestHeaders() (headers map[string]string) {
headers = make(map[string]string)