mirrors/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-05-11 02:16:36 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

check api status

Browse source
This commit is contained in:
Michael Yang 2023-07-11 13:05:51 -07:00
parent 2a66a1164a
commit e243329e2e
3 changed files with 38 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

28
api/client.go
View file

@ -10,6 +10,20 @@ import (
"net/url"
)
type StatusError struct {
StatusCode int
Status string
Message string
}
func (e StatusError) Error() string {
if e.Message != "" {
return fmt.Sprintf("%s: %s", e.Status, e.Message)
}
return e.Status
}
type Client struct {
base url.URL
}
@ -25,7 +39,7 @@ func NewClient(hosts ...string) *Client {
}
}
func (c *Client) stream(ctx context.Context, method, path string, data any, callback func([]byte) error) error {
func (c *Client) stream(ctx context.Context, method, path string, data any, fn func([]byte) error) error {
var buf *bytes.Buffer
if data != nil {
bts, err := json.Marshal(data)
@ -53,7 +67,7 @@ func (c *Client) stream(ctx context.Context, method, path string, data any, call
scanner := bufio.NewScanner(response.Body)
for scanner.Scan() {
var errorResponse struct {
Error string `json:"error"`
Error string `json:"error,omitempty"`
}
bts := scanner.Bytes()
@ -61,11 +75,15 @@ func (c *Client) stream(ctx context.Context, method, path string, data any, call
return fmt.Errorf("unmarshal: %w", err)
}
if len(errorResponse.Error) > 0 {
return fmt.Errorf("stream: %s", errorResponse.Error)
if response.StatusCode >= 400 {
return StatusError{
StatusCode: response.StatusCode,
Status: response.Status,
Message: errorResponse.Error,
}
}
if err := callback(bts); err != nil {
if err := fn(bts); err != nil {
return err
}
}