2024-04-24 22:19:02 +00:00
|
|
|
package routes
|
|
|
|
|
|
|
|
import (
|
2024-06-23 08:24:36 +00:00
|
|
|
"github.com/mudler/LocalAI/core/config"
|
|
|
|
"github.com/mudler/LocalAI/core/http/endpoints/jina"
|
2024-04-24 22:19:02 +00:00
|
|
|
|
|
|
|
"github.com/gofiber/fiber/v2"
|
2024-06-23 08:24:36 +00:00
|
|
|
"github.com/mudler/LocalAI/pkg/model"
|
2024-04-24 22:19:02 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func RegisterJINARoutes(app *fiber.App,
|
|
|
|
cl *config.BackendConfigLoader,
|
|
|
|
ml *model.ModelLoader,
|
2024-09-17 03:29:07 +00:00
|
|
|
appConfig *config.ApplicationConfig) {
|
2024-04-24 22:19:02 +00:00
|
|
|
|
|
|
|
// POST endpoint to mimic the reranking
|
|
|
|
app.Post("/v1/rerank", jina.JINARerankEndpoint(cl, ml, appConfig))
|
|
|
|
}
|