@reasoningco/infer 0.0.2

package/dist/llm.js

@@ -0,0 +1,311 @@
+import { InferError } from './errors.js';
+// ── OpenRouter Provider ──
+export class OpenRouterProvider {
+    apiKey;
+    model;
+    baseUrl;
+    constructor(config) {
+        this.apiKey = config.apiKey;
+        this.model = config.model ?? 'anthropic/claude-haiku-4-5';
+        this.baseUrl = config.baseUrl ?? 'https://openrouter.ai/api/v1';
+    }
+    async chat(messages, options) {
+        const timeout = options?.timeout ?? 10000;
+        const controller = new AbortController();
+        const timer = setTimeout(() => controller.abort(), timeout);
+        try {
+            const response = await fetch(`${this.baseUrl}/chat/completions`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    'Authorization': `Bearer ${this.apiKey}`,
+                },
+                body: JSON.stringify({
+                    model: this.model,
+                    messages,
+                    temperature: options?.temperature ?? 0,
+                    max_tokens: options?.maxTokens ?? 1000,
+                }),
+                signal: controller.signal,
+            });
+            if (!response.ok) {
+                const status = response.status;
+                if (status === 429)
+                    throw new InferError('RATE_LIMITED', 'planner', 'Rate limited by LLM provider');
+                throw new InferError('LLM_UNAVAILABLE', 'planner', `LLM provider returned ${status}`);
+            }
+            const data = await response.json();
+            return data.choices?.[0]?.message?.content ?? '';
+        }
+        catch (error) {
+            if (error instanceof InferError)
+                throw error;
+            if (error?.name === 'AbortError') {
+                throw new InferError('LLM_TIMEOUT', 'planner', 'LLM request timed out');
+            }
+            throw new InferError('LLM_UNAVAILABLE', 'planner', 'LLM request failed');
+        }
+        finally {
+            clearTimeout(timer);
+        }
+    }
+}
+// ── OpenAI Provider ──
+export class OpenAIProvider {
+    apiKey;
+    model;
+    baseUrl;
+    constructor(config) {
+        this.apiKey = config.apiKey;
+        this.model = config.model ?? 'gpt-4o';
+        this.baseUrl = config.baseUrl ?? 'https://api.openai.com/v1';
+    }
+    async chat(messages, options) {
+        const timeout = options?.timeout ?? 10000;
+        const controller = new AbortController();
+        const timer = setTimeout(() => controller.abort(), timeout);
+        try {
+            const response = await fetch(`${this.baseUrl}/chat/completions`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    'Authorization': `Bearer ${this.apiKey}`,
+                },
+                body: JSON.stringify({
+                    model: this.model,
+                    messages,
+                    temperature: options?.temperature ?? 0,
+                    max_tokens: options?.maxTokens ?? 1000,
+                }),
+                signal: controller.signal,
+            });
+            if (!response.ok) {
+                const status = response.status;
+                if (status === 429)
+                    throw new InferError('RATE_LIMITED', 'planner', 'Rate limited');
+                throw new InferError('LLM_UNAVAILABLE', 'planner', `OpenAI returned ${status}`);
+            }
+            const data = await response.json();
+            return data.choices?.[0]?.message?.content ?? '';
+        }
+        catch (error) {
+            if (error instanceof InferError)
+                throw error;
+            if (error?.name === 'AbortError')
+                throw new InferError('LLM_TIMEOUT', 'planner', 'Timed out');
+            throw new InferError('LLM_UNAVAILABLE', 'planner', 'OpenAI request failed');
+        }
+        finally {
+            clearTimeout(timer);
+        }
+    }
+}
+// ── Anthropic Provider ──
+export class AnthropicProvider {
+    apiKey;
+    model;
+    baseUrl;
+    constructor(config) {
+        this.apiKey = config.apiKey;
+        this.model = config.model ?? 'claude-sonnet-4-6';
+        this.baseUrl = config.baseUrl ?? 'https://api.anthropic.com/v1';
+    }
+    async chat(messages, options) {
+        const timeout = options?.timeout ?? 10000;
+        const controller = new AbortController();
+        const timer = setTimeout(() => controller.abort(), timeout);
+        try {
+            // Separate system message from user/assistant messages
+            const systemMsg = messages.find(m => m.role === 'system')?.content ?? '';
+            const chatMessages = messages.filter(m => m.role !== 'system');
+            const response = await fetch(`${this.baseUrl}/messages`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    'x-api-key': this.apiKey,
+                    'anthropic-version': '2023-06-01',
+                },
+                body: JSON.stringify({
+                    model: this.model,
+                    system: systemMsg,
+                    messages: chatMessages,
+                    max_tokens: options?.maxTokens ?? 1000,
+                    temperature: options?.temperature ?? 0,
+                }),
+                signal: controller.signal,
+            });
+            if (!response.ok) {
+                const status = response.status;
+                if (status === 429)
+                    throw new InferError('RATE_LIMITED', 'planner', 'Rate limited');
+                throw new InferError('LLM_UNAVAILABLE', 'planner', `Anthropic returned ${status}`);
+            }
+            const data = await response.json();
+            return data.content?.[0]?.text ?? '';
+        }
+        catch (error) {
+            if (error instanceof InferError)
+                throw error;
+            if (error?.name === 'AbortError')
+                throw new InferError('LLM_TIMEOUT', 'planner', 'Timed out');
+            throw new InferError('LLM_UNAVAILABLE', 'planner', 'Anthropic request failed');
+        }
+        finally {
+            clearTimeout(timer);
+        }
+    }
+}
+// ── Ollama Provider ──
+export class OllamaProvider {
+    model;
+    baseUrl;
+    constructor(config) {
+        this.model = config.model ?? 'llama3';
+        this.baseUrl = config.baseUrl ?? 'http://localhost:11434';
+    }
+    async chat(messages, options) {
+        const timeout = options?.timeout ?? 30000;
+        const controller = new AbortController();
+        const timer = setTimeout(() => controller.abort(), timeout);
+        try {
+            const response = await fetch(`${this.baseUrl}/api/chat`, {
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({
+                    model: this.model,
+                    messages,
+                    stream: false,
+                    options: {
+                        temperature: options?.temperature ?? 0,
+                        num_predict: options?.maxTokens ?? 1000,
+                    },
+                }),
+                signal: controller.signal,
+            });
+            if (!response.ok)
+                throw new InferError('LLM_UNAVAILABLE', 'planner', `Ollama returned ${response.status}`);
+            const data = await response.json();
+            return data.message?.content ?? '';
+        }
+        catch (error) {
+            if (error instanceof InferError)
+                throw error;
+            if (error?.name === 'AbortError')
+                throw new InferError('LLM_TIMEOUT', 'planner', 'Timed out');
+            throw new InferError('LLM_UNAVAILABLE', 'planner', 'Ollama request failed');
+        }
+        finally {
+            clearTimeout(timer);
+        }
+    }
+}
+// ── Provider Factory ──
+export function createLLMProvider(config) {
+    switch (config.provider) {
+        case 'openrouter':
+            if (!config.apiKey)
+                throw new Error('apiKey required for OpenRouter');
+            return new OpenRouterProvider({ apiKey: config.apiKey, model: config.model, baseUrl: config.baseUrl });
+        case 'openai':
+            if (!config.apiKey)
+                throw new Error('apiKey required for OpenAI');
+            return new OpenAIProvider({ apiKey: config.apiKey, model: config.model, baseUrl: config.baseUrl });
+        case 'anthropic':
+            if (!config.apiKey)
+                throw new Error('apiKey required for Anthropic');
+            return new AnthropicProvider({ apiKey: config.apiKey, model: config.model, baseUrl: config.baseUrl });
+        case 'ollama':
+            return new OllamaProvider({ model: config.model, baseUrl: config.baseUrl });
+        default:
+            throw new Error(`Unknown LLM provider: ${config.provider}`);
+    }
+}
+// ── System Prompt Auto-Generation (from ChefOS smartPlanner.ts buildCatalogSummary) ──
+export function buildCatalogSummary(catalog) {
+    const lines = [
+        'You are a query planner for a database with these tables:',
+        '',
+    ];
+    for (const [id, ds] of Object.entries(catalog.datasets)) {
+        lines.push(`**${id}** (${ds.table.schema}.${ds.table.name}${ds.estimatedRows ? `, ~${formatRowCount(ds.estimatedRows)} rows` : ''})`);
+        // Columns
+        const colDescs = ds.selectableColumns.map(col => {
+            const type = ds.columnTypes[col] || 'text';
+            return `${col} (${type})`;
+        });
+        lines.push(`  Columns: ${colDescs.join(', ')}`);
+        // Metrics
+        if (ds.metrics.length > 0) {
+            const metricDescs = ds.metrics.map(m => `${m.id} = ${m.expression}${m.description ? ` — "${m.description}"` : ''}`);
+            lines.push(`  Metrics: ${metricDescs.join(', ')}`);
+        }
+        // Dimensions
+        const groupableDims = ds.dimensions.filter(d => d.groupable);
+        if (groupableDims.length > 0) {
+            lines.push(`  Group by: ${groupableDims.map(d => d.column).join(', ')}`);
+        }
+        // Entities
+        if (ds.entities.length > 0) {
+            lines.push(`  Entities: ${ds.entities.map(e => `${e.label} (${e.key})`).join(', ')}`);
+        }
+        // Joins
+        if (ds.joins && ds.joins.length > 0) {
+            lines.push(`  Joins: ${ds.joins.map(j => `→ ${j.table} via ${j.localColumn}`).join(', ')}`);
+        }
+        // Date column
+        if (ds.dateColumn) {
+            lines.push(`  Date column: ${ds.dateColumn}`);
+        }
+        lines.push('');
+    }
+    lines.push('Given a user question, return a JSON query plan with this shape:');
+    lines.push('```json');
+    lines.push(JSON.stringify({
+        dataset: "dataset_id",
+        metrics: ["metric_id"],
+        groupBy: ["column"],
+        filters: [{ field: "column", op: "=", value: "value" }],
+        dateRange: { mode: "relative", phrase: "last 30 days" },
+        orderBy: [{ field: "column", dir: "desc" }],
+        limit: 50,
+        answerMode: "aggregate_table",
+    }, null, 2));
+    lines.push('```');
+    lines.push('');
+    lines.push('Rules:');
+    lines.push('- Only use datasets, columns, and metrics listed above');
+    lines.push('- Use answerMode "scalar" for single-number questions like "how many orders"');
+    lines.push('- Use answerMode "aggregate_table" for grouped data like "revenue by month"');
+    lines.push('- Use answerMode "rows" for listing individual records');
+    lines.push('- Date ranges: use "relative" with natural phrases like "last 7 days", "this month", "last quarter"');
+    lines.push('- Return ONLY the JSON object, no explanation');
+    return lines.join('\n');
+}
+// ── JSON Extraction (from ChefOS reportPlanner.ts) ──
+export function extractJSON(text) {
+    // Strip markdown fences
+    let cleaned = text.trim();
+    const fenceMatch = cleaned.match(/```(?:json)?\s*([\s\S]*?)```/);
+    if (fenceMatch) {
+        cleaned = fenceMatch[1].trim();
+    }
+    // Try to find JSON object
+    const jsonMatch = cleaned.match(/\{[\s\S]*\}/);
+    if (jsonMatch) {
+        try {
+            return JSON.parse(jsonMatch[0]);
+        }
+        catch {
+            throw new InferError('LLM_INVALID_RESPONSE', 'planner', 'Failed to parse LLM JSON output');
+        }
+    }
+    throw new InferError('LLM_INVALID_RESPONSE', 'planner', 'No JSON found in LLM response');
+}
+function formatRowCount(n) {
+    if (n >= 1_000_000)
+        return `${(n / 1_000_000).toFixed(1)}M`;
+    if (n >= 1_000)
+        return `${(n / 1_000).toFixed(0)}K`;
+    return String(n);
+}
+//# sourceMappingURL=llm.js.map

package/dist/llm.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm.js","sourceRoot":"","sources":["../src/llm.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAQzC,4BAA4B;AAC5B,MAAM,OAAO,kBAAkB;IACrB,MAAM,CAAS;IACf,KAAK,CAAS;IACd,OAAO,CAAS;IAExB,YAAY,MAA4D;QACtE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,4BAA4B,CAAC;QAC1D,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,8BAA8B,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAsB,EAAE,OAAoB;QACrD,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,KAAK,CAAC;QAC1C,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,OAAO,CAAC,CAAC;QAE5D,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,mBAAmB,EAAE;gBAC/D,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;iBACzC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,QAAQ;oBACR,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,CAAC;oBACtC,UAAU,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI;iBACvC,CAAC;gBACF,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;gBAC/B,IAAI,MAAM,KAAK,GAAG;oBAAE,MAAM,IAAI,UAAU,CAAC,cAAc,EAAE,SAAS,EAAE,8BAA8B,CAAC,CAAC;gBACpG,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,SAAS,EAAE,yBAAyB,MAAM,EAAE,CAAC,CAAC;YACxF,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QACnD,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,KAAK,YAAY,UAAU;gBAAE,MAAM,KAAK,CAAC;YAC7C,IAAI,KAAK,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;gBACjC,MAAM,IAAI,UAAU,CAAC,aAAa,EAAE,SAAS,EAAE,uBAAuB,CAAC,CAAC;YAC1E,CAAC;YACD,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,SAAS,EAAE,oBAAoB,CAAC,CAAC;QAC3E,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;CACF;AAED,wBAAwB;AACxB,MAAM,OAAO,cAAc;IACjB,MAAM,CAAS;IACf,KAAK,CAAS;IACd,OAAO,CAAS;IAExB,YAAY,MAA4D;QACtE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,QAAQ,CAAC;QACtC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,2BAA2B,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAsB,EAAE,OAAoB;QACrD,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,KAAK,CAAC;QAC1C,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,OAAO,CAAC,CAAC;QAE5D,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,mBAAmB,EAAE;gBAC/D,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;iBACzC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,QAAQ;oBACR,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,CAAC;oBACtC,UAAU,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI;iBACvC,CAAC;gBACF,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;gBAC/B,IAAI,MAAM,KAAK,GAAG;oBAAE,MAAM,IAAI,UAAU,CAAC,cAAc,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC;gBACpF,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,SAAS,EAAE,mBAAmB,MAAM,EAAE,CAAC,CAAC;YAClF,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QACnD,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,KAAK,YAAY,UAAU;gBAAE,MAAM,KAAK,CAAC;YAC7C,IAAI,KAAK,EAAE,IAAI,KAAK,YAAY;gBAAE,MAAM,IAAI,UAAU,CAAC,aAAa,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;YAC9F,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,SAAS,EAAE,uBAAuB,CAAC,CAAC;QAC9E,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;CACF;AAED,2BAA2B;AAC3B,MAAM,OAAO,iBAAiB;IACpB,MAAM,CAAS;IACf,KAAK,CAAS;IACd,OAAO,CAAS;IAExB,YAAY,MAA4D;QACtE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,mBAAmB,CAAC;QACjD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,8BAA8B,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAsB,EAAE,OAAoB;QACrD,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,KAAK,CAAC;QAC1C,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,OAAO,CAAC,CAAC;QAE5D,IAAI,CAAC;YACH,uDAAuD;YACvD,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,EAAE,OAAO,IAAI,EAAE,CAAC;YACzE,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;YAE/D,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,EAAE;gBACvD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,IAAI,CAAC,MAAM;oBACxB,mBAAmB,EAAE,YAAY;iBAClC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,MAAM,EAAE,SAAS;oBACjB,QAAQ,EAAE,YAAY;oBACtB,UAAU,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI;oBACtC,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,CAAC;iBACvC,CAAC;gBACF,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;gBAC/B,IAAI,MAAM,KAAK,GAAG;oBAAE,MAAM,IAAI,UAAU,CAAC,cAAc,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC;gBACpF,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,SAAS,EAAE,sBAAsB,MAAM,EAAE,CAAC,CAAC;YACrF,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;QACvC,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,KAAK,YAAY,UAAU;gBAAE,MAAM,KAAK,CAAC;YAC7C,IAAI,KAAK,EAAE,IAAI,KAAK,YAAY;gBAAE,MAAM,IAAI,UAAU,CAAC,aAAa,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;YAC9F,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,SAAS,EAAE,0BAA0B,CAAC,CAAC;QACjF,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;CACF;AAED,wBAAwB;AACxB,MAAM,OAAO,cAAc;IACjB,KAAK,CAAS;IACd,OAAO,CAAS;IAExB,YAAY,MAA4C;QACtD,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,QAAQ,CAAC;QACtC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,wBAAwB,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAsB,EAAE,OAAoB;QACrD,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,KAAK,CAAC;QAC1C,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,OAAO,CAAC,CAAC;QAE5D,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,EAAE;gBACvD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,QAAQ;oBACR,MAAM,EAAE,KAAK;oBACb,OAAO,EAAE;wBACP,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,CAAC;wBACtC,WAAW,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI;qBACxC;iBACF,CAAC;gBACF,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE;gBAAE,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,SAAS,EAAE,mBAAmB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YAC3G,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,OAAO,IAAI,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QACrC,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,KAAK,YAAY,UAAU;gBAAE,MAAM,KAAK,CAAC;YAC7C,IAAI,KAAK,EAAE,IAAI,KAAK,YAAY;gBAAE,MAAM,IAAI,UAAU,CAAC,aAAa,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;YAC9F,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,SAAS,EAAE,uBAAuB,CAAC,CAAC;QAC9E,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;CACF;AAED,yBAAyB;AACzB,MAAM,UAAU,iBAAiB,CAAC,MAAyB;IACzD,QAAQ,MAAM,CAAC,QAAQ,EAAE,CAAC;QACxB,KAAK,YAAY;YACf,IAAI,CAAC,MAAM,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;YACtE,OAAO,IAAI,kBAAkB,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;QACzG,KAAK,QAAQ;YACX,IAAI,CAAC,MAAM,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAClE,OAAO,IAAI,cAAc,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;QACrG,KAAK,WAAW;YACd,IAAI,CAAC,MAAM,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;YACrE,OAAO,IAAI,iBAAiB,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;QACxG,KAAK,QAAQ;YACX,OAAO,IAAI,cAAc,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;QAC9E;YACE,MAAM,IAAI,KAAK,CAAC,yBAAyB,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAChE,CAAC;AACH,CAAC;AAED,wFAAwF;AACxF,MAAM,UAAU,mBAAmB,CAAC,OAAqB;IACvD,MAAM,KAAK,GAAa;QACtB,2DAA2D;QAC3D,EAAE;KACH,CAAC;IAEF,KAAK,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACxD,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,GAAG,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,MAAM,cAAc,CAAC,EAAE,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAEtI,UAAU;QACV,MAAM,QAAQ,GAAG,EAAE,CAAC,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;YAC9C,MAAM,IAAI,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC;YAC3C,OAAO,GAAG,GAAG,KAAK,IAAI,GAAG,CAAC;QAC5B,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,IAAI,CAAC,cAAc,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEhD,UAAU;QACV,IAAI,EAAE,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,WAAW,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACpH,KAAK,CAAC,IAAI,CAAC,cAAc,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACrD,CAAC;QAED,aAAa;QACb,MAAM,aAAa,GAAG,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAC7D,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,KAAK,CAAC,IAAI,CAAC,eAAe,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3E,CAAC;QAED,WAAW;QACX,IAAI,EAAE,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,KAAK,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACxF,CAAC;QAED,QAAQ;QACR,IAAI,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpC,KAAK,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC9F,CAAC;QAED,cAAc;QACd,IAAI,EAAE,CAAC,UAAU,EAAE,CAAC;YAClB,KAAK,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC;QAChD,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,kEAAkE,CAAC,CAAC;IAC/E,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACtB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;QACxB,OAAO,EAAE,YAAY;QACrB,OAAO,EAAE,CAAC,WAAW,CAAC;QACtB,OAAO,EAAE,CAAC,QAAQ,CAAC;QACnB,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;QACvD,SAAS,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,cAAc,EAAE;QACvD,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC;QAC3C,KAAK,EAAE,EAAE;QACT,UAAU,EAAE,iBAAiB;KAC9B,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACb,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACrB,KAAK,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;IACrE,KAAK,CAAC,IAAI,CAAC,8EAA8E,CAAC,CAAC;IAC3F,KAAK,CAAC,IAAI,CAAC,6EAA6E,CAAC,CAAC;IAC1F,KAAK,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;IACrE,KAAK,CAAC,IAAI,CAAC,qGAAqG,CAAC,CAAC;IAClH,KAAK,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;IAE5D,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,uDAAuD;AACvD,MAAM,UAAU,WAAW,CAAC,IAAY;IACtC,wBAAwB;IACxB,IAAI,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAC1B,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;IACjE,IAAI,UAAU,EAAE,CAAC;QACf,OAAO,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IACjC,CAAC;IAED,0BAA0B;IAC1B,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;IAC/C,IAAI,SAAS,EAAE,CAAC;QACd,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,UAAU,CAAC,sBAAsB,EAAE,SAAS,EAAE,iCAAiC,CAAC,CAAC;QAC7F,CAAC;IACH,CAAC;IAED,MAAM,IAAI,UAAU,CAAC,sBAAsB,EAAE,SAAS,EAAE,+BAA+B,CAAC,CAAC;AAC3F,CAAC;AAED,SAAS,cAAc,CAAC,CAAS;IAC/B,IAAI,CAAC,IAAI,SAAS;QAAE,OAAO,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IAC5D,IAAI,CAAC,IAAI,KAAK;QAAE,OAAO,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IACpD,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;AACnB,CAAC"}

package/dist/query-validator.d.ts

@@ -0,0 +1,6 @@
+import type { InferCatalog, ValidatedSmartPlan } from './types.js';
+export declare function validateQueryPlan(rawPlan: unknown, catalog: InferCatalog, options?: {
+    maxLimit?: number;
+    defaultLimit?: number;
+}): ValidatedSmartPlan;
+//# sourceMappingURL=query-validator.d.ts.map

package/dist/query-validator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-validator.d.ts","sourceRoot":"","sources":["../src/query-validator.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,YAAY,EAAgC,kBAAkB,EAG/D,MAAM,YAAY,CAAC;AAKpB,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,OAAO,EAChB,OAAO,EAAE,YAAY,EACrB,OAAO,CAAC,EAAE;IAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;IAAC,YAAY,CAAC,EAAE,MAAM,CAAA;CAAE,GACrD,kBAAkB,CA0IpB"}

package/dist/query-validator.js

@@ -0,0 +1,136 @@
+import { InferError } from './errors.js';
+import { SmartQueryPlanSchema } from './schemas.js';
+import { resolveDateRange } from './dates.js';
+import { allowlistedMetricIds, allowlistedGroupByColumns } from './types.js';
+export function validateQueryPlan(rawPlan, catalog, options) {
+    const maxLimit = options?.maxLimit ?? 200;
+    const defaultLimit = options?.defaultLimit ?? 200;
+    // 1. Parse through Zod
+    const parseResult = SmartQueryPlanSchema.safeParse(rawPlan);
+    if (!parseResult.success) {
+        throw new InferError('PLAN_INVALID', 'validator', 'Invalid query plan structure', {
+            zodErrors: parseResult.error.flatten(),
+        });
+    }
+    const plan = parseResult.data;
+    // 2. Validate dataset exists in catalog
+    const dataset = catalog.datasets[plan.dataset];
+    if (!dataset) {
+        throw new InferError('PLAN_OUT_OF_SCOPE', 'validator', `Unknown dataset: ${plan.dataset}`);
+    }
+    // 3. Determine answer mode
+    const answerMode = plan.answerMode ?? 'rows';
+    // 4. Validate select columns (for "rows" mode)
+    const select = [];
+    if (answerMode === 'rows' && plan.select && plan.select.length > 0) {
+        for (const col of plan.select) {
+            if (!dataset.selectableColumns.includes(col)) {
+                throw new InferError('PLAN_INVALID', 'validator', `Column not selectable: ${col}`);
+            }
+            // Check sensitive columns
+            if (dataset.sensitiveColumns.includes(col)) {
+                throw new InferError('PLAN_SENSITIVE_COLUMN', 'validator', `Access to sensitive column denied`);
+            }
+            select.push(col);
+        }
+    }
+    // 5. Validate metrics
+    const allowedMetrics = allowlistedMetricIds(dataset);
+    const metrics = [];
+    if (plan.metrics && plan.metrics.length > 0) {
+        for (const metricId of plan.metrics) {
+            if (!allowedMetrics.has(metricId)) {
+                throw new InferError('PLAN_INVALID', 'validator', `Unknown metric: ${metricId}`);
+            }
+            const metricDef = dataset.metrics.find(m => m.id === metricId);
+            if (metricDef)
+                metrics.push(metricDef);
+        }
+    }
+    // 6. Validate groupBy columns
+    const allowedGroupBy = allowlistedGroupByColumns(dataset);
+    const groupBy = [];
+    if (plan.groupBy && plan.groupBy.length > 0) {
+        for (const col of plan.groupBy) {
+            if (!allowedGroupBy.has(col)) {
+                throw new InferError('PLAN_INVALID', 'validator', `Column not groupable: ${col}`);
+            }
+            if (dataset.sensitiveColumns.includes(col)) {
+                throw new InferError('PLAN_SENSITIVE_COLUMN', 'validator', `Access to sensitive column denied`);
+            }
+            groupBy.push(col);
+        }
+    }
+    // 7. Validate filters
+    const filters = [];
+    let paramIndex = 1;
+    if (plan.filters && plan.filters.length > 0) {
+        for (const filter of plan.filters) {
+            if (!dataset.filterableColumns.includes(filter.field)) {
+                throw new InferError('PLAN_INVALID', 'validator', `Column not filterable: ${filter.field}`);
+            }
+            if (dataset.sensitiveColumns.includes(filter.field)) {
+                throw new InferError('PLAN_SENSITIVE_COLUMN', 'validator', `Access to sensitive column denied`);
+            }
+            filters.push({
+                field: filter.field,
+                op: filter.op,
+                value: filter.value,
+                paramIndex: paramIndex++,
+            });
+        }
+    }
+    // 8. Resolve date range
+    let dateRange;
+    let timeMode = dataset.timeConfig.defaultMode;
+    if (plan.dateRange) {
+        if (plan.dateRange.mode === 'none') {
+            timeMode = 'none';
+        }
+        else if (plan.dateRange.mode === 'latest') {
+            timeMode = 'latest';
+        }
+        else if (plan.dateRange.mode === 'relative' && plan.dateRange.phrase) {
+            const resolved = resolveDateRange({ phrase: plan.dateRange.phrase });
+            if (resolved) {
+                dateRange = resolved;
+                timeMode = 'relative';
+            }
+        }
+        else if (plan.dateRange.mode === 'explicit' && plan.dateRange.startDate && plan.dateRange.endDate) {
+            dateRange = { startDate: plan.dateRange.startDate, endDate: plan.dateRange.endDate };
+            timeMode = 'absolute';
+        }
+    }
+    // 9. Validate orderBy
+    const orderBy = [];
+    if (plan.orderBy && plan.orderBy.length > 0) {
+        for (const ob of plan.orderBy) {
+            // orderBy can reference select columns, groupBy columns, or metric ids
+            const validField = dataset.selectableColumns.includes(ob.field) ||
+                groupBy.includes(ob.field) ||
+                metrics.some(m => m.id === ob.field);
+            if (!validField) {
+                // Silently skip invalid orderBy rather than throwing
+                continue;
+            }
+            orderBy.push({ field: ob.field, dir: ob.dir === 'desc' ? 'desc' : 'asc' });
+        }
+    }
+    // 10. Cap limit
+    const limit = Math.min(plan.limit ?? defaultLimit, maxLimit);
+    return {
+        dataset,
+        answerMode,
+        select,
+        metrics,
+        groupBy,
+        filters,
+        resolvedEntities: [], // populated later by entity resolver
+        dateRange,
+        timeMode,
+        orderBy,
+        limit,
+    };
+}
+//# sourceMappingURL=query-validator.js.map

package/dist/query-validator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-validator.js","sourceRoot":"","sources":["../src/query-validator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAMzC,OAAO,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAA8C,MAAM,YAAY,CAAC;AAC1F,OAAO,EAAE,oBAAoB,EAAE,yBAAyB,EAAuB,MAAM,YAAY,CAAC;AAElG,MAAM,UAAU,iBAAiB,CAC/B,OAAgB,EAChB,OAAqB,EACrB,OAAsD;IAEtD,MAAM,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,GAAG,CAAC;IAC1C,MAAM,YAAY,GAAG,OAAO,EAAE,YAAY,IAAI,GAAG,CAAC;IAElD,uBAAuB;IACvB,MAAM,WAAW,GAAG,oBAAoB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAC5D,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;QACzB,MAAM,IAAI,UAAU,CAAC,cAAc,EAAE,WAAW,EAAE,8BAA8B,EAAE;YAChF,SAAS,EAAE,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE;SACvC,CAAC,CAAC;IACL,CAAC;IACD,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC;IAE9B,wCAAwC;IACxC,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC/C,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,UAAU,CAAC,mBAAmB,EAAE,WAAW,EAAE,oBAAoB,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAC7F,CAAC;IAED,2BAA2B;IAC3B,MAAM,UAAU,GAAe,IAAI,CAAC,UAAU,IAAI,MAAM,CAAC;IAEzD,+CAA+C;IAC/C,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,UAAU,KAAK,MAAM,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACnE,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAC9B,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC7C,MAAM,IAAI,UAAU,CAAC,cAAc,EAAE,WAAW,EAAE,0BAA0B,GAAG,EAAE,CAAC,CAAC;YACrF,CAAC;YACD,0BAA0B;YAC1B,IAAI,OAAO,CAAC,gBAAgB,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC3C,MAAM,IAAI,UAAU,CAAC,uBAAuB,EAAE,WAAW,EAAE,mCAAmC,CAAC,CAAC;YAClG,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;IAED,sBAAsB;IACtB,MAAM,cAAc,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAC;IACrD,MAAM,OAAO,GAAgB,EAAE,CAAC;IAChC,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5C,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACpC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAClC,MAAM,IAAI,UAAU,CAAC,cAAc,EAAE,WAAW,EAAE,mBAAmB,QAAQ,EAAE,CAAC,CAAC;YACnF,CAAC;YACD,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,QAAQ,CAAC,CAAC;YAC/D,IAAI,SAAS;gBAAE,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,8BAA8B;IAC9B,MAAM,cAAc,GAAG,yBAAyB,CAAC,OAAO,CAAC,CAAC;IAC1D,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5C,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC/B,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC7B,MAAM,IAAI,UAAU,CAAC,cAAc,EAAE,WAAW,EAAE,yBAAyB,GAAG,EAAE,CAAC,CAAC;YACpF,CAAC;YACD,IAAI,OAAO,CAAC,gBAAgB,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC3C,MAAM,IAAI,UAAU,CAAC,uBAAuB,EAAE,WAAW,EAAE,mCAAmC,CAAC,CAAC;YAClG,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;IAED,sBAAsB;IACtB,MAAM,OAAO,GAAkC,EAAE,CAAC;IAClD,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5C,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;gBACtD,MAAM,IAAI,UAAU,CAAC,cAAc,EAAE,WAAW,EAAE,0BAA0B,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;YAC9F,CAAC;YACD,IAAI,OAAO,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;gBACpD,MAAM,IAAI,UAAU,CAAC,uBAAuB,EAAE,WAAW,EAAE,mCAAmC,CAAC,CAAC;YAClG,CAAC;YACD,OAAO,CAAC,IAAI,CAAC;gBACX,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,UAAU,EAAE,UAAU,EAAE;aACzB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,wBAAwB;IACxB,IAAI,SAA6D,CAAC;IAClE,IAAI,QAAQ,GAAa,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC;IAExD,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;QACnB,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACnC,QAAQ,GAAG,MAAM,CAAC;QACpB,CAAC;aAAM,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC5C,QAAQ,GAAG,QAAQ,CAAC;QACtB,CAAC;aAAM,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC;YACvE,MAAM,QAAQ,GAAG,gBAAgB,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;YACrE,IAAI,QAAQ,EAAE,CAAC;gBACb,SAAS,GAAG,QAAQ,CAAC;gBACrB,QAAQ,GAAG,UAAU,CAAC;YACxB,CAAC;QACH,CAAC;aAAM,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,CAAC,SAAS,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;YACpG,SAAS,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;YACrF,QAAQ,GAAG,UAAU,CAAC;QACxB,CAAC;IACH,CAAC;IAED,sBAAsB;IACtB,MAAM,OAAO,GAAoB,EAAE,CAAC;IACpC,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5C,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC9B,uEAAuE;YACvE,MAAM,UAAU,GAAG,OAAO,CAAC,iBAAiB,CAAC,QAAQ,CAAC,EAAE,CAAC,KAAK,CAAC;gBAC7D,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,KAAK,CAAC;gBAC1B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,KAAK,CAAC,CAAC;YACvC,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,qDAAqD;gBACrD,SAAS;YACX,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;QAC7E,CAAC;IACH,CAAC;IAED,gBAAgB;IAChB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,YAAY,EAAE,QAAQ,CAAC,CAAC;IAE7D,OAAO;QACL,OAAO;QACP,UAAU;QACV,MAAM;QACN,OAAO;QACP,OAAO;QACP,OAAO;QACP,gBAAgB,EAAE,EAAE,EAAE,qCAAqC;QAC3D,SAAS;QACT,QAAQ;QACR,OAAO;QACP,KAAK;KACN,CAAC;AACJ,CAAC"}

package/dist/schemas.d.ts

@@ -0,0 +1,170 @@
+import { z } from "zod";
+export declare const FilterSchema: z.ZodObject<{
+    field: z.ZodString;
+    op: z.ZodEnum<["=", "!=", ">", "<", ">=", "<=", "in", "not_in", "like", "ilike"]>;
+    value: z.ZodUnion<[z.ZodString, z.ZodNumber, z.ZodBoolean, z.ZodArray<z.ZodUnion<[z.ZodString, z.ZodNumber]>, "many">]>;
+}, "strip", z.ZodTypeAny, {
+    value: string | number | boolean | (string | number)[];
+    field: string;
+    op: "=" | "!=" | ">" | "<" | ">=" | "<=" | "in" | "not_in" | "like" | "ilike";
+}, {
+    value: string | number | boolean | (string | number)[];
+    field: string;
+    op: "=" | "!=" | ">" | "<" | ">=" | "<=" | "in" | "not_in" | "like" | "ilike";
+}>;
+export declare const EntityFilterSchema: z.ZodObject<{
+    resolver: z.ZodString;
+    input: z.ZodString;
+    outputColumn: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    input: string;
+    resolver: string;
+    outputColumn: string;
+}, {
+    input: string;
+    resolver: string;
+    outputColumn: string;
+}>;
+export declare const DateRangeSchema: z.ZodObject<{
+    mode: z.ZodEnum<["relative", "explicit", "latest", "none"]>;
+    phrase: z.ZodOptional<z.ZodString>;
+    startDate: z.ZodOptional<z.ZodString>;
+    endDate: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    mode: "latest" | "none" | "relative" | "explicit";
+    startDate?: string | undefined;
+    endDate?: string | undefined;
+    phrase?: string | undefined;
+}, {
+    mode: "latest" | "none" | "relative" | "explicit";
+    startDate?: string | undefined;
+    endDate?: string | undefined;
+    phrase?: string | undefined;
+}>;
+export declare const OrderBySchema: z.ZodObject<{
+    field: z.ZodString;
+    dir: z.ZodDefault<z.ZodEnum<["asc", "desc"]>>;
+}, "strip", z.ZodTypeAny, {
+    field: string;
+    dir: "asc" | "desc";
+}, {
+    field: string;
+    dir?: "asc" | "desc" | undefined;
+}>;
+export declare const SmartQueryPlanSchema: z.ZodObject<{
+    dataset: z.ZodString;
+    select: z.ZodDefault<z.ZodOptional<z.ZodArray<z.ZodString, "many">>>;
+    metrics: z.ZodDefault<z.ZodOptional<z.ZodArray<z.ZodString, "many">>>;
+    groupBy: z.ZodDefault<z.ZodOptional<z.ZodArray<z.ZodString, "many">>>;
+    filters: z.ZodDefault<z.ZodOptional<z.ZodArray<z.ZodObject<{
+        field: z.ZodString;
+        op: z.ZodEnum<["=", "!=", ">", "<", ">=", "<=", "in", "not_in", "like", "ilike"]>;
+        value: z.ZodUnion<[z.ZodString, z.ZodNumber, z.ZodBoolean, z.ZodArray<z.ZodUnion<[z.ZodString, z.ZodNumber]>, "many">]>;
+    }, "strip", z.ZodTypeAny, {
+        value: string | number | boolean | (string | number)[];
+        field: string;
+        op: "=" | "!=" | ">" | "<" | ">=" | "<=" | "in" | "not_in" | "like" | "ilike";
+    }, {
+        value: string | number | boolean | (string | number)[];
+        field: string;
+        op: "=" | "!=" | ">" | "<" | ">=" | "<=" | "in" | "not_in" | "like" | "ilike";
+    }>, "many">>>;
+    entityFilters: z.ZodDefault<z.ZodOptional<z.ZodArray<z.ZodObject<{
+        resolver: z.ZodString;
+        input: z.ZodString;
+        outputColumn: z.ZodString;
+    }, "strip", z.ZodTypeAny, {
+        input: string;
+        resolver: string;
+        outputColumn: string;
+    }, {
+        input: string;
+        resolver: string;
+        outputColumn: string;
+    }>, "many">>>;
+    dateRange: z.ZodOptional<z.ZodObject<{
+        mode: z.ZodEnum<["relative", "explicit", "latest", "none"]>;
+        phrase: z.ZodOptional<z.ZodString>;
+        startDate: z.ZodOptional<z.ZodString>;
+        endDate: z.ZodOptional<z.ZodString>;
+    }, "strip", z.ZodTypeAny, {
+        mode: "latest" | "none" | "relative" | "explicit";
+        startDate?: string | undefined;
+        endDate?: string | undefined;
+        phrase?: string | undefined;
+    }, {
+        mode: "latest" | "none" | "relative" | "explicit";
+        startDate?: string | undefined;
+        endDate?: string | undefined;
+        phrase?: string | undefined;
+    }>>;
+    orderBy: z.ZodDefault<z.ZodOptional<z.ZodArray<z.ZodObject<{
+        field: z.ZodString;
+        dir: z.ZodDefault<z.ZodEnum<["asc", "desc"]>>;
+    }, "strip", z.ZodTypeAny, {
+        field: string;
+        dir: "asc" | "desc";
+    }, {
+        field: string;
+        dir?: "asc" | "desc" | undefined;
+    }>, "many">>>;
+    limit: z.ZodOptional<z.ZodNumber>;
+    answerMode: z.ZodDefault<z.ZodOptional<z.ZodEnum<["rows", "aggregate_table", "scalar"]>>>;
+}, "strip", z.ZodTypeAny, {
+    select: string[];
+    dataset: string;
+    metrics: string[];
+    groupBy: string[];
+    filters: {
+        value: string | number | boolean | (string | number)[];
+        field: string;
+        op: "=" | "!=" | ">" | "<" | ">=" | "<=" | "in" | "not_in" | "like" | "ilike";
+    }[];
+    entityFilters: {
+        input: string;
+        resolver: string;
+        outputColumn: string;
+    }[];
+    orderBy: {
+        field: string;
+        dir: "asc" | "desc";
+    }[];
+    answerMode: "rows" | "aggregate_table" | "scalar";
+    dateRange?: {
+        mode: "latest" | "none" | "relative" | "explicit";
+        startDate?: string | undefined;
+        endDate?: string | undefined;
+        phrase?: string | undefined;
+    } | undefined;
+    limit?: number | undefined;
+}, {
+    dataset: string;
+    select?: string[] | undefined;
+    metrics?: string[] | undefined;
+    groupBy?: string[] | undefined;
+    filters?: {
+        value: string | number | boolean | (string | number)[];
+        field: string;
+        op: "=" | "!=" | ">" | "<" | ">=" | "<=" | "in" | "not_in" | "like" | "ilike";
+    }[] | undefined;
+    entityFilters?: {
+        input: string;
+        resolver: string;
+        outputColumn: string;
+    }[] | undefined;
+    dateRange?: {
+        mode: "latest" | "none" | "relative" | "explicit";
+        startDate?: string | undefined;
+        endDate?: string | undefined;
+        phrase?: string | undefined;
+    } | undefined;
+    orderBy?: {
+        field: string;
+        dir?: "asc" | "desc" | undefined;
+    }[] | undefined;
+    limit?: number | undefined;
+    answerMode?: "rows" | "aggregate_table" | "scalar" | undefined;
+}>;
+export type SmartQueryPlanInput = z.input<typeof SmartQueryPlanSchema>;
+export type SmartQueryPlanOutput = z.output<typeof SmartQueryPlanSchema>;
+//# sourceMappingURL=schemas.d.ts.map

package/dist/schemas.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schemas.d.ts","sourceRoot":"","sources":["../src/schemas.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAIxB,eAAO,MAAM,YAAY;;;;;;;;;;;;EASvB,CAAC;AAIH,eAAO,MAAM,kBAAkB;;;;;;;;;;;;EAI7B,CAAC;AAIH,eAAO,MAAM,eAAe;;;;;;;;;;;;;;;EAK1B,CAAC;AAIH,eAAO,MAAM,aAAa;;;;;;;;;EAGxB,CAAC;AAIH,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAW/B,CAAC;AAIH,MAAM,MAAM,mBAAmB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC;AACvE,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,oBAAoB,CAAC,CAAC"}