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

fleet/_async/client.py

@@ -560,6 +560,24 @@ class AsyncFleet:
             logger.info("No tasks found to export")
             return None
 
+    async def import_single_task(self, task: Task):
+        """Import a single task.
+
+        Args:
+            task: Task object to import
+
+        Returns:
+            Response from the API, or None if the import failed
+        """
+        try:
+            response = await self.client.request(
+                "POST", "/v1/tasks", json=task.model_dump()
+            )
+            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):
         """Import tasks from a JSON file.
 
@@ -578,14 +596,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)
+
+        # 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,6 +566,22 @@ class Fleet:
             logger.info("No tasks found to export")
             return None
 
+    def import_single_task(self, task: Task):
+        """Import a single task.
+
+        Args:
+            task: Task object to import
+
+        Returns:
+            Response from the API, or None if the import failed
+        """
+        try:
+            response = self.client.request("POST", "/v1/tasks", json=task.model_dump())
+            return response
+        except Exception as e:
+            logger.error(f"Failed to import task {task.key}: {e}")
+            return None
+
     def import_tasks(self, filename: str):
         """Import tasks from a JSON file.
 
@@ -584,19 +600,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(self.import_single_task, 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.52.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fleet-python
-Version: 0.2.51
+Version: 0.2.52
 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.52.dist-info}/RECORD

@@ -21,7 +21,7 @@ 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=eLo_yie4xUAV2SgeyNIIY-CUhdhp0K5zHr-XEQUw-S8,27444
 fleet/config.py,sha256=uY02ZKxVoXqVDta-0IMWaYJeE1CTXF_fA9NI6QUutmU,319
 fleet/exceptions.py,sha256=fUmPwWhnT8SR97lYsRq0kLHQHKtSh2eJS0VQ2caSzEI,5055
 fleet/global_client.py,sha256=frrDAFNM2ywN0JHLtlm9qbE1dQpnQJsavJpb7xSR_bU,1072
@@ -30,7 +30,7 @@ fleet/tasks.py,sha256=uR94wq7SG4ymJYLq24F2iYn0el58C2Czb2oK-8rIs8s,12344
 fleet/types.py,sha256=L4Y82xICf1tzyCLqhLYUgEoaIIS5h9T05TyFNHSWs3s,652
 fleet/_async/__init__.py,sha256=7C_JaEHoqZ4cddsCmlJ4z-UaU6Kr2CBZSgwx5B6fqnc,6765
 fleet/_async/base.py,sha256=oisVTQsx0M_yTmyQJc3oij63uKZ97MHz-xYFsWXxQE8,9202
-fleet/_async/client.py,sha256=aGbfT9PYsGcZFLlEV26-1GaPhU9WTQ8nCBNnqAmPiG4,27196
+fleet/_async/client.py,sha256=UIF0t-GgLdJeZLJqzsE9gVuzEesj7O8wWPVgjF45tso,27917
 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.52.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.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,,