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

client/ollama: Do take a *Client

Browse Source
This commit is contained in:
Blake Mizerany 2024-03-31 10:52:56 -07:00
parent e1d457c73e
commit f6e02d4bc7
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
client/ollama/ollama.go
View File

@ -46,7 +46,7 @@ func (c *Client) Build(ctx context.Context, ref string, modelfile []byte, source
// Push requests the remote Ollama service to push a model to the server. // Push requests the remote Ollama service to push a model to the server.
func (c *Client) Push(ctx context.Context, ref string) error { func (c *Client) Push(ctx context.Context, ref string) error {
_, err := Do[empty.Message](ctx, "POST", c.BaseURL+"/v1/push", apitype.PushRequest{Name: ref}) _, err := Do[empty.Message](ctx, c, "POST", "/v1/push", apitype.PushRequest{Name: ref})
return err return err
} }
@ -92,12 +92,13 @@ func (e *Error) Error() string {
return b.String() return b.String()
} }
func Do[Res any](ctx context.Context, method, urlStr string, in any) (*Res, error) { func Do[Res any](ctx context.Context, c *Client, method, path string, in any) (*Res, error) {
var body bytes.Buffer var body bytes.Buffer
// TODO(bmizerany): pool and reuse this buffer AND the encoder // TODO(bmizerany): pool and reuse this buffer AND the encoder
if err := encodeJSON(&body, in); err != nil { if err := encodeJSON(&body, in); err != nil {
return nil, err return nil, err
} }
urlStr := c.BaseURL + path
req, err := http.NewRequestWithContext(ctx, method, urlStr, &body) req, err := http.NewRequestWithContext(ctx, method, urlStr, &body)
if err != nil { if err != nil {
return nil, err return nil, err