parsagon 0.14.31__py3-none-any.whl → 0.14.32__py3-none-any.whl

parsagon/edit.py

@@ -1,4 +1,10 @@
-from parsagon.api import edit_program_sketch, get_pipeline, get_pipeline_code_parts, update_pipeline
+from parsagon.api import (
+    edit_program_sketch,
+    get_pipeline,
+    get_pipeline_code_parts,
+    update_pipeline,
+    add_examples_to_custom_function,
+)
 from parsagon.executor import Executor, custom_functions_to_descriptions
 from parsagon.print import assistant_print
 from rich.prompt import Confirm, Prompt
@@ -7,7 +13,10 @@ from rich.prompt import Confirm, Prompt
 def edit_program(task, program_name, variables={}, verbose=False):
     pipeline = get_pipeline(program_name)
     pipeline_code_parts = get_pipeline_code_parts(pipeline["id"])
-    custom_functions = {custom_function["call_id"]: custom_function["code"] for custom_function in pipeline_code_parts["custom_functions"]}
+    custom_functions = {
+        custom_function["call_id"]: custom_function["code"]
+        for custom_function in pipeline_code_parts["custom_functions"]
+    }
     program_sketches = edit_program_sketch(pipeline_code_parts["program_sketch"], task)
     full_program = program_sketches["full"]
     abridged_program = program_sketches["abridged"]
@@ -28,7 +37,7 @@ def edit_program(task, program_name, variables={}, verbose=False):
 
     program_name_input = Prompt.ask(
         f'Type "{program_name}" to update this program, or press enter without typing a name to CANCEL',
-        choices=[program_name, ""]
+        choices=[program_name, ""],
     )
     if not program_name_input:
         assistant_print("Discarded edits.")
@@ -45,10 +54,12 @@ def edit_program(task, program_name, variables={}, verbose=False):
             assistant_print(f"  Saving function{description}...")
             add_examples_to_custom_function(pipeline_id, call_id, custom_function, True)
         assistant_print(f"Finalizing program...")
-        update_pipeline(pipeline_id, {"program_sketch": full_program, "abridged_sketch": abridged_program, "pseudocode": pseudocode})
+        update_pipeline(
+            pipeline_id, {"program_sketch": full_program, "abridged_sketch": abridged_program, "pseudocode": pseudocode}
+        )
         assistant_print(f"Saved.")
     except Exception as e:
-        error_message = "An error occurred while saving the program. The program was not updated."
+        error_message = f"An error occurred while saving the program. The program was not updated. {repr(e)}"
         assistant_print(error_message)
         return {"success": False, "outcome": error_message}
 

parsagon/executor.py

@@ -168,8 +168,14 @@ class Executor:
         code = self.function_bank[call_id]
         args_dict["call_id"] = call_id
         args_str = ", ".join(f"{k}={repr(v)}" for k, v in args_dict.items())
-        code = f"{code}\noutput = {name}{call_id}({args_str})"
-        context = {"driver": self.driver, "PARSAGON_API_KEY": settings.get_api_key()}
+        code = f"import time\nimport lxml.html\nfrom urllib.parse import urljoin\nfrom selenium.webdriver.common.by import By\n{code}\noutput = {name}{call_id}({args_str})"
+        context = {
+            "driver": self.driver,
+            "PARSAGON_API_KEY": settings.get_api_key(),
+            "parsagon_elem_data": {"idx": self.max_elem_ids[self.driver.current_window_handle]},
+            "parsagon_log": [],
+            "parsagon_warnings": [],
+        }
         exec(code, context, context)
         return context["output"]
 

{parsagon-0.14.31.dist-info → parsagon-0.14.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: parsagon
-Version: 0.14.31
+Version: 0.14.32
 Summary: Allows you to create browser automations with natural language
 Author-email: Sandy Suh <sandy@parsagon.io>
 Project-URL: Homepage, https://parsagon.io
@@ -8,23 +8,23 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Requires-Dist: selenium ==4.9.1
-Requires-Dist: lxml ==4.9.2
-Requires-Dist: httpx ==0.27.0
-Requires-Dist: psutil ==5.9.6
-Requires-Dist: rich ==13.6.0
-Requires-Dist: PyVirtualDisplay ==3.0
-Requires-Dist: selenium-wire ==5.1.0
-Requires-Dist: cssselect ==1.2.0
-Requires-Dist: undetected-chromedriver ==3.5.5
-Requires-Dist: webdriver-manager ==4.0.1
-Requires-Dist: jsonpath-ng ==1.5.3
-Requires-Dist: simplejson ==3.19.1
-Requires-Dist: pypdf ==4.2.0
+Requires-Dist: selenium==4.9.1
+Requires-Dist: lxml==4.9.2
+Requires-Dist: httpx==0.27.0
+Requires-Dist: psutil==5.9.6
+Requires-Dist: rich==13.6.0
+Requires-Dist: PyVirtualDisplay==3.0
+Requires-Dist: selenium-wire==5.1.0
+Requires-Dist: cssselect==1.2.0
+Requires-Dist: undetected-chromedriver==3.5.5
+Requires-Dist: webdriver-manager==4.0.1
+Requires-Dist: jsonpath-ng==1.5.3
+Requires-Dist: simplejson==3.19.1
+Requires-Dist: pypdf==4.2.0
 Provides-Extra: dev
-Requires-Dist: pytest ==7.3.2 ; extra == 'dev'
-Requires-Dist: pytest-mock ==3.11.1 ; extra == 'dev'
-Requires-Dist: pre-commit ==3.6.0 ; extra == 'dev'
+Requires-Dist: pytest==7.3.2; extra == "dev"
+Requires-Dist: pytest-mock==3.11.1; extra == "dev"
+Requires-Dist: pre-commit==3.6.0; extra == "dev"
 
 # parsagon
 

{parsagon-0.14.31.dist-info → parsagon-0.14.32.dist-info}/RECORD

@@ -3,9 +3,9 @@ parsagon/api.py,sha256=UuHPK1HsKF0pvqUeBGlv2tDHHK5-i0PP9Di3bolcr2g,10383
 parsagon/assistant.py,sha256=V3NL6UdDqe74W_X3wPQ1qwFuJRvHzitXtOt-XSCXvds,4065
 parsagon/create.py,sha256=igrJN8jVP5p3P9-dPSabrHGOQOlG2Flx7ep4SqEDB14,4373
 parsagon/custom_function.py,sha256=oEj28qItaHUnsvLIHD7kg5QL3J3aO6rW6xKKP-H-Drs,770
-parsagon/edit.py,sha256=aLeAcp1hFHm29qh6do21cfYykK76CsHC4tc9Uh2_2bk,3006
+parsagon/edit.py,sha256=5gtnx0gNB7Gvz8ET00SczE-ZS0TomN1um6uObP-OObE,3120
 parsagon/exceptions.py,sha256=tG1vnpmUN1GdJ1GSpe1MaWH3zWmFLZCwtOfEGu8qPP0,910
-parsagon/executor.py,sha256=ATdRivS1I4NaEwjlxKyHmTNfQp6YQQHLi6nhS3n9o9Q,27475
+parsagon/executor.py,sha256=GoY87mkZGl7feab0qF3ja0BRIlDTvGlDktyCytb3Rj8,27786
 parsagon/gui_entry.py,sha256=bqG9K0CArXWWwDGoT8aV17YLNM8MfjSf6SJ_B3QbNeA,671
 parsagon/highlights.js,sha256=2UDfUApblU9xtGgTLCq4X7rHRV0wcqDSSFZPmJS6fJg,16643
 parsagon/main.py,sha256=WPxrT1ZyPe6wNSFP7GnnKIGisgPM2dFieHP2TC5TGVQ,9467
@@ -25,8 +25,8 @@ parsagon/tests/test_invalid_args.py,sha256=TAFdHGy92lUxjljPrtODOuEGVss6rn-F5GvEK
 parsagon/tests/test_pipeline_operations.py,sha256=aEwZNtIwOl9X7jdLDLB4YEdgMp7_x8PXCINAE7RT4NY,805
 parsagon/tests/test_print.py,sha256=BG7f55YDBoL0S7k291-so_Gje_hUAQOkB-jh-bEYsJY,198
 parsagon/tests/test_secrets.py,sha256=Ctsscl2tmMTZcFAy5dnyqUlgTov2UharZgLpbRCLdEg,2662
-parsagon-0.14.31.dist-info/METADATA,sha256=xwcrP85QhuvOmL1PZyZZFnDRN6Gpe0jls0btFnxaLLM,2461
-parsagon-0.14.31.dist-info/WHEEL,sha256=Wyh-_nZ0DJYolHNn1_hMa4lM7uDedD_RGVwbmTjyItk,91
-parsagon-0.14.31.dist-info/entry_points.txt,sha256=I1UlPUb4oY2k9idkI8kvdkEcrjKGRSOl5pMbA6uu6kw,48
-parsagon-0.14.31.dist-info/top_level.txt,sha256=ih5uYQzW4qjhRKppys-WiHLIbXVZ99YdqDcfAtlcQwk,9
-parsagon-0.14.31.dist-info/RECORD,,
+parsagon-0.14.32.dist-info/METADATA,sha256=bjuVJitevEmIBWOvrCptqaVhVs7QUu1mCt0aPWAKNN4,2442
+parsagon-0.14.32.dist-info/WHEEL,sha256=Mdi9PDNwEZptOjTlUcAth7XJDFtKrHYaQMPulZeBCiQ,91
+parsagon-0.14.32.dist-info/entry_points.txt,sha256=I1UlPUb4oY2k9idkI8kvdkEcrjKGRSOl5pMbA6uu6kw,48
+parsagon-0.14.32.dist-info/top_level.txt,sha256=ih5uYQzW4qjhRKppys-WiHLIbXVZ99YdqDcfAtlcQwk,9
+parsagon-0.14.32.dist-info/RECORD,,

{parsagon-0.14.31.dist-info → parsagon-0.14.32.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (71.1.0)
+Generator: setuptools (73.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any