diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-25 07:39:35 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-25 07:39:35 +0800 |
| commit | 2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f (patch) | |
| tree | 5e72e72d592025c00a46cebd0eb7012ccd765702 /models.yaml | |
| parent | ed71901611247d41daed8112a5106b42eb12395b (diff) | |
| download | aichat-2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f.tar.gz | |
feat: support rag-dedicated clients (jina and voyageai) (#645)
Diffstat (limited to 'models.yaml')
| -rw-r--r-- | models.yaml | 109 |
1 files changed, 108 insertions, 1 deletions
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 |