mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-11-17 05:33:42 +08:00
✨ feat: support deepseek (#60)
This commit is contained in:
36
providers/deepseek/base.go
Normal file
36
providers/deepseek/base.go
Normal file
@@ -0,0 +1,36 @@
|
||||
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
|
||||
}
|
||||
Reference in New Issue
Block a user