Fix mxbai rerank default model
This commit is contained in:
parent
68457f251e
commit
ed809759d5
3 changed files with 5 additions and 4 deletions
|
|
@ -164,7 +164,7 @@ Reranking retrieves more candidates than needed, then uses a cross-encoder to re
|
||||||
reranking:
|
reranking:
|
||||||
model:
|
model:
|
||||||
provider: mxbai # or cohere, zeroentropy, vllm
|
provider: mxbai # or cohere, zeroentropy, vllm
|
||||||
name: mxbai-rerank-base-v1
|
name: mixedbread-ai/mxbai-rerank-base-v2
|
||||||
```
|
```
|
||||||
|
|
||||||
**When to use reranking:**
|
**When to use reranking:**
|
||||||
|
|
@ -291,7 +291,7 @@ search:
|
||||||
reranking:
|
reranking:
|
||||||
model:
|
model:
|
||||||
provider: mxbai
|
provider: mxbai
|
||||||
name: mxbai-rerank-base-v1
|
name: mixedbread-ai/mxbai-rerank-base-v2
|
||||||
```
|
```
|
||||||
|
|
||||||
### Long-Form Content (Articles, Reports)
|
### Long-Form Content (Articles, Reports)
|
||||||
|
|
|
||||||
|
|
@ -10,7 +10,7 @@ class MxBAIReranker(RerankerBase):
|
||||||
model_name = (
|
model_name = (
|
||||||
Config.reranking.model.name
|
Config.reranking.model.name
|
||||||
if Config.reranking.model
|
if Config.reranking.model
|
||||||
else "mxbai-rerank-base-v2"
|
else "mixedbread-ai/mxbai-rerank-base-v2"
|
||||||
)
|
)
|
||||||
self._client = MxbaiRerankV2(model_name, disable_transformers_warnings=True)
|
self._client = MxbaiRerankV2(model_name, disable_transformers_warnings=True)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -31,7 +31,8 @@ async def test_reranker_base():
|
||||||
from haiku.rag.config import Config
|
from haiku.rag.config import Config
|
||||||
|
|
||||||
reranker = RerankerBase()
|
reranker = RerankerBase()
|
||||||
assert reranker._model == Config.reranking.model
|
expected_model = Config.reranking.model.name if Config.reranking.model else None
|
||||||
|
assert reranker._model == expected_model
|
||||||
|
|
||||||
with pytest.raises(NotImplementedError):
|
with pytest.raises(NotImplementedError):
|
||||||
await reranker.rerank("query", [])
|
await reranker.rerank("query", [])
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue