Purple Flea Tools for Vercel AI SDK

# Install the Vercel AI SDK (if not already present) + Purple Flea adapter
npm install ai @purpleflea/vercel-ai-sdk zod

# Add your Purple Flea API key to your environment
echo "PURPLE_FLEA_API_KEY=pf_your_key_here" >> .env.local

import { tool } from 'ai';
import { z } from 'zod';
import { PurpleFleas } from '@purpleflea/vercel-ai-sdk';

const flea = new PurpleFleas({
  apiKey: process.env.PURPLE_FLEA_API_KEY,
});

export const cryptoTools = {

  // ── Wallet ──────────────────────────────────────────────
  createWallet: tool({
    description: 'Create a non-custodial BIP-39 crypto wallet for the agent',
    parameters: z.object({}),
    execute: async () => flea.wallet.create(),
  }),

  walletBalance: tool({
    description: 'Fetch confirmed balances across all supported chains for a wallet address',
    parameters: z.object({
      address: z.string().describe('Wallet public address or ENS name'),
    }),
    execute: async (params) => flea.wallet.balance(params.address),
  }),

  // ── Trading ──────────────────────────────────────────────
  openTrade: tool({
    description: 'Open a perpetual futures position via Hyperliquid DEX',
    parameters: z.object({
      market:   z.string().describe('Ticker symbol, e.g. BTC, ETH, SOL'),
      size:     z.number().positive().describe('Position size in USD'),
      side:     z.enum(['long', 'short']),
      leverage: z.number().default(1).describe('Leverage multiplier 1–50'),
    }),
    execute: async (params) => flea.trading.openPosition(params),
  }),

  // ── Casino ───────────────────────────────────────────────
  casinoFlip: tool({
    description: 'Flip a provably fair coin — returns heads or tails plus payout',
    parameters: z.object({
      amount: z.number().positive().describe('Wager in USDC'),
      choice: z.enum(['heads', 'tails']),
    }),
    execute: async (params) => flea.casino.coinFlip(params),
  }),

  // ── Domains ──────────────────────────────────────────────
  domainCheck: tool({
    description: 'Check availability and price for a .crypto / .eth / .x domain name',
    parameters: z.object({
      name: z.string().describe('Full domain name including TLD, e.g. myagent.crypto'),
    }),
    execute: async (params) => flea.domains.check(params.name),
  }),
};

import { streamText } from 'ai';
import { openai } from '@ai-sdk/openai';
import { cryptoTools } from '@/lib/crypto-tools';

export const runtime = 'nodejs'; // or 'edge'

export async function POST(req: Request) {
  const { messages } = await req.json();

  const result = await streamText({
    model: openai('gpt-4o'),
    system: 'You are a crypto-native AI agent. You can create wallets, open perpetual futures positions on Hyperliquid, play provably fair casino games, and register Web3 domains. Always confirm transaction details before executing.',
    messages,
    tools: {
      ...cryptoTools,
      // spread any other tools here
    },
    maxSteps: 5, // allow multi-step tool use
  });

  return result.toDataStreamResponse();
}

'use client';
import { useChat } from 'ai/react';

export default function Page() {
  const { messages, input, handleInputChange, handleSubmit } = useChat({
    api: '/api/chat',
  });

  return (
    <div>
      {messages.map(m => (
        <div key={m.id}>
          <strong>{m.role}</strong>: {m.content}
        </div>
      ))}
      <form onSubmit={handleSubmit}>
        <input value={input} onChange={handleInputChange}
          placeholder="Ask me to open a BTC trade or flip a coin..." />
        <button type="submit">Send</button>
      </form>
    </div>
  );
}