fleet-python 0.2.51__tar.gz → 0.2.53__tar.gz

{fleet_python-0.2.51 → fleet_python-0.2.53}/PKG-INFO

@@ -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 → fleet_python-0.2.53}/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_python-0.2.51 → fleet_python-0.2.53}/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_python-0.2.51 → fleet_python-0.2.53}/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 → fleet_python-0.2.53}/fleet_python.egg-info/PKG-INFO

@@ -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 → fleet_python-0.2.53}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "fleet-python"
-version = "0.2.51"
+version = "0.2.53"
 description = "Python SDK for Fleet environments"
 authors = [
     {name = "Fleet AI", email = "nic@fleet.so"},