parsagon 0.14.15__tar.gz → 0.14.17__tar.gz

{parsagon-0.14.15 → parsagon-0.14.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: parsagon
-Version: 0.14.15
+Version: 0.14.17
 Summary: Allows you to create browser automations with natural language
 Author-email: Sandy Suh <sandy@parsagon.io>
 Project-URL: Homepage, https://parsagon.io

{parsagon-0.14.15 → parsagon-0.14.17}/pyproject.toml

@@ -16,7 +16,7 @@ line-length = 120
 
 [project]
 name = "parsagon"
-version = "0.14.15"
+version = "0.14.17"
 description = "Allows you to create browser automations with natural language"
 readme = "README.md"
 requires-python = ">=3.8"

{parsagon-0.14.15 → parsagon-0.14.17}/src/parsagon/__init__.py

@@ -3,3 +3,4 @@ from parsagon.create import create_program as create
 from parsagon.runs import run, batch_runs
 from parsagon.api import get_str_about_data, get_bool_about_data, get_json_about_data
 from parsagon.assistant import get_page_html, get_page_text
+from parsagon.executor import Executor

{parsagon-0.14.15 → parsagon-0.14.17}/src/parsagon/assistant.py

@@ -65,11 +65,11 @@ def assist(verbose=False):
             break
 
 
-def get_page_html(url, headless=False, use_uc=False, timeout=None):
+def get_page_html(url, headless=False, use_uc=False, page_load_timeout=None, script_timeout=None):
     browser_print(f"Checking what {url} looks like...")
-    executor = Executor("", headless=headless, use_uc=use_uc)
-    if timeout:
-        executor.driver.set_page_load_timeout(timeout)
+    executor = Executor(
+        "", headless=headless, use_uc=use_uc, page_load_timeout=page_load_timeout, script_timeout=script_timeout
+    )
     try:
         executor.goto(url)
         html = executor.get_visible_html()
@@ -78,11 +78,11 @@ def get_page_html(url, headless=False, use_uc=False, timeout=None):
     return html
 
 
-def get_page_text(url, headless=False, use_uc=False, timeout=None):
+def get_page_text(url, headless=False, use_uc=False, page_load_timeout=None, script_timeout=None):
     browser_print(f"Checking what {url} looks like...")
-    executor = Executor("", headless=headless, use_uc=use_uc)
-    if timeout:
-        executor.driver.set_page_load_timeout(timeout)
+    executor = Executor(
+        "", headless=headless, use_uc=use_uc, page_load_timeout=page_load_timeout, script_timeout=script_timeout
+    )
     try:
         executor.goto(url)
         text = executor.driver.execute_script("return document.body.innerText;")

{parsagon-0.14.15 → parsagon-0.14.17}/src/parsagon/executor.py

@@ -64,7 +64,16 @@ class Executor:
     Executes code produced by GPT with the proper context.  Records custom_function usage along the way.
     """
 
-    def __init__(self, task, headless=False, infer=False, use_uc=False, function_bank={}):
+    def __init__(
+        self,
+        task,
+        headless=False,
+        infer=False,
+        use_uc=False,
+        page_load_timeout=None,
+        script_timeout=None,
+        function_bank={},
+    ):
         self.task = task
         self.headless = headless
         self.function_bank = function_bank
@@ -79,6 +88,10 @@ class Executor:
             chrome_options = webdriver.ChromeOptions()
             chrome_options.add_argument("--start-maximized")
             self.driver = webdriver.Chrome(service=ChromeService(driver_executable_path), options=chrome_options)
+        if page_load_timeout:
+            self.driver.set_page_load_timeout(page_load_timeout)
+        if script_timeout:
+            self.driver.set_script_timeout(script_timeout)
         self.max_elem_ids = defaultdict(int)
         self.execution_context = {
             "custom_assert": self.custom_assert,

{parsagon-0.14.15 → parsagon-0.14.17}/src/parsagon.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: parsagon
-Version: 0.14.15
+Version: 0.14.17
 Summary: Allows you to create browser automations with natural language
 Author-email: Sandy Suh <sandy@parsagon.io>
 Project-URL: Homepage, https://parsagon.io