TravelAI

Agent

Travel Info

Select your travel dates and locations

Start Date

End Date

Departure Location

Arrival Location

Trip Goals

What are your goals for this trip?

Flight Preferences

Choose your preferred flight options

Preferred Flight Times

Morning

Afternoon

Evening

Price vs. Flight Time Priority

Prioritize PricePrioritize Convenience

Accommodation Preferences

Accommodation Type

HotelAirbnb

Attraction Preferences

Interests

Museums & CultureSports & RecreationNightlife & EntertainmentFood & DiningShoppingNature & OutdoorsHistorical Sites

How It Works

This travel planner uses AI agents to help plan your trip. Here's how the code works:

1. Assemble Agent

The agent is is defined with a name, a set of instructions, a model, and a collection of enabled tools:

src/mastra/agents/index.ts

1import { Agent } from '@mastra/core';
2import { anthropic } from '@ai-sdk/anthropic';
3
4export const travelAgent = new Agent({
5  name: 'travelAgent',
6  instructions:
7    'You are an expert travel agent responsible for finding a flight, hotel,
8    and three attractions for a user. You will be given a set of user
9    preferences along with some tools and you will need to find the best
10    options for them.',
11  model: anthropic('claude-3-5-sonnet-20240620'),
12  enabledTools: {
13    searchFlights: true,
14    searchHotels: true,
15    searchAttractions: true,
16  },
17});

2. Agent Tools

Tools are just functions that can be used by the agent. You can define these functions and then enable them for any agent you create. These tools can also be used within workflows. Here is the code for the searchFlights tool

mastra/agents/index.ts

1export const searchFlights = createTool({
2  label: 'Get Flight Info',
3  schema: z.object({
4    startDate: z.string(),
5    endDate: z.string(),
6    origin: z.string(),
7    destination: z.string(),
8  }),
9  description: 'Fetches flight information for a given date range, origin and destination',
10  executor: async ({ data: { startDate, endDate, origin, destination } }) => {
11    console.log('Using tool to fetch flight information: ', startDate, endDate, origin, destination);
12    const flights = await getFlights(startDate, endDate, origin, destination);
13    return {
14      flights: flights as Flight[],
15    };
16  },
17});
18

Note: The label and description of the tool are both important to help the model decide when to call the tool. It is also a good idea to pass in information in the Agent instructions that tells the model when certain tools should be used.

The getFlights function just makes a GET request to a flight search API and then pulls out only the data we want to pass back to the model.

app/travel-results.tsx

1export const getFlights = async (startDate: string, endDate: string, origin: string, destination: string) => {
2  const url = `https://booking-com15.p.rapidapi.com/api/v1/flights/searchFlights?fromId=${origin}&toId=${destination}&departDate=${startDate}&returnDate=${endDate}&pageNo=1&adults=1&sort=BEST&cabinClass=ECONOMY&currency_code=USD`;
3  const options = {
4    method: 'GET',
5    headers: {
6      'x-rapidapi-key': process.env.RAPID_API_KEY || '',
7      'x-rapidapi-host': 'booking-com15.p.rapidapi.com',
8    },
9  };
10
11  try {
12    const response = await fetch(url, options);
13    const result = await response.json();
14
15    return result?.data?.flightOffers.map(
16      (flight: FlightApiResponse): Flight => ({
17        airline: flight.segments[0].legs[0].carriersData[0].name,
18        flightNumber: `${flight.segments[0].legs[0].flightInfo.carrierInfo.marketingCarrier}${flight.segments[0].legs[0].flightInfo.flightNumber}`,
19        departureAirport: flight.segments[0].departureAirport.code,
20        departureCity: flight.segments[0].departureAirport.cityName,
21        departureTime: new Date(flight.segments[0].departureTime),
22        arrivalAirport: flight.segments[0].arrivalAirport.code,
23        arrivalCity: flight.segments[0].arrivalAirport.cityName,
24        arrivalTime: new Date(flight.segments[0].arrivalTime),
25        duration: `${Math.floor(flight.segments[0].totalTime / 60)}h ${flight.segments[0].totalTime % 60}m`,
26        price: flight.priceBreakdown.total.units + flight.priceBreakdown.total.nanos / 1000000000,
27      }),
28    );
29  } catch (error) {
30    console.error(error);
31  }
32};