53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
package ali
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"one-api/providers/base"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// 定义供应商工厂
|
|
type AliProviderFactory struct{}
|
|
|
|
// 创建 AliProvider
|
|
// https://dashscope.aliyuncs.com/api/v1/services/aigc/text-generation/generation
|
|
func (f AliProviderFactory) Create(c *gin.Context) base.ProviderInterface {
|
|
return &AliProvider{
|
|
BaseProvider: base.BaseProvider{
|
|
BaseURL: "https://dashscope.aliyuncs.com",
|
|
ChatCompletions: "/api/v1/services/aigc/text-generation/generation",
|
|
Embeddings: "/api/v1/services/embeddings/text-embedding/text-embedding",
|
|
Context: c,
|
|
},
|
|
}
|
|
}
|
|
|
|
type AliProvider struct {
|
|
base.BaseProvider
|
|
}
|
|
|
|
func (p *AliProvider) GetFullRequestURL(requestURL string, modelName string) string {
|
|
baseURL := strings.TrimSuffix(p.GetBaseURL(), "/")
|
|
|
|
if modelName == "qwen-vl-plus" {
|
|
requestURL = "/api/v1/services/aigc/multimodal-generation/generation"
|
|
}
|
|
|
|
return fmt.Sprintf("%s%s", baseURL, requestURL)
|
|
}
|
|
|
|
// 获取请求头
|
|
func (p *AliProvider) GetRequestHeaders() (headers map[string]string) {
|
|
headers = make(map[string]string)
|
|
p.CommonRequestHeaders(headers)
|
|
headers["Authorization"] = fmt.Sprintf("Bearer %s", p.Channel.Key)
|
|
if p.Channel.Other != "" {
|
|
headers["X-DashScope-Plugin"] = p.Channel.Other
|
|
}
|
|
|
|
return headers
|
|
}
|