mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-11-17 05:33:42 +08:00
✨ feat: support moonshot (#63)
This commit is contained in:
36
providers/moonshot/base.go
Normal file
36
providers/moonshot/base.go
Normal file
@@ -0,0 +1,36 @@
|
||||
package moonshot
|
||||
|
||||
import (
|
||||
"one-api/common/requester"
|
||||
"one-api/model"
|
||||
"one-api/providers/base"
|
||||
"one-api/providers/openai"
|
||||
)
|
||||
|
||||
type MoonshotProviderFactory struct{}
|
||||
|
||||
// 创建 MoonshotProvider
|
||||
func (f MoonshotProviderFactory) Create(channel *model.Channel) base.ProviderInterface {
|
||||
config := getMoonshotConfig()
|
||||
return &MoonshotProvider{
|
||||
OpenAIProvider: openai.OpenAIProvider{
|
||||
BaseProvider: base.BaseProvider{
|
||||
Config: config,
|
||||
Channel: channel,
|
||||
Requester: requester.NewHTTPRequester(*channel.Proxy, openai.RequestErrorHandle),
|
||||
},
|
||||
BalanceAction: false,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func getMoonshotConfig() base.ProviderConfig {
|
||||
return base.ProviderConfig{
|
||||
BaseURL: "https://api.moonshot.cn",
|
||||
ChatCompletions: "/v1/chat/completions",
|
||||
}
|
||||
}
|
||||
|
||||
type MoonshotProvider struct {
|
||||
openai.OpenAIProvider
|
||||
}
|
||||
Reference in New Issue
Block a user