fleet-python 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

fleet/env/models.py

@@ -0,0 +1,127 @@
+# generated by datamodel-codegen:
+#   filename:  openapi_2.json
+#   timestamp: 2025-07-08T20:02:11+00:00
+
+from __future__ import annotations
+
+from enum import Enum
+from typing import Any, Dict, List, Optional, Union
+
+from pydantic import BaseModel, Field
+
+
+class BrowserDescribeResponse(BaseModel):
+    success: bool = Field(..., title="Success")
+    url: str = Field(..., title="Url")
+    devtools_url: str = Field(..., title="Devtools Url")
+
+
+class BrowserStartRequest(BaseModel):
+    start_page: Optional[str] = Field("about:blank", title="Start Page")
+    resolution: Optional[str] = Field("1920x1080", title="Resolution")
+
+
+class BrowserStartResponse(BaseModel):
+    success: bool = Field(..., title="Success")
+    message: str = Field(..., title="Message")
+
+
+class CreateSnapshotsResponse(BaseModel):
+    success: bool = Field(..., title="Success")
+    initial_snapshot_path: Optional[str] = Field(None, title="Initial Snapshot Path")
+    final_snapshot_path: Optional[str] = Field(None, title="Final Snapshot Path")
+    message: str = Field(..., title="Message")
+
+
+class HealthResponse(BaseModel):
+    status: str = Field(..., title="Status")
+    timestamp: str = Field(..., title="Timestamp")
+    service: str = Field(..., title="Service")
+
+
+class LogActionRequest(BaseModel):
+    action_type: str = Field(..., title="Action Type")
+    sql: Optional[str] = Field(None, title="Sql")
+    args: Optional[str] = Field(None, title="Args")
+    path: Optional[str] = Field(None, title="Path")
+    raw_payload: Optional[str] = Field(None, title="Raw Payload")
+
+
+class LogActionResponse(BaseModel):
+    success: bool = Field(..., title="Success")
+    message: str = Field(..., title="Message")
+
+
+class QueryRequest(BaseModel):
+    query: str = Field(..., title="Query")
+    args: Optional[List] = Field(None, title="Args")
+    read_only: Optional[bool] = Field(True, title="Read Only")
+
+
+class QueryResponse(BaseModel):
+    success: bool = Field(..., title="Success")
+    columns: Optional[List[str]] = Field(None, title="Columns")
+    rows: Optional[List[List]] = Field(None, title="Rows")
+    rows_affected: Optional[int] = Field(None, title="Rows Affected")
+    last_insert_id: Optional[int] = Field(None, title="Last Insert Id")
+    error: Optional[str] = Field(None, title="Error")
+    message: str = Field(..., title="Message")
+
+
+class ResetRequest(BaseModel):
+    seed: Optional[int] = Field(None, title="Seed")
+    timestamp: Optional[str] = Field(None, title="Timestamp")
+
+
+class ResetResponse(BaseModel):
+    success: bool = Field(..., title="Success")
+    message: str = Field(..., title="Message")
+
+
+class ResourceMode(Enum):
+    ro = "ro"
+    rw = "rw"
+
+
+class ResourceType(Enum):
+    sqlite = "sqlite"
+    cdp = "cdp"
+
+
+class TableSchema(BaseModel):
+    name: str = Field(..., title="Name")
+    sql: str = Field(..., title="Sql")
+    columns: List[Dict[str, Any]] = Field(..., title="Columns")
+
+
+class TimestampResponse(BaseModel):
+    timestamp: str = Field(..., title="Timestamp")
+
+
+class ValidationError(BaseModel):
+    loc: List[Union[str, int]] = Field(..., title="Location")
+    msg: str = Field(..., title="Message")
+    type: str = Field(..., title="Error Type")
+
+
+class DescribeResponse(BaseModel):
+    success: bool = Field(..., title="Success")
+    resource_name: str = Field(..., title="Resource Name")
+    tables: Optional[List[TableSchema]] = Field(None, title="Tables")
+    error: Optional[str] = Field(None, title="Error")
+    message: str = Field(..., title="Message")
+
+
+class HTTPValidationError(BaseModel):
+    detail: Optional[List[ValidationError]] = Field(None, title="Detail")
+
+
+class Resource(BaseModel):
+    name: str = Field(..., title="Name")
+    type: ResourceType
+    mode: ResourceMode
+    label: Optional[str] = Field(None, title="Label")
+
+
+class ResourcesResponse(BaseModel):
+    resources: List[Resource] = Field(..., title="Resources")

fleet/models.py

@@ -0,0 +1,109 @@
+# generated by datamodel-codegen:
+#   filename:  openapi.json
+#   timestamp: 2025-07-08T18:21:47+00:00
+
+from __future__ import annotations
+
+from enum import Enum
+from typing import Dict, List, Optional, Union
+
+from pydantic import BaseModel, Field
+
+
+class Environment(BaseModel):
+    env_key: str = Field(..., title="Env Key")
+    name: str = Field(..., title="Name")
+    description: Optional[str] = Field(..., title="Description")
+    default_version: Optional[str] = Field(..., title="Default Version")
+    versions: Dict[str, str] = Field(..., title="Versions")
+
+
+class Instance(BaseModel):
+    instance_id: str = Field(..., title="Instance Id")
+    env_key: str = Field(..., title="Env Key")
+    version: str = Field(..., title="Version")
+    status: str = Field(..., title="Status")
+    subdomain: str = Field(..., title="Subdomain")
+    created_at: str = Field(..., title="Created At")
+    updated_at: str = Field(..., title="Updated At")
+    terminated_at: Optional[str] = Field(None, title="Terminated At")
+    team_id: str = Field(..., title="Team Id")
+    region: str = Field(..., title="Region")
+
+
+class InstanceRequest(BaseModel):
+    env_key: str = Field(..., title="Env Key")
+    version: Optional[str] = Field(None, title="Version")
+    region: Optional[str] = Field("us-east-2", title="Region")
+    seed: Optional[int] = Field(None, title="Seed")
+    timestamp: Optional[int] = Field(None, title="Timestamp")
+    p_error: Optional[float] = Field(None, title="P Error")
+    avg_latency: Optional[float] = Field(None, title="Avg Latency")
+    run_id: Optional[str] = Field(None, title="Run Id")
+    task_id: Optional[str] = Field(None, title="Task Id")
+
+
+class InstanceStatus(Enum):
+    pending = "pending"
+    running = "running"
+    stopped = "stopped"
+    error = "error"
+
+
+class ManagerURLs(BaseModel):
+    api: str = Field(..., title="Api")
+    docs: str = Field(..., title="Docs")
+    reset: str = Field(..., title="Reset")
+    diff: str = Field(..., title="Diff")
+    snapshot: str = Field(..., title="Snapshot")
+    execute_verifier_function: str = Field(..., title="Execute Verifier Function")
+    execute_verifier_function_with_upload: str = Field(
+        ..., title="Execute Verifier Function With Upload"
+    )
+
+
+class ValidationError(BaseModel):
+    loc: List[Union[str, int]] = Field(..., title="Location")
+    msg: str = Field(..., title="Message")
+    type: str = Field(..., title="Error Type")
+
+
+class HTTPValidationError(BaseModel):
+    detail: Optional[List[ValidationError]] = Field(None, title="Detail")
+
+
+class InstanceURLs(BaseModel):
+    root: str = Field(..., title="Root")
+    app: str = Field(..., title="App")
+    api: Optional[str] = Field(None, title="Api")
+    health: Optional[str] = Field(None, title="Health")
+    api_docs: Optional[str] = Field(None, title="Api Docs")
+    manager: ManagerURLs
+
+
+class InstanceResponse(BaseModel):
+    instance_id: str = Field(..., title="Instance Id")
+    env_key: str = Field(..., title="Env Key")
+    version: str = Field(..., title="Version")
+    status: str = Field(..., title="Status")
+    subdomain: str = Field(..., title="Subdomain")
+    created_at: str = Field(..., title="Created At")
+    updated_at: str = Field(..., title="Updated At")
+    terminated_at: Optional[str] = Field(None, title="Terminated At")
+    team_id: str = Field(..., title="Team Id")
+    region: str = Field(..., title="Region")
+    urls: InstanceURLs
+    health: Optional[bool] = Field(None, title="Health")
+
+
+class InstanceRecord(BaseModel):
+    instance_id: str
+    env_key: str
+    version: str
+    status: str
+    subdomain: str
+    created_at: str
+    updated_at: str
+    terminated_at: Optional[str] = None
+    team_id: str
+    region: str

fleet/resources/base.py

@@ -0,0 +1,23 @@
+from abc import ABC
+from ..env.models import Resource as ResourceModel, ResourceType, ResourceMode
+
+
+class Resource(ABC):
+    def __init__(self, resource: ResourceModel):
+        self.resource = resource
+
+    @property
+    def uri(self) -> str:
+        return f"{self.resource.type}://{self.resource.name}"
+
+    @property
+    def name(self) -> str:
+        return self.resource.name
+
+    @property
+    def type(self) -> ResourceType:
+        return self.resource.type
+
+    @property
+    def mode(self) -> ResourceMode:
+        return self.resource.mode

fleet/resources/browser.py

@@ -0,0 +1,18 @@
+from ..env.models import Resource as ResourceModel
+from ..env.models import BrowserDescribeResponse
+from .base import Resource
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from ..env.base import AsyncWrapper
+
+
+class AsyncBrowserResource(Resource):
+    def __init__(self, resource: ResourceModel, client: "AsyncWrapper"):
+        super().__init__(resource)
+        self.client = client
+
+    async def describe(self) -> BrowserDescribeResponse:
+        response = await self.client.request("GET", "/resource/cdp/describe")
+        return BrowserDescribeResponse(**response.json())

fleet/resources/sqlite.py

@@ -0,0 +1,41 @@
+from typing import Any, List, Optional
+from ..env.models import Resource as ResourceModel
+from ..env.models import DescribeResponse, QueryRequest, QueryResponse
+from .base import Resource
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from ..env.base import AsyncWrapper
+
+
+class AsyncSQLiteResource(Resource):
+    def __init__(self, resource: ResourceModel, client: "AsyncWrapper"):
+        super().__init__(resource)
+        self.client = client
+
+    async def describe(self) -> DescribeResponse:
+        """Describe the SQLite database schema."""
+        response = await self.client.request(
+            "GET", f"/resource/sqlite/{self.resource.name}/describe"
+        )
+        return DescribeResponse(**response.json())
+
+    async def query(
+        self, query: str, args: Optional[List[Any]] = None
+    ) -> QueryResponse:
+        return await self._query(query, args, read_only=True)
+
+    async def exec(self, query: str, args: Optional[List[Any]] = None) -> QueryResponse:
+        return await self._query(query, args, read_only=False)
+
+    async def _query(
+        self, query: str, args: Optional[List[Any]] = None, read_only: bool = True
+    ) -> QueryResponse:
+        request = QueryRequest(query=query, args=args, read_only=read_only)
+        response = await self.client.request(
+            "POST",
+            f"/resource/sqlite/{self.resource.name}/query",
+            json=request.model_dump(),
+        )
+        return QueryResponse(**response.json())

{fleet_python-0.1.0.dist-info → fleet_python-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fleet-python
-Version: 0.1.0
+Version: 0.2.0
 Summary: Python SDK for Fleet environments
 Author-email: Fleet AI <nic@fleet.so>
 License: Apache-2.0
@@ -22,6 +22,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: aiohttp>=3.8.0
 Requires-Dist: pydantic>=2.0.0
+Requires-Dist: httpx>=0.27.0
 Requires-Dist: typing-extensions>=4.0.0
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0.0; extra == "dev"

fleet_python-0.2.0.dist-info/RECORD

@@ -0,0 +1,19 @@
+examples/browser_control_example.py,sha256=2nEzUc83bliqR9jYTenkrLTGoKSkWahjP4WpiaHIMxw,1520
+examples/quickstart.py,sha256=AnLlLQYRfrP7y2J69d1HZRlZsjJT-KeBu897MoNfqYM,4671
+fleet/__init__.py,sha256=F8esCp6DcdAc4JAhWlCd_G3pBKRBzuLuVZnElWv5GGQ,987
+fleet/base.py,sha256=lgKhPZhLotP4Iqn7Z4nQTfCvuD3bT37MoucTrSRE2zs,2006
+fleet/client.py,sha256=xMyKc49YwTAh1F5Wbk5hdaB4F6wlrl41a3EjyBdjFUk,5660
+fleet/exceptions.py,sha256=yG3QWprCw1OnF-vdFBFJWE4m3ftBLBng31Dr__VbjI4,2249
+fleet/models.py,sha256=Jf6Zmk689TPXhTSnVENK_VCw0VsujWzEWsN3T29MQ0k,3713
+fleet/env/__init__.py,sha256=ScTHui5BAFTTKKZ6TxS71D5r-JYMxxFEkcT_AcAXDsM,145
+fleet/env/base.py,sha256=bm6BGd81TnTDqE_qG6S50Xhikf9DwNqEEk1uKFY7dEk,1540
+fleet/env/client.py,sha256=yVS4cblGlARWeoZC69Uz_KR2wlmYC19d-BsQalLB9kw,8043
+fleet/env/models.py,sha256=mxRoQEzswMEjwdEN-gn2ylG9Z8IZH-Gqs6hVG0fkSVs,3883
+fleet/resources/base.py,sha256=rzFXBoFqtkM0HZpwqN9NHiwQbmeKOGOYh7G4REaHKGc,553
+fleet/resources/browser.py,sha256=mC1g7xpyvEK8vDGXNn0MXGc8TGE83PIW3KOOj9i3rIA,591
+fleet/resources/sqlite.py,sha256=hZA2qcd7SqjsACJEkKzXQwlJCtvNZ9yom2HCjKy7lOs,1484
+fleet_python-0.2.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+fleet_python-0.2.0.dist-info/METADATA,sha256=qrpy0oFm5wfUvW7VVqbrdZX566B-CB1iJSp4spH9gmg,3069
+fleet_python-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fleet_python-0.2.0.dist-info/top_level.txt,sha256=AOyXOrBXUjPcH4BumElz_D95kiWKNIpUbUPFP_9gCLk,15
+fleet_python-0.2.0.dist-info/RECORD,,

fleet/config.py

@@ -1,125 +0,0 @@
-"""Fleet SDK Configuration Management."""
-
-import os
-import logging
-from typing import Optional, Dict, Any
-from pydantic import BaseModel, Field, validator
-from .exceptions import FleetAuthenticationError, FleetConfigurationError
-
-
-logger = logging.getLogger(__name__)
-
-
-class FleetConfig(BaseModel):
-    """Fleet SDK Configuration."""
-    
-    api_key: Optional[str] = Field(None, description="Fleet API key")
-    base_url: str = Field(default="https://fleet.new", description="Fleet API base URL (hardcoded)")
-    
-    @validator('api_key')
-    def validate_api_key(cls, v):
-        """Validate API key format."""
-        if v is not None and not _is_valid_api_key(v):
-            raise FleetAuthenticationError(
-                "Invalid API key format. Fleet API keys should start with 'sk_' followed by alphanumeric characters."
-            )
-        return v
-    
-    @validator('base_url')
-    def validate_base_url(cls, v):
-        """Validate base URL format."""
-        if not v.startswith(('http://', 'https://')):
-            raise FleetConfigurationError("Base URL must start with 'http://' or 'https://'")
-        return v.rstrip('/')
-    
-    def mask_sensitive_data(self) -> Dict[str, Any]:
-        """Return config dict with sensitive data masked."""
-        data = self.dict()
-        if data.get('api_key'):
-            data['api_key'] = _mask_api_key(data['api_key'])
-        return data
-    
-    class Config:
-        """Pydantic configuration."""
-        extra = 'allow'
-
-
-def get_config(**kwargs: Any) -> FleetConfig:
-    """Get Fleet configuration from environment variables.
-    
-    Loads FLEET_API_KEY from environment variables. The base URL is hardcoded to https://fleet.new.
-    
-    Args:
-        **kwargs: Override specific configuration values
-        
-    Returns:
-        FleetConfig instance
-        
-    Raises:
-        FleetAuthenticationError: If API key is invalid
-        FleetConfigurationError: If configuration is invalid
-    """
-    # Load from environment variables
-    config_data = _load_env_config()
-    
-    # Apply any overrides
-    config_data.update(kwargs)
-    
-    # Create and validate configuration
-    try:
-        config = FleetConfig(**config_data)
-        return config
-        
-    except Exception as e:
-        if isinstance(e, (FleetAuthenticationError, FleetConfigurationError)):
-            raise
-        raise FleetConfigurationError(f"Invalid configuration: {e}")
-
-
-def _load_env_config() -> Dict[str, Any]:
-    """Load configuration from environment variables."""
-    env_mapping = {
-        'FLEET_API_KEY': 'api_key',
-        # base_url is hardcoded, not configurable via env var
-    }
-    
-    config = {}
-    for env_var, config_key in env_mapping.items():
-        value = os.getenv(env_var)
-        if value is not None:
-            config[config_key] = value
-    
-    return config
-
-
-def _is_valid_api_key(api_key: str) -> bool:
-    """Validate API key format."""
-    if not api_key:
-        return False
-    
-    # Fleet API keys start with 'sk_' followed by alphanumeric characters
-    # This is a basic format check - actual validation happens on the server
-    if not api_key.startswith('sk_'):
-        return False
-    
-    # Check if the rest contains only alphanumeric characters and underscores
-    key_part = api_key[3:]  # Remove 'sk_' prefix
-    if not key_part or not key_part.replace('_', '').isalnum():
-        return False
-    
-    # Minimum length check
-    if len(api_key) < 20:
-        return False
-    
-    return True
-
-
-def _mask_api_key(api_key: str) -> str:
-    """Mask API key for logging."""
-    if not api_key:
-        return api_key
-    
-    if len(api_key) < 8:
-        return '*' * len(api_key)
-    
-    return api_key[:4] + '*' * (len(api_key) - 8) + api_key[-4:]