Switch to generating flights at agent boot time

Author: anishathalye

README.md

@@ -32,11 +32,7 @@ This repository contains an example of a production-grade AI Agent that is integ
     ```console
     $ hatch run create-vector-database
     ```
-7. Generate flight schedule data:
-    ```console
-    $ hatch run generate-flights
-    ```
-8. Install frontend dependencies:
+7. Install frontend dependencies:
     ```console
     $ cd frontend
     $ npm install

pyproject.toml

@@ -32,7 +32,6 @@ create-demo-project = "python -m airline_agent.cleanlab_utils.create_new_demo_pr
 open-project = "python -m airline_agent.cleanlab_utils.open_project {args}"
 fetch-pages = "python -m airline_agent.data_preparation.fetch_pages {args}"
 create-vector-database = "python -m airline_agent.preprocessing.create_vector_database {args}"
-generate-flights = "python scripts/generate_flights.py {args}"
 backend-server = "python -m airline_agent.backend.app {args}"
 
 

src/airline_agent/backend/services/airline_chat.py

@@ -106,9 +106,7 @@ def get_cleanlab_project() -> Project:
     kb_path=str(pathlib.Path(__file__).parents[4] / "data/kb.json"),
     vector_index_path=str(pathlib.Path(__file__).parents[4] / "data/vector-db"),
 )
-booking = BookingTools(
-    flights_path=str(pathlib.Path(__file__).parents[4] / "data/flights.json"),
-)
+booking = BookingTools()
 project = get_cleanlab_project()
 agent = create_agent(kb, booking)
 

src/airline_agent/constants.py

@@ -7,6 +7,7 @@
 DEMO_DATE = date(2025, 11, 5)
 DEMO_DATETIME = datetime(2025, 11, 5, 14, 0, 0, tzinfo=ZoneInfo("America/Los_Angeles"))
 FLIGHT_DATA_DATE = DEMO_DATE + timedelta(days=7)
+FLIGHT_DATA_NUM_DAYS = 8
 
 OFFICIAL_DEMO_PROJECT_ID = "3aae1f96-2dda-492f-8c86-17d453d3c298"  # to copy configuration from
 STAGING_DEMO_PROJECT_ID = "6de236e4-c6e7-456c-b248-872236010992"