Add 8999999

enhanced_websocket_handler.py

@@ -402,10 +402,8 @@ async def get_hybrid_response(user_message: str, context: str, config: dict, kno
     # Search for relevant documents directly using LanceDB
     try:
         logger.info(f"🔍 Searching documents for: {user_message}")
-        # Use the search_documents function instead of LangChain tool
-        from rag_service import search_documents
-        # Search for relevant documents
-        docs = search_documents(user_message, limit=3)
+        from rag_service import search_documents_async
+        docs = await search_documents_async(user_message, limit=3)
         if docs:
             context = "\n\n".join([doc["content"] for doc in docs])
             sources = [doc["source"] for doc in docs]