fleet-python 0.2.23__tar.gz → 0.2.25__tar.gz

{fleet_python-0.2.23 → fleet_python-0.2.25}/PKG-INFO

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

{fleet_python-0.2.23 → fleet_python-0.2.25}/fleet/__init__.py

@@ -24,7 +24,7 @@ from .exceptions import (
 )
 from .client import Fleet, SyncEnv
 from ._async.client import AsyncFleet, AsyncEnv
-from .models import InstanceResponse
+from .models import InstanceResponse, Environment
 from .instance.models import Resource, ResetResponse
 
 # Import sync verifiers with explicit naming
@@ -62,7 +62,8 @@ __all__ = [
     "AsyncFleet",
     "AsyncEnv",
     # Models
-    "InstanceRecord", 
+    "InstanceResponse",
+    "SyncEnv", 
     "Resource",
     "ResetResponse",
     # Task models

{fleet_python-0.2.23 → fleet_python-0.2.25}/fleet/_async/client.py

@@ -29,6 +29,7 @@ from ..models import (
     VerifiersCheckResponse,
     VerifiersExecuteResponse,
     TaskListResponse,
+    AccountResponse,
 )
 from .tasks import Task
 
@@ -124,6 +125,7 @@ class AsyncEnv(EnvironmentBase):
         key: str,
         function_name: str,
         args: tuple, 
+        args_array: list,
         kwargs: dict, 
         timeout: Optional[int] = 30,
         needs_upload: bool = True,
@@ -135,6 +137,7 @@ class AsyncEnv(EnvironmentBase):
             key,
             function_name,
             args, 
+            args_array,
             kwargs, 
             timeout,
             needs_upload
@@ -199,6 +202,7 @@ class AsyncFleet:
             json=request.model_dump(),
             base_url=region_base_url,
         )
+
         instance = AsyncEnv(client=self.client, **response.json())
         await instance.instance.load()
         return instance
@@ -268,6 +272,15 @@ class AsyncFleet:
         
         return tasks
 
+    async def account(self) -> AccountResponse:
+        """Get account information including instance limits and usage.
+
+        Returns:
+            AccountResponse containing team_id, team_name, instance_limit, and instance_count
+        """
+        response = await self.client.request("GET", "/v1/account")
+        return AccountResponse(**response.json())
+
 
 # Shared
 async def _delete_instance(client: AsyncWrapper, instance_id: str) -> InstanceResponse:
@@ -289,6 +302,7 @@ async def _execute_verifier_remote(
     key: str,
     function_name: str,
     args: tuple,
+    args_array: list,
     kwargs: dict,
     timeout: Optional[int] = 30,
     needs_upload: bool = True,
@@ -304,6 +318,7 @@ async def _execute_verifier_remote(
         "key": key,
         "sha256": bundle_sha,
         "args": args_kwargs_b64,
+        "args_array": args_array,
         "function_name": function_name,
         "timeout": timeout,
         "region": "us-west-1",  # TODO: make configurable

{fleet_python-0.2.23 → fleet_python-0.2.25}/fleet/_async/env/client.py

@@ -1,5 +1,5 @@
 from ..client import AsyncFleet, AsyncEnv
-from ...models import Environment as EnvironmentModel
+from ...models import Environment as EnvironmentModel, AccountResponse
 from typing import List, Optional
 
 
@@ -23,3 +23,7 @@ async def list_instances_async(
 
 async def get_async(instance_id: str) -> AsyncEnv:
     return await AsyncFleet().instance(instance_id)
+
+
+async def account_async() -> AccountResponse:
+    return await AsyncFleet().account()

{fleet_python-0.2.23 → fleet_python-0.2.25}/fleet/_async/models.py

@@ -319,3 +319,9 @@ class InstanceResponse(BaseModel):
     env_variables: Optional[Dict[str, Any]] = Field(None, title='Env Variables')
     urls: Optional[InstanceURLs] = Field(None, title='Urls')
     health: Optional[bool] = Field(None, title='Health')
+
+class AccountResponse(BaseModel):
+    team_id: str = Field(..., title='Team Id')
+    team_name: str = Field(..., title='Team Name')
+    instance_limit: int = Field(..., title='Instance Limit')
+    instance_count: int = Field(..., title='Instance Count')

{fleet_python-0.2.23 → fleet_python-0.2.25}/fleet/_async/verifiers/verifier.py

@@ -129,6 +129,10 @@ class AsyncVerifierFunction:
                 "Please provide a synchronous version of your verifier."
             )
         
+        args_array = list(args)
+        args_array.append({"env": env.instance_id})
+        args = tuple(args_array)
+        
         try:
             # Check if bundle needs to be uploaded
             bundle_sha, needs_upload = await self._check_bundle_status(env)
@@ -144,6 +148,7 @@ class AsyncVerifierFunction:
                     key=self.key,
                     function_name=self.func.__name__,
                     args=args,
+                    args_array=args_array,
                     kwargs=kwargs,
                     needs_upload=True
                 )

{fleet_python-0.2.23 → fleet_python-0.2.25}/fleet/client.py

@@ -29,6 +29,7 @@ from .models import (
     VerifiersCheckResponse,
     VerifiersExecuteResponse,
     TaskListResponse,
+    AccountResponse,
 )
 from .tasks import Task
 
@@ -124,6 +125,7 @@ class SyncEnv(EnvironmentBase):
         key: str,
         function_name: str,
         args: tuple, 
+        args_array: list,
         kwargs: dict, 
         timeout: Optional[int] = 30,
         needs_upload: bool = True,
@@ -135,6 +137,7 @@ class SyncEnv(EnvironmentBase):
             key,
             function_name,
             args, 
+            args_array,
             kwargs, 
             timeout,
             needs_upload
@@ -199,6 +202,7 @@ class Fleet:
             json=request.model_dump(),
             base_url=region_base_url,
         )
+
         instance = SyncEnv(client=self.client, **response.json())
         instance.instance.load()
         return instance
@@ -268,6 +272,15 @@ class Fleet:
         
         return tasks
 
+    def account(self) -> AccountResponse:
+        """Get account information including instance limits and usage.
+
+        Returns:
+            AccountResponse containing team_id, team_name, instance_limit, and instance_count
+        """
+        response = self.client.request("GET", "/v1/account")
+        return AccountResponse(**response.json())
+
 
 # Shared
 def _delete_instance(client: SyncWrapper, instance_id: str) -> InstanceResponse:
@@ -289,6 +302,7 @@ def _execute_verifier_remote(
     key: str,
     function_name: str,
     args: tuple,
+    args_array: list,
     kwargs: dict,
     timeout: Optional[int] = 30,
     needs_upload: bool = True,
@@ -304,6 +318,7 @@ def _execute_verifier_remote(
         "key": key,
         "sha256": bundle_sha,
         "args": args_kwargs_b64,
+        "args_array": args_array,
         "function_name": function_name,
         "timeout": timeout,
         "region": "us-west-1",  # TODO: make configurable

{fleet_python-0.2.23 → fleet_python-0.2.25}/fleet/config.py

@@ -1,5 +1,5 @@
 DEFAULT_MAX_RETRIES = 5
-DEFAULT_TIMEOUT = 300.0
+DEFAULT_TIMEOUT = 60.0
 
 GLOBAL_BASE_URL = "https://orchestrator.fleetai.com"
 REGION_BASE_URL = {

{fleet_python-0.2.23 → fleet_python-0.2.25}/fleet/env/__init__.py

@@ -1,6 +1,6 @@
 """Fleet env module - convenience functions for environment management."""
 
-from .client import make, list_envs, list_regions, get, list_instances
+from .client import make, list_envs, list_regions, get, list_instances, account
 
 # Import async versions from _async
 from .._async.env.client import (
@@ -9,6 +9,7 @@ from .._async.env.client import (
     list_regions_async,
     get_async,
     list_instances_async,
+    account_async,
 )
 
 __all__ = [
@@ -22,4 +23,6 @@ __all__ = [
     "list_regions_async",
     "list_instances_async",
     "get_async",
+    "account",
+    "account_async",
 ]

{fleet_python-0.2.23 → fleet_python-0.2.25}/fleet/env/client.py

@@ -1,5 +1,5 @@
 from ..client import Fleet, SyncEnv
-from ..models import Environment as EnvironmentModel
+from ..models import Environment as EnvironmentModel, AccountResponse
 from typing import List, Optional
 
 
@@ -23,3 +23,7 @@ def list_instances(
 
 def get(instance_id: str) -> SyncEnv:
     return Fleet().instance(instance_id)
+
+
+def account() -> AccountResponse:
+    return Fleet().account()

{fleet_python-0.2.23 → fleet_python-0.2.25}/fleet/models.py

@@ -319,3 +319,9 @@ class InstanceResponse(BaseModel):
     env_variables: Optional[Dict[str, Any]] = Field(None, title='Env Variables')
     urls: Optional[InstanceURLs] = Field(None, title='Urls')
     health: Optional[bool] = Field(None, title='Health')
+
+class AccountResponse(BaseModel):
+    team_id: str = Field(..., title='Team Id')
+    team_name: str = Field(..., title='Team Name')
+    instance_limit: int = Field(..., title='Instance Limit')
+    instance_count: int = Field(..., title='Instance Count')

{fleet_python-0.2.23 → fleet_python-0.2.25}/fleet/verifiers/verifier.py

@@ -128,6 +128,10 @@ class SyncVerifierFunction:
                 "Please provide a synchronous version of your verifier."
             )
         
+        args_array = list(args)
+        args_array.append({"env": env.instance_id})
+        args = tuple(args_array)
+        
         try:
             # Check if bundle needs to be uploaded
             bundle_sha, needs_upload = self._check_bundle_status(env)
@@ -143,6 +147,7 @@ class SyncVerifierFunction:
                     key=self.key,
                     function_name=self.func.__name__,
                     args=args,
+                    args_array=args_array,
                     kwargs=kwargs,
                     needs_upload=True
                 )

{fleet_python-0.2.23 → fleet_python-0.2.25}/fleet_python.egg-info/PKG-INFO

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

{fleet_python-0.2.23 → fleet_python-0.2.25}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "fleet-python"
-version = "0.2.23"
+version = "0.2.25"
 description = "Python SDK for Fleet environments"
 authors = [
     {name = "Fleet AI", email = "nic@fleet.so"},
@@ -99,6 +99,7 @@ testpaths = ["tests"]
 "list_regions_async" = "list_regions"
 "list_instances_async" = "list_instances"
 "get_async" = "get"
+"account_async" = "account"
 "async def" = "def"
 "from fleet.verifiers" = "from ..verifiers"
 "await asyncio.sleep" = "time.sleep"