From 193255cf540a05c7acc7c1aef50458ba7677a1b2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=89=A1=E4=B8=B9=E5=87=A4=E5=87=B0?= <87239270+sunrise0o0@users.noreply.github.com> Date: Sun, 9 Feb 2025 08:21:06 +0800 Subject: [PATCH] Update embedings.ts embedding models information --- src/renderer/src/config/embedings.ts | 224 ++++++++++++++++++++++++--- 1 file changed, 206 insertions(+), 18 deletions(-) diff --git a/src/renderer/src/config/embedings.ts b/src/renderer/src/config/embedings.ts index d687dd39ef..041a3e797c 100644 --- a/src/renderer/src/config/embedings.ts +++ b/src/renderer/src/config/embedings.ts @@ -1,39 +1,227 @@ export const EMBEDDING_MODELS = [ { - id: 'BAAI/bge-m3', - max_context: 8000 + "id": "Doubao-embedding", + "max_context": 4095 }, { - id: 'Pro/BAAI/bge-m3', - max_context: 8000 + "id": "Doubao-embedding-vision", + "max_context": 8191 }, { - id: 'BAAI/bge-large-zh-v1.5', - max_context: 512 + "id": "Doubao-embedding-large", + "max_context": 4095 }, { - id: 'BAAI/bge-large-en-v1.5', - max_context: 512 + "id": "text-embedding-v3", + "max_context": 8192 }, { - id: 'netease-youdao/bce-embedding-base_v1', - max_context: 512 + "id": "text-embedding-v2", + "max_context": 2048 }, { - id: 'tao-8k', - max_context: 8192 + "id": "text-embedding-v1", + "max_context": 2048 }, { - id: 'embedding-v1', - max_context: 384 + "id": "text-embedding-async-v2", + "max_context": 2048 }, { - id: 'bge-large-zh', - max_context: 512 + "id": "text-embedding-async-v1", + "max_context": 2048 }, { - id: 'bge-large-en', - max_context: 512 + "id": "text-embedding-3-small", + "max_context": 8191 + }, + { + "id": "text-embedding-3-large", + "max_context": 8191 + }, + { + "id": "text-embedding-ada-002", + "max_context": 8191 + }, + { + "id": "Embedding-V1", + "max_context": 384 + }, + { + "id": "tao-8k", + "max_context": 8192 + }, + { + "id": "embedding-2", + "max_context": 1024 + }, + { + "id": "embedding-3", + "max_context": 2048 + }, + { + "id": "hunyuan-embedding", + "max_context": 1024 + }, + { + "id": "Baichuan-Text-Embedding", + "max_context": 512 + }, + { + "id": "M2-BERT-80M-2K-Retrieval", + "max_context": 2048 + }, + { + "id": "M2-BERT-80M-8K-Retrieval", + "max_context": 8192 + }, + { + "id": "M2-BERT-80M-32K-Retrieval", + "max_context": 32768 + }, + { + "id": "UAE-Large-v1", + "max_context": 512 + }, + { + "id": "BGE-Large-EN-v1.5", + "max_context": 512 + }, + { + "id": "BGE-Base-EN-v1.5", + "max_context": 512 + }, + { + "id": "jina-embedding-b-en-v1", + "max_context": 512 + }, + { + "id": "jina-embeddings-v2-base-en", + "max_context": 8191 + }, + { + "id": "jina-embeddings-v2-base-zh", + "max_context": 8191 + }, + { + "id": "jina-embeddings-v2-base-de", + "max_context": 8191 + }, + { + "id": "jina-embeddings-v2-base-code", + "max_context": 8191 + }, + { + "id": "jina-embeddings-v2-base-es", + "max_context": 8191 + }, + { + "id": "jina-colbert-v1-en", + "max_context": 8191 + }, + { + "id": "jina-reranker-v1-base-en", + "max_context": 8191 + }, + { + "id": "jina-reranker-v1-turbo-en", + "max_context": 8191 + }, + { + "id": "jina-reranker-v1-tiny-en", + "max_context": 8191 + }, + { + "id": "jina-clip-v1", + "max_context": 8191 + }, + { + "id": "jina-reranker-v2-base-multilingual", + "max_context": 8191 + }, + { + "id": "reader-lm-1.5b", + "max_context": 256000 + }, + { + "id": "reader-lm-0.5b", + "max_context": 256000 + }, + { + "id": "jina-colbert-v2", + "max_context": 8191 + }, + { + "id": "jina-embeddings-v3", + "max_context": 8191 + }, + { + "id": "BAAI/bge-m3", + "max_context": 8191 + }, + { + "id": "netease-youdao/bce-embedding-base_v1", + "max_context": 512 + }, + { + "id": "BAAI/bge-large-zh-v1.5", + "max_context": 512 + }, + { + "id": "BAAI/bge-large-en-v1.5", + "max_context": 512 + }, + { + "id": "Pro/BAAI/bge-m3", + "max_context": 8191 + }, + { + "id": "nomic-embed-text-v1", + "max_context": 8192 + }, + { + "id": "nomic-embed-text-v1.5", + "max_context": 8192 + }, + { + "id": "gte-multilingual-base", + "max_context": 8192 + }, + { + "id": "embedding-query", + "max_context": 4000 + }, + { + "id": "embedding-passage", + "max_context": 4000 + }, + { + "id": "embed-english-v3.0", + "max_context": 512 + }, + { + "id": "embed-english-light-v3.0", + "max_context": 512 + }, + { + "id": "embed-multilingual-v3.0", + "max_context": 512 + }, + { + "id": "embed-multilingual-light-v3.0", + "max_context": 512 + }, + { + "id": "embed-english-v2.0", + "max_context": 512 + }, + { + "id": "embed-english-light-v2.0", + "max_context": 512 + }, + { + "id": "embed-multilingual-v2.0", + "max_context": 256 } ]