go-ollama: changed development to dev container and added streaming capabilities to methods v0.1.1

list_models.go

@@ -26,10 +26,10 @@ type ListModelsResponse struct {
 	} `json:"models"`
 }
 
-func (o Ollama) ListModels() (ListModelsResponse, int, error) {
+func (o Ollama) ListModels() (ListModelsResponse, error) {
 	req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("%s/tags", o.baseUrl), nil)
 	if err != nil {
-		return ListModelsResponse{}, -1, err
+		return ListModelsResponse{}, err
 	}
 
 	for key, val := range o.customHeaders {
@@ -38,17 +38,17 @@ func (o Ollama) ListModels() (ListModelsResponse, int, error) {
 
 	resp, err := http.DefaultClient.Do(req)
 	if err != nil {
-		return ListModelsResponse{}, -1, err
+		return ListModelsResponse{}, err
 	}
 	defer resp.Body.Close()
 
-	if resp.StatusCode != 200 {
-		return ListModelsResponse{}, resp.StatusCode, errors.New("status code is not 200")
+	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{}, -1, err
+		return ListModelsResponse{}, err
 	}
-	return respBody, resp.StatusCode, nil
+	return respBody, nil
 }