fleet-python 0.2.52__py3-none-any.whl → 0.2.54__py3-none-any.whl

examples/import_tasks.py

@@ -0,0 +1,16 @@
+import asyncio
+import fleet
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
+async def main():
+    print(f"Importing tasks... {(await fleet.env.account_async()).team_name}")
+    await fleet._async.import_tasks(
+        "6bb6c6b6-36a8-4407-ba66-6908e42069c8.json", project_key="amazon"
+    )
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

fleet/__init__.py

@@ -51,10 +51,17 @@ from ._async.tasks import (
     Task,
     load_tasks as load_tasks_async,
     load_tasks_from_file as load_tasks_from_file_async,
+    import_task as import_task_async,
+    import_tasks as import_tasks_async,
 )
 
-# Import sync load_tasks function
-from .tasks import load_tasks, load_tasks_from_file
+# Import sync task functions
+from .tasks import (
+    load_tasks,
+    load_tasks_from_file,
+    import_task,
+    import_tasks,
+)
 
 # Import shared types
 from .types import VerifierFunction
@@ -104,6 +111,12 @@ __all__ = [
     # Module-level functions (async is default)
     "load_tasks",
     "load_tasks_async",
+    "load_tasks_from_file",
+    "load_tasks_from_file_async",
+    "import_task",
+    "import_task_async",
+    "import_tasks",
+    "import_tasks_async",
     # Version
     "__version__",
 ]

fleet/_async/__init__.py

@@ -86,6 +86,7 @@ __all__ = [
     "load_task_from_string",
     "load_task_from_json",
     "export_tasks",
+    "import_task",
     "import_tasks",
     "account",
     # Version
@@ -193,13 +194,33 @@ async def export_tasks(
     return await _async_global_client.get_client().export_tasks(env_key, filename)
 
 
-async def import_tasks(filename: str):
+async def import_task(task, project_key: Optional[str] = None):
+    """Import a single task.
+
+    Args:
+        task: Task object to import
+        project_key: Optional project key to associate with the task
+
+    Example:
+        task = fleet.Task(key="my-task", prompt="Do something", env_id="my-env")
+        await fleet.import_task(task)
+        await fleet.import_task(task, project_key="my-project")
+    """
+    return await _async_global_client.get_client().import_single_task(task, project_key)
+
+
+async def import_tasks(filename: str, project_key: Optional[str] = None):
     """Import tasks from a JSON file.
 
+    Args:
+        filename: Path to the JSON file of Task objects to import
+        project_key: Optional project key to associate with the tasks
+
     Example:
         await fleet.import_tasks("tasks.json")
+        await fleet.import_tasks("tasks.json", project_key="my-project")
     """
-    return await _async_global_client.get_client().import_tasks(filename)
+    return await _async_global_client.get_client().import_tasks(filename, project_key)
 
 
 async def account() -> AccountResponse:

fleet/_async/client.py

@@ -328,7 +328,9 @@ class AsyncFleet:
         task_json = json.loads(task_string)
         return await self.load_task_from_json(task_json)
 
-    async def load_task_from_json(self, task_json: Dict) -> Task:
+    async def load_task_from_json(
+        self, task_json: Dict, raise_on_verifier_error: bool = False
+    ) -> Task:
         verifier = None
         verifier_code = task_json.get("verifier_func") or task_json.get("verifier_code")
 
@@ -356,9 +358,11 @@ class AsyncFleet:
                     verifier_sha=task_json.get("verifier_sha", ""),
                 )
         except Exception as e:
-            logger.warning(
-                f"Failed to create verifier {task_json.get('key', task_json.get('id'))}: {e}"
-            )
+            error_msg = f"Failed to create verifier {task_json.get('key', task_json.get('id'))}: {e}"
+            if raise_on_verifier_error:
+                raise ValueError(error_msg) from e
+            else:
+                logger.warning(error_msg)
 
         task = Task(
             key=task_json.get("key", task_json.get("id")),
@@ -560,40 +564,71 @@ class AsyncFleet:
             logger.info("No tasks found to export")
             return None
 
-    async def import_single_task(self, task: Task):
+    async def import_single_task(self, task: Task, project_key: Optional[str] = None):
         """Import a single task.
 
         Args:
             task: Task object to import
+            project_key: Optional project key to associate with the task
 
         Returns:
             Response from the API, or None if the import failed
         """
         try:
+            # Validate that verifier_func exists
+            if not task.verifier_func:
+                raise ValueError(
+                    f"Task {task.key} is missing verifier_func. "
+                    "All tasks must have a verifier_func to be imported."
+                )
+
+            params = {}
+            if project_key:
+                params["project_key"] = project_key
             response = await self.client.request(
-                "POST", "/v1/tasks", json=task.model_dump()
+                "POST", "/v1/tasks", json=task.model_dump(), params=params
             )
             return response
         except Exception as e:
             logger.error(f"Failed to import task {task.key}: {e}")
             return None
 
-    async def import_tasks(self, filename: str):
+    async def import_tasks(self, filename: str, project_key: Optional[str] = None):
         """Import tasks from a JSON file.
 
         Args:
             filename: Path to the JSON file of Task objects to import
+            project_key: Optional project key to associate with the tasks
 
         Returns:
             List[Task] containing imported Task objects
+
+        Raises:
+            ValueError: If any task is missing verifier_func or has invalid verifier code
         """
         with open(filename, "r", encoding="utf-8") as f:
             tasks_data = json.load(f)
 
-        # Create tasks from the loaded data
+        # Create tasks from the loaded data using load_task_from_json
+        # This will validate and create verifiers properly
         tasks = []
         for task_data in tasks_data:
-            task = Task(**task_data)
+            # Validate that verifier_func exists
+            verifier_code = task_data.get("verifier_func") or task_data.get(
+                "verifier_code"
+            )
+            if not verifier_code:
+                task_key = task_data.get("key", task_data.get("id", "unknown"))
+                raise ValueError(
+                    f"Task {task_key} is missing verifier_func. "
+                    "All tasks must have a verifier_func to be imported."
+                )
+
+            # Use load_task_from_json to properly create and validate the task
+            # Pass raise_on_verifier_error=True to fail fast on invalid verifier code
+            task = await self.load_task_from_json(
+                task_data, raise_on_verifier_error=True
+            )
             tasks.append(task)
 
         # Use semaphore to limit concurrency to 20
@@ -601,7 +636,7 @@ class AsyncFleet:
 
         async def import_with_semaphore(task):
             async with semaphore:
-                return await self.import_single_task(task)
+                return await self.import_single_task(task, project_key)
 
         # Use asyncio.gather to parallelize the imports
         responses = await asyncio.gather(

fleet/_async/tasks.py

@@ -332,3 +332,44 @@ async def update_task(
     return await client.update_task(
         task_key=task_key, prompt=prompt, verifier_code=verifier_code
     )
+
+
+async def import_task(task: Task, project_key: Optional[str] = None):
+    """Convenience function to import a single task.
+
+    Args:
+        task: Task object to import
+        project_key: Optional project key to associate with the task
+
+    Returns:
+        Response from the API, or None if the import failed
+
+    Examples:
+        task = fleet.Task(key="my-task", prompt="Do something", env_id="my-env")
+        response = await fleet.import_task(task)
+        response = await fleet.import_task(task, project_key="my-project")
+    """
+    from .global_client import get_client
+
+    client = get_client()
+    return await client.import_single_task(task, project_key=project_key)
+
+
+async def import_tasks(filename: str, project_key: Optional[str] = None):
+    """Convenience function to import tasks from a JSON file.
+
+    Args:
+        filename: Path to the JSON file of Task objects to import
+        project_key: Optional project key to associate with the tasks
+
+    Returns:
+        List of responses from the API for successfully imported tasks
+
+    Examples:
+        responses = await fleet.import_tasks("tasks.json")
+        responses = await fleet.import_tasks("tasks.json", project_key="my-project")
+    """
+    from .global_client import get_client
+
+    client = get_client()
+    return await client.import_tasks(filename, project_key=project_key)

fleet/client.py

@@ -326,7 +326,9 @@ class Fleet:
         task_json = json.loads(task_string)
         return self.load_task_from_json(task_json)
 
-    def load_task_from_json(self, task_json: Dict) -> Task:
+    def load_task_from_json(
+        self, task_json: Dict, raise_on_verifier_error: bool = False
+    ) -> Task:
         verifier = None
         verifier_code = task_json.get("verifier_func") or task_json.get("verifier_code")
 
@@ -354,9 +356,11 @@ class Fleet:
                     verifier_sha=task_json.get("verifier_sha", ""),
                 )
         except Exception as e:
-            logger.warning(
-                f"Failed to create verifier {task_json.get('key', task_json.get('id'))}: {e}"
-            )
+            error_msg = f"Failed to create verifier {task_json.get('key', task_json.get('id'))}: {e}"
+            if raise_on_verifier_error:
+                raise ValueError(error_msg) from e
+            else:
+                logger.warning(error_msg)
 
         task = Task(
             key=task_json.get("key", task_json.get("id")),
@@ -566,43 +570,76 @@ class Fleet:
             logger.info("No tasks found to export")
             return None
 
-    def import_single_task(self, task: Task):
+    def import_single_task(self, task: Task, project_key: Optional[str] = None):
         """Import a single task.
 
         Args:
             task: Task object to import
+            project_key: Optional project key to associate with the task
 
         Returns:
             Response from the API, or None if the import failed
         """
         try:
-            response = self.client.request("POST", "/v1/tasks", json=task.model_dump())
+            # Validate that verifier_func exists
+            if not task.verifier_func:
+                raise ValueError(
+                    f"Task {task.key} is missing verifier_func. "
+                    "All tasks must have a verifier_func to be imported."
+                )
+
+            params = {}
+            if project_key:
+                params["project_key"] = project_key
+            response = self.client.request(
+                "POST", "/v1/tasks", json=task.model_dump(), params=params
+            )
             return response
         except Exception as e:
             logger.error(f"Failed to import task {task.key}: {e}")
             return None
 
-    def import_tasks(self, filename: str):
+    def import_tasks(self, filename: str, project_key: Optional[str] = None):
         """Import tasks from a JSON file.
 
         Args:
             filename: Path to the JSON file of Task objects to import
+            project_key: Optional project key to associate with the tasks
 
         Returns:
             List[Task] containing imported Task objects
+
+        Raises:
+            ValueError: If any task is missing verifier_func or has invalid verifier code
         """
         with open(filename, "r", encoding="utf-8") as f:
             tasks_data = json.load(f)
 
-        # Create tasks from the loaded data
+        # Create tasks from the loaded data using load_task_from_json
+        # This will validate and create verifiers properly
         tasks = []
         for task_data in tasks_data:
-            task = Task(**task_data)
+            # Validate that verifier_func exists
+            verifier_code = task_data.get("verifier_func") or task_data.get(
+                "verifier_code"
+            )
+            if not verifier_code:
+                task_key = task_data.get("key", task_data.get("id", "unknown"))
+                raise ValueError(
+                    f"Task {task_key} is missing verifier_func. "
+                    "All tasks must have a verifier_func to be imported."
+                )
+
+            # Use load_task_from_json to properly create and validate the task
+            # Pass raise_on_verifier_error=True to fail fast on invalid verifier code
+            task = self.load_task_from_json(task_data, raise_on_verifier_error=True)
             tasks.append(task)
 
         # Use ThreadPoolExecutor to parallelize the imports with max 20 workers
         with concurrent.futures.ThreadPoolExecutor(max_workers=20) as executor:
-            responses = list(executor.map(self.import_single_task, tasks))
+            responses = list(
+                executor.map(lambda t: self.import_single_task(t, project_key), tasks)
+            )
 
         # Filter out None values (failed imports)
         return [r for r in responses if r is not None]

fleet/tasks.py

@@ -335,3 +335,44 @@ def update_task(
     return client.update_task(
         task_key=task_key, prompt=prompt, verifier_code=verifier_code
     )
+
+
+def import_task(task: Task, project_key: Optional[str] = None):
+    """Convenience function to import a single task.
+
+    Args:
+        task: Task object to import
+        project_key: Optional project key to associate with the task
+
+    Returns:
+        Response from the API, or None if the import failed
+
+    Examples:
+        task = fleet.Task(key="my-task", prompt="Do something", env_id="my-env")
+        response = fleet.import_task(task)
+        response = fleet.import_task(task, project_key="my-project")
+    """
+    from .global_client import get_client
+
+    client = get_client()
+    return client.import_single_task(task, project_key=project_key)
+
+
+def import_tasks(filename: str, project_key: Optional[str] = None):
+    """Convenience function to import tasks from a JSON file.
+
+    Args:
+        filename: Path to the JSON file of Task objects to import
+        project_key: Optional project key to associate with the tasks
+
+    Returns:
+        List of responses from the API for successfully imported tasks
+
+    Examples:
+        responses = fleet.import_tasks("tasks.json")
+        responses = fleet.import_tasks("tasks.json", project_key="my-project")
+    """
+    from .global_client import get_client
+
+    client = get_client()
+    return client.import_tasks(filename, project_key=project_key)

{fleet_python-0.2.52.dist-info → fleet_python-0.2.54.dist-info}/METADATA

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

{fleet_python-0.2.52.dist-info → fleet_python-0.2.54.dist-info}/RECORD

@@ -12,6 +12,7 @@ examples/example_task.py,sha256=dhG6STAkNsTdHs9cO1RFH9WfuvRmq5bRC211hTeFrk8,7088
 examples/example_tasks.py,sha256=xTL8UWVAuolSX6swskfrAcmDrLIzn45dJ7YPWCwoEBU,514
 examples/example_verifier.py,sha256=0vwNITIG3m4CkSPwIxNXcGx9TqrxEsCGqK2A8keKZMM,2392
 examples/gemini_example.py,sha256=qj9WDazQTYNiRHNeUg9Tjkp33lJMwbx8gDfpFe1sDQo,16180
+examples/import_tasks.py,sha256=InZdnnL8KfTy8JkuZPlsgFvdU7h2LrdpQNa2XA9VR88,343
 examples/json_tasks_example.py,sha256=CYPESGGtOo0fmsDdLidujTfsE4QlJHw7rOhyVqPJ_Ls,5329
 examples/nova_act_example.py,sha256=rH23Lp74Okf0rn8ynMdWjK2aviEf5NLPH4k_53Pyxho,831
 examples/openai_example.py,sha256=dEWERrTEP5xBiGkLkQjBQGd2NqoxX6gcW6XteBPsWFQ,8231
@@ -19,22 +20,22 @@ examples/openai_simple_example.py,sha256=HmiufucrAZne7tHq9uoEsDWlEhjNC265bQAyIGB
 examples/query_builder_example.py,sha256=-cOMfWGNifYfYEt_Ds73XpwATZvFDL6F4KTkVxdMjzg,3951
 examples/quickstart.py,sha256=1VT39IRRhemsJgxi0O0gprdpcw7HB4pYO97GAYagIcg,3788
 examples/test_cdp_logging.py,sha256=AkCwQCgOTQEI8w3v0knWK_4eXMph7L9x07wj9yIYM10,2836
-fleet/__init__.py,sha256=Mkdeh45N47lnSv73Eehj92cGU-AImUitvDWJLFhEp0Y,3844
+fleet/__init__.py,sha256=fwIcuaJPPdLMdKPsazfz4EAPeJchl9MBN4nxs0YlOjw,4117
 fleet/base.py,sha256=bc-340sTpq_DJs7yQ9d2pDWnmJFmA1SwDB9Lagvqtb4,9182
-fleet/client.py,sha256=eLo_yie4xUAV2SgeyNIIY-CUhdhp0K5zHr-XEQUw-S8,27444
+fleet/client.py,sha256=qG2nowkeDqitqDhAbYwoGtnXkau3nTwmkUGzN0iPgOk,29201
 fleet/config.py,sha256=uY02ZKxVoXqVDta-0IMWaYJeE1CTXF_fA9NI6QUutmU,319
 fleet/exceptions.py,sha256=fUmPwWhnT8SR97lYsRq0kLHQHKtSh2eJS0VQ2caSzEI,5055
 fleet/global_client.py,sha256=frrDAFNM2ywN0JHLtlm9qbE1dQpnQJsavJpb7xSR_bU,1072
 fleet/models.py,sha256=d9eish0KO3t4VCNu8h8Q_6K1Xj-crYo5Fejtle0Ur28,13056
-fleet/tasks.py,sha256=uR94wq7SG4ymJYLq24F2iYn0el58C2Czb2oK-8rIs8s,12344
+fleet/tasks.py,sha256=uzGKwbLDKlNpuPjUTeF7N2FG_tez3JwiyHYLYt5rBSI,13660
 fleet/types.py,sha256=L4Y82xICf1tzyCLqhLYUgEoaIIS5h9T05TyFNHSWs3s,652
-fleet/_async/__init__.py,sha256=7C_JaEHoqZ4cddsCmlJ4z-UaU6Kr2CBZSgwx5B6fqnc,6765
+fleet/_async/__init__.py,sha256=ruGtdMl9BRQRsJygNszPT89ZbMy6AYv1mmeYoB--r60,7557
 fleet/_async/base.py,sha256=oisVTQsx0M_yTmyQJc3oij63uKZ97MHz-xYFsWXxQE8,9202
-fleet/_async/client.py,sha256=UIF0t-GgLdJeZLJqzsE9gVuzEesj7O8wWPVgjF45tso,27917
+fleet/_async/client.py,sha256=SbateFBfFDRju-gHtFr8TQdR0thBNA0rWt0RuUse9s0,29637
 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
-fleet/_async/tasks.py,sha256=hrtvKlqepQOFnlJv6h5S06UWNWtn2hQlgSp3eYj5m30,12226
+fleet/_async/tasks.py,sha256=t51pK4gBZZOmtJs35wjEorxKIgDeIvPxwCtGRs00Tpg,13590
 fleet/_async/env/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fleet/_async/env/client.py,sha256=C5WG5Ir_McXaFPZNdkQjj0w4V7xMIcw3QyVP5g-3kVk,1237
 fleet/_async/instance/__init__.py,sha256=PtmJq8J8bh0SOQ2V55QURz5GJfobozwtQoqhaOk3_tI,515
@@ -67,10 +68,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.52.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+fleet_python-0.2.54.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.52.dist-info/METADATA,sha256=5XJkEis8IQs2W8ITyyU7a1LmZ3XqzrweVTFHCltP7zw,3304
-fleet_python-0.2.52.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fleet_python-0.2.52.dist-info/top_level.txt,sha256=_3DSmTohvSDf3AIP_BYfGzhwO1ECFwuzg83X-wHCx3Y,23
-fleet_python-0.2.52.dist-info/RECORD,,
+fleet_python-0.2.54.dist-info/METADATA,sha256=tkesOoS66aveH_fYtJvJoYi2ErjrkiApcXbx1bUh1Eo,3304
+fleet_python-0.2.54.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fleet_python-0.2.54.dist-info/top_level.txt,sha256=_3DSmTohvSDf3AIP_BYfGzhwO1ECFwuzg83X-wHCx3Y,23
+fleet_python-0.2.54.dist-info/RECORD,,