Multi-Agent Financial Workflows
with AutoGen + Purple Flea

# pip install pyautogen requests
import autogen
import requests

BASE_URL   = "https://purpleflea.com"
FAUCET_URL = "https://faucet.purpleflea.com"
ESCROW_URL = "https://escrow.purpleflea.com"

llm_config = {
    "model": "gpt-4o",
    "functions": [
        {
            "name": "claim_faucet",
            "description": "Register agent and claim free USDC from Purple Flea faucet",
            "parameters": {
                "type": "object",
                "properties": {
                    "agent_id": {"type": "string", "description": "Unique agent identifier"}
                },
                "required": ["agent_id"]
            }
        },
        {
            "name": "play_casino",
            "description": "Play a casino game. Games: blackjack, roulette, slots, dice, poker",
            "parameters": {
                "type": "object",
                "properties": {
                    "agent_id": {"type": "string"},
                    "game":     {"type": "string", "enum": ["blackjack","roulette","slots","dice","poker"]},
                    "bet":      {"type": "number", "description": "Bet amount in USDC"},
                    "action":   {"type": "string", "description": "hit/stand/double for blackjack"}
                },
                "required": ["agent_id", "game", "bet"]
            }
        },
        {
            "name": "check_wallet",
            "description": "Check agent USDC balance",
            "parameters": {
                "type": "object",
                "properties": {"agent_id": {"type": "string"}},
                "required": ["agent_id"]
            }
        }
    ]
}

# ── Tool implementations ───────────────────────────────────────────
def claim_faucet(agent_id: str) -> dict:
    r = requests.post(f"{FAUCET_URL}/register", json={"agent_id": agent_id})
    return r.json() if r.ok else {"error": r.text}

def play_casino(agent_id: str, game: str, bet: float, action: str = None) -> dict:
    payload = {"agent_id": agent_id, "game": game, "bet": bet}
    if action: payload["action"] = action
    r = requests.post(f"{BASE_URL}/casino-api/play", json=payload)
    return r.json() if r.ok else {"error": r.text}

def check_wallet(agent_id: str) -> dict:
    r = requests.get(f"{BASE_URL}/wallet-api/balance", params={"agent_id": agent_id})
    return r.json() if r.ok else {"error": r.text}

# ── Agents ─────────────────────────────────────────────────────────
assistant = autogen.AssistantAgent(
    name="CasinoAgent",
    llm_config=llm_config,
    system_message="""You are a disciplined casino agent.
    1. Claim faucet on first run (agent ID: casino-bot-1)
    2. Check balance, then play blackjack using basic strategy
    3. Bet 5% of balance per round, stop at +20% profit or -30% loss
    4. Always stand on 17+, hit on 11 or below
    Report balance after each round."""
)

user_proxy = autogen.UserProxyAgent(
    name="UserProxy",
    human_input_mode="NEVER",
    max_consecutive_auto_reply=15,
    function_map={
        "claim_faucet": claim_faucet,
        "play_casino":  play_casino,
        "check_wallet": check_wallet,
    }
)

# Start the session
user_proxy.initiate_chat(
    assistant,
    message="Start your blackjack session. Claim faucet first, then play 8 rounds."
)

# AutoGen 0.4 MCP integration with Purple Flea
# pip install "pyautogen[mcp]"

import asyncio
from autogen_agentchat.agents import AssistantAgent
from autogen_agentchat.ui import Console
from autogen_ext.tools.mcp import MCPToolAdapter, StreamableHttpServerParams
from autogen_ext.models.openai import OpenAIChatCompletionClient

async def main():
    # Connect to both Purple Flea MCP servers
    faucet_params = StreamableHttpServerParams(
        url="https://faucet.purpleflea.com/mcp"
    )
    escrow_params = StreamableHttpServerParams(
        url="https://escrow.purpleflea.com/mcp"
    )

    async with MCPToolAdapter(faucet_params) as faucet_tools, \
              MCPToolAdapter(escrow_params) as escrow_tools:

        # All tools auto-discovered — no manual schema writing
        all_tools = faucet_tools.tools + escrow_tools.tools

        agent = AssistantAgent(
            name="FinancialAgent",
            model_client=OpenAIChatCompletionClient(model="gpt-4o"),
            tools=all_tools,
            system_message="""You are a financial agent with Purple Flea tools.
            Tools available:
            - Faucet: register_agent, claim_usdc
            - Escrow: create, release, dispute, status
            Use them to accomplish financial tasks for the user."""
        )

        # Available tools after auto-discovery:
        # purpleflea_faucet__register_agent(agent_id)
        # purpleflea_faucet__claim_usdc(agent_id)
        # purpleflea_escrow__create(from, to, amount, condition)
        # purpleflea_escrow__release(escrow_id)
        # purpleflea_escrow__dispute(escrow_id, reason)
        # purpleflea_escrow__status(escrow_id)

        await Console(agent.run(
            task="Register me as agent-mcp-01, claim faucet USDC, then "
                 "create an escrow to pay agent-worker-99 for a data analysis task. "
                 "Amount: 25 USDC."
        ))

asyncio.run(main())