kanshan/graphrag
1
0
Fork 0
mirror of https://github.com/microsoft/graphrag.git synced 2026-02-07 11:42:33 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Address comments

Browse Source
This commit is contained in:
Alonso Guevara 2024-10-21 16:12:07 -06:00
parent f9de673d16
commit e16710ebd5
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
graphrag/query/structured_search/drift_search/search.py
View File

@ -120,7 +120,7 @@ class DRIFTSearch(BaseSearch[DRIFTSearchContextBuilder]):
i["intermediate_answer"] for i in response if "intermediate_answer" in i
]
if len(intermediate_answers) == 0:
if not intermediate_answers:
error_msg = "No intermediate answers found in primer response. Ensure that the primer response includes intermediate answers."
raise RuntimeError(error_msg)

2
graphrag/query/structured_search/drift_search/state.py
View File

@ -44,7 +44,7 @@ class QueryState:
for follow_up in follow_ups:
if isinstance(follow_up, str):
follow_up = DriftAction(query=follow_up)
else:
elif not isinstance(follow_up, DriftAction):
log.warning(
"Follow-up action is not a string, found type: %s", type(follow_up)
)