31 lines
667 B
Go
31 lines
667 B
Go
package baichuan
|
|
|
|
import (
|
|
"one-api/providers/base"
|
|
"one-api/providers/openai"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// 定义供应商工厂
|
|
type BaichuanProviderFactory struct{}
|
|
|
|
// 创建 BaichuanProvider
|
|
// https://platform.baichuan-ai.com/docs/api
|
|
func (f BaichuanProviderFactory) Create(c *gin.Context) base.ProviderInterface {
|
|
return &BaichuanProvider{
|
|
OpenAIProvider: openai.OpenAIProvider{
|
|
BaseProvider: base.BaseProvider{
|
|
BaseURL: "https://api.baichuan-ai.com",
|
|
ChatCompletions: "/v1/chat/completions",
|
|
Embeddings: "/v1/embeddings",
|
|
Context: c,
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
type BaichuanProvider struct {
|
|
openai.OpenAIProvider
|
|
}
|