diff --git a/controller/channel-billing.go b/controller/channel-billing.go
index de3fc5f9..b1926545 100644
--- a/controller/channel-billing.go
+++ b/controller/channel-billing.go
@@ -37,6 +37,58 @@ type OpenAIUsageResponse struct {
TotalUsage float64 `json:"total_usage"` // unit: 0.01 dollar
}
+type OpenAISBUsageResponse struct {
+ Msg string `json:"msg"`
+ Data *struct {
+ Credit string `json:"credit"`
+ } `json:"data"`
+}
+
+func GetResponseBody(method, url string, channel *model.Channel) ([]byte, error) {
+ client := &http.Client{}
+ req, err := http.NewRequest(method, url, nil)
+ if err != nil {
+ return nil, err
+ }
+ auth := fmt.Sprintf("Bearer %s", channel.Key)
+ req.Header.Add("Authorization", auth)
+ res, err := client.Do(req)
+ if err != nil {
+ return nil, err
+ }
+ body, err := io.ReadAll(res.Body)
+ if err != nil {
+ return nil, err
+ }
+ err = res.Body.Close()
+ if err != nil {
+ return nil, err
+ }
+ return body, nil
+}
+
+func updateChannelOpenAISBBalance(channel *model.Channel) (float64, error) {
+ url := fmt.Sprintf("https://api.openai-sb.com/sb-api/user/status?api_key=%s", channel.Key)
+ body, err := GetResponseBody("GET", url, channel)
+ if err != nil {
+ return 0, err
+ }
+ response := OpenAISBUsageResponse{}
+ err = json.Unmarshal(body, &response)
+ if err != nil {
+ return 0, err
+ }
+ if response.Data == nil {
+ return 0, errors.New(response.Msg)
+ }
+ balance, err := strconv.ParseFloat(response.Data.Credit, 64)
+ if err != nil {
+ return 0, err
+ }
+ channel.UpdateBalance(balance)
+ return balance, nil
+}
+
func updateChannelBalance(channel *model.Channel) (float64, error) {
baseURL := common.ChannelBaseURLs[channel.Type]
switch channel.Type {
@@ -48,27 +100,14 @@ func updateChannelBalance(channel *model.Channel) (float64, error) {
return 0, errors.New("尚未实现")
case common.ChannelTypeCustom:
baseURL = channel.BaseURL
+ case common.ChannelTypeOpenAISB:
+ return updateChannelOpenAISBBalance(channel)
default:
return 0, errors.New("尚未实现")
}
url := fmt.Sprintf("%s/v1/dashboard/billing/subscription", baseURL)
- client := &http.Client{}
- req, err := http.NewRequest("GET", url, nil)
- if err != nil {
- return 0, err
- }
- auth := fmt.Sprintf("Bearer %s", channel.Key)
- req.Header.Add("Authorization", auth)
- res, err := client.Do(req)
- if err != nil {
- return 0, err
- }
- body, err := io.ReadAll(res.Body)
- if err != nil {
- return 0, err
- }
- err = res.Body.Close()
+ body, err := GetResponseBody("GET", url, channel)
if err != nil {
return 0, err
}
@@ -84,20 +123,7 @@ func updateChannelBalance(channel *model.Channel) (float64, error) {
startDate = now.AddDate(0, 0, -100).Format("2006-01-02")
}
url = fmt.Sprintf("%s/v1/dashboard/billing/usage?start_date=%s&end_date=%s", baseURL, startDate, endDate)
- req, err = http.NewRequest("GET", url, nil)
- if err != nil {
- return 0, err
- }
- req.Header.Add("Authorization", auth)
- res, err = client.Do(req)
- if err != nil {
- return 0, err
- }
- body, err = io.ReadAll(res.Body)
- if err != nil {
- return 0, err
- }
- err = res.Body.Close()
+ body, err = GetResponseBody("GET", url, channel)
if err != nil {
return 0, err
}
diff --git a/web/src/components/ChannelsTable.js b/web/src/components/ChannelsTable.js
index be0bba16..999027fc 100644
--- a/web/src/components/ChannelsTable.js
+++ b/web/src/components/ChannelsTable.js
@@ -27,6 +27,13 @@ function renderType(type) {
return ;
}
+function renderBalance(type, balance) {
+ if (type === 5) {
+ return {balance.toFixed(2)}
+ }
+ return ${balance.toFixed(2)}
+}
+
const ChannelsTable = () => {
const [channels, setChannels] = useState([]);
const [loading, setLoading] = useState(true);
@@ -336,7 +343,7 @@ const ChannelsTable = () => {
${channel.balance.toFixed(2)}}
+ trigger={renderBalance(channel.type, channel.balance)}
basic
/>