agent_test.py

# examples/agent_test.py
# Test hiring a robot using the Python RTPClient

import asyncio
import os
from dotenv import load_dotenv
load_dotenv()

from rtp import RTPClient


async def main():
    client = RTPClient(
        wallet=os.getenv("WALLET_ADDRESS"),
        gateway_url="https://gateway.spraay.app"
    )

    print("\n💧 RTP Agent Test (Python)\n" + "─" * 40)

    # Discover robots
    print("🔍 Discovering robots with 'pick' capability...")
    robots = await client.discover(capability="pick")

    if not robots:
        print("❌ No robots found")
        return

    robot = robots[0]
    print(f"✅ Found: {robot['name']} ({robot['robot_id']})")
    print(f"💰 Price: {robot['price_per_task']} USDC\n")

    # Run test sequence
    tasks = [
        ("pick",  {"item": "SKU-001", "from_location": "bin_A1"}),
        ("scan",  {"target": "barcode"}),
        ("place", {"to_location": "conveyor_1"}),
        ("move",  {"angle": 45}),
        ("move",  {"angle": 90}),
    ]

    passed = 0
    failed = 0

    for task_name, params in tasks:
        print(f"▶️  Testing: {task_name}")
        try:
            result = await client.hire(
                robot,
                task=task_name,
                parameters=params,
                timeout_seconds=15
            )
            if result["status"] == "COMPLETED":
                print(f"✅ {task_name} PASSED — {result['result']['output']}")
                passed += 1
            else:
                print(f"❌ {task_name} FAILED — {result['result'].get('error')}")
                failed += 1
        except Exception as e:
            print(f"❌ {task_name} ERROR — {e}")
            failed += 1

        await asyncio.sleep(1)

    print(f"\n📊 Results: {passed} passed, {failed} failed")
    if failed == 0:
        print("🎉 All tasks completed — robot is RTP ready!\n")


if __name__ == "__main__":
    asyncio.run(main())