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:
Kevin Hu
2025-03-31 15:33:19 +08:00
committed by GitHub
parent 1985ff7918
commit 7d9dd1e5d3
9 changed files with 5 additions and 131 deletions

View File

@ -228,6 +228,7 @@ class TestChunksRetrieval:
else:
assert expected_message in res["message"]
@pytest.mark.skip
@pytest.mark.parametrize(
"payload, expected_code, expected_message",
[

View File

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

View File

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