ExternalVendorCode/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-01-30 08:04:13 +00:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

setup ws upgrade

Browse Source 
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2024-10-31 19:09:28 +01:00
parent 25c1fe3583
commit 08a358b481
1 changed files with 9 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
core/http/app.go
View File

@ -88,6 +88,15 @@ func API(application *application.Application) (*fiber.App, error) {
router := fiber.New(fiberCfg) router := fiber.New(fiberCfg)
router.Use(func(c *fiber.Ctx) error {
if websocket.IsWebSocketUpgrade(c) {
// Returns true if the client requested upgrade to the WebSocket protocol
return c.Next()
}
return nil
})
router.Hooks().OnListen(func(listenData fiber.ListenData) error { router.Hooks().OnListen(func(listenData fiber.ListenData) error {
scheme := "http" scheme := "http"
if listenData.TLS { if listenData.TLS {
@ -180,26 +189,6 @@ func API(application *application.Application) (*fiber.App, error) {
Browse: true, Browse: true,
})) }))
app.Use(func(c *fiber.Ctx) error {
if websocket.IsWebSocketUpgrade(c) {
// Returns true if the client requested upgrade to the WebSocket protocol
c.Next()
}
return nil
})
// app.Use("/v1/realtime", func(c *fiber.Ctx) error {
// fmt.Println("Hit upgrade from http")
// // IsWebSocketUpgrade returns true if the client
// // requested upgrade to the WebSocket protocol.
// if websocket.IsWebSocketUpgrade(c) {
// c.Locals("allowed", true)
// return c.Next()
// }
// return fiber.ErrUpgradeRequired
// })
// Define a custom 404 handler // Define a custom 404 handler
// Note: keep this at the bottom! // Note: keep this at the bottom!
router.Use(notFoundHandler) router.Use(notFoundHandler)