@danceiny/gotry 0.0.1-rc.10

package/dist/src/mock-llm.js

@@ -0,0 +1,103 @@
+import { parseFlightPackToSpec } from './unified.js';
+const SCRIPT = [
+    {
+        when: '请给我做机票和酒店的行程规划和推荐',
+        calendar: {
+            year: 2026,
+            assertedWeekdays: {
+                '2026-07-17': 'fri',
+                '2026-07-18': 'sat',
+                '2026-07-31': 'fri',
+                '2026-08-01': 'sat',
+                '2026-08-09': 'sun',
+                '2026-08-10': 'mon'
+            }
+        },
+        profile: {
+            companions: [
+                'girlfriend(先在普吉)'
+            ]
+        }
+    },
+    {
+        when: '工作时间',
+        profile: {
+            workWindow: {
+                homeTzOffsetMin: 240,
+                startMin: 600,
+                endMin: 1140,
+                workdays: [
+                    0,
+                    1,
+                    2,
+                    3,
+                    4
+                ],
+                evidence: '用户原话:我的工作时间是UTC+4的早上10点到下午7点'
+            }
+        }
+    },
+    {
+        when: '订了酒店',
+        profile: {
+            bookedResources: [
+                {
+                    kind: 'hotel',
+                    ref: 'The Title East Wing Rawai(7.18-23, 5 晚)',
+                    window: '2026-07-18~07-23'
+                }
+            ]
+        }
+    }
+];
+export function createMockLlm(flightPackPath, slotScript = []) {
+    let stepIdx = 0;
+    return {
+        async extractFacts (history, _state) {
+            const last = history[history.length - 1]?.text ?? '';
+            const step = SCRIPT[stepIdx];
+            if (step && last.includes(step.when)) {
+                stepIdx++;
+                return {
+                    calendar: step.calendar,
+                    profile: step.profile,
+                    assumptions: Object.keys({
+                        ...step.profile
+                    }).map((f)=>({
+                            field: f,
+                            source: 'user-verbatim'
+                        }))
+                };
+            }
+            return {
+                assumptions: []
+            };
+        },
+        async extractSpec (history, state) {
+            if (!state.profile.workWindow || !state.profile.bookedResources) return null;
+            if (!flightPackPath) return null;
+            const { readFile } = await import('node:fs/promises');
+            const pack = JSON.parse(await readFile(flightPackPath, 'utf-8'));
+            const spec = parseFlightPackToSpec(pack);
+            spec.workWindow = {
+                homeTzOffsetMin: state.profile.workWindow.homeTzOffsetMin,
+                startMin: state.profile.workWindow.startMin,
+                endMin: state.profile.workWindow.endMin,
+                workdays: state.profile.workWindow.workdays
+            };
+            spec.budgetCny = 9000;
+            return spec;
+        },
+        async extractSlots (history) {
+            const last = history[history.length - 1]?.text ?? '';
+            const hit = slotScript.find((s)=>last.includes(s.when));
+            return hit ? structuredClone(hit.extraction) : null;
+        },
+        async polishQuestion (q) {
+            return `【${q.key}】${q.text}\n(为什么问:${q.why})`;
+        }
+    };
+}
+
+
+//# sourceURL=/Users/bytedance/work/gotry/ts/src/mock-llm.ts

package/dist/src/model.js

@@ -0,0 +1,134 @@
+export function hhmmToMin(s) {
+    const [h, m] = s.split(':');
+    return Number(h) * 60 + Number(m);
+}
+export function minToHhmm(m) {
+    return `${String(Math.floor(m / 60)).padStart(2, '0')}:${String(m % 60).padStart(2, '0')}`;
+}
+export function requiredUsableHours(mot) {
+    return mot.baseUsableHours + mot.escapeHoursPerWeight * (mot.weights['escape_rest'] ?? 0);
+}
+export const WAKE_PENALTY_BEFORE_5 = 30;
+export const WAKE_PENALTY_BEFORE_6 = 25;
+export const WAKE_PENALTY_BEFORE_630 = 15;
+export const TRANSFER_PENALTY = 8;
+export const LATE_ARRIVAL_PENALTY = 10;
+export const LONG_D2D_PENALTY = 10;
+export const DAY_END_MIN = 21 * 60;
+export const DAY2_START_MIN = 9 * 60;
+export const DAY2_QUALITY = 0.9;
+export const LATEST_ARRIVE_STAY_MIN = 18 * 60;
+export function parseMotivation(d) {
+    const hard = d['hard'] ?? {};
+    const rawWeights = d['weights'] ?? d;
+    const weights = {};
+    for (const [k, v] of Object.entries(rawWeights)){
+        if (typeof v === 'number') weights[k] = v;
+    }
+    return {
+        weights,
+        wakeFloorMin: hhmmToMin(String(hard['wake_not_before'] ?? '06:30')),
+        minArrivalEnergyPct: Number(hard['min_arrival_energy_pct'] ?? 40),
+        baseUsableHours: 4.0,
+        escapeHoursPerWeight: 2.0
+    };
+}
+export function parseRequest(d) {
+    const home = (d['home'] ?? {})['hubs'] ?? {};
+    const homeHubAccess = {};
+    for (const [hub, acc] of Object.entries(home)){
+        homeHubAccess[hub] = {
+            hub,
+            toHubMin: Number(acc['to_hub_min'])
+        };
+    }
+    return {
+        note: String(d['note'] ?? ''),
+        motivation: parseMotivation(d['motivation']),
+        windowDays: Number(d['window_days']),
+        budgetCny: Number(d['budget_cny']),
+        homeHubAccess
+    };
+}
+function parseService(d) {
+    return {
+        id: String(d['id']),
+        depMin: hhmmToMin(String(d['dep'])),
+        arrMin: hhmmToMin(String(d['arr'])),
+        priceCny: Number(d['price_cny'])
+    };
+}
+function parseTransfer(d) {
+    return {
+        mode: String(d['mode']),
+        minutes: Number(d['min']),
+        priceCny: Number(d['price_cny'])
+    };
+}
+export function parseCandidate(d) {
+    return {
+        id: String(d['id']),
+        name: String(d['name']),
+        hub: String(d['hub']),
+        bufferOutMin: Number(d['buffer_out_min'] ?? 60),
+        bufferRetMin: Number(d['buffer_ret_min'] ?? 60),
+        servicesOut: d['services_out'].map(parseService),
+        servicesRet: d['services_ret'].map(parseService),
+        destTransfers: d['dest_transfers'].map(parseTransfer),
+        stayCnyPerNight: Number(d['stay_cny_per_night']),
+        localDailyCny: Number(d['local_daily_cny']),
+        minDaysForPurpose: Number(d['min_days_for_purpose']),
+        imageryMatch: Number(d['imagery_match']),
+        bestMonths: (d['best_months'] ?? []).map(Number)
+    };
+}
+export function evaluateChoice(cand, req, ch) {
+    const access = req.homeHubAccess[cand.hub];
+    const wake = ch.outService.depMin - cand.bufferOutMin - access.toHubMin;
+    const arriveStay = ch.outService.arrMin + ch.outTransfer.minutes;
+    const d2dOut = arriveStay - wake;
+    let energy = 100;
+    if (wake < 5 * 60) energy -= WAKE_PENALTY_BEFORE_5;
+    else if (wake < 6 * 60) energy -= WAKE_PENALTY_BEFORE_6;
+    else if (wake < hhmmToMin('06:30')) energy -= WAKE_PENALTY_BEFORE_630;
+    energy -= 2 * TRANSFER_PENALTY;
+    if (arriveStay > 21 * 60) energy -= LATE_ARRIVAL_PENALTY;
+    if (d2dOut > 6 * 60) energy -= LONG_D2D_PENALTY;
+    energy = Math.max(0, energy);
+    const day1Raw = Math.max(0, DAY_END_MIN - arriveStay) / 60;
+    const day1 = day1Raw * (0.5 + energy / 200);
+    const leaveStayRet = ch.retService.depMin - cand.bufferRetMin - ch.retTransfer.minutes;
+    const day2 = Math.max(0, leaveStayRet - DAY2_START_MIN) / 60 * DAY2_QUALITY;
+    const midDays = Math.max(0, ch.days - 2);
+    const usable = day1 + day2 + midDays * 8.0 * DAY2_QUALITY;
+    const money = ch.outService.priceCny + ch.retService.priceCny + ch.outTransfer.priceCny + ch.retTransfer.priceCny + cand.stayCnyPerNight * (ch.days - 1) + cand.localDailyCny * ch.days;
+    return {
+        moneyCny: money,
+        wakeMin: wake,
+        arriveStayMin: arriveStay,
+        doorToDoorOutMin: d2dOut,
+        energyArrivalPct: energy,
+        usableHours: usable,
+        usableDay1Hours: day1,
+        usableDay2Hours: day2,
+        departHomeRetMin: leaveStayRet,
+        arriveHomeRetMin: ch.retService.arrMin + access.toHubMin
+    };
+}
+export function trueCostToDict(t) {
+    return {
+        money_cny: t.moneyCny,
+        wake: minToHhmm(t.wakeMin),
+        arrive_stay: minToHhmm(t.arriveStayMin),
+        door_to_door_out: `${Math.floor(t.doorToDoorOutMin / 60)}h${String(t.doorToDoorOutMin % 60).padStart(2, '0')}m`,
+        energy_arrival_pct: t.energyArrivalPct,
+        usable_hours: Math.round(t.usableHours * 10) / 10,
+        usable_day1_hours: Math.round(t.usableDay1Hours * 10) / 10,
+        usable_day2_hours: Math.round(t.usableDay2Hours * 10) / 10,
+        leave_stay_return: minToHhmm(t.departHomeRetMin),
+        arrive_home_return: minToHhmm(Math.min(1440, t.arriveHomeRetMin))
+    };
+}
+
+
+//# sourceURL=/Users/bytedance/work/gotry/ts/src/model.ts

package/dist/src/slot-spec.js

@@ -0,0 +1,165 @@
+import { parseAbsoluteDate, ymd } from './time-anchor.js';
+const WEEKDAY_ZH = {
+    日: 0,
+    一: 1,
+    二: 2,
+    三: 3,
+    四: 4,
+    五: 5,
+    六: 6
+};
+function addDaysYmd(today, n) {
+    const [y, m, d] = today.split('-').map(Number);
+    return ymd(new Date(y, m - 1, d + n));
+}
+function splitPlusSuffix(expr) {
+    const m = expr.match(/^(.*?)\+(\d+)天?$/);
+    if (m && m[1]) return {
+        base: m[1],
+        plusDays: Number(m[2])
+    };
+    return {
+        base: expr,
+        plusDays: 0
+    };
+}
+function neighborDayOffset(expr) {
+    if (expr === '今天') return 0;
+    if (expr === '明天') return 1;
+    if (expr === '后天') return 2;
+    if (expr === '大后天') return 3;
+    return null;
+}
+function weekdayOffsetDays(expr, today) {
+    const m = expr.match(/^(本|下|下下)周([日一二三四五六])$/);
+    if (!m || !m[2]) return null;
+    const target = WEEKDAY_ZH[m[2]];
+    if (target === undefined) return null;
+    const [y, mm, d] = today.split('-').map(Number);
+    const todayMonOffset = (new Date(y, mm - 1, d).getDay() + 6) % 7;
+    const targetMonOffset = (target + 6) % 7;
+    const weekShift = m[1] === '本' ? 0 : m[1] === '下' ? 7 : 14;
+    return weekShift + targetMonOffset - todayMonOffset;
+}
+function monthFragmentDate(expr, today) {
+    if (![
+        '下个月初',
+        '下个月中旬',
+        '下个月下旬'
+    ].includes(expr)) return null;
+    const [y, m] = today.split('-').map(Number);
+    const day = expr === '下个月初' ? 1 : expr === '下个月中旬' ? 11 : 21;
+    return ymd(new Date(y, m, day));
+}
+export function resolveSlotDate(expr, anchor) {
+    const raw = expr.trim();
+    const year = Number(anchor.today.slice(0, 4));
+    const absolute = parseAbsoluteDate(raw, year);
+    if (absolute) return {
+        raw,
+        date: absolute,
+        kind: 'absolute'
+    };
+    const { base, plusDays } = splitPlusSuffix(raw);
+    const baseAbs = parseAbsoluteDate(base, year);
+    if (baseAbs) return {
+        raw,
+        date: addDaysYmd(baseAbs, plusDays),
+        kind: 'absolute'
+    };
+    const month = monthFragmentDate(base, anchor.today);
+    if (month) return {
+        raw,
+        date: addDaysYmd(month, plusDays),
+        kind: 'card'
+    };
+    const neighbor = neighborDayOffset(base);
+    if (neighbor !== null) return {
+        raw,
+        date: addDaysYmd(anchor.today, neighbor + plusDays),
+        kind: 'card'
+    };
+    const offset = weekdayOffsetDays(base, anchor.today);
+    if (offset !== null) return {
+        raw,
+        date: addDaysYmd(anchor.today, offset + plusDays),
+        kind: 'card'
+    };
+    return {
+        raw,
+        date: null,
+        kind: 'unresolved'
+    };
+}
+const DATE_FIELDS = {
+    requisition: [
+        'start_date',
+        'end_date'
+    ],
+    flight: [
+        'departure_date',
+        'return_date'
+    ],
+    hotel: [
+        'check_in_date',
+        'check_out_date'
+    ]
+};
+export function resolveSlots(ext, anchor) {
+    const out = {
+        unresolved: []
+    };
+    for (const domain of ext.domains){
+        const slots = ext.slots[domain];
+        if (!slots) continue;
+        for (const field of DATE_FIELDS[domain] ?? []){
+            const value = slots[field];
+            if (typeof value !== 'string') continue;
+            const res = resolveSlotDate(value, anchor);
+            const bucket = out[domain] ??= {};
+            bucket[field] = res;
+            if (res.kind === 'unresolved') out.unresolved.push({
+                field: `${domain}.${field}`,
+                raw: res.raw
+            });
+        }
+    }
+    return out;
+}
+export function specDateMismatches(spec, resolved) {
+    const mismatches = [];
+    const primary = [
+        {
+            field: 'requisition.start_date',
+            res: resolved.requisition?.start_date
+        },
+        {
+            field: 'flight.departure_date',
+            res: resolved.flight?.departure_date
+        },
+        {
+            field: 'hotel.check_in_date',
+            res: resolved.hotel?.check_in_date
+        }
+    ];
+    const dated = primary.filter((p)=>p.res?.date != null);
+    if (dated.length === 0) return mismatches;
+    const datedSegs = (spec.segments ?? []).filter((s)=>typeof s.date === 'string' && s.date);
+    if (datedSegs.length !== 1) return mismatches;
+    const seg = datedSegs[0];
+    const slotDates = new Set(dated.map((s)=>s.res.date));
+    if (!slotDates.has(seg.date)) {
+        mismatches.push({
+            specAt: `segments[${(spec.segments ?? []).indexOf(seg)}].date`,
+            field: dated.map((s)=>s.field).join('|'),
+            specDate: seg.date,
+            slotDate: [
+                ...slotDates
+            ].join('|')
+        });
+    }
+    return mismatches;
+}
+
+
+//# sourceURL=/Users/bytedance/work/gotry/ts/src/slot-spec.ts