mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-29 14:46:38 +08:00
37 lines
822 B
Go
37 lines
822 B
Go
package deepseek
|
|
|
|
import (
|
|
"one-api/common/requester"
|
|
"one-api/model"
|
|
"one-api/providers/base"
|
|
"one-api/providers/openai"
|
|
)
|
|
|
|
type DeepseekProviderFactory struct{}
|
|
|
|
// 创建 DeepseekProvider
|
|
func (f DeepseekProviderFactory) Create(channel *model.Channel) base.ProviderInterface {
|
|
config := getDeepseekConfig()
|
|
return &DeepseekProvider{
|
|
OpenAIProvider: openai.OpenAIProvider{
|
|
BaseProvider: base.BaseProvider{
|
|
Config: config,
|
|
Channel: channel,
|
|
Requester: requester.NewHTTPRequester(*channel.Proxy, openai.RequestErrorHandle),
|
|
},
|
|
BalanceAction: false,
|
|
},
|
|
}
|
|
}
|
|
|
|
func getDeepseekConfig() base.ProviderConfig {
|
|
return base.ProviderConfig{
|
|
BaseURL: "https://api.deepseek.com",
|
|
ChatCompletions: "/v1/chat/completions",
|
|
}
|
|
}
|
|
|
|
type DeepseekProvider struct {
|
|
openai.OpenAIProvider
|
|
}
|