♻️ refactor: provider refactor (#41)

* ♻️ refactor: provider refactor
* 完善百度/讯飞的函数调用,现在可以在`lobe-chat`中正常调用函数了
This commit is contained in:
Buer
2024-01-19 02:47:10 +08:00
committed by GitHub
parent 0bfe1f5779
commit ef041e28a1
96 changed files with 4339 additions and 3276 deletions

View File

@@ -3,7 +3,6 @@ package openaisb
import (
"errors"
"fmt"
"one-api/common"
"one-api/model"
"strconv"
)
@@ -13,15 +12,14 @@ func (p *OpenaiSBProvider) Balance(channel *model.Channel) (float64, error) {
fullRequestURL = fmt.Sprintf("%s?api_key=%s", fullRequestURL, channel.Key)
headers := p.GetRequestHeaders()
client := common.NewClient()
req, err := client.NewRequest("GET", fullRequestURL, common.WithHeader(headers))
req, err := p.Requester.NewRequest("GET", fullRequestURL, p.Requester.WithHeader(headers))
if err != nil {
return 0, err
}
// 发送请求
var response OpenAISBUsageResponse
_, errWithCode := common.SendRequest(req, &response, false, p.Channel.Proxy)
_, errWithCode := p.Requester.SendRequest(req, &response, false)
if err != nil {
return 0, errors.New(errWithCode.OpenAIError.Message)
}

View File

@@ -1,18 +1,17 @@
package openaisb
import (
"one-api/model"
"one-api/providers/base"
"one-api/providers/openai"
"github.com/gin-gonic/gin"
)
type OpenaiSBProviderFactory struct{}
// 创建 OpenaiSBProvider
func (f OpenaiSBProviderFactory) Create(c *gin.Context) base.ProviderInterface {
func (f OpenaiSBProviderFactory) Create(channel *model.Channel) base.ProviderInterface {
return &OpenaiSBProvider{
OpenAIProvider: openai.CreateOpenAIProvider(c, "https://api.openai-sb.com"),
OpenAIProvider: openai.CreateOpenAIProvider(channel, "https://api.openai-sb.com"),
}
}