File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -86,13 +86,13 @@ SEARCH docs SIMILAR TO 'hello' LIMIT 5
8686### Cross-encoder reranking (RERANK default)
8787
8888```
89- cross-encoder /ms-marco-MiniLM-L-6-v2
89+ Xenova /ms-marco-MiniLM-L-6-v2
9090```
9191
9292| Model | Notes |
9393| ---| ---|
94- | ` cross-encoder /ms-marco-MiniLM-L-6-v2` | Default. Fast passage reranker |
95- | ` cross-encoder /ms-marco-MiniLM-L-12-v2` | Larger, higher quality |
94+ | ` Xenova /ms-marco-MiniLM-L-6-v2` | Default. Fast passage reranker |
95+ | ` Xenova /ms-marco-MiniLM-L-12-v2` | Larger, higher quality |
9696| ` BAAI/bge-reranker-base ` | Strong general-purpose reranker |
9797| ` BAAI/bge-reranker-large ` | Highest quality, slower |
9898
Original file line number Diff line number Diff line change @@ -377,15 +377,15 @@ SEARCH articles SIMILAR TO 'attention mechanism in transformers' LIMIT 10 USING
377377Custom cross-encoder model:
378378``` sql
379379SEARCH articles SIMILAR TO ' semantic search' LIMIT 5
380- RERANK MODEL ' cross-encoder /ms-marco-MiniLM-L-6-v2'
380+ RERANK MODEL ' Xenova /ms-marco-MiniLM-L-6-v2'
381381```
382382
383- ** Default cross-encoder model:** ` cross-encoder /ms-marco-MiniLM-L-6-v2`
383+ ** Default cross-encoder model:** ` Xenova /ms-marco-MiniLM-L-6-v2`
384384
385385| Model | Notes |
386386| ---| ---|
387- | ` cross-encoder /ms-marco-MiniLM-L-6-v2` | Default. Fast and accurate for passage reranking |
388- | ` cross-encoder /ms-marco-MiniLM-L-12-v2` | Larger, higher quality, slower |
387+ | ` Xenova /ms-marco-MiniLM-L-6-v2` | Default. Fast and accurate for passage reranking |
388+ | ` Xenova /ms-marco-MiniLM-L-12-v2` | Larger, higher quality, slower |
389389| ` BAAI/bge-reranker-base ` | BGE reranker, strong general-purpose performance |
390390| ` BAAI/bge-reranker-large ` | Highest quality BGE reranker, slower |
391391
Original file line number Diff line number Diff line change @@ -431,7 +431,7 @@ SEARCH medical_records SIMILAR TO 'neuromuscular junction antibody acetylcholine
431431 AND sub_specialty IN ('neuromuscular', 'neuroimmunology')
432432 AND severity IN ('high', 'critical')
433433 AND year >= 2022
434- RERANK MODEL 'cross-encoder /ms-marco-MiniLM-L-6-v2'
434+ RERANK MODEL 'Xenova /ms-marco-MiniLM-L-6-v2'
435435```
436436
437437---
Original file line number Diff line number Diff line change @@ -1003,10 +1003,10 @@ def test_rerank_flag_set(self):
10031003
10041004 def test_rerank_with_model (self ):
10051005 node = parse (
1006- "SEARCH col SIMILAR TO 'q' LIMIT 5 RERANK MODEL 'cross-encoder /ms-marco-MiniLM-L-6-v2'"
1006+ "SEARCH col SIMILAR TO 'q' LIMIT 5 RERANK MODEL 'Xenova /ms-marco-MiniLM-L-6-v2'"
10071007 )
10081008 assert node .rerank is True
1009- assert node .rerank_model == "cross-encoder /ms-marco-MiniLM-L-6-v2"
1009+ assert node .rerank_model == "Xenova /ms-marco-MiniLM-L-6-v2"
10101010
10111011 def test_rerank_default_false (self ):
10121012 node = parse ("SEARCH col SIMILAR TO 'q' LIMIT 5" )
@@ -1032,12 +1032,12 @@ def test_rerank_with_where(self):
10321032 def test_rerank_with_hybrid_where_and_model (self ):
10331033 node = parse (
10341034 "SEARCH col SIMILAR TO 'q' LIMIT 5 USING HYBRID WHERE year > 2020 "
1035- "RERANK MODEL 'cross-encoder /ms-marco-MiniLM-L-6-v2'"
1035+ "RERANK MODEL 'Xenova /ms-marco-MiniLM-L-6-v2'"
10361036 )
10371037 assert node .hybrid is True
10381038 assert node .query_filter is not None
10391039 assert node .rerank is True
1040- assert node .rerank_model == "cross-encoder /ms-marco-MiniLM-L-6-v2"
1040+ assert node .rerank_model == "Xenova /ms-marco-MiniLM-L-6-v2"
10411041
10421042 def test_rerank_lowercase (self ):
10431043 node = parse ("SEARCH col SIMILAR TO 'q' LIMIT 5 rerank" )
You can’t perform that action at this time.
0 commit comments