From dd354bd3962bc37733bd9668ee19b2f9ad72e703 Mon Sep 17 00:00:00 2001
From: Yash Parmar <82636823+Yash-1511@users.noreply.github.com>
Date: Fri, 12 Apr 2024 20:06:39 +0530
Subject: [PATCH] FEAT: cohere rerank 3 model added (#3431)

---
 .../model_providers/cohere/rerank/_position.yaml              | 4 ++++
 .../model_providers/cohere/rerank/rerank-english-v3.0.yaml    | 4 ++++
 .../cohere/rerank/rerank-multilingual-v3.0.yaml               | 4 ++++
 3 files changed, 12 insertions(+)
 create mode 100644 api/core/model_runtime/model_providers/cohere/rerank/_position.yaml
 create mode 100644 api/core/model_runtime/model_providers/cohere/rerank/rerank-english-v3.0.yaml
 create mode 100644 api/core/model_runtime/model_providers/cohere/rerank/rerank-multilingual-v3.0.yaml

diff --git a/api/core/model_runtime/model_providers/cohere/rerank/_position.yaml b/api/core/model_runtime/model_providers/cohere/rerank/_position.yaml
new file mode 100644
index 0000000000..4dd58fc170
--- /dev/null
+++ 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
new file mode 100644
index 0000000000..3779f0b6c2
--- /dev/null
+++ 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
new file mode 100644
index 0000000000..4f6690ba76
--- /dev/null
+++ 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