mirror of
https://github.com/lordmathis/llamactl.git
synced 2025-11-06 00:54:23 +00:00
32 lines
711 B
Go
32 lines
711 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(handler *Handler) *chi.Mux {
|
|
r := chi.NewRouter()
|
|
r.Use(middleware.Logger)
|
|
|
|
r.Get("/swagger/*", httpSwagger.Handler(
|
|
httpSwagger.URL("/swagger/doc.json"),
|
|
))
|
|
|
|
// Define routes
|
|
r.Route("/api/v1", func(r chi.Router) {
|
|
r.Get("/server/help", handler.HelpHandler())
|
|
r.Get("/server/version", handler.VersionHandler())
|
|
r.Get("/server/devices", handler.ListDevicesHandler())
|
|
|
|
// Launch and stop handlers
|
|
// r.Post("/server/launch/{model}", launchHandler)
|
|
// r.Post("/server/stop/{model}", stopHandler)
|
|
})
|
|
|
|
return r
|
|
}
|