Fix E2E tests

neo4j · Dec 18, 2024 · 23a8001 · 23a8001
1 parent bd34e1a
commit 23a8001
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/e2e/test_graphrag_e2e.py b/tests/e2e/test_graphrag_e2e.py
@@ -72,7 +72,8 @@ def test_graphrag_happy_path(
 biology
 
 Answer:
-"""
+""",
+        None,
     )
     assert isinstance(result, RagResultModel)
     assert result.answer == "some text"
@@ -117,7 +118,8 @@ def test_graphrag_happy_path_return_context(
 biology
 
 Answer:
-"""
+""",
+        None,
     )
     assert isinstance(result, RagResultModel)
     assert result.answer == "some text"
@@ -163,7 +165,8 @@ def test_graphrag_happy_path_examples(
 biology
 
 Answer:
-"""
+""",
+        None,
     )
     assert result.answer == "some text"