From 09a8db10d432c7fc31e533d88358e04133890957 Mon Sep 17 00:00:00 2001 From: Joshua <138381132+joshua20231026@users.noreply.github.com> Date: Fri, 19 Jan 2024 18:11:55 +0800 Subject: [PATCH] Add jina-embeddings-v2-base-de model configuration (#2094) --- .../jina/text_embedding/jina-embeddings-v2-base-de.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-de.yaml diff --git a/api/core/model_runtime/model_providers/jina/text_embedding/jina-embeddings-v2-base-de.yaml b/api/core/model_runtime/model_providers/jina/text_embedding/jina-embeddings-v2-base-de.yaml new file mode 100644 index 0000000000..09f7023acb --- /dev/null +++ b/api/core/model_runtime/model_providers/jina/text_embedding/jina-embeddings-v2-base-de.yaml @@ -0,0 +1,9 @@ +model: jina-embeddings-v2-base-de +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 57aa3d4a13..50238fbcde 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', 'jina-embeddings-v2-base-zh'] + models: list[str] = ['jina-embeddings-v2-base-en', 'jina-embeddings-v2-small-en', 'jina-embeddings-v2-base-zh', 'jina-embeddings-v2-base-de'] def _invoke(self, model: str, credentials: dict, texts: list[str], user: Optional[str] = None) \