go-ollama: first release v0.1.0

delete_model.go

@@ -0,0 +1,41 @@
+package ollama
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"net/http"
+)
+
+type DeleteModelRequest struct {
+	Model string `json:"model"`
+}
+
+func (o Ollama) DeleteModel(reqBody DeleteModelRequest) (int, error) {
+	reqBodyBytes, err := json.Marshal(reqBody)
+	if err != nil {
+		return -1, err
+	}
+
+	req, err := http.NewRequest(http.MethodDelete, fmt.Sprintf("%s/delete", o.baseUrl), bytes.NewReader(reqBodyBytes))
+	if err != nil {
+		return -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 -1, err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != 200 {
+		return resp.StatusCode, errors.New("status code is not 200")
+	}
+	return resp.StatusCode, nil
+}