mirror of
https://github.com/mudler/LocalAI.git
synced 2024-06-07 19:40:48 +00:00
20 lines
482 B
Go
20 lines
482 B
Go
|
package routes
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/go-skynet/LocalAI/core/config"
|
||
|
|
"github.com/go-skynet/LocalAI/core/http/endpoints/jina"
|
||
|
|
|
||
|
|
"github.com/go-skynet/LocalAI/pkg/model"
|
||
|
|
"github.com/gofiber/fiber/v2"
|
||
|
|
)
|
||
|
|
|
||
|
|
func RegisterJINARoutes(app *fiber.App,
|
||
|
|
cl *config.BackendConfigLoader,
|
||
|
|
ml *model.ModelLoader,
|
||
|
|
appConfig *config.ApplicationConfig,
|
||
|
|
auth func(*fiber.Ctx) error) {
|
||
|
|
|
||
|
|
// POST endpoint to mimic the reranking
|
||
|
|
app.Post("/v1/rerank", jina.JINARerankEndpoint(cl, ml, appConfig))
|
||
|
|
}
|