From 1a2eacc5a6934557730fec8cc6bd90b9c5fc2fe3 Mon Sep 17 00:00:00 2001
From: Ricky <5317425+rickythink@users.noreply.github.com>
Date: Tue, 16 Jan 2024 12:25:42 +0800
Subject: [PATCH] Add jina-embeddings-v2-base-zh model configuration (#2049)

---
 .../jina/text_embedding/jina-embeddings-v2-base-zh.yaml  | 9 +++++++++
 .../jina/text_embedding/text_embedding.py                | 2 +-
 2 files changed, 10 insertions(+), 1 deletion(-)
 create mode 100644 api/core/model_runtime/model_providers/jina/text_embedding/jina-embeddings-v2-base-zh.yaml

diff --git a/api/core/model_runtime/model_providers/jina/text_embedding/jina-embeddings-v2-base-zh.yaml b/api/core/model_runtime/model_providers/jina/text_embedding/jina-embeddings-v2-base-zh.yaml
new file mode 100644
index 0000000000..2a66b4729b
--- /dev/null
+++ b/api/core/model_runtime/model_providers/jina/text_embedding/jina-embeddings-v2-base-zh.yaml
@@ -0,0 +1,9 @@
+model: jina-embeddings-v2-base-zh
+model_type: text-embedding
+model_properties:
+  context_size: 8192
+  max_chunks: 2048
+pricing:
+  input: '0.001'
+  unit: '0.001'
+  currency: USD
diff --git a/api/core/model_runtime/model_providers/jina/text_embedding/text_embedding.py b/api/core/model_runtime/model_providers/jina/text_embedding/text_embedding.py
index 7cd1c3e593..57aa3d4a13 100644
--- a/api/core/model_runtime/model_providers/jina/text_embedding/text_embedding.py
+++ b/api/core/model_runtime/model_providers/jina/text_embedding/text_embedding.py
@@ -17,7 +17,7 @@ class JinaTextEmbeddingModel(TextEmbeddingModel):
     Model class for Jina text embedding model.
     """
     api_base: str = 'https://api.jina.ai/v1/embeddings'
-    models: list[str] = ['jina-embeddings-v2-base-en', 'jina-embeddings-v2-small-en']
+    models: list[str] = ['jina-embeddings-v2-base-en', 'jina-embeddings-v2-small-en', 'jina-embeddings-v2-base-zh']
 
     def _invoke(self, model: str, credentials: dict,
                 texts: list[str], user: Optional[str] = None) \