mirror of
https://github.com/lordmathis/llamactl.git
synced 2025-11-06 09:04:27 +00:00
32 lines
703 B
Go
32 lines
703 B
Go
package llamactl
|
|
|
|
import (
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
httpSwagger "github.com/swaggo/http-swagger"
|
|
|
|
_ "llamactl/docs"
|
|
)
|
|
|
|
func SetupRouter() *chi.Mux {
|
|
r := chi.NewRouter()
|
|
r.Use(middleware.Logger)
|
|
|
|
r.Get("/swagger/*", httpSwagger.Handler(
|
|
httpSwagger.URL("/swagger/doc.json"), // The URL pointing to API definition
|
|
))
|
|
|
|
// 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
|
|
}
|