FEAT: cohere rerank 3 model added (langgenius#3431)

gcc10055 · Apr 15, 2024 · 082ee89 · 082ee89
1 parent bdbc6d2
commit 082ee89
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/api/core/model_runtime/model_providers/cohere/rerank/_position.yaml b/api/core/model_runtime/model_providers/cohere/rerank/_position.yaml
@@ -0,0 +1,4 @@
+- rerank-english-v2.0
+- rerank-english-v3.0
+- rerank-multilingual-v2.0
+- rerank-multilingual-v3.0
diff --git a/api/core/model_runtime/model_providers/cohere/rerank/rerank-english-v3.0.yaml b/api/core/model_runtime/model_providers/cohere/rerank/rerank-english-v3.0.yaml
@@ -0,0 +1,4 @@
+model: rerank-english-v3.0
+model_type: rerank
+model_properties:
+  context_size: 5120
diff --git a/api/core/model_runtime/model_providers/cohere/rerank/rerank-multilingual-v3.0.yaml b/api/core/model_runtime/model_providers/cohere/rerank/rerank-multilingual-v3.0.yaml
@@ -0,0 +1,4 @@
+model: rerank-multilingual-v3.0
+model_type: rerank
+model_properties:
+  context_size: 5120