Update retriever.py

retriever.py

@@ -38,25 +38,43 @@ class GuestInfoRetrieverTool(Tool):
             for guest in guest_dataset
         ]
         
-        # Create retriever
-        self.retriever = BM25Retriever.from_documents(self.docs)
-        print(" Guest retriever ready")
+        # Create retriever with k parameter for better results
+        self.retriever = BM25Retriever.from_documents(self.docs, k=10)
+        print("✅ Guest retriever ready")
 
     def forward(self, query: str):
-        print(f" Searching guest database for: '{query}'")
+        print(f"🔍 Searching guest database for: '{query}'")
         
         # Safety check
         if self.retriever is None:
             return "ERROR: Guest database not initialized."
         
-        results = self.retriever.get_relevant_documents(query)
-        
-        if results:
-            print(f" Found {len(results)} matches")
-            return "\n\n".join([doc.page_content for doc in results[:3]])
-        else:
-            print(" No matches found")
-            return "No matching guest information found."
+        try:
+            # Use invoke() instead of get_relevant_documents()
+            results = self.retriever.invoke(query)
+            
+            if results:
+                print(f"✅ Found {len(results)} matches")
+                # Return more results
+                return "\n\n".join([doc.page_content for doc in results[:5]])
+            else:
+                print("❌ No matches found")
+                return "No matching guest information found."
+                
+        except AttributeError:
+            # Fallback for older LangChain versions
+            try:
+                results = self.retriever.get_relevant_documents(query)
+                if results:
+                    print(f"✅ Found {len(results)} matches (using old method)")
+                    return "\n\n".join([doc.page_content for doc in results[:5]])
+                else:
+                    print("❌ No matches found")
+                    return "No matching guest information found."
+            except Exception as e:
+                return f"ERROR: Failed to retrieve documents - {str(e)}"
+        except Exception as e:
+            return f"ERROR: {str(e)}"
 
 
 def load_guest_dataset():