Update firecrawler.py

firecrawler.py

@@ -12,45 +12,39 @@ api_key = os.getenv("api_key")
 class FireCrawlTool(Tool):
     name = "firecrawl_website_qa"
     description = """
-    This tool returns informatuion about websites using an API call"""
+    This tool scrapes websites using an API call"""
     
     inputs = {
-        "websites": {
+        "website": {
             "type": "string",
-            "description": "A singlular or multiple website addresses",
+            "description": "A singlular website address",
         }
     }
     
     output_type = "string"
 
 
-    def forward(self, websites: str):
+    def forward(self, website: str):
     
         # Initialize the FirecrawlApp with the API key
         app = FirecrawlApp(api_key = api_key)
     
         # Convert the websites string to a list
         # Split by commas, newlines, or spaces and strip whitespace
-        websites_list = [
-            website.strip() 
-            for website in websites.replace(',', ' ').replace('\n', ' ').split()
-            if website.strip()
-        ]
-    
-        # Define the extraction schema
-        class ExtractSchema(BaseModel):
-            company_name: str
-            website_purpose: str
+        #websites_list = [
+            #website.strip() 
+            #for website in websites.replace(',', ' ').replace('\n', ' ').split()
+            #if website.strip()
+        #]
 
-        prompt_string = 'Extract the website purpose and company name.'
-    
-        # Execute the API call
-        data = app.extract(
-            websites_list,
-            {
-                'prompt': prompt_string,
-                'schema': ExtractSchema.model_json_schema(),
+        # Scrape a website:
+        scrape_result = app.scrape_url(website, 
+            params={
+                'location': {
+                    'country': 'AU'
+                }
             }
         )
     
-        return data
+    
+        return scrape_result