parsagon 0.9.5__py3-none-any.whl → 0.9.7__py3-none-any.whl

parsagon/api.py

@@ -149,11 +149,11 @@ def create_custom_function(pipeline_id, call_id, custom_function):
     )
 
 
-def add_examples_to_custom_function(pipeline_id, call_id, custom_function):
+def add_examples_to_custom_function(pipeline_id, call_id, custom_function, remove_old_examples):
     _api_call(
         httpx.post,
         "/transformers/custom-function/add-examples/",
-        json={"pipeline": pipeline_id, "call_id": call_id, **custom_function.to_json()},
+        json={"pipeline": pipeline_id, "call_id": call_id, "remove_old_examples": remove_old_examples, **custom_function.to_json()},
     )
 
 

parsagon/main.py

@@ -103,6 +103,11 @@ def get_args():
         action="store_true",
         help="let Parsagon infer all elements to be scraped",
     )
+    parser_update.add_argument(
+        "--replace",
+        action="store_true",
+        help="remove old example data while updating the program",
+    )
     parser_update.set_defaults(func=update)
 
     # Run
@@ -233,7 +238,7 @@ def create(task=None, program_name=None, headless=False, infer=False, verbose=Fa
     logger.info("Done.")
 
 
-def update(program_name, variables={}, headless=False, infer=False, verbose=False):
+def update(program_name, variables={}, headless=False, infer=False, replace=False, verbose=False):
     pipeline = get_pipeline(program_name)
     abridged_program = pipeline["abridged_sketch"]
     # Make the program runnable
@@ -245,6 +250,14 @@ def update(program_name, variables={}, headless=False, infer=False, verbose=Fals
     executor = Executor(headless=headless, infer=infer)
     executor.execute(abridged_program)
 
+    while True:
+        program_name_input = input(f"Type \"{program_name}\" to update this program, or press enter without typing a name to CANCEL: ")
+        if not program_name_input:
+            logger.info("Canceled update.")
+            return
+        if program_name_input == program_name:
+            break
+
     pipeline_id = pipeline["id"]
     try:
         for call_id, custom_function in executor.custom_functions.items():
@@ -254,7 +267,7 @@ def update(program_name, variables={}, headless=False, infer=False, verbose=Fals
             if verbose:
                 description += debug_suffix
             logger.info(f"  Saving function{description}...")
-            add_examples_to_custom_function(pipeline_id, call_id, custom_function)
+            add_examples_to_custom_function(pipeline_id, call_id, custom_function, replace)
         logger.info(f"Saved.")
     except Exception as e:
         print(e)

{parsagon-0.9.5.dist-info → parsagon-0.9.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: parsagon
-Version: 0.9.5
+Version: 0.9.7
 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.9.5.dist-info → parsagon-0.9.7.dist-info}/RECORD

@@ -1,10 +1,10 @@
 parsagon/__init__.py,sha256=YRvTX1ekTKcPXTKXLgr6-CrRC0DFc11L3gFSPI_Ohc8,54
-parsagon/api.py,sha256=gyyltByLwj6jz7Kv4fdlqJOU9E5Lj5AV2ZRNiuKoYPk,6227
+parsagon/api.py,sha256=yy_82ZvZU7ytl9t02lskeMGAoYLUQTBFr6Zo8F-4Vgo,6292
 parsagon/custom_function.py,sha256=oEj28qItaHUnsvLIHD7kg5QL3J3aO6rW6xKKP-H-Drs,770
 parsagon/exceptions.py,sha256=NYpFaSLZplBTv9fov_1LKPzDPIqb7Ffe7IunnjntxvA,819
 parsagon/executor.py,sha256=lkV9KvljNH-9nQwIxZbGJMSbKkU7MzeI6yiGxBQo1C0,15759
 parsagon/highlights.js,sha256=eLzQT7UABVLLinvWfnXruHjCKZK9If3w742VsoMCO_o,14871
-parsagon/main.py,sha256=GaaIaShgWTpbYhjizUydVsNH93wENfiid_E3HbC4C9E,11323
+parsagon/main.py,sha256=vL1-MJPAO5yASFk0B3iAymA8u7dNfUqpLerhQ0PMi5A,11818
 parsagon/settings.py,sha256=s5_MsDMFM5tB8U8tfHaFnKibCoEqPnAu8b_ueg07Ftw,2947
 parsagon/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 parsagon/tests/api_mocks.py,sha256=M8xhiyPa1dI8Vx-odDk7ETopfFAfcjfAf-ApmSqgvfw,3127
@@ -13,8 +13,8 @@ parsagon/tests/conftest.py,sha256=KMlHohc0QT77HzumraIojzKeqroyxarnaT6naJDNvEc,42
 parsagon/tests/test_executor.py,sha256=n3cmh84r74siSeJqUeAIwjjnNzDVPEdxcvYAeJ4hNX8,645
 parsagon/tests/test_invalid_args.py,sha256=kOjMpbZvviR1CwvXReteZMxBvuhq_rOv5Tm1muBSzNk,676
 parsagon/tests/test_pipeline_operations.py,sha256=TpBKCuRA8LHYWx3PD_k9mYCSsA_9SZjrOX-rS4mE8XE,1089
-parsagon-0.9.5.dist-info/METADATA,sha256=vwLRmI2DmO-O34kB8DMGbmHcZ4pqt0dyDw1yWEjI7HE,2260
-parsagon-0.9.5.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-parsagon-0.9.5.dist-info/entry_points.txt,sha256=I1UlPUb4oY2k9idkI8kvdkEcrjKGRSOl5pMbA6uu6kw,48
-parsagon-0.9.5.dist-info/top_level.txt,sha256=ih5uYQzW4qjhRKppys-WiHLIbXVZ99YdqDcfAtlcQwk,9
-parsagon-0.9.5.dist-info/RECORD,,
+parsagon-0.9.7.dist-info/METADATA,sha256=D75z0KUZFsoWY-JoY_FZIpSZxfR_D7VGru0aepBnFhc,2260
+parsagon-0.9.7.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+parsagon-0.9.7.dist-info/entry_points.txt,sha256=I1UlPUb4oY2k9idkI8kvdkEcrjKGRSOl5pMbA6uu6kw,48
+parsagon-0.9.7.dist-info/top_level.txt,sha256=ih5uYQzW4qjhRKppys-WiHLIbXVZ99YdqDcfAtlcQwk,9
+parsagon-0.9.7.dist-info/RECORD,,