fleet-python 0.2.51__py3-none-any.whl → 0.2.53__py3-none-any.whl

fleet/_async/__init__.py

@@ -193,13 +193,18 @@ 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_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

@@ -560,11 +560,34 @@ class AsyncFleet:
             logger.info("No tasks found to export")
             return None
 
-    async def import_tasks(self, filename: str):
+    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:
+            params = {}
+            if project_key:
+                params["project_key"] = project_key
+            response = await 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
+
+    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
@@ -578,14 +601,20 @@ class AsyncFleet:
             task = Task(**task_data)
             tasks.append(task)
 
-        for task in tasks:
-            try:
-                response = await self.client.request(
-                    "POST", "/v1/tasks", json=task.model_dump()
-                )
-            except Exception as e:
-                logger.error(f"Failed to import task {task.key}: {e}")
-                continue
+        # Use semaphore to limit concurrency to 20
+        semaphore = asyncio.Semaphore(20)
+
+        async def import_with_semaphore(task):
+            async with semaphore:
+                return await self.import_single_task(task, project_key)
+
+        # Use asyncio.gather to parallelize the imports
+        responses = await asyncio.gather(
+            *[import_with_semaphore(task) for task in tasks]
+        )
+
+        # Filter out None values (failed imports)
+        return [r for r in responses if r is not None]
 
     async def account(self) -> AccountResponse:
         """Get account information including instance limits and usage.

fleet/client.py

@@ -566,11 +566,32 @@ class Fleet:
             logger.info("No tasks found to export")
             return None
 
-    def import_tasks(self, filename: str):
+    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:
+            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, 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
@@ -584,19 +605,12 @@ class Fleet:
             task = Task(**task_data)
             tasks.append(task)
 
-        responses = []
-
-        for task in tasks:
-            try:
-                response = self.client.request(
-                    "POST", "/v1/tasks", json=task.model_dump()
-                )
-                responses.append(response)
-            except Exception as e:
-                logger.error(f"Failed to import task {task.key}: {e}")
-                continue
+        # Use ThreadPoolExecutor to parallelize the imports with max 20 workers
+        with concurrent.futures.ThreadPoolExecutor(max_workers=20) as executor:
+            responses = list(executor.map(lambda t: self.import_single_task(t, project_key), tasks))
 
-        return responses
+        # Filter out None values (failed imports)
+        return [r for r in responses if r is not None]
 
     def account(self) -> AccountResponse:
         """Get account information including instance limits and usage.

{fleet_python-0.2.51.dist-info → fleet_python-0.2.53.dist-info}/METADATA

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

{fleet_python-0.2.51.dist-info → fleet_python-0.2.53.dist-info}/RECORD

@@ -21,16 +21,16 @@ examples/quickstart.py,sha256=1VT39IRRhemsJgxi0O0gprdpcw7HB4pYO97GAYagIcg,3788
 examples/test_cdp_logging.py,sha256=AkCwQCgOTQEI8w3v0knWK_4eXMph7L9x07wj9yIYM10,2836
 fleet/__init__.py,sha256=Mkdeh45N47lnSv73Eehj92cGU-AImUitvDWJLFhEp0Y,3844
 fleet/base.py,sha256=bc-340sTpq_DJs7yQ9d2pDWnmJFmA1SwDB9Lagvqtb4,9182
-fleet/client.py,sha256=xuo9aqX9bTMzVtk_2I6hS7-9p9WPENIWpQAj76J7dXE,27021
+fleet/client.py,sha256=zu05ApXhJrKX7UiMF8TOY1yNWiHRGlYRAMBmhaCsOSo,27806
 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/types.py,sha256=L4Y82xICf1tzyCLqhLYUgEoaIIS5h9T05TyFNHSWs3s,652
-fleet/_async/__init__.py,sha256=7C_JaEHoqZ4cddsCmlJ4z-UaU6Kr2CBZSgwx5B6fqnc,6765
+fleet/_async/__init__.py,sha256=x2esWn8yf2AMfzHRmSHtFkXcxbrSSXCZdJyrFnJldEo,7033
 fleet/_async/base.py,sha256=oisVTQsx0M_yTmyQJc3oij63uKZ97MHz-xYFsWXxQE8,9202
-fleet/_async/client.py,sha256=aGbfT9PYsGcZFLlEV26-1GaPhU9WTQ8nCBNnqAmPiG4,27196
+fleet/_async/client.py,sha256=TZJsEg43IiYPnSA3S48GRpwAYg1fehcHVbYHA3yUHXY,28266
 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
@@ -67,10 +67,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.51.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+fleet_python-0.2.53.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.51.dist-info/METADATA,sha256=MODNsxRtnsaN4On2g4xQU23B2hwHfIsPPhD213guVrQ,3304
-fleet_python-0.2.51.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fleet_python-0.2.51.dist-info/top_level.txt,sha256=_3DSmTohvSDf3AIP_BYfGzhwO1ECFwuzg83X-wHCx3Y,23
-fleet_python-0.2.51.dist-info/RECORD,,
+fleet_python-0.2.53.dist-info/METADATA,sha256=SaC_PN_pQ1GG1g3RdaitFUy0PJWpD-1-hCIOT4sqcn8,3304
+fleet_python-0.2.53.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fleet_python-0.2.53.dist-info/top_level.txt,sha256=_3DSmTohvSDf3AIP_BYfGzhwO1ECFwuzg83X-wHCx3Y,23
+fleet_python-0.2.53.dist-info/RECORD,,