matt/ollama37
1
0
Fork 0
mirror of https://github.com/dogkeeper886/ollama37.git synced 2025-12-11 08:17:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

tcp socket

Browse Source
This commit is contained in:
Michael Yang
2023-07-06 10:56:08 -07:00
committed by Jeffrey Morgan
parent 3c3c09a523
commit 5079282120
2 changed files with 3 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
api/client.go
View File

@@ -11,12 +11,8 @@ import (
)
type Client struct {
Name string
Version string
URL string
HTTP http.Client
Headers http.Header
PrivateKey []byte
}
func checkError(resp *http.Response, body []byte) error {
@@ -26,8 +22,7 @@ func checkError(resp *http.Response, body []byte) error {
apiError := Error{Code: int32(resp.StatusCode)}
err := json.Unmarshal(body, &apiError)
if err != nil {
if err := json.Unmarshal(body, &apiError); err != nil {
// Use the full body as the message if we fail to decode a response.
apiError.Message = string(body)
}
@@ -57,10 +52,6 @@ func (c *Client) stream(ctx context.Context, method string, path string, reqData
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Accept", "application/json")
for k, v := range c.Headers {
req.Header[k] = v
}
res, err := c.HTTP.Do(req)
if err != nil {
return err
@@ -103,10 +94,6 @@ func (c *Client) do(ctx context.Context, method string, path string, reqData any
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Accept", "application/json")
for k, v := range c.Headers {
req.Header[k] = v
}
respObj, err := c.HTTP.Do(req)
if err != nil {
return err