♻️ refactor: provider refactor (#41)

* ♻️ refactor: provider refactor
* 完善百度/讯飞的函数调用，现在可以在`lobe-chat`中正常调用函数了

providers/baidu/base.go

@@ -4,35 +4,65 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
-	"one-api/common"
+	"net/http"
+	"one-api/common/requester"
+	"one-api/model"
 	"one-api/providers/base"
+	"one-api/types"
 	"strings"
 	"sync"
 	"time"
-
-	"github.com/gin-gonic/gin"
 )
 
 // 定义供应商工厂
 type BaiduProviderFactory struct{}
 
-// 创建 BaiduProvider
+var baiduTokenStore sync.Map
 
-func (f BaiduProviderFactory) Create(c *gin.Context) base.ProviderInterface {
+// 创建 BaiduProvider
+type BaiduProvider struct {
+	base.BaseProvider
+}
+
+func (f BaiduProviderFactory) Create(channel *model.Channel) base.ProviderInterface {
 	return &BaiduProvider{
 		BaseProvider: base.BaseProvider{
-			BaseURL:         "https://aip.baidubce.com",
-			ChatCompletions: "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat",
-			Embeddings:      "/rpc/2.0/ai_custom/v1/wenxinworkshop/embeddings",
-			Context:         c,
+			Config:    getConfig(),
+			Channel:   channel,
+			Requester: requester.NewHTTPRequester(channel.Proxy, requestErrorHandle),
 		},
 	}
 }
 
-var baiduTokenStore sync.Map
+func getConfig() base.ProviderConfig {
+	return base.ProviderConfig{
+		BaseURL:         "https://aip.baidubce.com",
+		ChatCompletions: "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat",
+		Embeddings:      "/rpc/2.0/ai_custom/v1/wenxinworkshop/embeddings",
+	}
+}
 
-type BaiduProvider struct {
-	base.BaseProvider
+// 请求错误处理
+func requestErrorHandle(resp *http.Response) *types.OpenAIError {
+	var baiduError *BaiduError
+	err := json.NewDecoder(resp.Body).Decode(baiduError)
+	if err != nil {
+		return nil
+	}
+
+	return errorHandle(baiduError)
+}
+
+// 错误处理
+func errorHandle(baiduError *BaiduError) *types.OpenAIError {
+	if baiduError.ErrorMsg == "" {
+		return nil
+	}
+	return &types.OpenAIError{
+		Message: baiduError.ErrorMsg,
+		Type:    "baidu_error",
+		Code:    baiduError.ErrorCode,
+	}
 }
 
 // 获取完整请求 URL
@@ -92,32 +122,21 @@ func (p *BaiduProvider) getBaiduAccessTokenHelper(apiKey string) (*BaiduAccessTo
 		return nil, errors.New("invalid baidu apikey")
 	}
 
-	client := common.NewClient()
-	url := fmt.Sprintf(p.BaseURL+"/oauth/2.0/token?grant_type=client_credentials&client_id=%s&client_secret=%s", parts[0], parts[1])
+	url := fmt.Sprintf(p.Config.BaseURL+"/oauth/2.0/token?grant_type=client_credentials&client_id=%s&client_secret=%s", parts[0], parts[1])
 
 	var headers = map[string]string{
 		"Content-Type": "application/json",
 		"Accept":       "application/json",
 	}
 
-	req, err := client.NewRequest("POST", url, common.WithHeader(headers))
+	req, err := p.Requester.NewRequest("POST", url, p.Requester.WithHeader(headers))
 	if err != nil {
 		return nil, err
 	}
-
-	httpClient := common.GetHttpClient(p.Channel.Proxy)
-	resp, err := httpClient.Do(req)
-	if err != nil {
-		return nil, err
-	}
-	common.PutHttpClient(httpClient)
-
-	defer resp.Body.Close()
-
 	var accessToken BaiduAccessToken
-	err = json.NewDecoder(resp.Body).Decode(&accessToken)
-	if err != nil {
-		return nil, err
+	_, errWithCode := p.Requester.SendRequest(req, &accessToken, false)
+	if errWithCode != nil {
+		return nil, errors.New(errWithCode.OpenAIError.Message)
 	}
 	if accessToken.Error != "" {
 		return nil, errors.New(accessToken.Error + ": " + accessToken.ErrorDescription)