fix: output of query and search task

Signed-off-by: thiswillbeyourgithub <[email protected]>

Author: thiswillbeyourgithub

wdoc/wdoc.py

@@ -760,13 +760,13 @@ def summary_task(self) -> wdocSummary:
         self.latest_cost = total_cost
         return results
 
-    def query_task(self, query) -> Any:
+    def query_task(self, query) -> Dict[str, Any]:
         self.task = "query"
-        self._query_or_search_task(query=query)
+        return self._query_or_search_task(query=query)
 
-    def search_task(self, query) -> Any:
+    def search_task(self, query) -> Dict[str, Any]:
         self.task = "search"
-        self._query_or_search_task(query=query)
+        return self._query_or_search_task(query=query)
 
     def _query_or_search_task(self, query: str) -> dict:
         # load embeddings for querying