mlow/lmcli
1
0
Fork 0
Code Issues 3 Pull Requests Packages Projects Releases Wiki Activity

Add openai response error handling

Browse Source
This commit is contained in:
Matt Low 2024-05-05 07:32:35 +00:00
parent 1b9a8f319c
commit bdaf6204f6
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/lmcli/provider/openai/openai.go
View File

@ -166,7 +166,14 @@ func (c *OpenAIClient) sendRequest(ctx context.Context, req *http.Request) (*htt
req.Header.Set("Authorization", "Bearer "+c.APIKey) req.Header.Set("Authorization", "Bearer "+c.APIKey)
client := &http.Client{} client := &http.Client{}
return client.Do(req.WithContext(ctx)) resp, err := client.Do(req.WithContext(ctx))
if resp.StatusCode != 200 {
bytes, _ := io.ReadAll(resp.Body)
return resp, fmt.Errorf("%v", string(bytes))
}
return resp, err
} }
func (c *OpenAIClient) CreateChatCompletion( func (c *OpenAIClient) CreateChatCompletion(