fleet-python 0.2.66b2__py3-none-any.whl → 0.2.105__py3-none-any.whl

fleet/__init__.py

@@ -14,7 +14,7 @@
 
 """Fleet Python SDK - Environment-based AI agent interactions."""
 
-from typing import Optional, List
+from typing import Any, Optional, List
 
 from .exceptions import (
     FleetError,
@@ -24,9 +24,9 @@ from .exceptions import (
     FleetInstanceLimitError,
     FleetConfigurationError,
 )
-from .client import Fleet, SyncEnv
-from ._async.client import AsyncFleet, AsyncEnv
-from .models import InstanceResponse, Environment
+from .client import Fleet, SyncEnv, Session
+from ._async.client import AsyncFleet, AsyncEnv, AsyncSession
+from .models import InstanceResponse, Environment, Run
 from .instance.models import Resource, ResetResponse
 
 # Import sync verifiers with explicit naming
@@ -73,7 +73,7 @@ from . import env
 from . import global_client as _global_client
 from ._async import global_client as _async_global_client
 
-__version__ = "0.1.0"
+__version__ = "0.2.103"
 
 __all__ = [
     # Core classes
@@ -86,6 +86,7 @@ __all__ = [
     "SyncEnv",
     "Resource",
     "ResetResponse",
+    "Run",
     # Task models
     "Task",
     "VerifierFunction",
@@ -121,6 +122,14 @@ __all__ = [
     "import_tasks_async",
     "get_task",
     "get_task_async",
+    # Session helpers
+    "session",
+    "session_async",
+    "Session",
+    "AsyncSession",
+    # Job helpers
+    "job",
+    "job_async",
     # Version
     "__version__",
 ]
@@ -161,3 +170,117 @@ def reset_client():
     """Reset both sync and async global clients."""
     _global_client.reset_client()
     _async_global_client.reset_client()
+
+
+def session(
+    session_id: Optional[str] = None,
+    job_id: Optional[str] = None,
+    config: Optional[Any] = None,
+    model: Optional[str] = None,
+    task_key: Optional[str] = None,
+    instance_id: Optional[str] = None,
+) -> Session:
+    """Start a new session for logging agent interactions (sync).
+
+    This returns a Session object. The session is created on the backend
+    when you call log() for the first time.
+
+    Args:
+        session_id: Optional existing session ID to resume
+        job_id: Optional job ID to associate with the session
+        config: Optional config object (e.g., GenerateContentConfig) to log
+        model: Optional model name to log
+        task_key: Optional Fleet task key
+        instance_id: Optional Fleet instance ID
+
+    Returns:
+        Session object with log(), complete(), and fail() methods
+
+    Example:
+        session = fleet.session(config=config, model="gpt-4", task_key="task_123")
+        session.log(history, response)
+        session.complete()
+    """
+    client = _global_client.get_client()
+    return client.start_session(
+        session_id=session_id,
+        job_id=job_id,
+        config=config,
+        model=model,
+        task_key=task_key,
+        instance_id=instance_id,
+    )
+
+
+def session_async(
+    session_id: Optional[str] = None,
+    job_id: Optional[str] = None,
+    config: Optional[Any] = None,
+    model: Optional[str] = None,
+    task_key: Optional[str] = None,
+    instance_id: Optional[str] = None,
+) -> AsyncSession:
+    """Start a new session for logging agent interactions (async).
+
+    This returns an AsyncSession object. The session is created on the backend
+    when you call log() for the first time.
+
+    Args:
+        session_id: Optional existing session ID to resume
+        job_id: Optional job ID to associate with the session
+        config: Optional config object (e.g., GenerateContentConfig) to log
+        model: Optional model name to log
+        task_key: Optional Fleet task key
+        instance_id: Optional Fleet instance ID
+
+    Returns:
+        AsyncSession object with log(), complete(), and fail() methods
+
+    Example:
+        session = fleet.session_async(config=config, model="gpt-4", task_key="task_123")
+        await session.log(history, response)
+        await session.complete()
+    """
+    client = _async_global_client.get_client()
+    return client.start_session(
+        session_id=session_id,
+        job_id=job_id,
+        config=config,
+        model=model,
+        task_key=task_key,
+        instance_id=instance_id,
+    )
+
+
+def job(name: Optional[str] = None) -> str:
+    """Create a new trace job (sync).
+
+    Args:
+        name: Name of the job (generated server-side if not provided)
+
+    Returns:
+        The job_id string
+
+    Example:
+        job_id = fleet.job("my-agent-run")
+        session = fleet.session(job_id=job_id, ...)
+    """
+    client = _global_client.get_client()
+    return client.trace_job(name=name)
+
+
+async def job_async(name: Optional[str] = None) -> str:
+    """Create a new trace job (async).
+
+    Args:
+        name: Name of the job (generated server-side if not provided)
+
+    Returns:
+        The job_id string
+
+    Example:
+        job_id = await fleet.job_async("my-agent-run")
+        session = await fleet.session_async(job_id=job_id, ...)
+    """
+    client = _async_global_client.get_client()
+    return await client.trace_job(name=name)

fleet/_async/__init__.py

@@ -24,8 +24,8 @@ from ..exceptions import (
     FleetInstanceLimitError,
     FleetConfigurationError,
 )
-from .client import AsyncFleet, AsyncEnv
-from ..models import InstanceResponse, Environment, AccountResponse
+from .client import AsyncFleet, AsyncEnv, AsyncSession
+from ..models import InstanceResponse, Environment, AccountResponse, Run
 from ..instance.models import Resource, ResetResponse
 
 # Import async verifiers
@@ -44,7 +44,7 @@ from ..types import VerifierFunction
 from .. import env
 from . import global_client as _async_global_client
 
-__version__ = "0.1.0"
+__version__ = "0.2.103"
 
 __all__ = [
     # Core classes
@@ -54,6 +54,7 @@ __all__ = [
     "InstanceResponse",
     "Resource",
     "ResetResponse",
+    "Run",
     # Task models
     "Task",
     "VerifierFunction",
@@ -90,6 +91,7 @@ __all__ = [
     "import_tasks",
     "account",
     "get_task",
+    "list_runs",
     # Version
     "__version__",
 ]
@@ -255,6 +257,28 @@ async def get_task(task_key: str, version_id: Optional[str] = None):
     )
 
 
+async def list_runs(
+    profile_id: Optional[str] = None, status: Optional[str] = "active"
+) -> List[Run]:
+    """List all runs (groups of instances by run_id) with aggregated statistics.
+
+    Args:
+        profile_id: Optional profile ID to filter runs by (use "self" for your own profile)
+        status: Filter by run status - "active" (default), "inactive", or "all"
+
+    Returns:
+        List[Run] containing run information with instance counts and timestamps
+
+    Example:
+        runs = await fleet.list_runs()
+        my_runs = await fleet.list_runs(profile_id="self")
+        all_runs = await fleet.list_runs(status="all")
+    """
+    return await _async_global_client.get_client().list_runs(
+        profile_id=profile_id, status=status
+    )
+
+
 def configure(
     api_key: Optional[str] = None,
     base_url: Optional[str] = None,

fleet/_async/base.py

@@ -2,6 +2,8 @@ import httpx
 from typing import Dict, Any, Optional
 import json
 import logging
+import time
+import uuid
 
 from ..models import InstanceResponse
 from ..config import GLOBAL_BASE_URL
@@ -20,6 +22,12 @@ from .exceptions import (
     FleetPermissionError,
 )
 
+# Import version
+try:
+    from .. import __version__
+except ImportError:
+    __version__ = "0.2.103"
+
 logger = logging.getLogger(__name__)
 
 
@@ -38,17 +46,20 @@ class BaseWrapper:
             base_url = GLOBAL_BASE_URL
         self.base_url = base_url
 
-    def get_headers(self) -> Dict[str, str]:
+    def get_headers(self, request_id: Optional[str] = None) -> Dict[str, str]:
         headers: Dict[str, str] = {
             "X-Fleet-SDK-Language": "Python",
-            "X-Fleet-SDK-Version": "1.0.0",
+            "X-Fleet-SDK-Version": __version__,
         }
         headers["Authorization"] = f"Bearer {self.api_key}"
-        # Debug log
-        import logging
 
-        logger = logging.getLogger(__name__)
-        logger.debug(f"Headers being sent: {headers}")
+        # Add request ID for idempotency (persists across retries)
+        if request_id:
+            headers["X-Request-ID"] = request_id
+
+        # Add timestamp for all requests
+        headers["X-Request-Timestamp"] = str(int(time.time() * 1000))
+
         return headers
 
 
@@ -67,11 +78,14 @@ class AsyncWrapper(BaseWrapper):
         **kwargs,
     ) -> httpx.Response:
         base_url = base_url or self.base_url
+        # Generate unique request ID that persists across retries
+        request_id = str(uuid.uuid4())
+        
         try:
             response = await self.httpx_client.request(
                 method,
                 f"{base_url}{url}",
-                headers=self.get_headers(),
+                headers=self.get_headers(request_id=request_id),
                 params=params,
                 json=json,
                 **kwargs,
@@ -93,8 +107,9 @@ class AsyncWrapper(BaseWrapper):
 
         # Debug log 500 errors
         if status_code == 500:
-            logger.error(f"Got 500 error from {response.url}")
-            logger.error(f"Response text: {response.text}")
+            # logger.error(f"Got 500 error from {response.url}")
+            # logger.error(f"Response text: {response.text}")
+            pass
 
         # Try to parse error response as JSON
         try: