55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
package ollama
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
)
|
|
|
|
type ListModelsResponse struct {
|
|
Models []struct {
|
|
Name string `json:"name"`
|
|
Model string `json:"model"`
|
|
RemoteModel string `json:"remote_model"`
|
|
RemoteHost string `json:"remote_host"`
|
|
ModifiedAt string `json:"modified_at"`
|
|
Size int `json:"size"`
|
|
Digest string `json:"digest"`
|
|
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"`
|
|
} `json:"models"`
|
|
}
|
|
|
|
func (o Ollama) ListModels() (ListModelsResponse, error) {
|
|
req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("%s/tags", o.baseUrl), nil)
|
|
if err != nil {
|
|
return ListModelsResponse{}, err
|
|
}
|
|
|
|
for key, val := range o.customHeaders {
|
|
req.Header.Set(key, val)
|
|
}
|
|
|
|
resp, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
return ListModelsResponse{}, err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return ListModelsResponse{}, errors.New("status code is not 200")
|
|
}
|
|
|
|
var respBody ListModelsResponse
|
|
if err := json.NewDecoder(resp.Body).Decode(&respBody); err != nil {
|
|
return ListModelsResponse{}, err
|
|
}
|
|
return respBody, nil
|
|
}
|