Add passenger count options to CLI and MCP

README.md

@@ -65,6 +65,9 @@ The MCP server provides two main tools:
 | `airlines`         | list   | Filter by airline codes (e.g., ['BA', 'AA'])        |
 | `sort_by`          | string | CHEAPEST, DURATION, DEPARTURE_TIME, or ARRIVAL_TIME |
 | `passengers`       | int    | Number of adult passengers                          |
+| `children`         | int    | Number of child passengers                          |
+| `infants_in_seat`  | int    | Number of infants traveling in their own seat       |
+| `infants_on_lap`   | int    | Number of lap infants                               |
 
 #### `search_dates` Parameters
 
@@ -82,6 +85,9 @@ The MCP server provides two main tools:
 | `airlines`         | list   | Filter by airline codes (e.g., ['BA', 'AA']) |
 | `sort_by_price`    | bool   | Sort results by price (lowest first)         |
 | `passengers`       | int    | Number of adult passengers                   |
+| `children`         | int    | Number of child passengers                   |
+| `infants_in_seat`  | int    | Number of infants traveling in their own seat |
+| `infants_on_lap`   | int    | Number of lap infants                        |
 
 ## Quick Start
 
@@ -185,6 +191,10 @@ fli dates JFK LHR \
 | `--class, -c`    | Cabin class           | `ECONOMY`, `BUSINESS`  |
 | `--stops, -s`    | Maximum stops         | `NON_STOP`, `ONE_STOP` |
 | `--sort, -o`     | Sort results by       | `CHEAPEST`, `DURATION` |
+| `--passengers`   | Number of adult passengers | `2`               |
+| `--children`     | Number of children    | `2`                    |
+| `--infants-in-seat` | Number of infants in seat | `1`             |
+| `--infants-on-lap` | Number of infants on lap | `1`               |
 | `--format`       | Output format         | `text`, `json`         |
 
 #### Dates Command (`fli dates`)
@@ -200,6 +210,10 @@ fli dates JFK LHR \
 | `--stops, -s`      | Maximum stops          | `NON_STOP`, `ONE_STOP` |
 | `--time`           | Departure time window  | `6-20`                 |
 | `--sort`           | Sort by price          | (flag)                 |
+| `--passengers`     | Number of adult passengers | `2`                |
+| `--children`       | Number of children     | `2`                    |
+| `--infants-in-seat` | Number of infants in seat | `1`                |
+| `--infants-on-lap` | Number of infants on lap | `1`                  |
 | `--[day]`          | Day filters            | `--monday`, `--friday` |
 | `--format`         | Output format          | `text`, `json`         |
 

docs/api/models.md

@@ -88,6 +88,13 @@ Type of trip for flight search.
 
 Configuration for passenger counts.
 
+Fields:
+
+- `adults`
+- `children`
+- `infants_in_seat`
+- `infants_on_lap`
+
 ::: fli.models.google_flights.PassengerInfo
 
 ### TimeRestrictions

docs/examples/advanced.md

@@ -35,6 +35,7 @@ filters = FlightSearchFilters(
     passenger_info=PassengerInfo(
         adults=2,
         children=1,
+        infants_in_seat=1,
         infants_on_lap=1
     ),
     flight_segments=[
@@ -280,6 +281,7 @@ filters = FlightSearchFilters(
     passenger_info=PassengerInfo(
         adults=2,
         children=1,
+        infants_in_seat=1,
         infants_on_lap=1
     ),
     flight_segments=[outbound, return_flight],

docs/guides/mcp.md

@@ -78,6 +78,9 @@ Search for flights between two airports on a specific date.
 | `airlines` | list | No | null | Filter by airline codes (e.g., ['BA', 'AA']) |
 | `sort_by` | string | No | CHEAPEST | CHEAPEST, DURATION, DEPARTURE_TIME, or ARRIVAL_TIME |
 | `passengers` | int | No | 1 | Number of adult passengers |
+| `children` | int | No | 0 | Number of child passengers |
+| `infants_in_seat` | int | No | 0 | Number of infants traveling in their own seat |
+| `infants_on_lap` | int | No | 0 | Number of lap infants |
 
 **Example Response:**
 
@@ -126,6 +129,9 @@ Find the cheapest travel dates between two airports within a date range.
 | `airlines` | list | No | null | Filter by airline codes (e.g., ['BA', 'AA']) |
 | `sort_by_price` | bool | No | false | Sort results by price (lowest first) |
 | `passengers` | int | No | 1 | Number of adult passengers |
+| `children` | int | No | 0 | Number of child passengers |
+| `infants_in_seat` | int | No | 0 | Number of infants traveling in their own seat |
+| `infants_on_lap` | int | No | 0 | Number of lap infants |
 
 **Example Response:**
 

examples/README.md

@@ -211,7 +211,7 @@ from fli.models import PassengerInfo
 PassengerInfo(adults=1)
 
 # Family with children
-PassengerInfo(adults=2, children=2, infants_on_lap=1)
+PassengerInfo(adults=2, children=2, infants_in_seat=1, infants_on_lap=1)
 ```
 
 ## Support

examples/complex_flight_search.py

@@ -25,7 +25,12 @@ def main():
     # Create detailed filters
     filters = FlightSearchFilters(
         trip_type=TripType.ONE_WAY,
-        passenger_info=PassengerInfo(adults=2, children=1, infants_on_lap=1),
+        passenger_info=PassengerInfo(
+            adults=2,
+            children=1,
+            infants_in_seat=1,
+            infants_on_lap=1,
+        ),
         flight_segments=[
             FlightSegment(
                 departure_airport=[[Airport.JFK, 0]],

examples/complex_round_trip_validation.py

@@ -80,7 +80,12 @@ def main():
     # Create filters with complex requirements
     filters = FlightSearchFilters(
         trip_type=TripType.ROUND_TRIP,
-        passenger_info=PassengerInfo(adults=2, children=1, infants_on_lap=1),
+        passenger_info=PassengerInfo(
+            adults=2,
+            children=1,
+            infants_in_seat=1,
+            infants_on_lap=1,
+        ),
         flight_segments=[outbound, return_flight],
         stops=MaxStops.ONE_STOP_OR_FEWER,
         seat_type=SeatType.BUSINESS,

fli/cli/commands/dates.py

@@ -80,6 +80,38 @@ def dates(
             help="Trip duration in days",
         ),
     ] = 3,
+    passengers: Annotated[
+        int,
+        typer.Option(
+            "--passengers",
+            help="Number of adult passengers",
+            min=1,
+        ),
+    ] = 1,
+    children: Annotated[
+        int,
+        typer.Option(
+            "--children",
+            help="Number of children",
+            min=0,
+        ),
+    ] = 0,
+    infants_in_seat: Annotated[
+        int,
+        typer.Option(
+            "--infants-in-seat",
+            help="Number of infants in seat",
+            min=0,
+        ),
+    ] = 0,
+    infants_on_lap: Annotated[
+        int,
+        typer.Option(
+            "--infants-on-lap",
+            help="Number of infants on lap",
+            min=0,
+        ),
+    ] = 0,
     airlines: Annotated[
         list[str] | None,
         typer.Option(
@@ -226,6 +258,10 @@ def dates(
             "end_date": end_date,
             "trip_duration": trip_duration,
             "is_round_trip": is_round_trip,
+            "passengers": passengers,
+            "children": children,
+            "infants_in_seat": infants_in_seat,
+            "infants_on_lap": infants_on_lap,
             "cabin_class": seat_type.name,
             "max_stops": stops.name,
             "departure_window": (
@@ -260,7 +296,12 @@ def dates(
         # Create search filters
         filters = DateSearchFilters(
             trip_type=trip_type,
-            passenger_info=PassengerInfo(adults=1),
+            passenger_info=PassengerInfo(
+                adults=passengers,
+                children=children,
+                infants_in_seat=infants_in_seat,
+                infants_on_lap=infants_on_lap,
+            ),
             flight_segments=segments,
             stops=stops,
             seat_type=seat_type,
@@ -320,6 +361,10 @@ def dates(
                         "end_date": end_date,
                         "trip_duration": trip_duration,
                         "is_round_trip": is_round_trip,
+                        "passengers": passengers,
+                        "children": children,
+                        "infants_in_seat": infants_in_seat,
+                        "infants_on_lap": infants_on_lap,
                         "cabin_class": cabin_class,
                         "max_stops": max_stops,
                         "departure_window": (

fli/cli/commands/flights.py

@@ -36,6 +36,10 @@ def _search_flights_core(
     departure_date: str,
     return_date: str | None = None,
     departure_window: str | tuple[int, int] | None = None,
+    passengers: int = 1,
+    children: int = 0,
+    infants_in_seat: int = 0,
+    infants_on_lap: int = 0,
     airlines: list[str] | None = None,
     cabin_class: str = "ECONOMY",
     max_stops: str = "ANY",
@@ -50,6 +54,10 @@ def _search_flights_core(
         "departure_date": departure_date,
         "return_date": return_date,
         "departure_window": None,
+        "passengers": passengers,
+        "children": children,
+        "infants_in_seat": infants_in_seat,
+        "infants_on_lap": infants_on_lap,
         "airlines": [airline.upper() for airline in airlines] if airlines else None,
         "cabin_class": cabin_class.upper(),
         "max_stops": max_stops.upper(),
@@ -96,7 +104,12 @@ def _search_flights_core(
         # Create search filters
         filters = FlightSearchFilters(
             trip_type=trip_type,
-            passenger_info=PassengerInfo(adults=1),
+            passenger_info=PassengerInfo(
+                adults=passengers,
+                children=children,
+                infants_in_seat=infants_in_seat,
+                infants_on_lap=infants_on_lap,
+            ),
             flight_segments=segments,
             stops=stops,
             seat_type=seat_type,
@@ -188,6 +201,38 @@ def flights(
             help="Departure time window in 24h format (e.g., 6-20)",
         ),
     ] = None,
+    passengers: Annotated[
+        int,
+        typer.Option(
+            "--passengers",
+            help="Number of adult passengers",
+            min=1,
+        ),
+    ] = 1,
+    children: Annotated[
+        int,
+        typer.Option(
+            "--children",
+            help="Number of children",
+            min=0,
+        ),
+    ] = 0,
+    infants_in_seat: Annotated[
+        int,
+        typer.Option(
+            "--infants-in-seat",
+            help="Number of infants in seat",
+            min=0,
+        ),
+    ] = 0,
+    infants_on_lap: Annotated[
+        int,
+        typer.Option(
+            "--infants-on-lap",
+            help="Number of infants on lap",
+            min=0,
+        ),
+    ] = 0,
     airlines: Annotated[
         list[str] | None,
         typer.Option(
@@ -251,6 +296,10 @@ def flights(
         departure_date=departure_date,
         return_date=return_date,
         departure_window=departure_window,
+        passengers=passengers,
+        children=children,
+        infants_in_seat=infants_in_seat,
+        infants_on_lap=infants_on_lap,
         airlines=airlines,
         cabin_class=cabin_class,
         max_stops=max_stops,

fli/mcp/server.py

@@ -240,6 +240,9 @@ class FlightSearchParams(BaseModel):
         ge=1,
         description="Number of adult passengers",
     )
+    children: int = Field(0, ge=0, description="Number of children")
+    infants_in_seat: int = Field(0, ge=0, description="Number of infants in seat")
+    infants_on_lap: int = Field(0, ge=0, description="Number of infants on lap")
     exclude_basic_economy: bool = Field(
         False, description="Exclude basic economy fares from results"
     )
@@ -275,6 +278,9 @@ class DateSearchParams(BaseModel):
         ge=1,
         description="Number of adult passengers",
     )
+    children: int = Field(0, ge=0, description="Number of children")
+    infants_in_seat: int = Field(0, ge=0, description="Number of infants in seat")
+    infants_on_lap: int = Field(0, ge=0, description="Number of infants on lap")
 
 
 # =============================================================================
@@ -359,7 +365,12 @@ def _execute_flight_search(params: FlightSearchParams) -> dict[str, Any]:
         # Create search filters
         filters = FlightSearchFilters(
             trip_type=trip_type,
-            passenger_info=PassengerInfo(adults=params.passengers),
+            passenger_info=PassengerInfo(
+                adults=params.passengers,
+                children=params.children,
+                infants_in_seat=params.infants_in_seat,
+                infants_on_lap=params.infants_on_lap,
+            ),
             flight_segments=segments,
             stops=max_stops,
             seat_type=cabin_class,
@@ -425,7 +436,12 @@ def _execute_date_search(params: DateSearchParams) -> dict[str, Any]:
         # Create search filters
         filters = DateSearchFilters(
             trip_type=trip_type,
-            passenger_info=PassengerInfo(adults=params.passengers),
+            passenger_info=PassengerInfo(
+                adults=params.passengers,
+                children=params.children,
+                infants_in_seat=params.infants_in_seat,
+                infants_on_lap=params.infants_on_lap,
+            ),
             flight_segments=segments,
             stops=max_stops,
             seat_type=cabin_class,
@@ -516,6 +532,18 @@ def search_flights(
         int | None,
         Field(description="Number of adult passengers", ge=1),
     ] = None,
+    children: Annotated[
+        int,
+        Field(description="Number of children", ge=0),
+    ] = 0,
+    infants_in_seat: Annotated[
+        int,
+        Field(description="Number of infants in seat", ge=0),
+    ] = 0,
+    infants_on_lap: Annotated[
+        int,
+        Field(description="Number of infants on lap", ge=0),
+    ] = 0,
     exclude_basic_economy: Annotated[
         bool,
         Field(description="Exclude basic economy fares from results"),
@@ -538,6 +566,9 @@ def search_flights(
         max_stops=max_stops,
         sort_by=sort_by,
         passengers=passengers or CONFIG.default_passengers,
+        children=children,
+        infants_in_seat=infants_in_seat,
+        infants_on_lap=infants_on_lap,
         exclude_basic_economy=exclude_basic_economy,
     )
     return _execute_flight_search(params)
@@ -595,6 +626,18 @@ def search_dates(
         int | None,
         Field(description="Number of adult passengers", ge=1),
     ] = None,
+    children: Annotated[
+        int,
+        Field(description="Number of children", ge=0),
+    ] = 0,
+    infants_in_seat: Annotated[
+        int,
+        Field(description="Number of infants in seat", ge=0),
+    ] = 0,
+    infants_on_lap: Annotated[
+        int,
+        Field(description="Number of infants on lap", ge=0),
+    ] = 0,
 ) -> dict[str, Any]:
     """Find the cheapest travel dates between two airports within a date range.
 
@@ -615,6 +658,9 @@ def search_dates(
         departure_window=effective_departure_window,
         sort_by_price=sort_by_price,
         passengers=passengers or CONFIG.default_passengers,
+        children=children,
+        infants_in_seat=infants_in_seat,
+        infants_on_lap=infants_on_lap,
     )
     return _execute_date_search(params)