55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
|
package provider
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/Tencent/WeKnora/internal/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
QianfanBaseURL = "https://qianfan.baidubce.com/v2"
|
||
|
|
)
|
||
|
|
|
||
|
|
// QianfanProvider 实现百度千帆的 Provider 接口
|
||
|
|
type QianfanProvider struct{}
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
Register(&QianfanProvider{})
|
||
|
|
}
|
||
|
|
|
||
|
|
// Info 返回百度千帆 provider 的元数据
|
||
|
|
func (p *QianfanProvider) Info() ProviderInfo {
|
||
|
|
return ProviderInfo{
|
||
|
|
Name: ProviderQianfan,
|
||
|
|
DisplayName: "百度千帆 Baidu Cloud",
|
||
|
|
Description: "ernie-5.0-thinking-preview, embedding-v1, bce-reranker-base, etc.",
|
||
|
|
DefaultURLs: map[types.ModelType]string{
|
||
|
|
types.ModelTypeKnowledgeQA: QianfanBaseURL,
|
||
|
|
types.ModelTypeEmbedding: QianfanBaseURL,
|
||
|
|
types.ModelTypeRerank: QianfanBaseURL,
|
||
|
|
types.ModelTypeVLLM: QianfanBaseURL,
|
||
|
|
},
|
||
|
|
ModelTypes: []types.ModelType{
|
||
|
|
types.ModelTypeKnowledgeQA,
|
||
|
|
types.ModelTypeEmbedding,
|
||
|
|
types.ModelTypeRerank,
|
||
|
|
types.ModelTypeVLLM,
|
||
|
|
},
|
||
|
|
RequiresAuth: true,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// ValidateConfig 验证百度千帆 provider 配置
|
||
|
|
func (p *QianfanProvider) ValidateConfig(config *Config) error {
|
||
|
|
if config.BaseURL == "" {
|
||
|
|
return fmt.Errorf("base URL is required for Qianfan provider")
|
||
|
|
}
|
||
|
|
if config.APIKey == "" {
|
||
|
|
return fmt.Errorf("API key is required for Qianfan provider")
|
||
|
|
}
|
||
|
|
if config.ModelName == "" {
|
||
|
|
return fmt.Errorf("model name is required")
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|