From 2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f Mon Sep 17 00:00:00 2001 From: sigoden Date: Tue, 25 Jun 2024 07:39:35 +0800 Subject: feat: support rag-dedicated clients (jina and voyageai) (#645) --- models.yaml | 109 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++- 1 file changed, 108 insertions(+), 1 deletion(-) (limited to 'models.yaml') diff --git a/models.yaml b/models.yaml index 51cbb76..0b24d78 100644 --- a/models.yaml +++ b/models.yaml @@ -1145,4 +1145,111 @@ input_price: 0.008 output_vector_size: 768 default_chunk_size: 1000 - max_batch_size: 100 \ No newline at end of file + max_batch_size: 100 + +- platform: jina + # docs: + # - https://jina.ai/ + # - https://api.jina.ai/redoc + models: + - name: jina-embeddings-v2-base-en + type: embedding + max_input_tokens: 8192 + input_price: 0.02 + output_vector_size: 768 + default_chunk_size: 1500 + max_batch_size: 100 + - name: jina-embeddings-v2-small-en + type: embedding + max_input_tokens: 8192 + input_price: 0.02 + output_vector_size: 512 + default_chunk_size: 1000 + max_batch_size: 100 + - name: jina-embeddings-v2-base-zsh + type: embedding + max_input_tokens: 8192 + input_price: 0.02 + output_vector_size: 768 + default_chunk_size: 1500 + max_batch_size: 100 + - name: jina-embeddings-v2-base-code + type: embedding + max_input_tokens: 8192 + input_price: 0.02 + output_vector_size: 768 + default_chunk_size: 1500 + max_batch_size: 100 + - name: jina-colbert-v1-en + type: embedding + max_input_tokens: 8192 + input_price: 0.02 + output_vector_size: 768 + default_chunk_size: 1500 + max_batch_size: 100 + - name: jina-reranker-v1-base-en + type: rerank + max_input_tokens: 8192 + input_price: 0.02 + - name: jina-reranker-v1-turbo-en + type: rerank + max_input_tokens: 8192 + input_price: 0.02 + - name: jina-colbert-v1-en + type: rerank + max_input_tokens: 8192 + input_price: 0.02 + - name: jina-reranker-v1-base-multilingual + type: rerank + max_input_tokens: 8192 + input_price: 0.02 + +- platform: voyageai + # docs: + # - https://docs.voyageai.com/docs/embeddings + # - https://docs.voyageai.com/docs/pricing + # - https://docs.voyageai.com/reference/embeddings-api + models: + - name: voyage-large-2-instruct + type: embedding + max_input_tokens: 16000 + input_price: 0.12 + output_vector_size: 1024 + default_chunk_size: 2000 + max_batch_size: 128 + - name: voyage-large-2 + type: embedding + max_input_tokens: 16000 + input_price: 0.12 + output_vector_size: 1536 + default_chunk_size: 3000 + max_batch_size: 128 + - name: voyage-multilingual-2 + type: embedding + max_input_tokens: 32000 + input_price: 0.12 + output_vector_size: 1024 + default_chunk_size: 2000 + max_batch_size: 128 + - name: voyage-code-2 + type: embedding + max_input_tokens: 16000 + input_price: 0.12 + output_vector_size: 1536 + default_chunk_size: 3000 + max_batch_size: 128 + - name: voyage-2 + type: embedding + max_input_tokens: 4000 + input_price: 0.1 + output_vector_size: 1024 + default_chunk_size: 2000 + max_batch_size: 128 + - name: rerank-1 + type: rerank + max_input_tokens: 8000 + input_price: 0.05 + - name: rerank-lite-1 + type: rerank + max_input_tokens: 4000 + input_price: 0.02 -- cgit v1.2.3