fleet-python 0.2.56__py3-none-any.whl → 0.2.58__py3-none-any.whl

examples/export_tasks.py

@@ -14,6 +14,13 @@ def main():
         help="Optional project key to filter tasks",
         default=None,
     )
+    parser.add_argument(
+        "--task-keys",
+        "-t",
+        nargs="+",
+        help="Optional list of task keys to export (space-separated)",
+        default=None,
+    )
     parser.add_argument(
         "--output",
         "-o",
@@ -23,6 +30,12 @@ def main():
 
     args = parser.parse_args()
 
+    # Validate that only one filter is specified
+    if args.project_key and args.task_keys:
+        parser.error(
+            "Cannot specify both --project-key and --task-keys. Use one or neither."
+        )
+
     # Get account info
     account = fleet.env.account()
     print(f"Exporting from team: {account.team_name}")
@@ -31,6 +44,11 @@ def main():
     if args.project_key:
         print(f"Loading tasks from project: {args.project_key}")
         tasks = fleet.load_tasks(project_key=args.project_key)
+    elif args.task_keys:
+        print(
+            f"Loading {len(args.task_keys)} specific task(s): {', '.join(args.task_keys)}"
+        )
+        tasks = fleet.load_tasks(keys=args.task_keys)
     else:
         print("Loading all tasks")
         tasks = fleet.load_tasks()

fleet/_async/client.py

@@ -333,9 +333,26 @@ class AsyncFleet:
     ) -> Task:
         verifier = None
         verifier_code = task_json.get("verifier_func") or task_json.get("verifier_code")
+        verifier_sha = task_json.get("verifier_sha", "")
+
+        # Check if verifier is a nested object with code inside
+        if not verifier_code and "verifier" in task_json:
+            verifier_obj = task_json["verifier"]
+            if isinstance(verifier_obj, dict):
+                verifier_code = verifier_obj.get("code")
+                # Also extract sha256 from nested verifier if not found at top level
+                if not verifier_sha:
+                    verifier_sha = verifier_obj.get("sha256", "")
 
         # Try to find verifier_id in multiple locations
         verifier_id = task_json.get("verifier_id")
+        
+        # Check nested verifier object for verifier_id
+        if not verifier_id and "verifier" in task_json:
+            verifier_obj = task_json["verifier"]
+            if isinstance(verifier_obj, dict):
+                verifier_id = verifier_obj.get("verifier_id")
+        
         if (
             not verifier_id
             and "metadata" in task_json
@@ -355,7 +372,7 @@ class AsyncFleet:
                     verifier_id=verifier_id,
                     verifier_key=task_json.get("key", task_json.get("id")),
                     verifier_code=verifier_code,
-                    verifier_sha=task_json.get("verifier_sha", ""),
+                    verifier_sha=verifier_sha,
                 )
         except Exception as e:
             error_msg = f"Failed to create verifier {task_json.get('key', task_json.get('id'))}: {e}"
@@ -377,6 +394,8 @@ class AsyncFleet:
             env_variables=task_json.get("env_variables", {}),
             verifier_func=verifier_code,  # Set verifier code
             verifier=verifier,  # Use created verifier or None
+            verifier_id=verifier_id,  # Set verifier_id so _rebuild_verifier works
+            verifier_sha=verifier_sha,  # Set verifier_sha
             metadata=task_json.get("metadata", {}),  # Default empty metadata
         )
         return task

fleet/client.py

@@ -331,9 +331,26 @@ class Fleet:
     ) -> Task:
         verifier = None
         verifier_code = task_json.get("verifier_func") or task_json.get("verifier_code")
+        verifier_sha = task_json.get("verifier_sha", "")
+
+        # Check if verifier is a nested object with code inside
+        if not verifier_code and "verifier" in task_json:
+            verifier_obj = task_json["verifier"]
+            if isinstance(verifier_obj, dict):
+                verifier_code = verifier_obj.get("code")
+                # Also extract sha256 from nested verifier if not found at top level
+                if not verifier_sha:
+                    verifier_sha = verifier_obj.get("sha256", "")
 
         # Try to find verifier_id in multiple locations
         verifier_id = task_json.get("verifier_id")
+        
+        # Check nested verifier object for verifier_id
+        if not verifier_id and "verifier" in task_json:
+            verifier_obj = task_json["verifier"]
+            if isinstance(verifier_obj, dict):
+                verifier_id = verifier_obj.get("verifier_id")
+        
         if (
             not verifier_id
             and "metadata" in task_json
@@ -353,7 +370,7 @@ class Fleet:
                     verifier_id=verifier_id,
                     verifier_key=task_json.get("key", task_json.get("id")),
                     verifier_code=verifier_code,
-                    verifier_sha=task_json.get("verifier_sha", ""),
+                    verifier_sha=verifier_sha,
                 )
         except Exception as e:
             error_msg = f"Failed to create verifier {task_json.get('key', task_json.get('id'))}: {e}"
@@ -375,6 +392,8 @@ class Fleet:
             env_variables=task_json.get("env_variables", {}),
             verifier_func=verifier_code,  # Set verifier code
             verifier=verifier,  # Use created verifier or None
+            verifier_id=verifier_id,  # Set verifier_id so _rebuild_verifier works
+            verifier_sha=verifier_sha,  # Set verifier_sha
             metadata=task_json.get("metadata", {}),  # Default empty metadata
         )
         return task

{fleet_python-0.2.56.dist-info → fleet_python-0.2.58.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fleet-python
-Version: 0.2.56
+Version: 0.2.58
 Summary: Python SDK for Fleet environments
 Author-email: Fleet AI <nic@fleet.so>
 License: Apache-2.0

{fleet_python-0.2.56.dist-info → fleet_python-0.2.58.dist-info}/RECORD

@@ -11,7 +11,7 @@ examples/example_sync.py,sha256=EkuWmUzB1ZsBJQk6ZRflB793rKsuRHeSg5HJZHVhBB0,975
 examples/example_task.py,sha256=dhG6STAkNsTdHs9cO1RFH9WfuvRmq5bRC211hTeFrk8,7088
 examples/example_tasks.py,sha256=xTL8UWVAuolSX6swskfrAcmDrLIzn45dJ7YPWCwoEBU,514
 examples/example_verifier.py,sha256=0vwNITIG3m4CkSPwIxNXcGx9TqrxEsCGqK2A8keKZMM,2392
-examples/export_tasks.py,sha256=BSRmsGIIk34QYanB8rPlaFbmn3MBxaf7TyFVrbsdPeU,2114
+examples/export_tasks.py,sha256=e5CaHga_1IV2W7PGPjQjOZtMrLAhoPNt74V6qRY99EE,2700
 examples/gemini_example.py,sha256=qj9WDazQTYNiRHNeUg9Tjkp33lJMwbx8gDfpFe1sDQo,16180
 examples/import_tasks.py,sha256=pb60dOXnOn-G1INQxV1wujeHmmY2oWz9Ddj9SbMm1pk,3139
 examples/json_tasks_example.py,sha256=CYPESGGtOo0fmsDdLidujTfsE4QlJHw7rOhyVqPJ_Ls,5329
@@ -23,7 +23,7 @@ examples/quickstart.py,sha256=1VT39IRRhemsJgxi0O0gprdpcw7HB4pYO97GAYagIcg,3788
 examples/test_cdp_logging.py,sha256=AkCwQCgOTQEI8w3v0knWK_4eXMph7L9x07wj9yIYM10,2836
 fleet/__init__.py,sha256=fwIcuaJPPdLMdKPsazfz4EAPeJchl9MBN4nxs0YlOjw,4117
 fleet/base.py,sha256=bc-340sTpq_DJs7yQ9d2pDWnmJFmA1SwDB9Lagvqtb4,9182
-fleet/client.py,sha256=0RwFXBdiBSK7ziTrUe2uzhHX14MxuboIQhMCoRMC300,30315
+fleet/client.py,sha256=gvJYF1Y3hG4DHM0PPuCHVbID-aSVK_oIT84aAY8lhQU,31244
 fleet/config.py,sha256=uY02ZKxVoXqVDta-0IMWaYJeE1CTXF_fA9NI6QUutmU,319
 fleet/exceptions.py,sha256=fUmPwWhnT8SR97lYsRq0kLHQHKtSh2eJS0VQ2caSzEI,5055
 fleet/global_client.py,sha256=frrDAFNM2ywN0JHLtlm9qbE1dQpnQJsavJpb7xSR_bU,1072
@@ -32,7 +32,7 @@ fleet/tasks.py,sha256=PqOlJBM5b-qwto_nrOXzLvfLJ7B4fGq05Eg-0kDNFv8,15681
 fleet/types.py,sha256=L4Y82xICf1tzyCLqhLYUgEoaIIS5h9T05TyFNHSWs3s,652
 fleet/_async/__init__.py,sha256=ruGtdMl9BRQRsJygNszPT89ZbMy6AYv1mmeYoB--r60,7557
 fleet/_async/base.py,sha256=oisVTQsx0M_yTmyQJc3oij63uKZ97MHz-xYFsWXxQE8,9202
-fleet/_async/client.py,sha256=uiYufTUAG3o_nseflp_DxW8yLe8JnprIxCwjZ4CsqoI,30751
+fleet/_async/client.py,sha256=NxYb4jVBLO3f3eL_1x7z-nkZusFHllFQdgUunUfCwSw,31680
 fleet/_async/exceptions.py,sha256=fUmPwWhnT8SR97lYsRq0kLHQHKtSh2eJS0VQ2caSzEI,5055
 fleet/_async/global_client.py,sha256=4WskpLHbsDEgWW7hXMD09W-brkp4euy8w2ZJ88594rQ,1103
 fleet/_async/models.py,sha256=GX-sRciZDenW2O7Qx9w_ftOkJyE4ph1-92WMq6lynHE,12856
@@ -69,10 +69,10 @@ fleet/verifiers/decorator.py,sha256=nAP3O8szXu7md_kpwpz91hGSUNEVLYjwZQZTkQlV1DM,
 fleet/verifiers/parse.py,sha256=qz9AfJrTbjlg-LU-lE8Ciqi7Yt2a8-cs17FdpjTLhMk,8550
 fleet/verifiers/sql_differ.py,sha256=TqTLWyK3uOyLbitT6HYzYEzuSFC39wcyhgk3rcm__k8,6525
 fleet/verifiers/verifier.py,sha256=_lcxXVm8e0xRrK2gNJy9up7pW1zOkPRY5n5lQ85S8jg,14197
-fleet_python-0.2.56.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+fleet_python-0.2.58.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
 scripts/fix_sync_imports.py,sha256=X9fWLTpiPGkSHsjyQUDepOJkxOqw1DPj7nd8wFlFqLQ,8368
 scripts/unasync.py,sha256=vWVQxRWX8SRZO5cmzEhpvnG_REhCWXpidIGIpWmEcvI,696
-fleet_python-0.2.56.dist-info/METADATA,sha256=i9F_dXu80Okx9Eic7EEMqaDVy_7L7cZNksRcC1KTCP4,3304
-fleet_python-0.2.56.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fleet_python-0.2.56.dist-info/top_level.txt,sha256=_3DSmTohvSDf3AIP_BYfGzhwO1ECFwuzg83X-wHCx3Y,23
-fleet_python-0.2.56.dist-info/RECORD,,
+fleet_python-0.2.58.dist-info/METADATA,sha256=wMsd79uNixxkzRCfBNmI2xdVsEz4TeHA2ZFnl_wwos4,3304
+fleet_python-0.2.58.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fleet_python-0.2.58.dist-info/top_level.txt,sha256=_3DSmTohvSDf3AIP_BYfGzhwO1ECFwuzg83X-wHCx3Y,23
+fleet_python-0.2.58.dist-info/RECORD,,