Setup routing

server/pkg/handlers.go

@@ -61,3 +61,42 @@ func ListDevicesHandler(w http.ResponseWriter, r *http.Request) {
 	w.Header().Set("Content-Type", "text/plain")
 	w.Write(output)
 }
+
+// func launchHandler(w http.ResponseWriter, r *http.Request) {
+// 	model := chi.URLParam(r, "model")
+// 	if model == "" {
+// 		http.Error(w, "Model parameter is required", http.StatusBadRequest)
+// 		return
+// 	}
+
+// 	cmd := execLLama(model)
+// 	if err := cmd.Start(); err != nil {
+// 		http.Error(w, "Failed to start llama server: "+err.Error(), http.StatusInternalServerError)
+// 		return
+// 	}
+
+// 	instances[model] = cmd
+// 	w.Write([]byte("Llama server started for model: " + model))
+// }
+
+// func stopHandler(w http.ResponseWriter, r *http.Request) {
+// 	model := chi.URLParam(r, "model")
+// 	if model == "" {
+// 		http.Error(w, "Model parameter is required", http.StatusBadRequest)
+// 		return
+// 	}
+
+// 	cmd, exists := instances[model]
+// 	if !exists {
+// 		http.Error(w, "No running instance for model: "+model, http.StatusNotFound)
+// 		return
+// 	}
+
+// 	if err := cmd.Process.Signal(os.Interrupt); err != nil {
+// 		http.Error(w, "Failed to stop llama server: "+err.Error(), http.StatusInternalServerError)
+// 		return
+// 	}
+
+// 	delete(instances, model)
+// 	w.Write([]byte("Llama server stopped for model: " + model))
+// }

server/pkg/routes.go

@@ -0,0 +1,24 @@
+package llamactl
+
+import (
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
+)
+
+func SetupRouter() *chi.Mux {
+	r := chi.NewRouter()
+	r.Use(middleware.Logger)
+
+	// Define routes
+	r.Route("/api/v1", func(r chi.Router) {
+		r.Get("/server/help", HelpHandler)
+		r.Get("/server/version", VersionHandler)
+		r.Get("/server/devices", ListDevicesHandler)
+
+		// Launch and stop handlers
+		// r.Post("/server/launch/{model}", launchHandler)
+		// r.Post("/server/stop/{model}", stopHandler)
+	})
+
+	return r
+}