imprint-mcp 0.4.10 → 0.4.11

package/examples/google-flights/_shared/flights_request.ts

@@ -19,8 +19,10 @@ function buildLeg(leg: any): any[] {
   out[1] = [[[leg?.dest, 0]]];
   out[2] = leg?.times ?? null;
   out[3] = leg?.stops ?? 0;
-  out[4] = leg?.alliances ?? null;
-  out[5] = leg?.carriers ?? null;
+  // Google uses slot 4 for included alliances and carrier codes. Slot 5 is an
+  // exclusion list; putting carrier codes there inverts the filter.
+  out[4] = leg?.includeAirlines ?? leg?.alliances ?? null;
+  out[5] = leg?.excludeAirlines ?? null;
   out[6] = leg?.date ?? null;
   out[7] = leg?.duration ?? null;
   out[8] = Array.isArray(leg?.selected)

package/examples/google-flights/search_flights/parser.ts

@@ -19,7 +19,13 @@ interface Itinerary {
   flight_token: string;
 }
 
+interface AirlineFilter {
+  code: string;
+  name: string;
+}
+
 const AIRPORT = /^[A-Z]{3}$/;
+const ALLIANCE_CODES = new Set(['ONEWORLD', 'SKYTEAM', 'STAR_ALLIANCE']);
 
 // A leg is [carrierCode, [carrierNames], [segments], originIATA, [departDate],
 // [departTime], destIATA, [arriveDate], [arriveTime], durationMinutes, ...].
@@ -79,6 +85,38 @@ function walk(node: unknown, found: unknown[][]): void {
   for (const child of node) walk(child, found);
 }
 
+function isPairList(node: unknown): node is string[][] {
+  return (
+    Array.isArray(node) &&
+    node.length > 0 &&
+    node.every(
+      (item) =>
+        Array.isArray(item) && typeof item[0] === 'string' && typeof item[1] === 'string',
+    )
+  );
+}
+
+function toFilters(pairs: string[][]): AirlineFilter[] {
+  return pairs.map((pair) => ({ code: pair[0] as string, name: pair[1] as string }));
+}
+
+function collectAirlineFilters(
+  node: unknown,
+  found: { alliances: AirlineFilter[]; carriers: AirlineFilter[] },
+): void {
+  if (!Array.isArray(node)) return;
+  if (
+    node.length >= 2 &&
+    isPairList(node[0]) &&
+    isPairList(node[1]) &&
+    node[0].some((pair) => ALLIANCE_CODES.has(pair[0] as string))
+  ) {
+    found.alliances = toFilters(node[0]);
+    found.carriers = toFilters(node[1]);
+  }
+  for (const child of node) collectAirlineFilters(child, found);
+}
+
 function normalize(it: unknown[]): Itinerary {
   const legs = legsOf(it);
   const priceTok = it[1] as unknown[];
@@ -158,6 +196,11 @@ export function extract(
 
   const found: unknown[][] = [];
   if (payload != null) walk(payload, found);
+  const availableAirlineFilters = {
+    alliances: [] as AirlineFilter[],
+    carriers: [] as AirlineFilter[],
+  };
+  if (payload != null) collectAirlineFilters(payload, availableAirlineFilters);
 
   const byToken = new Map<string, Itinerary>();
   for (const it of found) {
@@ -175,5 +218,11 @@ export function extract(
   return {
     count: itineraries.length,
     itineraries,
+    resultScope: {
+      exhaustive: false,
+      note:
+        'Google Flights GetShoppingResults returns a limited sorted subset. A carrier can be available in availableAirlineFilters without appearing in itineraries; call search_flights again with airlines=<code> to fetch that carrier.',
+    },
+    availableAirlineFilters,
   };
 }

package/examples/google-flights/search_flights/request-transform.ts

@@ -1,15 +1,13 @@
 // Adapter around the shared FlightsFrontendService body builder.
 // The tool exposes flat snake_case params (origin, destination, departure_date,
 // max_stops, …); the shared encoder consumes a structured camelCase shape
-// ({ tripType, legs:[{origin,dest,date,times,stops,alliances,carriers,duration}],
+// ({ tripType, legs:[{origin,dest,date,times,stops,includeAirlines,duration}],
 // maxPrice, bags }). We map between them here and delegate the byte-for-byte
 // positional encoding to the shared module (required reuse).
 import { transform as sharedTransform } from '../_shared/flights_request.ts';
 
 type Params = Record<string, string | number | boolean | undefined | null>;
 
-const ALLIANCES = new Set(['ONEWORLD', 'SKYTEAM', 'STAR_ALLIANCE']);
-
 function mapTripType(v: unknown): number {
   if (v == null || v === '') return 1;
   if (typeof v === 'number') return v;
@@ -42,18 +40,14 @@ function parseTimes(v: unknown): number[] | null {
   return [Number(m[1]), Number(m[2]), 0, 23];
 }
 
-function parseAirlines(v: unknown): { alliances: string[] | null; carriers: string[] | null } {
-  if (v == null || v === '') return { alliances: null, carriers: null };
-  const parts = String(v)
+function parseAirlines(v: unknown): string[] | null {
+  if (v == null || v === '') return null;
+  const includeAirlines = String(v)
     .split(',')
     .map((x) => x.trim())
-    .filter(Boolean);
-  const alliances = parts.filter((p) => ALLIANCES.has(p.toUpperCase())).map((p) => p.toUpperCase());
-  const carriers = parts.filter((p) => !ALLIANCES.has(p.toUpperCase()));
-  return {
-    alliances: alliances.length ? alliances : null,
-    carriers: carriers.length ? carriers : null,
-  };
+    .filter(Boolean)
+    .map((p) => p.toUpperCase());
+  return includeAirlines.length ? includeAirlines : null;
 }
 
 function num(v: unknown): number | undefined {
@@ -73,7 +67,7 @@ export function transform(
   const hasReturnDate = p.return_date != null && String(p.return_date).trim() !== '';
   const tripType = requestedTripType === 1 && !hasReturnDate ? 2 : requestedTripType;
   const stops = p.max_stops != null && p.max_stops !== '' ? mapStops(p.max_stops) : 0;
-  const { alliances, carriers } = parseAirlines(p.airlines);
+  const includeAirlines = parseAirlines(p.airlines);
   const maxDur = num(p.max_duration);
   const duration = maxDur != null ? [maxDur] : null;
 
@@ -87,8 +81,7 @@ export function transform(
       date: p.departure_date ? String(p.departure_date) : null,
       times: parseTimes(p.outbound_times),
       stops,
-      alliances,
-      carriers,
+      includeAirlines,
       duration,
     },
   ];
@@ -101,8 +94,7 @@ export function transform(
       date: String(p.return_date),
       times: parseTimes(p.return_times),
       stops,
-      alliances,
-      carriers,
+      includeAirlines,
       duration,
     });
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "imprint-mcp",
-  "version": "0.4.10",
+  "version": "0.4.11",
   "description": "Teach an AI agent how to use any website. Once. Records a real browser session + narration; generates a deterministic MCP tool plus a DOM-replay playbook fallback.",
   "type": "module",
   "exports": {