RunRelay Flights API

import { Client } from "@modelcontextprotocol/sdk/client/index.js";
import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";

const API_KEY = process.env.RUNRELAY_API_KEY!;
const BASE_URL = "https://api.runrelay.io";

async function bookFlight() {
  // ── 1. Create transport with auth ──────────────────────────────────
  const transport = new SSEClientTransport(new URL(`${BASE_URL}/sse`), {
    requestInit: {
      headers: {
        Authorization: `Bearer ${API_KEY}`,
      },
    },
  });

  // ── 2. Connect MCP client ──────────────────────────────────────────
  const client = new Client(
    { name: "my-flight-agent", version: "1.0.0" },
    { capabilities: {} }
  );

  await client.connect(transport);
  console.log("✅ Connected to RunRelay MCP");

  // ── 3. Search for flights ──────────────────────────────────────────
  const searchResult = await client.callTool({
    name: "search_flights",
    arguments: {
      origin: "BCN",
      destination: "LIS",
      departureDate: "2025-03-20",
      passengers: { adults: 1, children: 0, infants: 0 },
      cabinClass: "economy",
      maxResults: 5,
    },
  });

  const { offers } = searchResult.content[0] as { offers: any[] };
  console.log(`✈️  Found ${offers.length} flights`);

  if (offers.length === 0) {
    console.log("No flights found");
    return;
  }

  const bestOffer = offers[0];
  console.log(`   Best: ${bestOffer.airline} — €${bestOffer.totalPrice}`);

  // ── 4. Get full details for the selected offer ─────────────────────
  const details = await client.callTool({
    name: "get_flight_details",
    arguments: { offerId: bestOffer.offerId },
  });
  console.log("📋 Fare conditions:", details.content[0]);

  // ── 5. Check seat map (optional) ──────────────────────────────────
  try {
    const seatMap = await client.callTool({
      name: "get_seat_map",
      arguments: { offerId: bestOffer.offerId },
    });
    console.log("💺 Seat map available:", seatMap.content[0]);
  } catch {
    console.log("💺 Seat map not available for this airline");
  }

  // ── 6. Book the flight ─────────────────────────────────────────────
  const booking = await client.callTool({
    name: "book_flight",
    arguments: {
      offerId: bestOffer.offerId,
      passengers: [
        {
          type: "adult",
          firstName: "Jane",
          lastName: "Smith",
          dateOfBirth: "1985-06-15",
          passport: {
            number: "AB123456",
            expiryDate: "2030-01-01",
            issuingCountry: "GB",
          },
          email: "jane@example.com",
          phone: "+44 7700 900000",
        },
      ],
      paymentMethod: {
        type: "card",
        savedMethodId: "pm_saved_123",
      },
    },
  });

  const result = booking.content[0] as any;

  if (result.bookingReference) {
    console.log(`🎉 Booked! Reference: ${result.bookingReference}`);
  } else if (result.escalationTicketId) {
    // HITLaaS: human operator will complete the booking
    console.log(`🤝 Escalated to human operator: ${result.escalationTicketId}`);
    console.log("   You'll be notified when the booking is confirmed.");
  }

  await client.close();
}

bookFlight().catch(console.error);

// One-time setup: create user profile & add traveler
// After this, book_flight auto-fills passenger data!

// Create user profile
await client.callTool({
  name: "setup_user",
  arguments: {
    userId: "user_jane_smith",
    preferences: {
      seatPreference: "window",
      defaultCabin: "economy",
      confirmationStyle: "always_confirm", // ask before booking
      mealPreference: "vegetarian",
    },
  },
});

// Add a saved traveler (done once, reused forever)
await client.callTool({
  name: "add_traveler",
  arguments: {
    userId: "user_jane_smith",
    traveler: {
      firstName: "Jane",
      lastName: "Smith",
      dateOfBirth: "1985-06-15",
      passport: {
        number: "AB123456",
        expiryDate: "2030-01-01",
        issuingCountry: "GB",
        nationality: "GB",
      },
      frequentFlyer: [
        { airline: "IB", number: "IBE123456" }, // Iberia Plus
        { airline: "TP", number: "TP789012" },  // TAP Miles&Go
      ],
    },
  },
});

// Now book_flight will auto-fill Jane's details:
await client.callTool({
  name: "book_flight",
  arguments: {
    offerId: "offer_xyz",
    userId: "user_jane_smith", // passengers auto-populated!
    paymentMethod: { type: "card", savedMethodId: "pm_saved_123" },
  },
});