refactor: Update LocalSeleniumCrawlerStrategy to execute JS code if provided

2024-09-01 16:34:51 +08:00
parent 2ba70b9501
commit 3caf48c9be
1 changed files with 1 additions and 0 deletions
--- a/crawl4ai/crawler_strategy.py
+++ b/crawl4ai/crawler_strategy.py
@@ -244,6 +244,7 @@ class LocalSeleniumCrawlerStrategy(CrawlerStrategy):
                driver.quit()
            
            # Execute JS code if provided
+            self.js_code = kwargs.get("js_code", self.js_code)
            if self.js_code and type(self.js_code) == str:
                self.driver.execute_script(self.js_code)
                # Optionally, wait for some condition after executing the JS code