Add jina-embeddings-v2-base-zh model configuration (#2049)

pull/2050/head
Ricky 2 years ago committed by GitHub
parent f7a2f7a727
commit 1a2eacc5a6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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

@ -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) \

Loading…
Cancel
Save