go-ollama: first release v0.1.0

push_model.go

@@ -0,0 +1,52 @@
+package ollama
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"net/http"
+)
+
+type PushModelRequest struct {
+	Model    string `json:"name"`
+	Insecure *bool  `json:"insecure,omitempty"`
+	Stream   *bool  `json:"stream,omitempty"`
+}
+
+type PushModelResponse struct {
+	Status string `json:"status"`
+}
+
+func (o Ollama) PushModel(reqBody PushModelRequest) (PushModelResponse, int, error) {
+	reqBodyBytes, err := json.Marshal(reqBody)
+	if err != nil {
+		return PushModelResponse{}, -1, err
+	}
+
+	req, err := http.NewRequest(http.MethodPost, fmt.Sprintf("%s/pull", o.baseUrl), bytes.NewReader(reqBodyBytes))
+	if err != nil {
+		return PushModelResponse{}, -1, err
+	}
+
+	for key, val := range o.customHeaders {
+		req.Header.Set(key, val)
+	}
+	req.Header.Set("Content-Type", "application/json")
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return PushModelResponse{}, -1, err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != 200 {
+		return PushModelResponse{}, resp.StatusCode, errors.New("status code is not 200")
+	}
+
+	var respBody PushModelResponse
+	if err := json.NewDecoder(resp.Body).Decode(&respBody); err != nil {
+		return PushModelResponse{}, -1, err
+	}
+	return respBody, resp.StatusCode, nil
+}