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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fleet-python
-Version: 0.2.52
+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.52 → 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.52 → fleet_python-0.2.53}/fleet/_async/client.py

@@ -560,29 +560,34 @@ 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:
+            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
@@ -601,7 +606,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_python-0.2.52 → fleet_python-0.2.53}/fleet/client.py

@@ -566,27 +566,32 @@ 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())
+            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
@@ -602,7 +607,7 @@ class Fleet:
 
         # 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_python-0.2.52 → 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.52
+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.52 → fleet_python-0.2.53}/pyproject.toml

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