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

fix: billing status code check

Browse Source
This commit is contained in:
ckt1031 2023-07-10 23:15:30 +08:00
parent 4b2cb573b6
commit 67b8e82457
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
controller/channel-billing.go
View File

@ -96,6 +96,9 @@ func GetResponseBody(method, url string, channel *model.Channel, headers http.He
if err != nil { if err != nil {
return nil, err return nil, err
} }
if res.StatusCode != http.StatusOK {
return nil, fmt.Errorf("status code: %d", res.StatusCode)
}
body, err := io.ReadAll(res.Body) body, err := io.ReadAll(res.Body)
if err != nil { if err != nil {
return nil, err return nil, err

2
controller/channel-test.go
View File

@ -55,7 +55,7 @@ func testChannel(channel *model.Channel, request ChatRequest) error {
return err return err
} }
if resp.StatusCode != 200 { if resp.StatusCode != http.StatusOK {
return errors.New("invalid status code: " + strconv.Itoa(resp.StatusCode)) return errors.New("invalid status code: " + strconv.Itoa(resp.StatusCode))
} }