fleet-python 0.2.22__py3-none-any.whl → 0.2.24__py3-none-any.whl

fleet/__init__.py

@@ -22,7 +22,7 @@ from .exceptions import (
     FleetInstanceLimitError,
     FleetConfigurationError,
 )
-from .client import Fleet, Environment
+from .client import Fleet, SyncEnv
 from ._async.client import AsyncFleet, AsyncEnv
 from .models import InstanceResponse
 from .instance.models import Resource, ResetResponse
@@ -58,7 +58,7 @@ __version__ = "0.1.0"
 __all__ = [
     # Core classes
     "Fleet",
-    "Environment",
+    "SyncEnv",
     "AsyncFleet",
     "AsyncEnv",
     # Models

fleet/_async/client.py

@@ -29,6 +29,7 @@ from ..models import (
     VerifiersCheckResponse,
     VerifiersExecuteResponse,
     TaskListResponse,
+    AccountResponse,
 )
 from .tasks import Task
 
@@ -268,6 +269,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:

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

@@ -317,5 +317,5 @@ class InstanceResponse(BaseModel):
     team_id: str = Field(..., title='Team Id')
     region: str = Field(..., title='Region')
     env_variables: Optional[Dict[str, Any]] = Field(None, title='Env Variables')
-    urls: InstanceURLs
+    urls: Optional[InstanceURLs] = Field(None, title='Urls')
     health: Optional[bool] = Field(None, title='Health')

fleet/_async/tasks.py

@@ -10,7 +10,7 @@ from uuid import UUID
 from pydantic import BaseModel, Field, validator
 
 # Import the shared VerifierFunction type that works for both async and sync
-from ..types import VerifierFunction
+from fleet.types import VerifierFunction
 
 
 class Task(BaseModel):

fleet/client.py

@@ -29,6 +29,7 @@ from .models import (
     VerifiersCheckResponse,
     VerifiersExecuteResponse,
     TaskListResponse,
+    AccountResponse,
 )
 from .tasks import Task
 
@@ -268,6 +269,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:

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

@@ -305,6 +305,14 @@ class TaskListResponse(BaseModel):
     total: int = Field(..., title='Total')
 
 
+class AccountResponse(BaseModel):
+    """Response model for account information."""
+    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')
+
+
 class InstanceResponse(BaseModel):
     instance_id: str = Field(..., title='Instance Id')
     env_key: str = Field(..., title='Env Key')
@@ -317,5 +325,5 @@ class InstanceResponse(BaseModel):
     team_id: str = Field(..., title='Team Id')
     region: str = Field(..., title='Region')
     env_variables: Optional[Dict[str, Any]] = Field(None, title='Env Variables')
-    urls: InstanceURLs
+    urls: Optional[InstanceURLs] = Field(None, title='Urls')
     health: Optional[bool] = Field(None, title='Health')

fleet/tasks.py

@@ -10,7 +10,7 @@ from uuid import UUID
 from pydantic import BaseModel, Field, validator
 
 # Import the shared VerifierFunction type that works for both async and sync
-from ..types import VerifierFunction
+from fleet.types import VerifierFunction
 
 
 class Task(BaseModel):

{fleet_python-0.2.22.dist-info → fleet_python-0.2.24.dist-info}/METADATA

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

{fleet_python-0.2.22.dist-info → fleet_python-0.2.24.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=m_Y_HYmMPWfWzUgt-jvm1ZP9kLeRlhy3sjNb0IDMWME,2284
+fleet/__init__.py,sha256=5js0jYu1wddNyfy4sJIcoVGMZNZeHHSAPJJzfFXN-Po,2276
 fleet/base.py,sha256=0yYuMN0lBkrfTTZBt5NQp5112xWgziuWEk4GuHJB1wE,9189
-fleet/client.py,sha256=SDrq3gfmbAa-F-wmewHN7PyoUc2SWKy2DdaYmJHx1dk,11591
+fleet/client.py,sha256=HJfUy9UlHw7610Cx1zhgIAl9Wu40OVH7hhpCvaTueJg,11970
 fleet/config.py,sha256=zd19st83NJdW9DdOq7Irpc0x-iUnMad0JOtAr_nD5DM,273
 fleet/exceptions.py,sha256=fUmPwWhnT8SR97lYsRq0kLHQHKtSh2eJS0VQ2caSzEI,5055
-fleet/models.py,sha256=gH3hpQM5awRqagSGPIsjOleB8AdFND8E4s9ZkZc5c_0,11372
-fleet/tasks.py,sha256=Zxz1qAq4C7MT1ZuqIeGp6gf6RCAYSkFDQtm6Je5UEk4,1644
+fleet/models.py,sha256=qDw9_yfn9t9VDOWaNPlnevTWIEuJQNZ0EQQTpuooEas,11716
+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=XOociP8kFb7PnSifxuURpqBC7Tk5C4o_8nhSrBfQMcQ,12315
 fleet/_async/exceptions.py,sha256=fUmPwWhnT8SR97lYsRq0kLHQHKtSh2eJS0VQ2caSzEI,5055
-fleet/_async/models.py,sha256=gH3hpQM5awRqagSGPIsjOleB8AdFND8E4s9ZkZc5c_0,11372
-fleet/_async/tasks.py,sha256=w-0vVGfEuCWRHMEJ73SN141J7Lz2pD_0v-nNG4TyJTU,1645
+fleet/_async/models.py,sha256=aPjcwbGfyEoDaS5Iip3cAyy3j5NDqhP0wyrvtVvF3M8,11410
+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
@@ -41,8 +41,8 @@ fleet/_async/resources/sqlite.py,sha256=DvDLRI5dJ7_v4WkHw-Zday1M_FQUdzAqUCy9FtfO
 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/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
@@ -60,10 +60,10 @@ fleet/verifiers/decorator.py,sha256=Q-KHhicnIYFwX7FX_VZguzNfu8ZslqNUeWxcS2CwNVY,
 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.22.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-scripts/fix_sync_imports.py,sha256=BIQfnaOoQ7bwR1c-pDqH9ifN47W1bwL7OafWVXZNkuA,4368
+fleet_python-0.2.24.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.22.dist-info/METADATA,sha256=jktRiagpdxi-xyw-RiDk4csZI9FagVeJ1t3eTJQN9uU,3297
-fleet_python-0.2.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fleet_python-0.2.22.dist-info/top_level.txt,sha256=_3DSmTohvSDf3AIP_BYfGzhwO1ECFwuzg83X-wHCx3Y,23
-fleet_python-0.2.22.dist-info/RECORD,,
+fleet_python-0.2.24.dist-info/METADATA,sha256=JcD86EF6s8M3Z7Meyq30c4eYp1-gP4nqrBkcf9gE4B8,3297
+fleet_python-0.2.24.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fleet_python-0.2.24.dist-info/top_level.txt,sha256=_3DSmTohvSDf3AIP_BYfGzhwO1ECFwuzg83X-wHCx3Y,23
+fleet_python-0.2.24.dist-info/RECORD,,

scripts/fix_sync_imports.py

@@ -50,6 +50,12 @@ def fix_file(filepath: Path) -> bool:
         content = content.replace('from ..models import', 'from .models import')
         content = content.replace('from ..config import', 'from .config import')
     
+    # Fix __init__.py imports - the class is called SyncEnv, not Environment
+    if rel_path.parts[0] == '__init__.py' and len(rel_path.parts) == 1:
+        content = content.replace('from .client import Fleet, Environment', 'from .client import Fleet, SyncEnv')
+        content = content.replace('"Environment",', '"SyncEnv",')
+        content = content.replace("'Environment',", "'SyncEnv',")
+    
     # Fix playwright imports for sync version
     if 'playwright' in str(filepath):
         # Fix the import statement