fix FAQRetrieverTool

faq_retriever.py

@@ -19,23 +19,26 @@ docs = [
 # Create the custom FAQ retriever tool
 class FAQRetrieverTool(Tool):
     name = "faq_retriever"
-    description = "Retrieves relevant internal company FAQ entries based on a user query."
+    description = "Answers company FAQ questions using internal documentation."
     inputs = {
         "query": {
             "type": "string",
-            "description": "User's question about internal company processes or services."
+            "description": "The user question to answer from internal FAQ documents."
         }
     }
     output_type = "string"
 
-    def __init__(self, documents):
-        self.retriever = BM25Retriever.from_documents(documents)
+    def __init__(self, docs):
+        self.is_initialized = True  # <-- questo mancava!
+        self.retriever = BM25Retriever.from_documents(docs)
 
-    def forward(self, query: str) -> str:
+    def forward(self, query: str):
         results = self.retriever.get_relevant_documents(query)
-        if not results:
-            return "No relevant FAQ entry found."
-        return "\n\n".join(doc.page_content for doc in results[:2])  # Return top 2
+        if results:
+            return "\n\n".join([doc.page_content for doc in results[:3]])
+        else:
+            return "Sorry, I couldn't find a matching FAQ."
+
 
 # Initialize the retriever tool
 faq_tool = FAQRetrieverTool(docs)