fleet-python 0.2.27__tar.gz → 0.2.28__tar.gz

{fleet_python-0.2.27 → fleet_python-0.2.28}/PKG-INFO

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

fleet_python-0.2.28/fleet/tasks.py

@@ -0,0 +1,82 @@
+"""Fleet SDK Task Model."""
+
+from __future__ import annotations
+
+import re
+from datetime import datetime
+from typing import Any, Dict, Optional
+from uuid import UUID
+
+from pydantic import BaseModel, Field, validator
+
+# Import the shared VerifierFunction type that works for both async and sync
+from fleet.types import VerifierFunction
+
+
+class Task(BaseModel):
+    """A task model representing a single task in the Fleet system."""
+    
+    key: str = Field(..., description="Unique task key identifier")
+    prompt: str = Field(..., description="Task prompt or instruction")
+    env_id: str = Field(..., description="Environment identifier")
+    env_variables: Optional[Dict[str, Any]] = Field(default_factory=dict, description="Environment variables")
+    created_at: Optional[datetime] = Field(None, description="Task creation timestamp")
+    version: Optional[str] = Field(None, description="Task version")
+    verifier_func: Optional[str] = Field(None, description="Verifier function code")
+    verifier: Optional[Any] = Field(None, description="Verifier function with decorator (async or sync)")
+    metadata: Optional[Dict[str, Any]] = Field(default_factory=dict, description="Additional task metadata")
+
+    @validator('key')
+    def validate_key_format(cls, v):
+        """Validate key follows kebab-case format."""
+        if not re.match(r'^[a-z0-9]+(-[a-z0-9]+)*$', v):
+            raise ValueError(f'Invalid task key format: {v}. Must follow kebab-case format.')
+        return v
+
+    @validator('created_at', pre=True, always=True)
+    def set_created_at(cls, v):
+        """Set created_at to current time if not provided."""
+        return v or datetime.now()
+    
+    @property
+    def env_key(self) -> str:
+        """Get the environment key combining env_id and version."""
+        if self.version:
+            return f"{self.env_id}:{self.version}"
+        return self.env_id
+
+    class Config:
+        """Pydantic model configuration."""
+        json_encoders = {
+            datetime: lambda v: v.isoformat(),
+        }
+        # Allow arbitrary types for the verifier field
+        arbitrary_types_allowed = True 
+
+    def verify(self, env, *args, **kwargs) -> float:
+        """Verify the task using the verifier function (sync version).
+        
+        For sync environments, calls the sync verifier directly.
+        For async verifiers, automatically runs them with asyncio.run().
+        """
+        if self.verifier:
+            return self.verifier.remote(env, *args, **kwargs)
+        else:
+            raise ValueError("No verifier function found for this task")
+    
+    def verify_async(self, *args, **kwargs) -> float:
+        """Verify the task using the verifier function (async version).
+        
+        For async environments, awaits the async verifier.
+        Works with both sync and async verifiers in async contexts.
+        """
+        if self.verifier:
+            result = self.verifier.remote(*args, **kwargs)
+            # If it's a coroutine, await it
+            import inspect
+            if inspect.iscoroutine(result):
+                return result
+            else:
+                return result
+        else:
+            raise ValueError("No verifier function found for this task")

{fleet_python-0.2.27 → fleet_python-0.2.28}/fleet_python.egg-info/PKG-INFO

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

{fleet_python-0.2.27 → fleet_python-0.2.28}/fleet_python.egg-info/SOURCES.txt

@@ -25,6 +25,7 @@ fleet/client.py
 fleet/config.py
 fleet/exceptions.py
 fleet/models.py
+fleet/tasks.py
 fleet/types.py
 fleet/_async/__init__.py
 fleet/_async/base.py

{fleet_python-0.2.27 → fleet_python-0.2.28}/pyproject.toml

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