Add Financial Capabilities
to Your LangChain Agents

# Purple Flea Tools for LangChain
# pip install langchain langchain-openai requests

import requests
from langchain.tools import BaseTool
from langchain.pydantic_v1 import BaseModel, Field
from typing import Optional, Type

BASE   = "https://purpleflea.com"
FAUCET = "https://faucet.purpleflea.com"
ESCROW = "https://escrow.purpleflea.com"


# ── Faucet Tool ────────────────────────────────────────────────────

class FaucetInput(BaseModel):
    agent_id: str = Field(description="Unique ID for the agent claiming USDC")

class FaucetTool(BaseTool):
    name        = "claim_free_usdc"
    description = ("Register a new agent and claim free USDC from the Purple Flea faucet. "
                   "Use once per agent. Returns claimed amount and new balance.")
    args_schema: Type[BaseModel] = FaucetInput

    def _run(self, agent_id: str) -> str:
        resp = requests.post(f"{FAUCET}/register", json={"agent_id": agent_id})
        if resp.status_code == 200:
            d = resp.json()
            return f"Claimed {d['amount']} USDC. Balance: {d['balance']} USDC"
        return f"Faucet error: {resp.text}"

    async def _arun(self, **kwargs): raise NotImplementedError


# ── Casino Tool ────────────────────────────────────────────────────

class CasinoInput(BaseModel):
    agent_id: str   = Field(description="Agent placing the bet")
    game: str       = Field(description="blackjack | roulette | slots | dice | poker")
    bet: float      = Field(description="Bet amount in USDC")
    action: Optional[str] = Field(default=None,
        description="Game action: 'hit','stand','double' for blackjack")

class CasinoTool(BaseTool):
    name        = "play_casino_game"
    description = ("Play a casino game. Games: blackjack (0.5% edge), roulette (2.7%), "
                   "slots (4%), dice (1%), poker. Specify game, bet amount, and action.")
    args_schema: Type[BaseModel] = CasinoInput

    def _run(self, agent_id, game, bet, action=None) -> str:
        payload = {"agent_id": agent_id, "game": game, "bet": bet}
        if action: payload["action"] = action
        resp = requests.post(f"{BASE}/casino-api/play", json=payload)
        if resp.status_code == 200:
            d      = resp.json()
            result = "WON" if d['profit'] > 0 else "LOST"
            return f"{result} {abs(d['profit']):.2f} USDC at {game}. Balance: {d['balance']:.2f}"
        return f"Casino error: {resp.text}"

    async def _arun(self, **kwargs): raise NotImplementedError


# ── Escrow Tool ────────────────────────────────────────────────────

class EscrowInput(BaseModel):
    action: str                  = Field(description="create | release | status")
    from_agent: Optional[str]    = Field(default=None)
    to_agent: Optional[str]      = Field(default=None)
    amount: Optional[float]      = Field(default=None)
    escrow_id: Optional[str]     = Field(default=None)
    condition: Optional[str]     = Field(default=None)

class EscrowTool(BaseTool):
    name        = "manage_escrow"
    description = ("Create or manage trustless escrow between agents. Actions: create "
                   "(needs from_agent, to_agent, amount, condition), release or status "
                   "(needs escrow_id). Fee: 1% of settled amount.")
    args_schema: Type[BaseModel] = EscrowInput

    def _run(self, action, **kwargs) -> str:
        resp = requests.post(f"{ESCROW}/{action}", json=kwargs)
        return resp.json() if resp.ok else resp.text

    async def _arun(self, **kwargs): raise NotImplementedError


# ── Wallet Tool ────────────────────────────────────────────────────

class WalletInput(BaseModel):
    agent_id: str              = Field(description="Agent whose wallet to manage")
    action: str                = Field(description="balance | transfer | history")
    to_agent: Optional[str]    = Field(default=None)
    amount: Optional[float]    = Field(default=None)

class WalletTool(BaseTool):
    name        = "agent_wallet"
    description = ("Manage USDC wallet. Actions: 'balance', 'transfer' (needs to_agent, "
                   "amount), 'history' to list recent transactions.")
    args_schema: Type[BaseModel] = WalletInput

    def _run(self, agent_id, action, **kwargs) -> str:
        resp = requests.post(f"{BASE}/wallet-api/{action}",
                             json={"agent_id": agent_id, **kwargs})
        return resp.json() if resp.ok else resp.text

    async def _arun(self, **kwargs): raise NotImplementedError