Skip to content

Commit a6fa692

Browse files
committed
handle empty query
1 parent 4475116 commit a6fa692

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

fast_graphrag/_graphrag.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ async def _query() -> TQueryResponse[GTNode, GTEdge, GTHash, GTChunk]:
157157
return get_event_loop().run_until_complete(_query())
158158

159159
async def async_query(
160-
self, query: str, params: Optional[QueryParam] = None
160+
self, query: Optional[str], params: Optional[QueryParam] = None
161161
) -> TQueryResponse[GTNode, GTEdge, GTHash, GTChunk]:
162162
"""Query the graph with a given input.
163163
@@ -168,6 +168,10 @@ async def async_query(
168168
Returns:
169169
TQueryResponse: The result of the query (response + context).
170170
"""
171+
if query is None or len(query) == 0:
172+
return TQueryResponse[GTNode, GTEdge, GTHash, GTChunk](
173+
response=PROMPTS["fail_response"], context=TContext([], [], [])
174+
)
171175
if params is None:
172176
params = QueryParam()
173177

0 commit comments

Comments
 (0)