mirror of
https://github.com/infiniflow/ragflow.git
synced 2025-12-08 20:42:30 +08:00
Refa: remove default build-in rerank model. (#6682)
### What problem does this PR solve? ### Type of change - [x] Refactoring - [x] Performance Improvement
This commit is contained in:
@ -228,6 +228,7 @@ class TestChunksRetrieval:
|
||||
else:
|
||||
assert expected_message in res["message"]
|
||||
|
||||
@pytest.mark.skip
|
||||
@pytest.mark.parametrize(
|
||||
"payload, expected_code, expected_message",
|
||||
[
|
||||
|
||||
@ -145,28 +145,6 @@ class TestAdvancedConfigurations:
|
||||
"name, embedding_model, expected_code",
|
||||
[
|
||||
("BAAI/bge-large-zh-v1.5", "BAAI/bge-large-zh-v1.5", 0),
|
||||
("BAAI/bge-base-en-v1.5", "BAAI/bge-base-en-v1.5", 0),
|
||||
("BAAI/bge-large-en-v1.5", "BAAI/bge-large-en-v1.5", 0),
|
||||
("BAAI/bge-small-en-v1.5", "BAAI/bge-small-en-v1.5", 0),
|
||||
("BAAI/bge-small-zh-v1.5", "BAAI/bge-small-zh-v1.5", 0),
|
||||
(
|
||||
"jinaai/jina-embeddings-v2-base-en",
|
||||
"jinaai/jina-embeddings-v2-base-en",
|
||||
0,
|
||||
),
|
||||
(
|
||||
"jinaai/jina-embeddings-v2-small-en",
|
||||
"jinaai/jina-embeddings-v2-small-en",
|
||||
0,
|
||||
),
|
||||
("nomic-ai/nomic-embed-text-v1.5", "nomic-ai/nomic-embed-text-v1.5", 0),
|
||||
(
|
||||
"sentence-transformers/all-MiniLM-L6-v2",
|
||||
"sentence-transformers/all-MiniLM-L6-v2",
|
||||
0,
|
||||
),
|
||||
("text-embedding-v2", "text-embedding-v2", 0),
|
||||
("text-embedding-v3", "text-embedding-v3", 0),
|
||||
(
|
||||
"maidalun1020/bce-embedding-base_v1",
|
||||
"maidalun1020/bce-embedding-base_v1",
|
||||
|
||||
@ -86,16 +86,6 @@ class TestDatasetUpdate:
|
||||
"embedding_model, expected_code, expected_message",
|
||||
[
|
||||
("BAAI/bge-large-zh-v1.5", 0, ""),
|
||||
("BAAI/bge-base-en-v1.5", 0, ""),
|
||||
("BAAI/bge-large-en-v1.5", 0, ""),
|
||||
("BAAI/bge-small-en-v1.5", 0, ""),
|
||||
("BAAI/bge-small-zh-v1.5", 0, ""),
|
||||
("jinaai/jina-embeddings-v2-base-en", 0, ""),
|
||||
("jinaai/jina-embeddings-v2-small-en", 0, ""),
|
||||
("nomic-ai/nomic-embed-text-v1.5", 0, ""),
|
||||
("sentence-transformers/all-MiniLM-L6-v2", 0, ""),
|
||||
("text-embedding-v2", 0, ""),
|
||||
("text-embedding-v3", 0, ""),
|
||||
("maidalun1020/bce-embedding-base_v1", 0, ""),
|
||||
(
|
||||
"other_embedding_model",
|
||||
|
||||
Reference in New Issue
Block a user