Update firecrawler.py

firecrawler.py

@@ -25,31 +25,31 @@ class FireCrawlTool(Tool):
 
     def forward(self, websites_string: str):
     
-    # Initialize the FirecrawlApp with the API key
-    app = FirecrawlApp(api_key = api_key)
+        # 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_string.replace(',', ' ').replace('\n', ' ').split()
-        if website.strip()
-    ]
+        # Convert the websites string to a list
+        # Split by commas, newlines, or spaces and strip whitespace
+        websites_list = [
+            website.strip() 
+            for website in websites_string.replace(',', ' ').replace('\n', ' ').split()
+            if website.strip()
+        ]
     
-    # Define the extraction schema
-    class ExtractSchema(BaseModel):
-        company_name: str
-        website_purpose: str
+        # Define the extraction schema
+        class ExtractSchema(BaseModel):
+            company_name: str
+            website_purpose: str
 
-    prompt_string = 'Extract the website purpose and company name.'
+        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(),
-        }
-    )
+        # Execute the API call
+        data = app.extract(
+            websites_list,
+            {
+                'prompt': prompt_string,
+                'schema': ExtractSchema.model_json_schema(),
+            }
+        )
     
-    return data
+        return data