From d53198d5ab9447de7a8062ae2e13532176aff153 Mon Sep 17 00:00:00 2001
From: Steve White <stwhite@mac.com>
Date: Fri, 28 Feb 2025 16:46:25 -0600
Subject: [PATCH] Make generate_search_queries async to fix runtime errors

---
 query/query_processor.py   |  4 ++--
 scripts/query_to_report.py | 14 ++++++++++++--
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/query/query_processor.py b/query/query_processor.py
index 6bb1128..ebd4166 100644
--- a/query/query_processor.py
+++ b/query/query_processor.py
@@ -71,7 +71,7 @@ class QueryProcessor:
             }
         }
     
-    def generate_search_queries(self, structured_query: Dict[str, Any], 
+    async def generate_search_queries(self, structured_query: Dict[str, Any], 
                                search_engines: List[str]) -> Dict[str, Any]:
         """
         Generate optimized search queries for different search engines.
@@ -87,7 +87,7 @@ class QueryProcessor:
         enhanced_query = structured_query['enhanced_query']
         
         # Generate search queries for each engine
-        search_queries = self.llm_interface.generate_search_queries(
+        search_queries = await self.llm_interface.generate_search_queries(
             enhanced_query, search_engines
         )
         
diff --git a/scripts/query_to_report.py b/scripts/query_to_report.py
index 2fff556..3bb33dc 100755
--- a/scripts/query_to_report.py
+++ b/scripts/query_to_report.py
@@ -75,8 +75,18 @@ async def query_to_report(
         search_executor = SearchExecutor()
         search_engines = search_executor.get_available_search_engines()
     
-    structured_query = query_processor.generate_search_queries(structured_query, search_engines)
-    logger.info(f"Generated search queries for engines: {', '.join(search_engines)}")
+    if not use_mock:
+        # Generate search queries for each engine
+        search_queries = await query_processor.generate_search_queries(
+            structured_query, 
+            search_engines or list(search_executor.available_handlers.keys())
+        )
+        structured_query['search_queries'] = search_queries
+        logger.info(f"Generated search queries for engines: {', '.join(search_queries.keys())}")
+    else:
+        # Use mock data
+        structured_query = await query_processor.generate_search_queries(structured_query, search_engines)
+        logger.info(f"Generated search queries for engines: {', '.join(search_engines)}")
     
     # Step 3: Execute search
     search_executor = SearchExecutor()