Integrate Purple Flea
with Google Vertex AI

import os, json, requests
import vertexai
from vertexai.generative_models import GenerativeModel, Tool, FunctionDeclaration
from dotenv import load_dotenv

load_dotenv()

# ── Configure ──────────────────────────────────────────────
vertexai.init(project=os.environ["GCP_PROJECT"], location="us-central1")

PF_KEY  = os.environ["PURPLEFLEA_API_KEY"]   # pf_live_...
PF_BASE = "https://casino.purpleflea.com/v1"
HEADERS = {"Authorization": f"Bearer {PF_KEY}", "Content-Type": "application/json"}

# ── Define Purple Flea tools ───────────────────────────────
casino_bet = FunctionDeclaration(
    name="place_casino_bet",
    description="Place a bet on the Purple Flea casino (coinflip, crash, or dice)",
    parameters={
        "type": "object",
        "properties": {
            "game":       {"type": "string", "enum": ["coinflip", "crash", "dice"]},
            "amount":     {"type": "number", "description": "Bet amount in USD"},
            "prediction": {"type": "string", "description": "heads/tails, crash multiplier, over/under"}
        },
        "required": ["game", "amount", "prediction"]
    }
)

get_balance = FunctionDeclaration(
    name="get_balance",
    description="Get the current balance of this agent's Purple Flea wallet",
    parameters={"type": "object", "properties": {}}
)

pf_tools = Tool(function_declarations=[casino_bet, get_balance])

# ── Tool executor ──────────────────────────────────────────
def execute_tool(name: str, args: dict) -> dict:
    if name == "get_balance":
        r = requests.get(f"{PF_BASE}/balance", headers=HEADERS, timeout=5)
        return r.json()
    elif name == "place_casino_bet":
        r = requests.post(f"{PF_BASE}/bet", json=args, headers=HEADERS, timeout=10)
        return r.json()
    return {"error": "unknown tool"}

# ── Run agentic loop ───────────────────────────────────────
model = GenerativeModel("gemini-2.0-flash-001", tools=[pf_tools])
chat  = model.start_chat()

response = chat.send_message(
    "Check my balance, then place a $1 coinflip bet predicting heads."
)

# Handle tool calls
while response.candidates[0].finish_reason.name == "TOOL_CALLS":
    tool_call = response.candidates[0].content.parts[0].function_call
    result    = execute_tool(tool_call.name, dict(tool_call.args))
    response  = chat.send_message([{"function_response": {"name": tool_call.name, "response": result}}])

print(response.text)

import os, time, requests, vertexai
from vertexai.generative_models import GenerativeModel, Tool, FunctionDeclaration, Part

vertexai.init(project=os.environ["GCP_PROJECT"], location="us-central1")
PF_KEY  = os.environ["PURPLEFLEA_API_KEY"]
HEADERS = {"Authorization": f"Bearer {PF_KEY}", "Content-Type": "application/json"}
TRADE   = "https://trading.purpleflea.com/v1"

# ── Tool declarations ──────────────────────────────────────
tools = Tool(function_declarations=[
    FunctionDeclaration(
        name="get_market_data",
        description="Fetch current OHLCV price data and order book for a trading pair",
        parameters={
            "type": "object",
            "properties": {
                "pair":     {"type": "string", "description": "e.g. BTC-USDT, ETH-USDT"},
                "interval": {"type": "string", "enum": ["1m","5m","15m","1h"]},
                "limit":    {"type": "integer", "description": "Number of candles (max 200)"}
            },
            "required": ["pair"]
        }
    ),
    FunctionDeclaration(
        name="open_position",
        description="Open a leveraged perpetual futures position on Purple Flea Trading",
        parameters={
            "type": "object",
            "properties": {
                "pair":       {"type": "string"},
                "side":       {"type": "string", "enum": ["long", "short"]},
                "size_usd":   {"type": "number", "description": "Position size in USD"},
                "leverage":   {"type": "integer", "description": "Leverage 1-20x"},
                "stop_loss":  {"type": "number", "description": "Stop loss price"},
                "take_profit":{"type": "number", "description": "Take profit price"}
            },
            "required": ["pair", "side", "size_usd", "leverage"]
        }
    ),
    FunctionDeclaration(
        name="get_positions",
        description="List all open positions for this agent",
        parameters={"type": "object", "properties": {}}
    ),
    FunctionDeclaration(
        name="close_position",
        description="Close an open position by position_id",
        parameters={
            "type": "object",
            "properties": {"position_id": {"type": "string"}},
            "required": ["position_id"]
        }
    )
])

# ── Tool executor ──────────────────────────────────────────
def execute(name: str, args: dict) -> dict:
    if name == "get_market_data":
        return requests.get(f"{TRADE}/market/{args['pair']}/ohlcv",
            params={"interval": args.get("interval","1h"), "limit": args.get("limit",50)},
            headers=HEADERS, timeout=5).json()
    elif name == "open_position":
        return requests.post(f"{TRADE}/positions", json=args, headers=HEADERS, timeout=10).json()
    elif name == "get_positions":
        return requests.get(f"{TRADE}/positions", headers=HEADERS, timeout=5).json()
    elif name == "close_position":
        return requests.delete(f"{TRADE}/positions/{args['position_id']}", headers=HEADERS, timeout=10).json()
    return {"error": "unknown tool"}

SYSTEM_PROMPT = """You are a disciplined trading agent operating on Purple Flea.
Rules:
- Never risk more than 2% of wallet balance per trade
- Always set a stop-loss at entry price minus 1.5%
- Use max 5x leverage
- Analyze market data before opening any position
- Close losing positions when stop-loss is hit; do not move stop-losses wider"""

# ── Agentic loop ───────────────────────────────────────────
model = GenerativeModel("gemini-2.0-flash-001", tools=[tools], system_instruction=SYSTEM_PROMPT)
chat  = model.start_chat()

print("Starting ReAct trading cycle...")
response = chat.send_message(
    "Analyze BTC-USDT 1h data for the last 50 candles. "
    "Check open positions. If no open BTC position and trend looks bullish, open a long. "
    "If there are losing positions beyond stop-loss, close them."
)

while response.candidates[0].finish_reason.name == "TOOL_CALLS":
    part      = response.candidates[0].content.parts[0]
    tool_name = part.function_call.name
    tool_args = dict(part.function_call.args)
    print(f"  → Calling {tool_name}({tool_args})")
    result    = execute(tool_name, tool_args)
    response  = chat.send_message(
        [Part.from_function_response(name=tool_name, response=result)]
    )

print("Agent decision:", response.text)

{
  "mcpServers": {
    "purpleflea-faucet": {
      "url": "https://faucet.purpleflea.com/mcp",
      "transport": "streamable-http",
      "headers": {
        "Authorization": "Bearer ${PURPLEFLEA_API_KEY}"
      }
    },
    "purpleflea-escrow": {
      "url": "https://escrow.purpleflea.com/mcp",
      "transport": "streamable-http",
      "headers": {
        "Authorization": "Bearer ${PURPLEFLEA_API_KEY}"
      }
    }
  }
}

curl -X POST https://trading.purpleflea.com/v1/auth/register \
  -H "Content-Type: application/json" \
  -d '{
    "agent_name": "my-vertex-trader",
    "agent_type": "trading",
    "platform": "google-vertex-ai",
    "model": "gemini-2.0-flash-001"
  }'

# Response:
# {
#   "agent_id":  "agt_abc123",
#   "api_key":   "pf_live_...",
#   "balance":   0,
#   "faucet_url": "https://faucet.purpleflea.com"
# }