mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-29 14:46:38 +08:00
* ✨ feat: Add support for retrieving model list from providers * 🔖 chore: Custom channel automatically get the model
37 lines
800 B
Go
37 lines
800 B
Go
package groq
|
|
|
|
import (
|
|
"one-api/common/requester"
|
|
"one-api/model"
|
|
"one-api/providers/base"
|
|
"one-api/providers/openai"
|
|
)
|
|
|
|
// 定义供应商工厂
|
|
type GroqProviderFactory struct{}
|
|
|
|
// 创建 GroqProvider
|
|
func (f GroqProviderFactory) Create(channel *model.Channel) base.ProviderInterface {
|
|
return &GroqProvider{
|
|
OpenAIProvider: openai.OpenAIProvider{
|
|
BaseProvider: base.BaseProvider{
|
|
Config: getConfig(),
|
|
Channel: channel,
|
|
Requester: requester.NewHTTPRequester(*channel.Proxy, openai.RequestErrorHandle),
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func getConfig() base.ProviderConfig {
|
|
return base.ProviderConfig{
|
|
BaseURL: "https://api.groq.com/openai",
|
|
ChatCompletions: "/v1/chat/completions",
|
|
ModelList: "/v1/models",
|
|
}
|
|
}
|
|
|
|
type GroqProvider struct {
|
|
openai.OpenAIProvider
|
|
}
|