Update services/chat_service.py

services/chat_service.py

@@ -199,34 +199,36 @@ class ChatService:
         try:
             print("-----------------------------")
             print("starting searches .... ")
-            # Run searches in parallel
-            product_task = asyncio.create_task(
-                self.data_service.search(query, top_k)
-            )
-            pdf_task = asyncio.create_task(
-                self.pdf_service.search(query, top_k)
-            )
-            faq_task = asyncio.create_task(
-                self.faq_service.search_faqs(query, top_k)
-            )
             
-            # Gather results
-            products, pdfs, faqs = await asyncio.gather(
-                product_task, pdf_task, faq_task
-            )
+            # Make sure the search methods are actually async
+            if not hasattr(self.data_service.search, '__await__'):
+                products = await asyncio.to_thread(self.data_service.search, query, top_k)
+            else:
+                products = await self.data_service.search(query, top_k)
+                
+            if not hasattr(self.pdf_service.search, '__await__'):
+                pdfs = await asyncio.to_thread(self.pdf_service.search, query, top_k)
+            else:
+                pdfs = await self.pdf_service.search(query, top_k)
+                
+            if not hasattr(self.faq_service.search_faqs, '__await__'):
+                faqs = await asyncio.to_thread(self.faq_service.search_faqs, query, top_k)
+            else:
+                faqs = await self.faq_service.search_faqs(query, top_k)
 
-            print(products)
-            return {
-                'products': products,
-                'documents': pdfs,
-                'faqs': faqs
+            results = {
+                'products': products or [],
+                'documents': pdfs or [],
+                'faqs': faqs or []
             }
             
+            print("Search results:", results)
+            return results
+            
         except Exception as e:
             logger.error(f"Error searching sources: {e}")
             return {'products': [], 'documents': [], 'faqs': []}
 
-
     async def generate_response(
         self,
         prompt: str,