Update tools/ai_tools.py

tools/ai_tools.py

@@ -70,10 +70,10 @@ class AITools():
                 project_id=os.environ["BROWSERBASE_PROJECT_ID"],
                 model_api_key=os.environ["MODEL_API_KEY"],
                 model_name=WEB_BROWSER_MODEL,
-                dom_settle_timeout_ms=5000,
-                headless=True,
-                self_heal=True,
-                wait_for_captcha_solves=True,
+                #dom_settle_timeout_ms=5000,
+                #headless=True,
+                #self_heal=True,
+                #wait_for_captcha_solves=True,
                 verbose=3
             )
             
@@ -83,6 +83,8 @@ class AITools():
             )
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
+        finally:
+            stagehand_tool.close()
     
     @tool("Image Analysis Tool")
     def image_analysis_tool(question: str, file_path: str) -> str: