Skip to content

Commit 23a8001

Browse files
Fix E2E tests
1 parent bd34e1a commit 23a8001

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tests/e2e/test_graphrag_e2e.py

+6-3
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,8 @@ def test_graphrag_happy_path(
7272
biology
7373
7474
Answer:
75-
"""
75+
""",
76+
None,
7677
)
7778
assert isinstance(result, RagResultModel)
7879
assert result.answer == "some text"
@@ -117,7 +118,8 @@ def test_graphrag_happy_path_return_context(
117118
biology
118119
119120
Answer:
120-
"""
121+
""",
122+
None,
121123
)
122124
assert isinstance(result, RagResultModel)
123125
assert result.answer == "some text"
@@ -163,7 +165,8 @@ def test_graphrag_happy_path_examples(
163165
biology
164166
165167
Answer:
166-
"""
168+
""",
169+
None,
167170
)
168171
assert result.answer == "some text"
169172

0 commit comments

Comments
 (0)