✨ 添加余额查询方法

providers/aiproxy/balance.go

@@ -0,0 +1,35 @@
+package aiproxy
+
+import (
+	"errors"
+	"fmt"
+	"one-api/common"
+	"one-api/model"
+)
+
+func (p *AIProxyProvider) Balance(channel *model.Channel) (float64, error) {
+	fullRequestURL := "https://aiproxy.io/api/report/getUserOverview"
+	headers := make(map[string]string)
+	headers["Api-Key"] = channel.Key
+
+	client := common.NewClient()
+	req, err := client.NewRequest("GET", fullRequestURL, common.WithHeader(headers))
+	if err != nil {
+		return 0, err
+	}
+
+	// 发送请求
+	var response AIProxyUserOverviewResponse
+	_, errWithCode := common.SendRequest(req, &response, false)
+	if errWithCode != nil {
+		return 0, errors.New(errWithCode.OpenAIError.Message)
+	}
+
+	if !response.Success {
+		return 0, fmt.Errorf("code: %d, message: %s", response.ErrorCode, response.Message)
+	}
+
+	channel.UpdateBalance(response.Data.TotalPoints)
+
+	return response.Data.TotalPoints, nil
+}

providers/aiproxy/base.go

@@ -0,0 +1,18 @@
+package aiproxy
+
+import (
+	"one-api/providers/openai"
+
+	"github.com/gin-gonic/gin"
+)
+
+type AIProxyProvider struct {
+	*openai.OpenAIProvider
+}
+
+// 创建 CreateAIProxyProvider
+func CreateAIProxyProvider(c *gin.Context) *AIProxyProvider {
+	return &AIProxyProvider{
+		OpenAIProvider: openai.CreateOpenAIProvider(c, "https://api.aiproxy.io"),
+	}
+}

providers/aiproxy/type.go

@@ -0,0 +1,10 @@
+package aiproxy
+
+type AIProxyUserOverviewResponse struct {
+	Success   bool   `json:"success"`
+	Message   string `json:"message"`
+	ErrorCode int    `json:"error_code"`
+	Data      struct {
+		TotalPoints float64 `json:"totalPoints"`
+	} `json:"data"`
+}