jeans/ollama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

client/ollama: report invalid server error response with raw bytes

Browse Source
This commit is contained in:
Blake Mizerany 2024-03-31 09:34:58 -07:00
parent cd5df121a5
commit e1d457c73e
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
client/ollama/ollama.go
View File

@ -5,6 +5,7 @@ import (
"cmp" "cmp"
"context" "context"
"encoding/json" "encoding/json"
"fmt"
"io" "io"
"io/fs" "io/fs"
"iter" "iter"
@ -109,8 +110,11 @@ func Do[Res any](ctx context.Context, method, urlStr string, in any) (*Res, erro
defer res.Body.Close() defer res.Body.Close()
if res.StatusCode/100 != 2 { if res.StatusCode/100 != 2 {
e, err := decodeJSON[Error](res.Body) var buf bytes.Buffer
body := io.TeeReader(res.Body, &buf)
e, err := decodeJSON[Error](body)
if err != nil { if err != nil {
err := fmt.Errorf("ollama: invalid error response from server (status %d): %q", res.StatusCode, buf.String())
return nil, err return nil, err
} }
return nil, e return nil, e