fleet-python 0.2.23__py3-none-any.whl → 0.2.25__py3-none-any.whl

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/_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/_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/_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/_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/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/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/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/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/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/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.dist-info → fleet_python-0.2.25.dist-info}/METADATA

@@ -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.dist-info → fleet_python-0.2.25.dist-info}/RECORD

@@ -15,22 +15,22 @@ examples/openai_example.py,sha256=I2vk_SJN9BkSRQCYRJfbtGJ-HJ2xzQj-lOjwqmLos5M,82
 examples/openai_simple_example.py,sha256=I42ytIwv0INgDO39pp1MOQSqsJz2YYH8GeNNBaUtq3A,1748
 examples/query_builder_example.py,sha256=Q3lUBETHpu1aS2FXAO79ADYqCxOjMMMZNgCcFVapiII,3918
 examples/quickstart.py,sha256=1VT39IRRhemsJgxi0O0gprdpcw7HB4pYO97GAYagIcg,3788
-fleet/__init__.py,sha256=5js0jYu1wddNyfy4sJIcoVGMZNZeHHSAPJJzfFXN-Po,2276
+fleet/__init__.py,sha256=2WHAk_ZRIxDs1uxrWf-sSeokiK4f_nDVHJJ1_VUFSPA,2306
 fleet/base.py,sha256=0yYuMN0lBkrfTTZBt5NQp5112xWgziuWEk4GuHJB1wE,9189
-fleet/client.py,sha256=SDrq3gfmbAa-F-wmewHN7PyoUc2SWKy2DdaYmJHx1dk,11591
-fleet/config.py,sha256=zd19st83NJdW9DdOq7Irpc0x-iUnMad0JOtAr_nD5DM,273
+fleet/client.py,sha256=K7tLdwvLx-ExGtNoIDqsuROb-egaM-OxGijI19vnUqY,12069
+fleet/config.py,sha256=rqR-y2TbZS-VYACaqrg-PUe0y0UDbR1ZNU1KGJZBwNQ,272
 fleet/exceptions.py,sha256=fUmPwWhnT8SR97lYsRq0kLHQHKtSh2eJS0VQ2caSzEI,5055
-fleet/models.py,sha256=aPjcwbGfyEoDaS5Iip3cAyy3j5NDqhP0wyrvtVvF3M8,11410
+fleet/models.py,sha256=trtM-3A2wW4pOg6EvbKxiukU2clSNFrj4Uax7l2uiIM,11665
 fleet/tasks.py,sha256=OScp0tHIbCmOpmCFzHATBmA7WiJMlehP3rvEfeMAoPk,1648
 fleet/types.py,sha256=eXeI8BFmiU5hln0PVvJbUZs7BSjl6wSqAtN9zaJT6yY,652
 fleet/_async/__init__.py,sha256=AJWCnuo7XKja4yBb8fK2wX7ntciLXQrpzdRHwjTRP6M,62
 fleet/_async/base.py,sha256=s0rYOtXsMJeitOvpa-Oh8ciLV226p_TIPp3fplzWvV4,9209
-fleet/_async/client.py,sha256=jByARVSmtsOS9Qlukwij_exPzIQW5TBMtIGKnLqlF00,11924
+fleet/_async/client.py,sha256=2LlMFO5cnLIwvunSixKRk3kJ-F55WufmoyutwgDYf64,12414
 fleet/_async/exceptions.py,sha256=fUmPwWhnT8SR97lYsRq0kLHQHKtSh2eJS0VQ2caSzEI,5055
-fleet/_async/models.py,sha256=aPjcwbGfyEoDaS5Iip3cAyy3j5NDqhP0wyrvtVvF3M8,11410
+fleet/_async/models.py,sha256=trtM-3A2wW4pOg6EvbKxiukU2clSNFrj4Uax7l2uiIM,11665
 fleet/_async/tasks.py,sha256=7ddJWkgulGW7dTPVQUUcl2L0HgsajmMQY6AUblHpG8A,1649
 fleet/_async/env/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-fleet/_async/env/client.py,sha256=PQGKcvnVTr3o2KNsgQM7ZBqaCjimv-wEn-XiYGm44Ko,753
+fleet/_async/env/client.py,sha256=MR2Ju7AudGWmztTVUPPut-ys3yqD7D00D_jsxZaRybA,858
 fleet/_async/instance/__init__.py,sha256=PtmJq8J8bh0SOQ2V55QURz5GJfobozwtQoqhaOk3_tI,515
 fleet/_async/instance/base.py,sha256=3qUBuUR8OVS36LzdP6KyZzngtwPKYO09HoY6Ekxp-KA,1625
 fleet/_async/instance/client.py,sha256=2ZNoBxRJXDQc8fohCLy_RNo-wmuUKXzJo5RZQeCQnkQ,6069
@@ -40,9 +40,9 @@ fleet/_async/resources/browser.py,sha256=oldoSiymJ1lJkADhpUG81ViOBDNyppX1jSoEwe9
 fleet/_async/resources/sqlite.py,sha256=DvDLRI5dJ7_v4WkHw-Zday1M_FQUdzAqUCy9FtfOY8w,26636
 fleet/_async/verifiers/__init__.py,sha256=Z2Ic77mw6-mhF5CmVrucmDnAGSTAtiejR_eZjTjPPA0,447
 fleet/_async/verifiers/bundler.py,sha256=A4yR3wBOcVZYFAv87CD58QlJn6L4QXeilrasnVm8n74,26185
-fleet/_async/verifiers/verifier.py,sha256=fhy1bs_n0Ki2xDtisZI86McIZ09-v5WUFo5tZCBRsvA,12153
-fleet/env/__init__.py,sha256=yYk657kVDnMWYbuPS_2o0kpVC99AYD85o5WG1jCaWSY,531
-fleet/env/client.py,sha256=pG5S1ylDN0pqUGQtRqayDSnYI7ZVpCFvTbtC4LuMRts,628
+fleet/_async/verifiers/verifier.py,sha256=-BNKVKVpg_x4VoPNafEEIGZrP0u7Nh-7OKN6juCiqqs,12322
+fleet/env/__init__.py,sha256=Wt6mO2dQnOdzdjOCvtiIP49wpSrmS7OaZ_OeHNrzqsY,595
+fleet/env/client.py,sha256=2FzA7qfR-B7tVdCNa4uQk10g8zXV-9MNjAqx2D5vQ7c,710
 fleet/instance/__init__.py,sha256=-Anms7Vw_FO8VBIvwnAnq4rQjwl_XNfAS-i7bypHMow,478
 fleet/instance/base.py,sha256=OYqzBwZFfTX9wlBGSG5gljqj98NbiJeKIfFJ3uj5I4s,1587
 fleet/instance/client.py,sha256=4x-3IZZNwlK02OJLKp7mhSymRgXk8NToRCDU3ZuDnJ0,5891
@@ -59,11 +59,11 @@ fleet/verifiers/db.py,sha256=tssmvJjDHuBIy8qlL_P5-UdmEFUw2DZcqLsWZ8ot3Xw,27766
 fleet/verifiers/decorator.py,sha256=Q-KHhicnIYFwX7FX_VZguzNfu8ZslqNUeWxcS2CwNVY,3386
 fleet/verifiers/parse.py,sha256=FYbzgX86dLAxoAZ3jXfu6zRPlNxgzjTkhO3yxMC0hZ4,5261
 fleet/verifiers/sql_differ.py,sha256=dmiGCFXVMEMbAX519OjhVqgA8ZvhnvdmC1BVpL7QCF0,6490
-fleet/verifiers/verifier.py,sha256=-iEnLi4BOpqMV8tJgSPcTi_mGWM3vMJiAYTmaJolo2c,12080
-fleet_python-0.2.23.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+fleet/verifiers/verifier.py,sha256=XVK6a17GnQjxjEkYgl05QKjlmqTA5sWf-eGzcM-QR7E,12249
+fleet_python-0.2.25.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
 scripts/fix_sync_imports.py,sha256=zangqElfqIvIc-E0S46m6R-51pTZ6gbd8oUVkvCGzJo,4768
 scripts/unasync.py,sha256=--Fmaae47o-dZ1HYgX1c3Nvi-rMjcFymTRlJcWWnmpw,725
-fleet_python-0.2.23.dist-info/METADATA,sha256=BnRRdSDuEiBGpMKUnjrB89CfR4SYbqgo7LhDy6yC6DI,3297
-fleet_python-0.2.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fleet_python-0.2.23.dist-info/top_level.txt,sha256=_3DSmTohvSDf3AIP_BYfGzhwO1ECFwuzg83X-wHCx3Y,23
-fleet_python-0.2.23.dist-info/RECORD,,
+fleet_python-0.2.25.dist-info/METADATA,sha256=K9TL_OP6vnVBzED_sXSpvZDli67CXfQ5FKOotwEMkKg,3297
+fleet_python-0.2.25.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fleet_python-0.2.25.dist-info/top_level.txt,sha256=_3DSmTohvSDf3AIP_BYfGzhwO1ECFwuzg83X-wHCx3Y,23
+fleet_python-0.2.25.dist-info/RECORD,,