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

fix rag ut failed cases

Browse source
This commit is contained in:
betterwang 2024-03-07 14:37:26 +08:00
parent 184b0127e5
commit f327798e00
4 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/metagpt/rag/engines/test_simple.py
View file

@ -97,7 +97,7 @@ class TestSimpleEngine:
mock_super_aretrieve = mocker.patch(
"metagpt.rag.engines.simple.RetrieverQueryEngine.aretrieve", new_callable=mocker.AsyncMock
)
mock_super_aretrieve.return_value = ["node_with_score"]
mock_super_aretrieve.return_value = [TextNode(text="node_with_score", metadata={"is_obj": False})]
# Setup
engine = SimpleEngine(retriever=mocker.MagicMock())
@ -109,7 +109,7 @@ class TestSimpleEngine:
# Assertions
mock_query_bundle.assert_called_once_with(test_query)
mock_super_aretrieve.assert_called_once_with("query_bundle")
assert result == ["node_with_score"]
assert result[0].text == "node_with_score"
def test_add_docs(self, mocker):
# Mock
@ -157,4 +157,4 @@ class TestSimpleEngine:
assert mock_retriever.add_nodes.call_count == 1
for node in mock_retriever.add_nodes.call_args[0][0]:
assert isinstance(node, TextNode)
assert "obj" in node.metadata
assert "is_obj" in node.metadata