switch to Fiber

apiv2/localai_fiber.go

@@ -0,0 +1,27 @@
+package apiv2
+
+import (
+	model "github.com/go-skynet/LocalAI/pkg/model"
+	"github.com/gofiber/fiber/v2"
+)
+
+func RegisterNewLocalAIFiberServer(configManager *ConfigManager, loader *model.ModelLoader, app *fiber.App) *LocalAIServer {
+	engine := NewLocalAIEngine(loader)
+	localAI := LocalAIServer{
+		configManager: configManager,
+		loader:        loader,
+		engine:        &engine,
+	}
+
+	v2Group := app.Group("/v2")
+	var mw []StrictMiddlewareFunc
+
+	// Use our validation middleware to check all requests against the
+	// OpenAPI schema.
+	// v2Group.Use(middleware.OapiRequestValidator(swagger))
+
+	// We now register our petStore above as the handler for the interface
+	RegisterHandlers(v2Group, NewStrictHandler(&localAI, mw))
+
+	return &localAI
+}

apiv2/localai_nethttp.go

@@ -1,23 +1,23 @@
 package apiv2
 
-import (
-	"net/http"
+// import (
+// 	"net/http"
 
-	"github.com/go-skynet/LocalAI/pkg/model"
-)
+// 	"github.com/go-skynet/LocalAI/pkg/model"
+// )
 
-func NewLocalAINetHTTPServer(configManager *ConfigManager, loader *model.ModelLoader, address string) *LocalAIServer {
-	engine := NewLocalAIEngine(loader)
-	localAI := LocalAIServer{
-		configManager: configManager,
-		loader:        loader,
-		engine:        &engine,
-	}
+// func NewLocalAINetHTTPServer(configManager *ConfigManager, loader *model.ModelLoader, address string) *LocalAIServer {
+// 	engine := NewLocalAIEngine(loader)
+// 	localAI := LocalAIServer{
+// 		configManager: configManager,
+// 		loader:        loader,
+// 		engine:        &engine,
+// 	}
 
-	var middlewares []StrictMiddlewareFunc
+// 	var middlewares []StrictMiddlewareFunc
 
-	http.Handle("/", Handler(NewStrictHandler(&localAI, middlewares)))
+// 	http.Handle("/", Handler(NewStrictHandler(&localAI, middlewares)))
 
-	http.ListenAndServe(address, nil)
-	return &localAI
-}
+// 	http.ListenAndServe(address, nil)
+// 	return &localAI
+// }