apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-14 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add index factory and chromadb

Browse source
This commit is contained in:
seehi 2024-02-28 13:42:41 +08:00 • committed by betterwang
parent 8f3d56d18c
commit 90ca74147d
19 changed files with 505 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/metagpt/rag/factories/test_ranker.py
View file

@ -18,9 +18,7 @@ class TestRankerFactory:
def test_get_rankers_with_no_configs(self, ranker_factory: RankerFactory, mock_llm, mocker):
mocker.patch.object(ranker_factory, "_extract_llm", return_value=mock_llm)
default_rankers = ranker_factory.get_rankers()
assert len(default_rankers) == 1
assert isinstance(default_rankers[0], LLMRerank)
ranker_factory._extract_llm.assert_called_once()
assert len(default_rankers) == 0
def test_get_rankers_with_configs(self, ranker_factory: RankerFactory, mock_llm):
mock_config = LLMRankerConfig(llm=mock_llm)