Update app.py

app.py

@@ -4,6 +4,8 @@ from langchain_community.tools import DuckDuckGoSearchRun
 from langchain_google_genai import ChatGoogleGenerativeAI
 import os
 import google.generativeai as genai
+from langchain.tools import BaseTool # Import BaseTool
+
 
 # Configure Gemini API Key (replace with your actual key)
 # Use Streamlit secrets for secure storage.
@@ -11,11 +13,23 @@ GOOGLE_API_KEY = st.secrets["GOOGLE_API_KEY"]
 genai.configure(api_key=GOOGLE_API_KEY)
 
 
-# Define Tools
+# Define Tools - Corrected: Wrap the tool in a list if it's a single tool
 search_tool = DuckDuckGoSearchRun()
 
+class ToolWrapper(BaseTool):
+    """Wrapper for tools to make them compatible with CrewAI."""
+    name: str
+    func: callable
+    description: str
+
+    def _run(self, *args, **kwargs):
+        return self.func(*args, **kwargs)
+
+    async def _arun(self, *args, **kwargs):
+        return await self.func(*args, **kwargs)
+
 
-# Define Agents
+# Define Agents - Corrected: Use ToolWrapper
 inventory_specialist = Agent(
     role='Inventory Specialist',
     goal='Identify the best matching homes from current inventory based on customer criteria.',
@@ -26,7 +40,7 @@ inventory_specialist = Agent(
     """,
     verbose=True,
     allow_delegation=False,
-    tools=[search_tool],
+    tools=[ToolWrapper(name=search_tool.name, func=search_tool.run, description=search_tool.description)],  # Corrected Tool Usage
     llm=ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.7, convert_system_message_to_human=True) # Use Gemini model
 )