Update firecrawler.py

firecrawler.py

@@ -1,29 +1,30 @@
 from firecrawl import FirecrawlApp
 from pydantic import BaseModel, Field
 from typing import List
-from smolagents import tool
+from smolagents import Tool
 import os
 
 
 #Fetch API key fior firecrawl
 api_key = os.getenv("api_key")
 
-@tool
-def execute_firecrawl(websites_string: str):
-    """
-    This tool scans a url and returns its intended purpose
+
+class FireCrawlTool(Tool):
+    name = "firecrawl_website_qa"
+    description = """
+    This tool returns informatuion about websites using an API call"""
     
-    Args:
-        websites_string (str): A string of websites separated by commas, spaces, or new lines
+    inputs = {
+        "websites": {
+            "type": "string",
+        }
+    }
     
-    Returns:
-        str: Information about a website
+    output_type = "string"
 
-    Example:
-        >>> # https://au.norton.com
-        >>> result = [{"website":{"purpose":"Our best protection, trusted by millions, now includes Genie AI Scam Detection. Advanced tech for advanced threats. Explore plans.","company_name":"Gen Digital Inc."}}]
 
-    """
+    def forward(self, websites_string: str):
+    
     # Initialize the FirecrawlApp with the API key
     app = FirecrawlApp(api_key = api_key)