fleet-python 0.2.41__py3-none-any.whl → 0.2.42__py3-none-any.whl

fleet/_async/client.py

@@ -32,6 +32,8 @@ from ..models import (
     TaskListResponse,
     AccountResponse,
     TaskRequest,
+    TaskResponse,
+    TaskUpdateRequest,
 )
 from .tasks import Task
 
@@ -501,6 +503,31 @@ class AsyncFleet:
         response = await self.client.request("GET", "/v1/account")
         return AccountResponse(**response.json())
 
+    async def update_task(
+        self,
+        task_key: str,
+        prompt: Optional[str] = None,
+        verifier_code: Optional[str] = None
+    ) -> TaskResponse:
+        """Update an existing task.
+
+        Args:
+            task_key: The key of the task to update
+            prompt: New prompt text for the task (optional)
+            verifier_code: Python code for task verification (optional)
+
+        Returns:
+            TaskResponse containing the updated task details
+        """
+        payload = TaskUpdateRequest(
+            prompt=prompt,
+            verifier_code=verifier_code
+        )
+        response = await self.client.request(
+            "PUT", f"/v1/tasks/{task_key}", json=payload.model_dump(exclude_none=True)
+        )
+        return TaskResponse(**response.json())
+
     async def _create_verifier_from_data(
         self, verifier_id: str, verifier_key: str, verifier_code: str, verifier_sha: str
     ) -> "AsyncVerifierFunction":

fleet/_async/models.py

@@ -156,6 +156,11 @@ class TaskRequest(BaseModel):
     env_variables: Optional[Dict[str, Any]] = Field(None, title="Env Variables")
 
 
+class TaskUpdateRequest(BaseModel):
+    prompt: Optional[str] = Field(None, title="Prompt")
+    verifier_code: Optional[str] = Field(None, title="Verifier Code")
+
+
 class VerifierData(BaseModel):
     verifier_id: str = Field(..., title="Verifier Id")
     key: str = Field(..., title="Key")

fleet/_async/tasks.py

@@ -201,8 +201,38 @@ async def load_tasks(
 
     client = get_client()
     return await client.load_tasks(
-        env_key=env_key, 
-        keys=keys, 
-        version=version, 
+        env_key=env_key,
+        keys=keys,
+        version=version,
         team_id=team_id
     )
+
+
+async def update_task(
+    task_key: str,
+    prompt: Optional[str] = None,
+    verifier_code: Optional[str] = None
+):
+    """Convenience function to update an existing task.
+
+    Args:
+        task_key: The key of the task to update
+        prompt: New prompt text for the task (optional)
+        verifier_code: Python code for task verification (optional)
+
+    Returns:
+        TaskResponse containing the updated task details
+
+    Examples:
+        response = await fleet.update_task("my-task", prompt="New prompt text")
+        response = await fleet.update_task("my-task", verifier_code="def verify(env): return True")
+    """
+    from .global_client import get_client
+    from ..models import TaskResponse
+
+    client = get_client()
+    return await client.update_task(
+        task_key=task_key,
+        prompt=prompt,
+        verifier_code=verifier_code
+    )

fleet/client.py

@@ -32,6 +32,8 @@ from .models import (
     TaskListResponse,
     AccountResponse,
     TaskRequest,
+    TaskResponse,
+    TaskUpdateRequest,
 )
 from .tasks import Task
 
@@ -497,6 +499,31 @@ class Fleet:
         response = self.client.request("GET", "/v1/account")
         return AccountResponse(**response.json())
 
+    def update_task(
+        self,
+        task_key: str,
+        prompt: Optional[str] = None,
+        verifier_code: Optional[str] = None
+    ) -> TaskResponse:
+        """Update an existing task.
+
+        Args:
+            task_key: The key of the task to update
+            prompt: New prompt text for the task (optional)
+            verifier_code: Python code for task verification (optional)
+
+        Returns:
+            TaskResponse containing the updated task details
+        """
+        payload = TaskUpdateRequest(
+            prompt=prompt,
+            verifier_code=verifier_code
+        )
+        response = self.client.request(
+            "PUT", f"/v1/tasks/{task_key}", json=payload.model_dump(exclude_none=True)
+        )
+        return TaskResponse(**response.json())
+
     def _create_verifier_from_data(
         self, verifier_id: str, verifier_key: str, verifier_code: str, verifier_sha: str
     ) -> "SyncVerifierFunction":

fleet/models.py

@@ -156,6 +156,11 @@ class TaskRequest(BaseModel):
     env_variables: Optional[Dict[str, Any]] = Field(None, title="Env Variables")
 
 
+class TaskUpdateRequest(BaseModel):
+    prompt: Optional[str] = Field(None, title="Prompt")
+    verifier_code: Optional[str] = Field(None, title="Verifier Code")
+
+
 class VerifierData(BaseModel):
     verifier_id: str = Field(..., title="Verifier Id")
     key: str = Field(..., title="Key")

fleet/tasks.py

@@ -232,8 +232,38 @@ def load_tasks(
 
     client = get_client()
     return client.load_tasks(
-        env_key=env_key, 
-        keys=keys, 
-        version=version, 
+        env_key=env_key,
+        keys=keys,
+        version=version,
         team_id=team_id
     )
+
+
+def update_task(
+    task_key: str,
+    prompt: Optional[str] = None,
+    verifier_code: Optional[str] = None
+):
+    """Convenience function to update an existing task.
+
+    Args:
+        task_key: The key of the task to update
+        prompt: New prompt text for the task (optional)
+        verifier_code: Python code for task verification (optional)
+
+    Returns:
+        TaskResponse containing the updated task details
+
+    Examples:
+        response = fleet.update_task("my-task", prompt="New prompt text")
+        response = fleet.update_task("my-task", verifier_code="def verify(env): return True")
+    """
+    from .global_client import get_client
+    from .models import TaskResponse
+
+    client = get_client()
+    return client.update_task(
+        task_key=task_key,
+        prompt=prompt,
+        verifier_code=verifier_code
+    )

{fleet_python-0.2.41.dist-info → fleet_python-0.2.42.dist-info}/METADATA

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

{fleet_python-0.2.41.dist-info → fleet_python-0.2.42.dist-info}/RECORD

@@ -21,20 +21,20 @@ examples/quickstart.py,sha256=1VT39IRRhemsJgxi0O0gprdpcw7HB4pYO97GAYagIcg,3788
 examples/test_cdp_logging.py,sha256=AkCwQCgOTQEI8w3v0knWK_4eXMph7L9x07wj9yIYM10,2836
 fleet/__init__.py,sha256=oxI2XvaiRMn15AZpoDHOvX26WlXALHXvqSRP0KkBpAY,3751
 fleet/base.py,sha256=bc-340sTpq_DJs7yQ9d2pDWnmJFmA1SwDB9Lagvqtb4,9182
-fleet/client.py,sha256=OKc68xmt819OxBQ-RfNslfDTHJtX0pPLxdGVdH1WAh8,22102
+fleet/client.py,sha256=T9Tvhq4aTEs1MjTJw26TkvSMFtQSYEi2hCVu0mdRGxc,22931
 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=9tDjgcgKPMnf-R_MDh-Ocp_UMbyJ8tJyjb15XqU0N94,12454
-fleet/tasks.py,sha256=9ass6zLyuMKC_2gTphZClhVR9RdJml-yeH_GbCu6WOE,8733
+fleet/models.py,sha256=li5Cii7ASUHCFMFeJIMklyicYczqPez768RxO0Q0F2o,12618
+fleet/tasks.py,sha256=TuPmCdgMnqBjJhqxLUL2CKgR0w-P5zNAPukHT_fQTyU,9575
 fleet/types.py,sha256=L4Y82xICf1tzyCLqhLYUgEoaIIS5h9T05TyFNHSWs3s,652
 fleet/_async/__init__.py,sha256=lrnDD6N9p0Oqpi_djxTnxh8I5F7nA7KNn0khciGmgpg,6747
 fleet/_async/base.py,sha256=oisVTQsx0M_yTmyQJc3oij63uKZ97MHz-xYFsWXxQE8,9202
-fleet/_async/client.py,sha256=fTzc5wtPntihAdydE_g-XIJYojcrfDvvmcOJUgR557k,22597
+fleet/_async/client.py,sha256=Etpho0Vvovv8l7Y4HFE2KGBb64rdE-LefFXux9QcWeI,23438
 fleet/_async/exceptions.py,sha256=fUmPwWhnT8SR97lYsRq0kLHQHKtSh2eJS0VQ2caSzEI,5055
 fleet/_async/global_client.py,sha256=4WskpLHbsDEgWW7hXMD09W-brkp4euy8w2ZJ88594rQ,1103
-fleet/_async/models.py,sha256=9tDjgcgKPMnf-R_MDh-Ocp_UMbyJ8tJyjb15XqU0N94,12454
-fleet/_async/tasks.py,sha256=M1jueEdqzr_q02YrOKgjNLf3C1wRWK5rsPp0bcoxkME,7450
+fleet/_async/models.py,sha256=li5Cii7ASUHCFMFeJIMklyicYczqPez768RxO0Q0F2o,12618
+fleet/_async/tasks.py,sha256=8r1jg_r-YpHCezsqlXKoT5RxBelyqax85iwG0kublPc,8317
 fleet/_async/env/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fleet/_async/env/client.py,sha256=9GOSkEWNncwTtiZNaJ2vNGrFCPutyan9lBNhD87dAzQ,1059
 fleet/_async/instance/__init__.py,sha256=PtmJq8J8bh0SOQ2V55QURz5GJfobozwtQoqhaOk3_tI,515
@@ -67,10 +67,10 @@ fleet/verifiers/decorator.py,sha256=nAP3O8szXu7md_kpwpz91hGSUNEVLYjwZQZTkQlV1DM,
 fleet/verifiers/parse.py,sha256=0bAbj9VvT__yU4ZVREUK-Tn9dukh9LCpmfVsgj1DfP4,8508
 fleet/verifiers/sql_differ.py,sha256=dmiGCFXVMEMbAX519OjhVqgA8ZvhnvdmC1BVpL7QCF0,6490
 fleet/verifiers/verifier.py,sha256=53oBWAf0yy3bZmZx9eH9AWIf65H7OP2UUm0YwWCL6Mc,14286
-fleet_python-0.2.41.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+fleet_python-0.2.42.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.41.dist-info/METADATA,sha256=kgKHMZOtj-gOUUcMJQPqp-2f-QYWLfZZwyMkjKd0Mzg,3304
-fleet_python-0.2.41.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fleet_python-0.2.41.dist-info/top_level.txt,sha256=_3DSmTohvSDf3AIP_BYfGzhwO1ECFwuzg83X-wHCx3Y,23
-fleet_python-0.2.41.dist-info/RECORD,,
+fleet_python-0.2.42.dist-info/METADATA,sha256=DW1jtGvGk6pKaOmWaBNBPMG0-BxpSUnGLg8DWOIMkqU,3304
+fleet_python-0.2.42.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fleet_python-0.2.42.dist-info/top_level.txt,sha256=_3DSmTohvSDf3AIP_BYfGzhwO1ECFwuzg83X-wHCx3Y,23
+fleet_python-0.2.42.dist-info/RECORD,,