ds2api/internal/httpapi/ollama/handler_routes.go

package ollama

import (
	"encoding/json"
	"net/http"
	"github.com/go-chi/chi/v5"
	"ds2api/internal/config"	
	"ds2api/internal/util"	
)

var WriteJSON = util.WriteJSON

type ConfigReader interface {
	ModelAliases() map[string]string
}

type Handler struct {
	Store       ConfigReader
}

type OllamaModelRequest struct {
    Model string `json:"model"`
}

func RegisterRoutes(r chi.Router, h *Handler) {
	r.Get("/api/version", h.GetVersion)
	r.Get("/api/tags", h.ListOllamaModels)
	r.Post("/api/show", h.GetOllamaModel)
}

func (h *Handler) GetVersion(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Content-Type", "application/json")
	w.WriteHeader(http.StatusOK)
	_, _ = w.Write([]byte(`{"version":"0.23.1"}`))	
}
func (h *Handler) ListOllamaModels(w http.ResponseWriter, r *http.Request) {
	WriteJSON(w, http.StatusOK, config.OllamaModelsResponse())
}
func (h *Handler) GetOllamaModel(w http.ResponseWriter, r *http.Request) {
	var payload OllamaModelRequest
    if err := json.NewDecoder(r.Body).Decode(&payload); err != nil {
        http.Error(w, "Invalid JSON body: "+err.Error(), http.StatusBadRequest)
        return
    }
    defer r.Body.Close()
	modelID := payload.Model
	model, ok := config.OllamaModelByID(h.Store, modelID)
	if !ok {
		http.Error(w, "Model not found.", http.StatusNotFound)
		return
	}
	WriteJSON(w, http.StatusOK, model)
}