@mrtdown/core 2.0.0-alpha.3 → 2.0.0-alpha.5

package/dist/llm/common/formatCurrentState.js

@@ -0,0 +1,342 @@
+import { gfmToMarkdown } from 'mdast-util-gfm';
+import { toMarkdown } from 'mdast-util-to-markdown';
+/**
+ * Format the current state as markdown for better LLM readability
+ */
+export function formatCurrentState(stateOrOptions) {
+    const options = stateOrOptions != null &&
+        typeof stateOrOptions === 'object' &&
+        'state' in stateOrOptions
+        ? stateOrOptions
+        : { state: stateOrOptions };
+    const state = options.state;
+    const evidence = options.evidence ?? [];
+    if (state == null && evidence.length === 0) {
+        return 'No state';
+    }
+    const services = state?.services ?? {};
+    const facilities = state?.facilities ?? {};
+    const impactEventIds = state?.impactEventIds ?? [];
+    const relevantEvidenceIds = state != null ? collectEvidenceIds(state) : [];
+    const root = {
+        type: 'root',
+        children: [],
+    };
+    // Evidence Section
+    root.children.push({
+        type: 'heading',
+        depth: 2,
+        children: [{ type: 'text', value: 'Evidence' }],
+    });
+    if (evidence.length === 0) {
+        root.children.push({
+            type: 'paragraph',
+            children: [{ type: 'text', value: 'None' }],
+        });
+    }
+    else {
+        for (const ev of evidence) {
+            root.children.push(...formatEvidenceItem(ev));
+        }
+    }
+    // Services Section
+    root.children.push({
+        type: 'heading',
+        depth: 2,
+        children: [{ type: 'text', value: 'Services' }],
+    });
+    const serviceKeys = Object.keys(services);
+    if (serviceKeys.length === 0) {
+        root.children.push({
+            type: 'paragraph',
+            children: [{ type: 'text', value: 'None' }],
+        });
+    }
+    else {
+        for (const key of serviceKeys) {
+            const svc = services[key];
+            root.children.push(...formatServiceSection(key, svc.serviceId, svc.effect, svc.scopes, svc.periods, svc.causes));
+        }
+    }
+    // Facilities Section
+    const facilitiesHeading = {
+        type: 'heading',
+        depth: 2,
+        children: [{ type: 'text', value: 'Facilities' }],
+    };
+    root.children.push(facilitiesHeading);
+    const facilityKeys = Object.keys(facilities);
+    if (facilityKeys.length === 0) {
+        root.children.push({
+            type: 'paragraph',
+            children: [{ type: 'text', value: 'None' }],
+        });
+    }
+    else {
+        for (const key of facilityKeys) {
+            const fac = facilities[key];
+            root.children.push(...formatFacilitySection(key, fac.stationId, fac.kind, fac.effect, fac.periods));
+        }
+    }
+    // Impact Event IDs Section
+    root.children.push({
+        type: 'heading',
+        depth: 2,
+        children: [{ type: 'text', value: 'Impact Event IDs' }],
+    });
+    if (impactEventIds.length === 0) {
+        root.children.push({
+            type: 'paragraph',
+            children: [{ type: 'text', value: 'None' }],
+        });
+    }
+    else {
+        root.children.push({
+            type: 'list',
+            ordered: false,
+            children: impactEventIds.map((id) => ({
+                type: 'listItem',
+                children: [
+                    {
+                        type: 'paragraph',
+                        children: [{ type: 'inlineCode', value: id }],
+                    },
+                ],
+            })),
+        });
+    }
+    // Relevant Evidence IDs Section
+    const evidenceHeading = {
+        type: 'heading',
+        depth: 2,
+        children: [{ type: 'text', value: 'Relevant Evidence IDs' }],
+    };
+    root.children.push(evidenceHeading);
+    if (relevantEvidenceIds.length === 0) {
+        root.children.push({
+            type: 'paragraph',
+            children: [{ type: 'text', value: 'None' }],
+        });
+    }
+    else {
+        root.children.push({
+            type: 'list',
+            ordered: false,
+            children: relevantEvidenceIds.map((id) => ({
+                type: 'listItem',
+                children: [
+                    {
+                        type: 'paragraph',
+                        children: [{ type: 'inlineCode', value: id }],
+                    },
+                ],
+            })),
+        });
+    }
+    return toMarkdown(root, { extensions: [gfmToMarkdown()] });
+}
+function formatEvidenceItem(ev) {
+    const children = [];
+    children.push({
+        type: 'heading',
+        depth: 3,
+        children: [
+            { type: 'inlineCode', value: ev.id },
+            { type: 'text', value: ` (${ev.type})` },
+        ],
+    });
+    children.push({
+        type: 'paragraph',
+        children: [
+            { type: 'text', value: 'Timestamp: ' },
+            { type: 'inlineCode', value: ev.ts },
+        ],
+    });
+    children.push({
+        type: 'paragraph',
+        children: [
+            { type: 'text', value: 'Source: ' },
+            { type: 'inlineCode', value: ev.render?.source ?? '—' },
+        ],
+    });
+    if ('sourceUrl' in ev) {
+        children.push({
+            type: 'paragraph',
+            children: [
+                { type: 'text', value: 'URL: ' },
+                {
+                    type: 'link',
+                    url: ev.sourceUrl,
+                    children: [{ type: 'text', value: ev.sourceUrl }],
+                },
+            ],
+        });
+    }
+    children.push({
+        type: 'paragraph',
+        children: [{ type: 'text', value: 'Text:' }],
+    });
+    children.push({
+        type: 'blockquote',
+        children: [
+            {
+                type: 'paragraph',
+                children: [{ type: 'text', value: ev.text }],
+            },
+        ],
+    });
+    return children;
+}
+function collectEvidenceIds(state) {
+    const ids = new Set();
+    for (const prov of Object.values(state.servicesProvenance)) {
+        if (prov.effect?.evidenceId)
+            ids.add(prov.effect.evidenceId);
+        if (prov.scopes?.evidenceId)
+            ids.add(prov.scopes.evidenceId);
+        if (prov.periods?.evidenceId)
+            ids.add(prov.periods.evidenceId);
+        if (prov.causes?.evidenceId)
+            ids.add(prov.causes.evidenceId);
+    }
+    for (const prov of Object.values(state.facilitiesProvenance)) {
+        if (prov.effect?.evidenceId)
+            ids.add(prov.effect.evidenceId);
+        if (prov.periods?.evidenceId)
+            ids.add(prov.periods.evidenceId);
+    }
+    return [...ids];
+}
+function formatServiceSection(key, serviceId, effect, scopes, periods, causes) {
+    const children = [];
+    children.push({
+        type: 'heading',
+        depth: 3,
+        children: [{ type: 'text', value: key }],
+    });
+    children.push({
+        type: 'paragraph',
+        children: [
+            { type: 'text', value: 'Service ID: ' },
+            { type: 'inlineCode', value: serviceId },
+        ],
+    });
+    children.push({
+        type: 'paragraph',
+        children: [
+            { type: 'text', value: 'Effect: ' },
+            { type: 'inlineCode', value: effect?.kind ?? 'null' },
+        ],
+    });
+    if (scopes.length > 0) {
+        children.push({
+            type: 'paragraph',
+            children: [{ type: 'text', value: 'Scopes:' }],
+        });
+        children.push({
+            type: 'list',
+            ordered: false,
+            children: scopes.map((scope) => ({
+                type: 'listItem',
+                children: [
+                    {
+                        type: 'paragraph',
+                        children: [{ type: 'inlineCode', value: formatScope(scope) }],
+                    },
+                ],
+            })),
+        });
+    }
+    if (periods.length > 0) {
+        children.push({
+            type: 'paragraph',
+            children: [{ type: 'text', value: 'Periods:' }],
+        });
+        children.push({
+            type: 'list',
+            ordered: false,
+            children: periods.map((period, i) => ({
+                type: 'listItem',
+                children: [
+                    {
+                        type: 'paragraph',
+                        children: [{ type: 'text', value: formatPeriod(period, i) }],
+                    },
+                ],
+            })),
+        });
+    }
+    if (causes.length > 0) {
+        children.push({
+            type: 'paragraph',
+            children: [
+                { type: 'text', value: 'Causes: ' },
+                { type: 'inlineCode', value: causes.join(', ') },
+            ],
+        });
+    }
+    return children;
+}
+function formatFacilitySection(key, stationId, kind, effect, periods) {
+    const children = [];
+    children.push({
+        type: 'heading',
+        depth: 3,
+        children: [{ type: 'text', value: key }],
+    });
+    children.push({
+        type: 'paragraph',
+        children: [
+            { type: 'text', value: 'Station: ' },
+            { type: 'inlineCode', value: stationId },
+            { type: 'text', value: ', Kind: ' },
+            { type: 'inlineCode', value: kind },
+        ],
+    });
+    children.push({
+        type: 'paragraph',
+        children: [
+            { type: 'text', value: 'Effect: ' },
+            { type: 'inlineCode', value: effect?.kind ?? 'null' },
+        ],
+    });
+    if (periods.length > 0) {
+        children.push({
+            type: 'paragraph',
+            children: [{ type: 'text', value: 'Periods:' }],
+        });
+        children.push({
+            type: 'list',
+            ordered: false,
+            children: periods.map((period, i) => ({
+                type: 'listItem',
+                children: [
+                    {
+                        type: 'paragraph',
+                        children: [{ type: 'text', value: formatPeriod(period, i) }],
+                    },
+                ],
+            })),
+        });
+    }
+    return children;
+}
+function formatScope(scope) {
+    switch (scope.type) {
+        case 'service.whole':
+            return scope.type;
+        case 'service.segment':
+            return `${scope.type} (${scope.fromStationId} → ${scope.toStationId})`;
+        case 'service.point':
+            return `${scope.type} (${scope.stationId})`;
+    }
+}
+function formatPeriod(period, index) {
+    switch (period.kind) {
+        case 'fixed':
+            return `[${index}] fixed: ${period.startAt} → ${period.endAt ?? 'ongoing'}`;
+        case 'recurring':
+            return `[${index}] recurring ${period.frequency}: ${period.startAt}–${period.endAt} ${period.timeWindow.startAt}–${period.timeWindow.endAt}`;
+    }
+}
+//# sourceMappingURL=formatCurrentState.js.map

package/dist/llm/common/formatCurrentState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"formatCurrentState.js","sourceRoot":"/","sources":["llm/common/formatCurrentState.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAC/C,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAWpD;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAChC,cAAmE;IAEnE,MAAM,OAAO,GACX,cAAc,IAAI,IAAI;QACtB,OAAO,cAAc,KAAK,QAAQ;QAClC,OAAO,IAAI,cAAc;QACvB,CAAC,CAAC,cAAc;QAChB,CAAC,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;IAChC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IAC5B,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;IAExC,IAAI,KAAK,IAAI,IAAI,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3C,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,MAAM,QAAQ,GAAG,KAAK,EAAE,QAAQ,IAAI,EAAE,CAAC;IACvC,MAAM,UAAU,GAAG,KAAK,EAAE,UAAU,IAAI,EAAE,CAAC;IAC3C,MAAM,cAAc,GAAG,KAAK,EAAE,cAAc,IAAI,EAAE,CAAC;IACnD,MAAM,mBAAmB,GAAG,KAAK,IAAI,IAAI,CAAC,CAAC,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAE3E,MAAM,IAAI,GAAS;QACjB,IAAI,EAAE,MAAM;QACZ,QAAQ,EAAE,EAAE;KACb,CAAC;IAEF,mBAAmB;IACnB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;QACjB,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,CAAC;QACR,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;KAChD,CAAC,CAAC;IAEH,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,WAAW;YACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;SAC5C,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,mBAAmB;IACnB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;QACjB,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,CAAC;QACR,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;KAChD,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,WAAW;YACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;SAC5C,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;YAC9B,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;YAC1B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAChB,GAAG,oBAAoB,CACrB,GAAG,EACH,GAAG,CAAC,SAAS,EACb,GAAG,CAAC,MAAM,EACV,GAAG,CAAC,MAAM,EACV,GAAG,CAAC,OAAO,EACX,GAAG,CAAC,MAAM,CACX,CACF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,qBAAqB;IACrB,MAAM,iBAAiB,GAAY;QACjC,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,CAAC;QACR,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;KAClD,CAAC;IACF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAEtC,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC7C,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,WAAW;YACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;SAC5C,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;YAC/B,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;YAC5B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAChB,GAAG,qBAAqB,CACtB,GAAG,EACH,GAAG,CAAC,SAAS,EACb,GAAG,CAAC,IAAI,EACR,GAAG,CAAC,MAAM,EACV,GAAG,CAAC,OAAO,CACZ,CACF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,2BAA2B;IAC3B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;QACjB,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,CAAC;QACR,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,kBAAkB,EAAE,CAAC;KACxD,CAAC,CAAC;IAEH,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAChC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,WAAW;YACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;SAC5C,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,KAAK;YACd,QAAQ,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,WAAW;wBACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;qBAC9C;iBACF;aACF,CAAC,CAAC;SACJ,CAAC,CAAC;IACL,CAAC;IAED,gCAAgC;IAChC,MAAM,eAAe,GAAY;QAC/B,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,CAAC;QACR,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC;KAC7D,CAAC;IACF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAEpC,IAAI,mBAAmB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,WAAW;YACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;SAC5C,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,KAAK;YACd,QAAQ,EAAE,mBAAmB,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBACzC,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,WAAW;wBACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;qBAC9C;iBACF;aACF,CAAC,CAAC;SACJ,CAAC,CAAC;IACL,CAAC;IAED,OAAO,UAAU,CAAC,IAAI,EAAE,EAAE,UAAU,EAAE,CAAC,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;AAC7D,CAAC;AAED,SAAS,kBAAkB,CAAC,EAAY;IACtC,MAAM,QAAQ,GAAqB,EAAE,CAAC;IAEtC,QAAQ,CAAC,IAAI,CAAC;QACZ,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,CAAC;QACR,QAAQ,EAAE;YACR,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;YACpC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,IAAI,GAAG,EAAE;SACzC;KACF,CAAC,CAAC;IAEH,QAAQ,CAAC,IAAI,CAAC;QACZ,IAAI,EAAE,WAAW;QACjB,QAAQ,EAAE;YACR,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE;YACtC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;SACrC;KACF,CAAC,CAAC;IAEH,QAAQ,CAAC,IAAI,CAAC;QACZ,IAAI,EAAE,WAAW;QACjB,QAAQ,EAAE;YACR,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE;YACnC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,EAAE,CAAC,MAAM,EAAE,MAAM,IAAI,GAAG,EAAE;SACxD;KACF,CAAC,CAAC;IAEH,IAAI,WAAW,IAAI,EAAE,EAAE,CAAC;QACtB,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,WAAW;YACjB,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE;gBAChC;oBACE,IAAI,EAAE,MAAM;oBACZ,GAAG,EAAE,EAAE,CAAC,SAAS;oBACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC,SAAS,EAAE,CAAC;iBAClD;aACF;SACF,CAAC,CAAC;IACL,CAAC;IAED,QAAQ,CAAC,IAAI,CAAC;QACZ,IAAI,EAAE,WAAW;QACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;KAC7C,CAAC,CAAC;IACH,QAAQ,CAAC,IAAI,CAAC;QACZ,IAAI,EAAE,YAAY;QAClB,QAAQ,EAAE;YACR;gBACE,IAAI,EAAE,WAAW;gBACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;aAC7C;SACF;KACF,CAAC,CAAC;IAEH,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAuB;IACjD,MAAM,GAAG,GAAG,IAAI,GAAG,EAAU,CAAC;IAC9B,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,EAAE,CAAC;QAC3D,IAAI,IAAI,CAAC,MAAM,EAAE,UAAU;YAAE,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAC7D,IAAI,IAAI,CAAC,MAAM,EAAE,UAAU;YAAE,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAC7D,IAAI,IAAI,CAAC,OAAO,EAAE,UAAU;YAAE,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC/D,IAAI,IAAI,CAAC,MAAM,EAAE,UAAU;YAAE,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAC/D,CAAC;IACD,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,oBAAoB,CAAC,EAAE,CAAC;QAC7D,IAAI,IAAI,CAAC,MAAM,EAAE,UAAU;YAAE,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAC7D,IAAI,IAAI,CAAC,OAAO,EAAE,UAAU;YAAE,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IACjE,CAAC;IACD,OAAO,CAAC,GAAG,GAAG,CAAC,CAAC;AAClB,CAAC;AAED,SAAS,oBAAoB,CAC3B,GAAW,EACX,SAAiB,EACjB,MAA+B,EAC/B,MAAsB,EACtB,OAAiB,EACjB,MAAgB;IAEhB,MAAM,QAAQ,GAAqB,EAAE,CAAC;IAEtC,QAAQ,CAAC,IAAI,CAAC;QACZ,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,CAAC;QACR,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;KACzC,CAAC,CAAC;IAEH,QAAQ,CAAC,IAAI,CAAC;QACZ,IAAI,EAAE,WAAW;QACjB,QAAQ,EAAE;YACR,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,cAAc,EAAE;YACvC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,SAAS,EAAE;SACzC;KACF,CAAC,CAAC;IAEH,QAAQ,CAAC,IAAI,CAAC;QACZ,IAAI,EAAE,WAAW;QACjB,QAAQ,EAAE;YACR,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE;YACnC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,IAAI,MAAM,EAAE;SACtD;KACF,CAAC,CAAC;IAEH,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,WAAW;YACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;SAC/C,CAAC,CAAC;QACH,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,KAAK;YACd,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAC/B,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,WAAW;wBACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC;qBAC9D;iBACF;aACF,CAAC,CAAC;SACJ,CAAC,CAAC;IACL,CAAC;IAED,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,WAAW;YACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;SAChD,CAAC,CAAC;QACH,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,KAAK;YACd,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,WAAW;wBACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC;qBAC7D;iBACF;aACF,CAAC,CAAC;SACJ,CAAC,CAAC;IACL,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,WAAW;YACjB,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE;gBACnC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;aACjD;SACF,CAAC,CAAC;IACL,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,qBAAqB,CAC5B,GAAW,EACX,SAAiB,EACjB,IAAY,EACZ,MAA+B,EAC/B,OAAiB;IAEjB,MAAM,QAAQ,GAAqB,EAAE,CAAC;IAEtC,QAAQ,CAAC,IAAI,CAAC;QACZ,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,CAAC;QACR,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;KACzC,CAAC,CAAC;IAEH,QAAQ,CAAC,IAAI,CAAC;QACZ,IAAI,EAAE,WAAW;QACjB,QAAQ,EAAE;YACR,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE;YACpC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,SAAS,EAAE;YACxC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE;YACnC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE;SACpC;KACF,CAAC,CAAC;IAEH,QAAQ,CAAC,IAAI,CAAC;QACZ,IAAI,EAAE,WAAW;QACjB,QAAQ,EAAE;YACR,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE;YACnC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,IAAI,MAAM,EAAE;SACtD;KACF,CAAC,CAAC;IAEH,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,WAAW;YACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;SAChD,CAAC,CAAC;QACH,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,KAAK;YACd,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBACpC,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,WAAW;wBACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC;qBAC7D;iBACF;aACF,CAAC,CAAC;SACJ,CAAC,CAAC;IACL,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,WAAW,CAAC,KAAmB;IACtC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,KAAK,eAAe;YAClB,OAAO,KAAK,CAAC,IAAI,CAAC;QACpB,KAAK,iBAAiB;YACpB,OAAO,GAAG,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,aAAa,MAAM,KAAK,CAAC,WAAW,GAAG,CAAC;QACzE,KAAK,eAAe;YAClB,OAAO,GAAG,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,SAAS,GAAG,CAAC;IAChD,CAAC;AACH,CAAC;AAED,SAAS,YAAY,CAAC,MAAc,EAAE,KAAa;IACjD,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,KAAK,OAAO;YACV,OAAO,IAAI,KAAK,YAAY,MAAM,CAAC,OAAO,MAAM,MAAM,CAAC,KAAK,IAAI,SAAS,EAAE,CAAC;QAC9E,KAAK,WAAW;YACd,OAAO,IAAI,KAAK,eAAe,MAAM,CAAC,SAAS,KAAK,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,UAAU,CAAC,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;IACjJ,CAAC;AACH,CAAC","sourcesContent":["import type { Heading, Root } from 'mdast';\nimport { gfmToMarkdown } from 'mdast-util-gfm';\nimport { toMarkdown } from 'mdast-util-to-markdown';\nimport type { IssueBundleState } from '../../repo/issue/helpers/deriveCurrentState.js';\nimport type { Evidence } from '../../schema/issue/evidence.js';\nimport type { Period } from '../../schema/issue/period.js';\nimport type { ServiceScope } from '../../schema/issue/serviceScope.js';\n\nexport type FormatCurrentStateOptions = {\n  state: IssueBundleState | null;\n  evidence?: Evidence[];\n};\n\n/**\n * Format the current state as markdown for better LLM readability\n */\nexport function formatCurrentState(\n  stateOrOptions: IssueBundleState | null | FormatCurrentStateOptions,\n): string {\n  const options: FormatCurrentStateOptions =\n    stateOrOptions != null &&\n    typeof stateOrOptions === 'object' &&\n    'state' in stateOrOptions\n      ? stateOrOptions\n      : { state: stateOrOptions };\n  const state = options.state;\n  const evidence = options.evidence ?? [];\n\n  if (state == null && evidence.length === 0) {\n    return 'No state';\n  }\n\n  const services = state?.services ?? {};\n  const facilities = state?.facilities ?? {};\n  const impactEventIds = state?.impactEventIds ?? [];\n  const relevantEvidenceIds = state != null ? collectEvidenceIds(state) : [];\n\n  const root: Root = {\n    type: 'root',\n    children: [],\n  };\n\n  // Evidence Section\n  root.children.push({\n    type: 'heading',\n    depth: 2,\n    children: [{ type: 'text', value: 'Evidence' }],\n  });\n\n  if (evidence.length === 0) {\n    root.children.push({\n      type: 'paragraph',\n      children: [{ type: 'text', value: 'None' }],\n    });\n  } else {\n    for (const ev of evidence) {\n      root.children.push(...formatEvidenceItem(ev));\n    }\n  }\n\n  // Services Section\n  root.children.push({\n    type: 'heading',\n    depth: 2,\n    children: [{ type: 'text', value: 'Services' }],\n  });\n\n  const serviceKeys = Object.keys(services);\n  if (serviceKeys.length === 0) {\n    root.children.push({\n      type: 'paragraph',\n      children: [{ type: 'text', value: 'None' }],\n    });\n  } else {\n    for (const key of serviceKeys) {\n      const svc = services[key];\n      root.children.push(\n        ...formatServiceSection(\n          key,\n          svc.serviceId,\n          svc.effect,\n          svc.scopes,\n          svc.periods,\n          svc.causes,\n        ),\n      );\n    }\n  }\n\n  // Facilities Section\n  const facilitiesHeading: Heading = {\n    type: 'heading',\n    depth: 2,\n    children: [{ type: 'text', value: 'Facilities' }],\n  };\n  root.children.push(facilitiesHeading);\n\n  const facilityKeys = Object.keys(facilities);\n  if (facilityKeys.length === 0) {\n    root.children.push({\n      type: 'paragraph',\n      children: [{ type: 'text', value: 'None' }],\n    });\n  } else {\n    for (const key of facilityKeys) {\n      const fac = facilities[key];\n      root.children.push(\n        ...formatFacilitySection(\n          key,\n          fac.stationId,\n          fac.kind,\n          fac.effect,\n          fac.periods,\n        ),\n      );\n    }\n  }\n\n  // Impact Event IDs Section\n  root.children.push({\n    type: 'heading',\n    depth: 2,\n    children: [{ type: 'text', value: 'Impact Event IDs' }],\n  });\n\n  if (impactEventIds.length === 0) {\n    root.children.push({\n      type: 'paragraph',\n      children: [{ type: 'text', value: 'None' }],\n    });\n  } else {\n    root.children.push({\n      type: 'list',\n      ordered: false,\n      children: impactEventIds.map((id) => ({\n        type: 'listItem',\n        children: [\n          {\n            type: 'paragraph',\n            children: [{ type: 'inlineCode', value: id }],\n          },\n        ],\n      })),\n    });\n  }\n\n  // Relevant Evidence IDs Section\n  const evidenceHeading: Heading = {\n    type: 'heading',\n    depth: 2,\n    children: [{ type: 'text', value: 'Relevant Evidence IDs' }],\n  };\n  root.children.push(evidenceHeading);\n\n  if (relevantEvidenceIds.length === 0) {\n    root.children.push({\n      type: 'paragraph',\n      children: [{ type: 'text', value: 'None' }],\n    });\n  } else {\n    root.children.push({\n      type: 'list',\n      ordered: false,\n      children: relevantEvidenceIds.map((id) => ({\n        type: 'listItem',\n        children: [\n          {\n            type: 'paragraph',\n            children: [{ type: 'inlineCode', value: id }],\n          },\n        ],\n      })),\n    });\n  }\n\n  return toMarkdown(root, { extensions: [gfmToMarkdown()] });\n}\n\nfunction formatEvidenceItem(ev: Evidence): Root['children'] {\n  const children: Root['children'] = [];\n\n  children.push({\n    type: 'heading',\n    depth: 3,\n    children: [\n      { type: 'inlineCode', value: ev.id },\n      { type: 'text', value: ` (${ev.type})` },\n    ],\n  });\n\n  children.push({\n    type: 'paragraph',\n    children: [\n      { type: 'text', value: 'Timestamp: ' },\n      { type: 'inlineCode', value: ev.ts },\n    ],\n  });\n\n  children.push({\n    type: 'paragraph',\n    children: [\n      { type: 'text', value: 'Source: ' },\n      { type: 'inlineCode', value: ev.render?.source ?? '—' },\n    ],\n  });\n\n  if ('sourceUrl' in ev) {\n    children.push({\n      type: 'paragraph',\n      children: [\n        { type: 'text', value: 'URL: ' },\n        {\n          type: 'link',\n          url: ev.sourceUrl,\n          children: [{ type: 'text', value: ev.sourceUrl }],\n        },\n      ],\n    });\n  }\n\n  children.push({\n    type: 'paragraph',\n    children: [{ type: 'text', value: 'Text:' }],\n  });\n  children.push({\n    type: 'blockquote',\n    children: [\n      {\n        type: 'paragraph',\n        children: [{ type: 'text', value: ev.text }],\n      },\n    ],\n  });\n\n  return children;\n}\n\nfunction collectEvidenceIds(state: IssueBundleState): string[] {\n  const ids = new Set<string>();\n  for (const prov of Object.values(state.servicesProvenance)) {\n    if (prov.effect?.evidenceId) ids.add(prov.effect.evidenceId);\n    if (prov.scopes?.evidenceId) ids.add(prov.scopes.evidenceId);\n    if (prov.periods?.evidenceId) ids.add(prov.periods.evidenceId);\n    if (prov.causes?.evidenceId) ids.add(prov.causes.evidenceId);\n  }\n  for (const prov of Object.values(state.facilitiesProvenance)) {\n    if (prov.effect?.evidenceId) ids.add(prov.effect.evidenceId);\n    if (prov.periods?.evidenceId) ids.add(prov.periods.evidenceId);\n  }\n  return [...ids];\n}\n\nfunction formatServiceSection(\n  key: string,\n  serviceId: string,\n  effect: { kind: string } | null,\n  scopes: ServiceScope[],\n  periods: Period[],\n  causes: string[],\n): Root['children'] {\n  const children: Root['children'] = [];\n\n  children.push({\n    type: 'heading',\n    depth: 3,\n    children: [{ type: 'text', value: key }],\n  });\n\n  children.push({\n    type: 'paragraph',\n    children: [\n      { type: 'text', value: 'Service ID: ' },\n      { type: 'inlineCode', value: serviceId },\n    ],\n  });\n\n  children.push({\n    type: 'paragraph',\n    children: [\n      { type: 'text', value: 'Effect: ' },\n      { type: 'inlineCode', value: effect?.kind ?? 'null' },\n    ],\n  });\n\n  if (scopes.length > 0) {\n    children.push({\n      type: 'paragraph',\n      children: [{ type: 'text', value: 'Scopes:' }],\n    });\n    children.push({\n      type: 'list',\n      ordered: false,\n      children: scopes.map((scope) => ({\n        type: 'listItem',\n        children: [\n          {\n            type: 'paragraph',\n            children: [{ type: 'inlineCode', value: formatScope(scope) }],\n          },\n        ],\n      })),\n    });\n  }\n\n  if (periods.length > 0) {\n    children.push({\n      type: 'paragraph',\n      children: [{ type: 'text', value: 'Periods:' }],\n    });\n    children.push({\n      type: 'list',\n      ordered: false,\n      children: periods.map((period, i) => ({\n        type: 'listItem',\n        children: [\n          {\n            type: 'paragraph',\n            children: [{ type: 'text', value: formatPeriod(period, i) }],\n          },\n        ],\n      })),\n    });\n  }\n\n  if (causes.length > 0) {\n    children.push({\n      type: 'paragraph',\n      children: [\n        { type: 'text', value: 'Causes: ' },\n        { type: 'inlineCode', value: causes.join(', ') },\n      ],\n    });\n  }\n\n  return children;\n}\n\nfunction formatFacilitySection(\n  key: string,\n  stationId: string,\n  kind: string,\n  effect: { kind: string } | null,\n  periods: Period[],\n): Root['children'] {\n  const children: Root['children'] = [];\n\n  children.push({\n    type: 'heading',\n    depth: 3,\n    children: [{ type: 'text', value: key }],\n  });\n\n  children.push({\n    type: 'paragraph',\n    children: [\n      { type: 'text', value: 'Station: ' },\n      { type: 'inlineCode', value: stationId },\n      { type: 'text', value: ', Kind: ' },\n      { type: 'inlineCode', value: kind },\n    ],\n  });\n\n  children.push({\n    type: 'paragraph',\n    children: [\n      { type: 'text', value: 'Effect: ' },\n      { type: 'inlineCode', value: effect?.kind ?? 'null' },\n    ],\n  });\n\n  if (periods.length > 0) {\n    children.push({\n      type: 'paragraph',\n      children: [{ type: 'text', value: 'Periods:' }],\n    });\n    children.push({\n      type: 'list',\n      ordered: false,\n      children: periods.map((period, i) => ({\n        type: 'listItem',\n        children: [\n          {\n            type: 'paragraph',\n            children: [{ type: 'text', value: formatPeriod(period, i) }],\n          },\n        ],\n      })),\n    });\n  }\n\n  return children;\n}\n\nfunction formatScope(scope: ServiceScope): string {\n  switch (scope.type) {\n    case 'service.whole':\n      return scope.type;\n    case 'service.segment':\n      return `${scope.type} (${scope.fromStationId} → ${scope.toStationId})`;\n    case 'service.point':\n      return `${scope.type} (${scope.stationId})`;\n  }\n}\n\nfunction formatPeriod(period: Period, index: number): string {\n  switch (period.kind) {\n    case 'fixed':\n      return `[${index}] fixed: ${period.startAt} → ${period.endAt ?? 'ongoing'}`;\n    case 'recurring':\n      return `[${index}] recurring ${period.frequency}: ${period.startAt}–${period.endAt} ${period.timeWindow.startAt}–${period.timeWindow.endAt}`;\n  }\n}\n"]}

package/dist/llm/common/tool.d.ts

@@ -0,0 +1,32 @@
+/**
+ * A generic class for LLM tools.
+ */
+export declare abstract class Tool<TParams = any> {
+    /**
+     * The name of the tool.
+     */
+    abstract name: string;
+    /**
+     * The description of the tool.
+     */
+    abstract description: string;
+    /**
+     * The JSON schema for the tool's parameters.
+     */
+    abstract get paramsSchema(): {
+        [key: string]: unknown;
+    };
+    /**
+     * Parse the parameters into a typed object.
+     * @param params
+     * @returns
+     */
+    abstract parseParams(params: unknown): TParams;
+    /**
+     * Run the tool with the given parameters.
+     * @param param
+     * @returns
+     */
+    abstract runner(param: TParams): Promise<string>;
+}
+export type ToolRegistry = Record<string, Tool>;

package/dist/llm/common/tool.js

@@ -0,0 +1,6 @@
+/**
+ * A generic class for LLM tools.
+ */
+export class Tool {
+}
+//# sourceMappingURL=tool.js.map

package/dist/llm/common/tool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool.js","sourceRoot":"/","sources":["llm/common/tool.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,OAAgB,IAAI;CA6BzB","sourcesContent":["/**\n * A generic class for LLM tools.\n */\nexport abstract class Tool<TParams = any> {\n  /**\n   * The name of the tool.\n   */\n  public abstract name: string;\n\n  /**\n   * The description of the tool.\n   */\n  public abstract description: string;\n\n  /**\n   * The JSON schema for the tool's parameters.\n   */\n  public abstract get paramsSchema(): { [key: string]: unknown };\n\n  /**\n   * Parse the parameters into a typed object.\n   * @param params\n   * @returns\n   */\n  public abstract parseParams(params: unknown): TParams;\n\n  /**\n   * Run the tool with the given parameters.\n   * @param param\n   * @returns\n   */\n  public abstract runner(param: TParams): Promise<string>;\n}\n\nexport type ToolRegistry = Record<string, Tool>;\n"]}

package/dist/llm/functions/extractClaimsFromNewEvidence/eval.test.d.ts

@@ -0,0 +1 @@
+import 'dotenv/config';