Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: handel error

Browse Source
This commit is contained in:
JustSong 2023-06-11 21:03:24 +08:00
parent 146dc840e6
commit d84415977d
2 changed files with 23 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
controller/channel-billing.go
View File

@ -37,13 +37,14 @@ type OpenAIUsageResponse struct {
TotalUsage float64 `json:"total_usage"` // unit: 0.01 dollar TotalUsage float64 `json:"total_usage"` // unit: 0.01 dollar
} }
type OpenaiSbUsageResponse struct { type OpenAISBUsageResponse struct {
Data struct { Msg string `json:"msg"`
Data *struct {
Credit string `json:"credit"` Credit string `json:"credit"`
} `json:"data"` } `json:"data"`
} }
func GetRequestBody(method, url string, channel *model.Channel) ([]byte, error) { func GetResponseBody(method, url string, channel *model.Channel) ([]byte, error) {
client := &http.Client{} client := &http.Client{}
req, err := http.NewRequest(method, url, nil) req, err := http.NewRequest(method, url, nil)
if err != nil { if err != nil {
@ -66,19 +67,21 @@ func GetRequestBody(method, url string, channel *model.Channel) ([]byte, error)
return body, nil return body, nil
} }
func getOpenaiSBCredit(channel *model.Channel) (float64, error) { func updateChannelOpenAISBBalance(channel *model.Channel) (float64, error) {
url := fmt.Sprintf("https://api.openai-sb.com/sb-api/user/status?api_key=%s", channel.Key) url := fmt.Sprintf("https://api.openai-sb.com/sb-api/user/status?api_key=%s", channel.Key)
body, err := GetRequestBody("GET", url, channel) body, err := GetResponseBody("GET", url, channel)
if err != nil { if err != nil {
return 0, err return 0, err
} }
subscription := OpenaiSbUsageResponse{} response := OpenAISBUsageResponse{}
err = json.Unmarshal(body, &subscription) err = json.Unmarshal(body, &response)
if err != nil { if err != nil {
return 0, err return 0, err
} }
if response.Data == nil {
balance, err := strconv.ParseFloat(subscription.Data.Credit, 64) return 0, errors.New(response.Msg)
}
balance, err := strconv.ParseFloat(response.Data.Credit, 64)
if err != nil { if err != nil {
return 0, err return 0, err
} }
@ -98,13 +101,13 @@ func updateChannelBalance(channel *model.Channel) (float64, error) {
case common.ChannelTypeCustom: case common.ChannelTypeCustom:
baseURL = channel.BaseURL baseURL = channel.BaseURL
case common.ChannelTypeOpenAISB: case common.ChannelTypeOpenAISB:
return getOpenaiSBCredit(channel) return updateChannelOpenAISBBalance(channel)
default: default:
return 0, errors.New("尚未实现") return 0, errors.New("尚未实现")
} }
url := fmt.Sprintf("%s/v1/dashboard/billing/subscription", baseURL) url := fmt.Sprintf("%s/v1/dashboard/billing/subscription", baseURL)
body, err := GetRequestBody("GET", url, channel) body, err := GetResponseBody("GET", url, channel)
if err != nil { if err != nil {
return 0, err return 0, err
} }
@ -120,7 +123,7 @@ func updateChannelBalance(channel *model.Channel) (float64, error) {
startDate = now.AddDate(0, 0, -100).Format("2006-01-02") 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) url = fmt.Sprintf("%s/v1/dashboard/billing/usage?start_date=%s&end_date=%s", baseURL, startDate, endDate)
body, err = GetRequestBody("GET", url, channel) body, err = GetResponseBody("GET", url, channel)
if err != nil { if err != nil {
return 0, err return 0, err
} }

9
web/src/components/ChannelsTable.js
View File

@ -27,6 +27,13 @@ function renderType(type) {
return <Label basic color={type2label[type].color}>{type2label[type].text}</Label>; return <Label basic color={type2label[type].color}>{type2label[type].text}</Label>;
} }
function renderBalance(type, balance) {
if (type === 5) {
return <span>{balance.toFixed(2)}</span>
}
return <span>${balance.toFixed(2)}</span>
}
const ChannelsTable = () => { const ChannelsTable = () => {
const [channels, setChannels] = useState([]); const [channels, setChannels] = useState([]);
const [loading, setLoading] = useState(true); const [loading, setLoading] = useState(true);
@ -336,7 +343,7 @@ const ChannelsTable = () => {
<Popup <Popup
content={channel.balance_updated_time ? renderTimestamp(channel.balance_updated_time) : '未更新'} content={channel.balance_updated_time ? renderTimestamp(channel.balance_updated_time) : '未更新'}
key={channel.id} key={channel.id}
trigger={<span>${channel.balance.toFixed(2)}</span>} trigger={renderBalance(channel.type, channel.balance)}
basic basic
/> />
</Table.Cell> </Table.Cell>