go-ollama: first release v0.1.0

show_model_details.go

@@ -0,0 +1,63 @@
+package ollama
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"net/http"
+)
+
+type ShowModelDetailsRequest struct {
+	Model   string `json:"model"`
+	Verbose *bool  `json:"verbose,omitempty"`
+}
+
+type ShowModelDetailsResponse struct {
+	Parameters string `json:"parameters"`
+	License    string `json:"license"`
+	ModifiedAt string `json:"modified_at"`
+	Details    struct {
+		Format            string   `json:"format"`
+		Family            string   `json:"family"`
+		Families          []string `json:"families"`
+		ParameterSize     string   `json:"parameter_size"`
+		QuantizationLevel string   `json:"quantization_level"`
+	} `json:"details"`
+	Template     string         `json:"template"`
+	Capabilities []string       `json:"capabilities"`
+	ModelInfo    map[string]any `json:"model_info"`
+}
+
+func (o Ollama) ShowModelDetails(reqBody ShowModelDetailsRequest) (ShowModelDetailsResponse, int, error) {
+	reqBodyBytes, err := json.Marshal(reqBody)
+	if err != nil {
+		return ShowModelDetailsResponse{}, -1, err
+	}
+
+	req, err := http.NewRequest(http.MethodPost, fmt.Sprintf("%s/show", o.baseUrl), bytes.NewReader(reqBodyBytes))
+	if err != nil {
+		return ShowModelDetailsResponse{}, -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 ShowModelDetailsResponse{}, -1, err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != 200 {
+		return ShowModelDetailsResponse{}, resp.StatusCode, errors.New("status code is not 200")
+	}
+
+	var respBody ShowModelDetailsResponse
+	if err := json.NewDecoder(resp.Body).Decode(&respBody); err != nil {
+		return ShowModelDetailsResponse{}, -1, err
+	}
+	return respBody, resp.StatusCode, nil
+}