cli.py

"""
cli.py - Headless CLI for the Agentic Wallet SDK.
Usage:
  python cli.py "Check my balance"   # single command
  python cli.py --interactive         # REPL
"""
import argparse, asyncio, os, sys
sys.path.insert(0, os.path.dirname(__file__))


async def run_once(instruction, keypair, network, model):
    from agent import create_agent
    key = os.getenv("OPENROUTER_API_KEY", "")
    if not key: print("ERROR: set OPENROUTER_API_KEY"); sys.exit(1)
    agent = await create_agent(keypair_path=keypair, network=network,
                                openrouter_api_key=key, model=model)
    try:
        r = await agent.run(instruction)
        print("\n" + "="*60)
        print("RESULT:", r["summary"])
        print(f"Iterations: {r.get('iterations')}")
        print("="*60)
    finally:
        await agent.close()


async def repl(keypair, network, model):
    from agent import create_agent
    key = os.getenv("OPENROUTER_API_KEY", "")
    if not key: print("ERROR: set OPENROUTER_API_KEY"); sys.exit(1)
    agent = await create_agent(keypair_path=keypair, network=network,
                                openrouter_api_key=key, model=model)
    print(f"\nAgentic Wallet REPL | {agent.wallet.get_public_key()[:20]}... | {network}")
    print("Type instruction or q to exit\n")
    try:
        while True:
            try: line = input("wallet> ").strip()
            except (EOFError, KeyboardInterrupt): break
            if not line: continue
            if line.lower() in ("q","quit","exit"): break
            r = await agent.run(line)
            print(f"\n-> {r['summary']}\n")
    finally:
        await agent.close()


def main():
    p = argparse.ArgumentParser(description="Agentic Wallet CLI")
    p.add_argument("instruction", nargs="?", default=None)
    p.add_argument("--interactive", action="store_true")
    p.add_argument("--keypair", default=None)
    p.add_argument("--network", default="devnet", choices=["devnet","mainnet-beta"])
    p.add_argument("--model",   default="anthropic/claude-3.5-haiku")
    a = p.parse_args()
    if a.interactive:  asyncio.run(repl(a.keypair, a.network, a.model))
    elif a.instruction: asyncio.run(run_once(a.instruction, a.keypair, a.network, a.model))
    else: p.print_help()


if __name__ == "__main__":
    main()