stwhite
/
ira
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make generate_search_queries async to fix runtime errors

This commit is contained in:
Steve White 2025-02-28 16:46:25 -06:00
parent 6b749b9cb6
commit d53198d5ab
2 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
query/query_processor.py
View File

@ -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]: search_engines: List[str]) -> Dict[str, Any]:
""" """
Generate optimized search queries for different search engines. Generate optimized search queries for different search engines.
@ -87,7 +87,7 @@ class QueryProcessor:
enhanced_query = structured_query['enhanced_query'] enhanced_query = structured_query['enhanced_query']
# Generate search queries for each engine # 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 enhanced_query, search_engines
) )

14
scripts/query_to_report.py
View File

@ -75,8 +75,18 @@ async def query_to_report(
search_executor = SearchExecutor() search_executor = SearchExecutor()
search_engines = search_executor.get_available_search_engines() search_engines = search_executor.get_available_search_engines()
structured_query = query_processor.generate_search_queries(structured_query, search_engines) if not use_mock:
logger.info(f"Generated search queries for engines: {', '.join(search_engines)}") # 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 # Step 3: Execute search
search_executor = SearchExecutor() search_executor = SearchExecutor()