50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
|
package ai
|
|||
|
|
|
|||
|
|
import "errors"
|
|||
|
|
|
|||
|
|
// 支持的 Provider 标识。
|
|||
|
|
const (
|
|||
|
|
ProviderSpark = "spark"
|
|||
|
|
ProviderDeepSeek = "deepseek"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// New 是 Provider 工厂：按标识与 API Key 创建对应实例。
|
|||
|
|
func New(provider, apiKey string) (Provider, error) {
|
|||
|
|
switch provider {
|
|||
|
|
case ProviderSpark:
|
|||
|
|
return newSpark(apiKey)
|
|||
|
|
case ProviderDeepSeek:
|
|||
|
|
return newDeepSeek(apiKey)
|
|||
|
|
default:
|
|||
|
|
return nil, errors.New("AI_PROVIDER_INVALID")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// newSpark 创建讯飞星火 Lite（OpenAI 兼容 HTTP 端点，APIPassword 鉴权，Lite 免费）。
|
|||
|
|
func newSpark(apiPassword string) (Provider, error) {
|
|||
|
|
if apiPassword == "" {
|
|||
|
|
return nil, errors.New("AI_KEY_MISSING")
|
|||
|
|
}
|
|||
|
|
return &openAICompatible{
|
|||
|
|
name: ProviderSpark,
|
|||
|
|
baseURL: "https://spark-api-open.xf-yun.com/v1",
|
|||
|
|
model: "lite",
|
|||
|
|
apiKey: apiPassword,
|
|||
|
|
client: newHTTPClient(),
|
|||
|
|
}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// newDeepSeek 创建 DeepSeek（OpenAI 兼容）。
|
|||
|
|
func newDeepSeek(apiKey string) (Provider, error) {
|
|||
|
|
if apiKey == "" {
|
|||
|
|
return nil, errors.New("AI_KEY_MISSING")
|
|||
|
|
}
|
|||
|
|
return &openAICompatible{
|
|||
|
|
name: ProviderDeepSeek,
|
|||
|
|
baseURL: "https://api.deepseek.com",
|
|||
|
|
model: "deepseek-chat",
|
|||
|
|
apiKey: apiKey,
|
|||
|
|
client: newHTTPClient(),
|
|||
|
|
}, nil
|
|||
|
|
}
|