fleet-python 0.2.66a2__tar.gz → 0.2.99__tar.gz

{fleet_python-0.2.66a2/fleet_python.egg-info → fleet_python-0.2.99}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fleet-python
-Version: 0.2.66a2
+Version: 0.2.99
 Summary: Python SDK for Fleet environments
 Author-email: Fleet AI <nic@fleet.so>
 License: Apache-2.0
@@ -26,6 +26,9 @@ Requires-Dist: httpx-retries>=0.4.0
 Requires-Dist: typing-extensions>=4.0.0
 Requires-Dist: modulegraph2>=0.2.0
 Requires-Dist: cloudpickle==3.1.1
+Provides-Extra: cli
+Requires-Dist: typer>=0.9.0; extra == "cli"
+Requires-Dist: rich>=10.0.0; extra == "cli"
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0.0; extra == "dev"
 Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
@@ -35,12 +38,22 @@ Requires-Dist: mypy>=1.0.0; extra == "dev"
 Requires-Dist: ruff>=0.1.0; extra == "dev"
 Requires-Dist: unasync>=0.6.0; extra == "dev"
 Requires-Dist: python-dotenv>=1.1.1; extra == "dev"
+Requires-Dist: typer>=0.9.0; extra == "dev"
+Requires-Dist: rich>=10.0.0; extra == "dev"
 Provides-Extra: playwright
 Requires-Dist: playwright>=1.40.0; extra == "playwright"
+Provides-Extra: eval
+Requires-Dist: aiohttp>=3.9.0; extra == "eval"
+Requires-Dist: google-genai>=1.0.0; extra == "eval"
+Requires-Dist: mcp==1.24.0; python_version >= "3.10" and extra == "eval"
 Dynamic: license-file
 
 # Fleet SDK
 
+[![PyPI version](https://img.shields.io/pypi/v/fleet-python.svg)](https://pypi.org/project/fleet-python/)
+[![Python versions](https://img.shields.io/pypi/pyversions/fleet-python.svg)](https://pypi.org/project/fleet-python/)
+[![License](https://img.shields.io/pypi/l/fleet-python.svg)](https://pypi.org/project/fleet-python/)
+
 The Fleet Python SDK provides programmatic access to Fleet's environment infrastructure.
 
 ## Installation

{fleet_python-0.2.66a2 → fleet_python-0.2.99}/README.md

@@ -1,5 +1,9 @@
 # Fleet SDK
 
+[![PyPI version](https://img.shields.io/pypi/v/fleet-python.svg)](https://pypi.org/project/fleet-python/)
+[![Python versions](https://img.shields.io/pypi/pyversions/fleet-python.svg)](https://pypi.org/project/fleet-python/)
+[![License](https://img.shields.io/pypi/l/fleet-python.svg)](https://pypi.org/project/fleet-python/)
+
 The Fleet Python SDK provides programmatic access to Fleet's environment infrastructure.
 
 ## Installation

{fleet_python-0.2.66a2 → fleet_python-0.2.99}/examples/export_tasks.py

@@ -27,6 +27,12 @@ def main():
         help="Optional task project key to filter tasks",
         default=None,
     )
+    parser.add_argument(
+        "--env-key",
+        "-e",
+        help="Optional environment key to filter tasks",
+        default=None,
+    )
     parser.add_argument(
         "--output",
         "-o",
@@ -42,12 +48,13 @@ def main():
             args.project_key is not None,
             args.task_keys is not None,
             args.task_project_key is not None,
+            args.env_key is not None,
         ]
     )
 
     if filters_specified > 1:
         parser.error(
-            "Cannot specify multiple filters. Use only one of --project-key, --task-keys, or --task-project-key."
+            "Cannot specify multiple filters. Use only one of --project-key, --task-keys, --task-project-key, or --env-key."
         )
 
     # Get account info
@@ -66,6 +73,9 @@ def main():
     elif args.task_project_key:
         print(f"Loading tasks from task project: {args.task_project_key}")
         tasks = fleet.load_tasks(task_project_key=args.task_project_key)
+    elif args.env_key:
+        print(f"Loading tasks from environment: {args.env_key}")
+        tasks = fleet.load_tasks(env_key=args.env_key)
     else:
         print("Loading all tasks")
         tasks = fleet.load_tasks()

fleet_python-0.2.99/examples/fetch_tasks.py

@@ -0,0 +1,230 @@
+import argparse
+import asyncio
+import json
+import sys
+from typing import List, Dict, Any, Optional, Tuple
+import fleet
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
+async def fetch_task(
+    task_key: str, semaphore: asyncio.Semaphore
+) -> Tuple[str, Optional[Dict[str, Any]], Optional[str]]:
+    """
+    Fetch a single task from the Fleet API.
+
+    Args:
+        task_key: Task key to fetch
+        semaphore: Semaphore to limit concurrent requests
+
+    Returns:
+        Tuple of (task_key, task_data_dict, error_message)
+    """
+    async with semaphore:
+        try:
+            # Use load_tasks with keys parameter to get Task objects
+            tasks = await fleet.load_tasks_async(keys=[task_key])
+            if tasks:
+                task = tasks[0]
+                # Convert to dict using model_dump() like export_tasks.py does
+                return task_key, task.model_dump(), None
+            else:
+                return task_key, None, "Task not found"
+        except Exception as e:
+            error_msg = f"{type(e).__name__}: {str(e)}"
+            return task_key, None, error_msg
+
+
+async def fetch_tasks_batch(
+    task_keys: List[str], max_concurrent: int = 20
+) -> Dict[str, Dict[str, Any]]:
+    """
+    Fetch multiple tasks concurrently from the Fleet API.
+
+    Args:
+        task_keys: List of task keys to fetch
+        max_concurrent: Maximum number of concurrent requests
+
+    Returns:
+        Dictionary mapping task_key to task data
+    """
+    print(f"\nFetching {len(task_keys)} task(s) from Fleet API...")
+    print(f"Max concurrent requests: {max_concurrent}")
+
+    semaphore = asyncio.Semaphore(max_concurrent)
+
+    # Fetch all tasks concurrently
+    results = await asyncio.gather(
+        *[fetch_task(key, semaphore) for key in task_keys],
+        return_exceptions=True,
+    )
+
+    # Process results
+    fetched_tasks = {}
+    errors = []
+
+    for result in results:
+        if isinstance(result, Exception):
+            errors.append(f"Unexpected error: {result}")
+            continue
+
+        task_key, task_data, error = result
+
+        if error:
+            errors.append(f"{task_key}: {error}")
+            print(f"  ✗ {task_key}: {error}")
+        elif task_data:
+            # Task data is already a dict from model_dump()
+            fetched_tasks[task_key] = task_data
+            print(f"  ✓ {task_key}")
+
+    print(f"\n✓ Successfully fetched {len(fetched_tasks)} task(s)")
+
+    if errors:
+        print(f"\n⚠ {len(errors)} error(s) occurred:")
+        for error in errors[:10]:  # Show first 10
+            print(f"  - {error}")
+        if len(errors) > 10:
+            print(f"  ... and {len(errors) - 10} more")
+
+    return fetched_tasks
+
+
+async def main():
+    parser = argparse.ArgumentParser(
+        description="Fetch tasks from Fleet API and update JSON file",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+Examples:
+  # Fetch tasks and update in-place
+  %(prog)s tasks.json
+
+  # Fetch tasks and save to a new file
+  %(prog)s tasks.json --output updated_tasks.json
+
+  # Limit concurrent requests
+  %(prog)s tasks.json --max-concurrent 10
+        """,
+    )
+
+    parser.add_argument("json_file", help="Path to JSON file containing tasks")
+    parser.add_argument(
+        "--output",
+        "-o",
+        help="Path to output JSON file (defaults to overwriting input)",
+        default=None,
+    )
+    parser.add_argument(
+        "--max-concurrent",
+        "-c",
+        type=int,
+        default=20,
+        help="Maximum number of concurrent API requests (default: 20)",
+    )
+
+    args = parser.parse_args()
+
+    # Load JSON file
+    print(f"Reading tasks from: {args.json_file}")
+    try:
+        with open(args.json_file, "r", encoding="utf-8") as f:
+            tasks_data = json.load(f)
+    except FileNotFoundError:
+        print(f"✗ Error: File '{args.json_file}' not found")
+        sys.exit(1)
+    except json.JSONDecodeError as e:
+        print(f"✗ Error: Invalid JSON in '{args.json_file}': {e}")
+        sys.exit(1)
+
+    if not isinstance(tasks_data, list):
+        print("✗ Error: JSON file must contain an array of tasks")
+        sys.exit(1)
+
+    print(f"Found {len(tasks_data)} task(s) in file")
+
+    # Extract task keys
+    task_keys = []
+    missing_keys = []
+
+    for i, task in enumerate(tasks_data):
+        task_key = task.get("key") or task.get("id")
+        if task_key:
+            task_keys.append(task_key)
+        else:
+            missing_keys.append(f"Task at index {i}")
+
+    if missing_keys:
+        print(f"\n⚠ Warning: {len(missing_keys)} task(s) missing key/id:")
+        for key in missing_keys[:10]:
+            print(f"  - {key}")
+        if len(missing_keys) > 10:
+            print(f"  ... and {len(missing_keys) - 10} more")
+
+    if not task_keys:
+        print("\n✗ Error: No valid task keys found in JSON file")
+        sys.exit(1)
+
+    print(f"\nExtracted {len(task_keys)} task key(s)")
+
+    # Get account info
+    account = await fleet.env.account_async()
+    print(f"Fetching from team: {account.team_name}")
+
+    # Fetch tasks from API
+    fetched_tasks = await fetch_tasks_batch(task_keys, args.max_concurrent)
+
+    if not fetched_tasks:
+        print("\n✗ Error: No tasks were successfully fetched")
+        sys.exit(1)
+
+    # Update tasks in the original data
+    updated_count = 0
+    not_found = []
+
+    print("\nUpdating task data...")
+    for i, task in enumerate(tasks_data):
+        task_key = task.get("key") or task.get("id")
+
+        if task_key in fetched_tasks:
+            # Replace entire task with fetched data
+            tasks_data[i] = fetched_tasks[task_key]
+            updated_count += 1
+        else:
+            not_found.append(task_key or f"index {i}")
+
+    print(f"✓ Updated {updated_count} task(s)")
+
+    if not_found:
+        print(f"\n⚠ Warning: {len(not_found)} task(s) not fetched:")
+        for key in not_found[:10]:
+            print(f"  - {key}")
+        if len(not_found) > 10:
+            print(f"  ... and {len(not_found) - 10} more")
+
+    # Write output
+    output_file = args.output or args.json_file
+    print(f"\nWriting updated tasks to: {output_file}")
+
+    try:
+        with open(output_file, "w", encoding="utf-8") as f:
+            json.dump(tasks_data, f, indent=2, ensure_ascii=False)
+        print(f"✓ Successfully wrote {len(tasks_data)} task(s) to '{output_file}'")
+    except Exception as e:
+        print(f"✗ Error writing output file: {e}")
+        sys.exit(1)
+
+    # Summary
+    print("\n" + "=" * 60)
+    print("Summary:")
+    print(f"  Total tasks in file: {len(tasks_data)}")
+    print(f"  Successfully fetched: {len(fetched_tasks)}")
+    print(f"  Updated in file: {updated_count}")
+    print(f"  Failed to fetch: {len(task_keys) - len(fetched_tasks)}")
+    print("=" * 60)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())
+

{fleet_python-0.2.66a2 → fleet_python-0.2.99}/examples/import_tasks.py

@@ -210,9 +210,18 @@ async def main():
         action="store_true",
         help="Skip the verifier sanity check (not recommended)",
     )
+    parser.add_argument(
+        "--sanity-check-only",
+        action="store_true",
+        help="Only run the sanity check without importing tasks",
+    )
 
     args = parser.parse_args()
 
+    # Validate conflicting flags
+    if args.skip_sanity_check and args.sanity_check_only:
+        parser.error("Cannot use --skip-sanity-check and --sanity-check-only together")
+
     # Load and parse the JSON file
     try:
         with open(args.json_file, "r", encoding="utf-8") as f:
@@ -286,6 +295,12 @@ async def main():
         success, errors = await run_verifier_sanity_check(tasks, client)
         if not success:
             sys.exit(1)
+
+        # If only doing sanity check, exit successfully here
+        if args.sanity_check_only:
+            print("\n✓ Sanity check complete! (--sanity-check-only)")
+            print("Tasks are ready to import.")
+            sys.exit(0)
     else:
         print("\n⚠️  Skipping sanity check (--skip-sanity-check)")