handle empty query

circlemind-ai · Dec 14, 2024 · a6fa692 · a6fa692
1 parent 4475116
commit a6fa692
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/fast_graphrag/_graphrag.py b/fast_graphrag/_graphrag.py
@@ -157,7 +157,7 @@ async def _query() -> TQueryResponse[GTNode, GTEdge, GTHash, GTChunk]:
         return get_event_loop().run_until_complete(_query())
 
     async def async_query(
-        self, query: str, params: Optional[QueryParam] = None
+        self, query: Optional[str], params: Optional[QueryParam] = None
     ) -> TQueryResponse[GTNode, GTEdge, GTHash, GTChunk]:
         """Query the graph with a given input.
 
@@ -168,6 +168,10 @@ async def async_query(
         Returns:
             TQueryResponse: The result of the query (response + context).
         """
+        if query is None or len(query) == 0:
+            return TQueryResponse[GTNode, GTEdge, GTHash, GTChunk](
+                response=PROMPTS["fail_response"], context=TContext([], [], [])
+            )
         if params is None:
             params = QueryParam()