🎨 添加工厂方法

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

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
}