From 3ccad33194c716237b315ebf1ad18e4bf1a3f3a9 Mon Sep 17 00:00:00 2001 From: takatost Date: Thu, 27 Jun 2024 13:45:35 +0800 Subject: [PATCH] feat: add jina new pre-defined rerankers, include: jina-reranker-v2 (#5657) --- .../model_runtime/model_providers/jina/rerank/_position.yaml | 5 +++++ .../jina/rerank/jina-reranker-v1-tiny-en.yaml | 4 ++++ .../jina/rerank/jina-reranker-v1-turbo-en.yaml | 4 ++++ .../jina/rerank/jina-reranker-v2-base-multilingual.yaml | 4 ++++ 4 files changed, 17 insertions(+) create mode 100644 api/core/model_runtime/model_providers/jina/rerank/_position.yaml create mode 100644 api/core/model_runtime/model_providers/jina/rerank/jina-reranker-v1-tiny-en.yaml create mode 100644 api/core/model_runtime/model_providers/jina/rerank/jina-reranker-v1-turbo-en.yaml create mode 100644 api/core/model_runtime/model_providers/jina/rerank/jina-reranker-v2-base-multilingual.yaml diff --git a/api/core/model_runtime/model_providers/jina/rerank/_position.yaml b/api/core/model_runtime/model_providers/jina/rerank/_position.yaml new file mode 100644 index 0000000000..c9ddaad758 --- /dev/null +++ b/api/core/model_runtime/model_providers/jina/rerank/_position.yaml @@ -0,0 +1,5 @@ +- jina-reranker-v2-base-multilingual +- jina-reranker-v1-base-en +- jina-reranker-v1-turbo-en +- jina-colbert-v1-en +- jina-reranker-v1-tiny-en diff --git a/api/core/model_runtime/model_providers/jina/rerank/jina-reranker-v1-tiny-en.yaml b/api/core/model_runtime/model_providers/jina/rerank/jina-reranker-v1-tiny-en.yaml new file mode 100644 index 0000000000..b81711195b --- /dev/null +++ b/api/core/model_runtime/model_providers/jina/rerank/jina-reranker-v1-tiny-en.yaml @@ -0,0 +1,4 @@ +model: jina-reranker-v1-tiny-en +model_type: rerank +model_properties: + context_size: 8192 diff --git a/api/core/model_runtime/model_providers/jina/rerank/jina-reranker-v1-turbo-en.yaml b/api/core/model_runtime/model_providers/jina/rerank/jina-reranker-v1-turbo-en.yaml new file mode 100644 index 0000000000..d05f4bb4a2 --- /dev/null +++ b/api/core/model_runtime/model_providers/jina/rerank/jina-reranker-v1-turbo-en.yaml @@ -0,0 +1,4 @@ +model: jina-reranker-v1-turbo-en +model_type: rerank +model_properties: + context_size: 8192 diff --git a/api/core/model_runtime/model_providers/jina/rerank/jina-reranker-v2-base-multilingual.yaml b/api/core/model_runtime/model_providers/jina/rerank/jina-reranker-v2-base-multilingual.yaml new file mode 100644 index 0000000000..acf576719c --- /dev/null +++ b/api/core/model_runtime/model_providers/jina/rerank/jina-reranker-v2-base-multilingual.yaml @@ -0,0 +1,4 @@ +model: jina-reranker-v2-base-multilingual +model_type: rerank +model_properties: + context_size: 8192