@memberjunction/db-auto-doc 2.118.0 → 2.120.0

package/README.md

@@ -21,6 +21,7 @@ Automatically generate comprehensive documentation for SQL Server, MySQL, and Po
 
 ### Advanced Features
 - **🔍 Relationship Discovery** - Automatically detect missing primary and foreign keys using statistical analysis and LLM validation
+- **🎯 Sample Query Generation** - Generate reference SQL queries for AI agents with alignment tracking
 - **🛡️ Granular Guardrails** - Multi-level resource controls (run, phase, iteration limits)
 - **⏸️ Resume Capability** - Pause and resume analysis from checkpoint state files
 - **📦 Programmatic API** - Use as a library in your own applications
@@ -84,7 +85,90 @@ This will:
 - Perform sanity checks
 - Save state to `db-doc-state.json`
 
-### 3. Export
+### 3. Generate Sample Queries (Optional)
+
+Generate reference SQL queries for AI agent training:
+
+```bash
+# During analysis (if enabled in config)
+db-auto-doc analyze  # Automatically generates queries
+
+# Or generate separately from existing state
+db-auto-doc generate-queries --from-state ./output/run-1/state.json
+
+# With custom settings
+db-auto-doc generate-queries --from-state ./output/run-1/state.json \
+  --queries-per-table 10 \
+  --max-execution-time 60000 \
+  --output-dir ./queries
+```
+
+This generates:
+- **sample-queries.json**: Full query specifications with SQL, metadata, and alignment info
+- **sample-queries-summary.json**: Execution statistics, token usage, and cost breakdown
+
+**Configuration Options:**
+```json
+{
+  "analysis": {
+    "sampleQueryGeneration": {
+      "enabled": true,              // Enable sample query generation
+      "queriesPerTable": 5,         // Number of queries per table
+      "maxTables": 10,              // Max tables to process (0 = all tables)
+      "tokenBudget": 100000,        // Token limit (0 = unlimited)
+      "maxExecutionTime": 30000,    // Query validation timeout (ms)
+      "includeMultiQueryPatterns": true,  // Generate related query patterns
+      "validateAlignment": true,    // Validate alignment between queries
+      "maxRowsInSample": 10         // Sample result rows to capture
+    }
+  }
+}
+```
+
+**Key Configuration Settings:**
+- **`maxTables`**: Controls table selection
+  - `10` (default) - Generate queries for top 10 most important tables
+  - `0` - Generate queries for **all tables** with data
+  - Custom value - Generate queries for top N tables
+
+- **`tokenBudget`**: Controls LLM token usage and cost
+  - `100000` (default) - Limit to 100K tokens (~$0.50-1.00 with GPT-4o)
+  - `0` - **Unlimited** token budget (useful with `maxTables: 0`)
+  - Custom value - Set specific token limit for cost control
+
+**Example Configurations:**
+
+*Cost-conscious (default):*
+```json
+{
+  "maxTables": 10,
+  "tokenBudget": 100000
+}
+```
+
+*Medium coverage (~25 tables):*
+```json
+{
+  "maxTables": 25,
+  "tokenBudget": 500000
+}
+```
+
+*Complete coverage (all tables):*
+```json
+{
+  "maxTables": 0,
+  "tokenBudget": 0
+}
+```
+
+**Model Recommendations:**
+- ✅ **GPT-4o** - Best balance of speed, cost, and quality (~$6-10 for 50 tables)
+- ✅ **Claude 3.5 Sonnet** - High quality, good reasoning about alignment
+- ⚠️ **GPT-5** - Very slow (reasoning model), doesn't support JSON format, expensive
+- ⚠️ **Groq** - Fast and cheap but may struggle with complex alignment
+
+### 4. Export
 
 ```bash
 db-auto-doc export --sql --markdown --html --csv --mermaid
@@ -103,7 +187,7 @@ Optionally apply directly to database:
 db-auto-doc export --sql --apply
 ```
 
-### 4. Check Status
+### 5. Check Status
 
 ```bash
 db-auto-doc status
@@ -116,7 +200,7 @@ Shows:
 - Token usage, cost, and duration
 - Guardrail status and warnings
 
-### 5. Resume Analysis
+### 6. Resume Analysis
 
 ```bash
 db-auto-doc analyze --resume ./db-doc-state.json
@@ -157,6 +241,41 @@ Triggered automatically when:
 - Tables lack primary key constraints
 - Insufficient foreign key relationships detected (below threshold)
 
+### Sample Query Generation
+
+DBAutoDoc can generate reference SQL queries for AI agents, solving the **query alignment problem** where multi-query patterns (summary + detail) have inconsistent filtering logic:
+
+**The Problem:**
+```sql
+-- Summary query
+SELECT COUNT(*) FROM Registrations  -- All registrations
+
+-- Detail query
+SELECT * FROM Registrations WHERE Status='Attended'  -- Only attended
+
+-- Result: Numbers don't match! Bad UX.
+```
+
+**The Solution:**
+DBAutoDoc generates "gold standard" reference queries with:
+- **Explicit Filtering Rules** - Documents filter logic for consistency
+- **Alignment Tracking** - Links related queries via `relatedQueryIds`
+- **Query Patterns** - Summary+Detail, Multi-Entity Drilldown, Time Series, etc.
+- **Validation** - Executes queries and validates results
+- **Few-Shot Training** - Use as examples for AI agent prompting
+
+**Two-Prompt Architecture:**
+1. **Planning Phase** - AI designs what queries to create (lightweight, ~4K tokens)
+2. **Generation Phase** - AI generates SQL for each query individually (~3K tokens each)
+
+This approach prevents JSON truncation issues while maintaining alignment context between related queries.
+
+**Use Cases:**
+- Training AI agents like Skip to generate consistent multi-query patterns
+- Creating reference examples for few-shot prompting
+- Documenting common query patterns for your database
+- Validating that related queries use consistent filtering logic
+
 ### Backpropagation
 
 After analyzing child tables, DBAutoDoc can detect insights about parent tables and trigger re-analysis:
@@ -253,6 +372,15 @@ This rich context enables AI to make accurate inferences.
       "schemaLevel": true,
       "crossSchema": true
     },
+    "sampleQueryGeneration": {
+      "enabled": true,
+      "queriesPerTable": 5,
+      "maxExecutionTime": 30000,
+      "includeMultiQueryPatterns": true,
+      "validateAlignment": true,
+      "tokenBudget": 100000,
+      "maxRowsInSample": 10
+    },
     "guardrails": {
       "enabled": true,
       "stopOnExceeded": true,
@@ -623,6 +751,16 @@ Typical costs (will vary by database size and complexity):
 
 **With Relationship Discovery**: Add 25-40% to token/cost estimates for databases with missing constraints.
 
+**With Sample Query Generation** (5 queries/table, GPT-4o):
+
+| Database Size | Tables | Additional Tokens | Additional Cost |
+|---------------|--------|-------------------|-----------------|
+| Small | 10-20 | ~100K | $0.50-1.00 |
+| Medium | 50-100 | ~500K | $2.50-5.00 |
+| Large | 200+ | ~2M | $10-20 |
+
+Note: Sample query generation uses ~6× more API calls than description generation (planning + individual SQL generation for each query), adding ~50% to total token usage.
+
 **Guardrails** help control costs by setting hard limits on token usage and runtime.
 
 ## Best Practices
@@ -638,6 +776,19 @@ Typical costs (will vary by database size and complexity):
 9. **Use appropriate models** - Balance cost vs. quality (GPT-4 vs. Groq)
 10. **Export multiple formats** - HTML for browsing, CSV for analysis, SQL for database
 
+### Sample Query Generation Best Practices
+
+1. **Use GPT-4o or Claude 3.5** - Best balance of quality, speed, and cost
+2. **Set token budget** - Prevents runaway costs (default: 100K tokens)
+3. **Start with 5 queries/table** - Good balance of coverage and cost
+4. **Enable alignment validation** - Ensures related queries use consistent logic
+5. **Review generated queries** - Verify SQL correctness before using for training
+6. **Use for few-shot prompting** - Include in AI agent system prompts as examples
+7. **Generate separately** - Use `generate-queries` command on existing state to avoid re-running full analysis
+8. **Focus on complex tables** - Skip simple lookup tables to save costs
+9. **Validate execution** - Enable `maxExecutionTime` to test queries run successfully
+10. **Document patterns** - Use generated queries to document common query patterns for your domain
+
 ## Troubleshooting
 
 ### "Connection failed"

package/dist/commands/generate-queries.d.ts

@@ -0,0 +1,17 @@
+/**
+ * Generate Queries command - Generate sample SQL queries from existing state
+ */
+import { Command } from '@oclif/core';
+export default class GenerateQueries extends Command {
+    static description: string;
+    static examples: string[];
+    static flags: {
+        'from-state': import("@oclif/core/lib/interfaces/parser.js").OptionFlag<string, import("@oclif/core/lib/interfaces/parser.js").CustomOptions>;
+        'output-dir': import("@oclif/core/lib/interfaces/parser.js").OptionFlag<string | undefined, import("@oclif/core/lib/interfaces/parser.js").CustomOptions>;
+        config: import("@oclif/core/lib/interfaces/parser.js").OptionFlag<string, import("@oclif/core/lib/interfaces/parser.js").CustomOptions>;
+        'queries-per-table': import("@oclif/core/lib/interfaces/parser.js").OptionFlag<number | undefined, import("@oclif/core/lib/interfaces/parser.js").CustomOptions>;
+        'max-execution-time': import("@oclif/core/lib/interfaces/parser.js").OptionFlag<number | undefined, import("@oclif/core/lib/interfaces/parser.js").CustomOptions>;
+    };
+    run(): Promise<void>;
+}
+//# sourceMappingURL=generate-queries.d.ts.map

package/dist/commands/generate-queries.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-queries.d.ts","sourceRoot":"","sources":["../../src/commands/generate-queries.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,OAAO,EAAS,MAAM,aAAa,CAAC;AAa7C,MAAM,CAAC,OAAO,OAAO,eAAgB,SAAQ,OAAO;IAClD,MAAM,CAAC,WAAW,SAA8D;IAEhF,MAAM,CAAC,QAAQ,WAIb;IAEF,MAAM,CAAC,KAAK;;;;;;MAsBV;IAEI,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CA4I3B"}

package/dist/commands/generate-queries.js

@@ -0,0 +1,182 @@
+"use strict";
+/**
+ * Generate Queries command - Generate sample SQL queries from existing state
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const core_1 = require("@oclif/core");
+const ora_1 = __importDefault(require("ora"));
+const chalk_1 = __importDefault(require("chalk"));
+const fs = __importStar(require("fs/promises"));
+const path = __importStar(require("path"));
+const config_loader_js_1 = require("../utils/config-loader.js");
+const Database_js_1 = require("../database/Database.js");
+const PromptEngine_js_1 = require("../prompts/PromptEngine.js");
+const SampleQueryGenerator_js_1 = require("../generators/SampleQueryGenerator.js");
+class GenerateQueries extends core_1.Command {
+    async run() {
+        const { flags } = await this.parse(GenerateQueries);
+        const spinner = (0, ora_1.default)();
+        try {
+            // Load configuration for database connection and AI settings
+            spinner.start('Loading configuration');
+            const config = await config_loader_js_1.ConfigLoader.load(flags.config);
+            spinner.succeed('Configuration loaded');
+            // Load existing state
+            spinner.start('Loading existing analysis state');
+            const stateJson = await fs.readFile(flags['from-state'], 'utf-8');
+            const state = JSON.parse(stateJson);
+            spinner.succeed(`State loaded: ${state.schemas.length} schemas, ${state.schemas.reduce((sum, s) => sum + s.tables.length, 0)} tables`);
+            // Determine output directory
+            const outputDir = flags['output-dir'] || path.dirname(flags['from-state']);
+            await fs.mkdir(outputDir, { recursive: true });
+            // Connect to database
+            spinner.start('Connecting to database');
+            const driverConfig = {
+                provider: config.database.provider || 'sqlserver',
+                host: config.database.server,
+                port: config.database.port,
+                database: config.database.database,
+                user: config.database.user,
+                password: config.database.password,
+                encrypt: config.database.encrypt,
+                trustServerCertificate: config.database.trustServerCertificate,
+                connectionTimeout: config.database.connectionTimeout,
+                requestTimeout: config.database.requestTimeout,
+                maxConnections: config.database.maxConnections,
+                minConnections: config.database.minConnections,
+                idleTimeoutMillis: config.database.idleTimeoutMillis
+            };
+            const db = new Database_js_1.DatabaseConnection(driverConfig);
+            await db.connect();
+            const testResult = await db.test();
+            if (!testResult.success) {
+                throw new Error(`Database connection failed: ${testResult.message}`);
+            }
+            spinner.succeed('Connected to database');
+            // Initialize prompt engine
+            spinner.start('Initializing AI prompt engine');
+            const promptsDir = path.join(__dirname, '../../prompts');
+            const promptEngine = new PromptEngine_js_1.PromptEngine(config.ai, promptsDir);
+            await promptEngine.initialize();
+            spinner.succeed('Prompt engine ready');
+            // Build query generation config
+            const queryConfig = {
+                enabled: true,
+                queriesPerTable: flags['queries-per-table'] || config.analysis.sampleQueryGeneration?.queriesPerTable || 5,
+                maxExecutionTime: flags['max-execution-time'] || config.analysis.sampleQueryGeneration?.maxExecutionTime || 30000,
+                includeMultiQueryPatterns: config.analysis.sampleQueryGeneration?.includeMultiQueryPatterns !== false,
+                validateAlignment: config.analysis.sampleQueryGeneration?.validateAlignment !== false,
+                tokenBudget: config.analysis.sampleQueryGeneration?.tokenBudget || 100000,
+                maxRowsInSample: config.analysis.sampleQueryGeneration?.maxRowsInSample || 10,
+                maxTables: config.analysis.sampleQueryGeneration?.maxTables // Optional, defaults to 10 in generator
+            };
+            // Use AI config from main settings
+            const model = config.ai.model;
+            const effortLevel = config.ai.effortLevel || 75;
+            const maxTokens = config.ai.maxTokens || 16000; // Use config value or default
+            // Determine output file paths for incremental writes
+            const queriesPath = path.join(outputDir, 'sample-queries.json');
+            const summaryPath = path.join(outputDir, 'sample-queries-summary.json');
+            // Create generator
+            spinner.start('Generating sample queries');
+            const generator = new SampleQueryGenerator_js_1.SampleQueryGenerator(queryConfig, promptEngine, db.getDriver(), model, effortLevel, maxTokens, queriesPath, // Pass queries output path for incremental writes
+            summaryPath // Pass summary output path for incremental writes
+            );
+            // Generate queries
+            const result = await generator.generateQueries(state.schemas);
+            await db.close();
+            if (result.success) {
+                // Save queries and summary (final write to ensure we have the complete set)
+                // Note: Both were already written incrementally during generation
+                await fs.writeFile(queriesPath, JSON.stringify(result.queries, null, 2), 'utf-8');
+                await fs.writeFile(summaryPath, JSON.stringify(result.summary, null, 2), 'utf-8');
+                spinner.succeed('Sample queries generated!');
+                this.log(chalk_1.default.green('\n✓ Query generation complete!'));
+                this.log(`  Total queries: ${result.summary.totalQueriesGenerated}`);
+                this.log(`  Validated: ${result.summary.queriesValidated}`);
+                this.log(`  Failed validation: ${result.summary.queriesFailed}`);
+                this.log(`  Tokens used: ${result.summary.tokensUsed.toLocaleString()}`);
+                this.log(`  Estimated cost: $${result.summary.estimatedCost.toFixed(2)}`);
+                this.log(`  Average confidence: ${(result.summary.averageConfidence * 100).toFixed(1)}%`);
+                this.log(`  Execution time: ${(result.summary.totalExecutionTime / 1000).toFixed(1)}s`);
+                this.log(`\n  Output files:`);
+                this.log(`    - ${queriesPath}`);
+                this.log(`    - ${summaryPath}`);
+                this.log(chalk_1.default.blue('\n  Query breakdown:'));
+                this.log(`    By type:`);
+                Object.entries(result.summary.queriesByType).forEach(([type, count]) => {
+                    this.log(`      ${type}: ${count}`);
+                });
+                this.log(`    By complexity:`);
+                Object.entries(result.summary.queriesByComplexity).forEach(([complexity, count]) => {
+                    this.log(`      ${complexity}: ${count}`);
+                });
+            }
+            else {
+                spinner.fail('Query generation failed');
+                this.error(result.errorMessage || 'Unknown error');
+            }
+        }
+        catch (error) {
+            spinner.fail('Command failed');
+            this.error(error.message);
+        }
+    }
+}
+GenerateQueries.description = 'Generate sample SQL queries from existing analysis state';
+GenerateQueries.examples = [
+    '$ db-auto-doc generate-queries --from-state ./output/run-1/state.json',
+    '$ db-auto-doc generate-queries --from-state ./output/run-1/state.json --output-dir ./queries',
+    '$ db-auto-doc generate-queries --from-state ./output/run-1/state.json --queries-per-table 10'
+];
+GenerateQueries.flags = {
+    'from-state': core_1.Flags.string({
+        description: 'Path to existing state.json file from previous analysis',
+        required: true
+    }),
+    'output-dir': core_1.Flags.string({
+        description: 'Output directory for generated queries',
+        required: false
+    }),
+    config: core_1.Flags.string({
+        char: 'c',
+        description: 'Path to config file (for database connection and AI settings)',
+        default: './config.json'
+    }),
+    'queries-per-table': core_1.Flags.integer({
+        description: 'Number of queries to generate per table',
+        required: false
+    }),
+    'max-execution-time': core_1.Flags.integer({
+        description: 'Maximum execution time for query validation (ms)',
+        required: false
+    })
+};
+exports.default = GenerateQueries;
+//# sourceMappingURL=generate-queries.js.map

package/dist/commands/generate-queries.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-queries.js","sourceRoot":"","sources":["../../src/commands/generate-queries.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,sCAA6C;AAC7C,8CAAsB;AACtB,kDAA0B;AAC1B,gDAAkC;AAClC,2CAA6B;AAC7B,gEAAyD;AACzD,yDAA6D;AAE7D,gEAA0D;AAC1D,mFAA6E;AAI7E,MAAqB,eAAgB,SAAQ,cAAO;IAiClD,KAAK,CAAC,GAAG;QACP,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;QACpD,MAAM,OAAO,GAAG,IAAA,aAAG,GAAE,CAAC;QAEtB,IAAI,CAAC;YACH,6DAA6D;YAC7D,OAAO,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;YACvC,MAAM,MAAM,GAAG,MAAM,+BAAY,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACrD,OAAO,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;YAExC,sBAAsB;YACtB,OAAO,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC;YACjD,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,OAAO,CAAC,CAAC;YAClE,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAA0B,CAAC;YAC7D,OAAO,CAAC,OAAO,CAAC,iBAAiB,KAAK,CAAC,OAAO,CAAC,MAAM,aAAa,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC;YAEvI,6BAA6B;YAC7B,MAAM,SAAS,GAAG,KAAK,CAAC,YAAY,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;YAC3E,MAAM,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAE/C,sBAAsB;YACtB,OAAO,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;YACxC,MAAM,YAAY,GAA4B;gBAC5C,QAAQ,EAAG,MAAM,CAAC,QAAQ,CAAC,QAA4D,IAAI,WAAW;gBACtG,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,MAAM;gBAC5B,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;gBAC1B,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,QAAQ;gBAClC,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;gBAC1B,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,QAAQ;gBAClC,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,OAAO;gBAChC,sBAAsB,EAAE,MAAM,CAAC,QAAQ,CAAC,sBAAsB;gBAC9D,iBAAiB,EAAE,MAAM,CAAC,QAAQ,CAAC,iBAAiB;gBACpD,cAAc,EAAE,MAAM,CAAC,QAAQ,CAAC,cAAc;gBAC9C,cAAc,EAAE,MAAM,CAAC,QAAQ,CAAC,cAAc;gBAC9C,cAAc,EAAE,MAAM,CAAC,QAAQ,CAAC,cAAc;gBAC9C,iBAAiB,EAAE,MAAM,CAAC,QAAQ,CAAC,iBAAiB;aACrD,CAAC;YAEF,MAAM,EAAE,GAAG,IAAI,gCAAkB,CAAC,YAAY,CAAC,CAAC;YAChD,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;YACnB,MAAM,UAAU,GAAG,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;YAEnC,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,+BAA+B,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;YACvE,CAAC;YACD,OAAO,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;YAEzC,2BAA2B;YAC3B,OAAO,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC;YAC/C,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;YACzD,MAAM,YAAY,GAAG,IAAI,8BAAY,CAAC,MAAM,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YAC7D,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;YAChC,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;YAEvC,gCAAgC;YAChC,MAAM,WAAW,GAAgC;gBAC/C,OAAO,EAAE,IAAI;gBACb,eAAe,EAAE,KAAK,CAAC,mBAAmB,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,qBAAqB,EAAE,eAAe,IAAI,CAAC;gBAC1G,gBAAgB,EAAE,KAAK,CAAC,oBAAoB,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,qBAAqB,EAAE,gBAAgB,IAAI,KAAK;gBACjH,yBAAyB,EAAE,MAAM,CAAC,QAAQ,CAAC,qBAAqB,EAAE,yBAAyB,KAAK,KAAK;gBACrG,iBAAiB,EAAE,MAAM,CAAC,QAAQ,CAAC,qBAAqB,EAAE,iBAAiB,KAAK,KAAK;gBACrF,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,qBAAqB,EAAE,WAAW,IAAI,MAAM;gBACzE,eAAe,EAAE,MAAM,CAAC,QAAQ,CAAC,qBAAqB,EAAE,eAAe,IAAI,EAAE;gBAC7E,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,qBAAqB,EAAE,SAAS,CAAE,wCAAwC;aACtG,CAAC;YAEF,mCAAmC;YACnC,MAAM,KAAK,GAAG,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC;YAC9B,MAAM,WAAW,GAAG,MAAM,CAAC,EAAE,CAAC,WAAW,IAAI,EAAE,CAAC;YAChD,MAAM,SAAS,GAAG,MAAM,CAAC,EAAE,CAAC,SAAS,IAAI,KAAK,CAAC,CAAE,8BAA8B;YAE/E,qDAAqD;YACrD,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,qBAAqB,CAAC,CAAC;YAChE,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,6BAA6B,CAAC,CAAC;YAExE,mBAAmB;YACnB,OAAO,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAC3C,MAAM,SAAS,GAAG,IAAI,8CAAoB,CACxC,WAAW,EACX,YAAY,EACZ,EAAE,CAAC,SAAS,EAAE,EACd,KAAK,EACL,WAAW,EACX,SAAS,EACT,WAAW,EAAG,kDAAkD;YAChE,WAAW,CAAG,kDAAkD;aACjE,CAAC;YAEF,mBAAmB;YACnB,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,eAAe,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAE9D,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC;YAEjB,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,4EAA4E;gBAC5E,kEAAkE;gBAClE,MAAM,EAAE,CAAC,SAAS,CAChB,WAAW,EACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,EACvC,OAAO,CACR,CAAC;gBAEF,MAAM,EAAE,CAAC,SAAS,CAChB,WAAW,EACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,EACvC,OAAO,CACR,CAAC;gBAEF,OAAO,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;gBAC7C,IAAI,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,gCAAgC,CAAC,CAAC,CAAC;gBACxD,IAAI,CAAC,GAAG,CAAC,oBAAoB,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,CAAC,CAAC;gBACrE,IAAI,CAAC,GAAG,CAAC,gBAAgB,MAAM,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAAC;gBAC5D,IAAI,CAAC,GAAG,CAAC,wBAAwB,MAAM,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;gBACjE,IAAI,CAAC,GAAG,CAAC,kBAAkB,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;gBACzE,IAAI,CAAC,GAAG,CAAC,sBAAsB,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBAC1E,IAAI,CAAC,GAAG,CAAC,yBAAyB,CAAC,MAAM,CAAC,OAAO,CAAC,iBAAiB,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBAC1F,IAAI,CAAC,GAAG,CAAC,qBAAqB,CAAC,MAAM,CAAC,OAAO,CAAC,kBAAkB,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBACxF,IAAI,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;gBAC9B,IAAI,CAAC,GAAG,CAAC,SAAS,WAAW,EAAE,CAAC,CAAC;gBACjC,IAAI,CAAC,GAAG,CAAC,SAAS,WAAW,EAAE,CAAC,CAAC;gBAEjC,IAAI,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC,CAAC;gBAC7C,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;gBACzB,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,EAAE;oBACrE,IAAI,CAAC,GAAG,CAAC,SAAS,IAAI,KAAK,KAAK,EAAE,CAAC,CAAC;gBACtC,CAAC,CAAC,CAAC;gBACH,IAAI,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;gBAC/B,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,EAAE,KAAK,CAAC,EAAE,EAAE;oBACjF,IAAI,CAAC,GAAG,CAAC,SAAS,UAAU,KAAK,KAAK,EAAE,CAAC,CAAC;gBAC5C,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;gBACxC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,IAAI,eAAe,CAAC,CAAC;YACrD,CAAC;QAEH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAC/B,IAAI,CAAC,KAAK,CAAE,KAAe,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;;AA3KM,2BAAW,GAAG,0DAA0D,CAAC;AAEzE,wBAAQ,GAAG;IAChB,uEAAuE;IACvE,8FAA8F;IAC9F,8FAA8F;CAC/F,CAAC;AAEK,qBAAK,GAAG;IACb,YAAY,EAAE,YAAK,CAAC,MAAM,CAAC;QACzB,WAAW,EAAE,yDAAyD;QACtE,QAAQ,EAAE,IAAI;KACf,CAAC;IACF,YAAY,EAAE,YAAK,CAAC,MAAM,CAAC;QACzB,WAAW,EAAE,wCAAwC;QACrD,QAAQ,EAAE,KAAK;KAChB,CAAC;IACF,MAAM,EAAE,YAAK,CAAC,MAAM,CAAC;QACnB,IAAI,EAAE,GAAG;QACT,WAAW,EAAE,+DAA+D;QAC5E,OAAO,EAAE,eAAe;KACzB,CAAC;IACF,mBAAmB,EAAE,YAAK,CAAC,OAAO,CAAC;QACjC,WAAW,EAAE,yCAAyC;QACtD,QAAQ,EAAE,KAAK;KAChB,CAAC;IACF,oBAAoB,EAAE,YAAK,CAAC,OAAO,CAAC;QAClC,WAAW,EAAE,kDAAkD;QAC/D,QAAQ,EAAE,KAAK;KAChB,CAAC;CACH,CAAC;kBA/BiB,eAAe"}

package/dist/commands/init.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"init.d.ts","sourceRoot":"","sources":["../../src/commands/init.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAMtC,MAAM,CAAC,OAAO,OAAO,IAAK,SAAQ,OAAO;IACvC,MAAM,CAAC,WAAW,SAAwC;IAE1D,MAAM,CAAC,QAAQ,WAA0B;IAEnC,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CAkK3B"}
+{"version":3,"file":"init.d.ts","sourceRoot":"","sources":["../../src/commands/init.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAMtC,MAAM,CAAC,OAAO,OAAO,IAAK,SAAQ,OAAO;IACvC,MAAM,CAAC,WAAW,SAAwC;IAE1D,MAAM,CAAC,QAAQ,WAA0B;IAEnC,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CAyN3B"}

package/dist/commands/init.js

@@ -129,6 +129,46 @@ class Init extends core_1.Command {
                 when: (answers) => answers.addContext
             }
         ]);
+        // Sample query generation options
+        const queryAnswers = await inquirer_1.default.prompt([
+            {
+                type: 'confirm',
+                name: 'enableSampleQueries',
+                message: 'Generate sample queries for each table?',
+                default: false
+            },
+            {
+                type: 'number',
+                name: 'queriesPerTable',
+                message: 'Number of queries to generate per table:',
+                default: 5,
+                when: (answers) => answers.enableSampleQueries,
+                validate: (input) => input > 0 || 'Must be greater than 0'
+            },
+            {
+                type: 'number',
+                name: 'maxTables',
+                message: 'Max number of tables to generate queries for (0 = all tables):',
+                default: 10,
+                when: (answers) => answers.enableSampleQueries,
+                validate: (input) => input >= 0 || 'Must be 0 or greater'
+            },
+            {
+                type: 'number',
+                name: 'tokenBudget',
+                message: 'Token budget for query generation (0 = unlimited):',
+                default: 100000,
+                when: (answers) => answers.enableSampleQueries,
+                validate: (input) => input >= 0 || 'Must be 0 or greater'
+            },
+            {
+                type: 'number',
+                name: 'maxExecutionTime',
+                message: 'Max execution time for query validation (ms):',
+                default: 30000,
+                when: (answers) => answers.enableSampleQueries
+            }
+        ]);
         // Create configuration
         const config = config_loader_js_1.ConfigLoader.createDefault();
         // Update with user inputs
@@ -158,6 +198,19 @@ class Init extends core_1.Command {
                 industryContext: contextAnswers.industryContext || undefined
             };
         }
+        // Add sample query generation config if enabled
+        if (queryAnswers.enableSampleQueries) {
+            config.analysis.sampleQueryGeneration = {
+                enabled: true,
+                queriesPerTable: queryAnswers.queriesPerTable || 5,
+                maxExecutionTime: queryAnswers.maxExecutionTime || 30000,
+                includeMultiQueryPatterns: true,
+                validateAlignment: true,
+                tokenBudget: queryAnswers.tokenBudget !== undefined ? queryAnswers.tokenBudget : 100000,
+                maxRowsInSample: 10,
+                maxTables: queryAnswers.maxTables !== undefined ? queryAnswers.maxTables : 10
+            };
+        }
         // Save configuration
         await config_loader_js_1.ConfigLoader.save(config, './config.json');
         this.log(chalk_1.default.green('\n✓ Configuration saved to config.json'));

package/dist/commands/init.js.map

@@ -1 +1 @@
-{"version":3,"file":"init.js","sourceRoot":"","sources":["../../src/commands/init.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;AAEH,sCAAsC;AACtC,wDAAgC;AAChC,kDAA0B;AAC1B,gEAAyD;AACzD,yDAA6D;AAE7D,MAAqB,IAAK,SAAQ,cAAO;IAKvC,KAAK,CAAC,GAAG;QACP,IAAI,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC,CAAC;QAEnD,yBAAyB;QACzB,MAAM,SAAS,GAAG,MAAM,kBAAQ,CAAC,MAAM,CAAC;YACtC;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,kBAAkB;gBAC3B,OAAO,EAAE,WAAW;aACrB;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,gBAAgB;gBACzB,QAAQ,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,2BAA2B;aAC7E;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,WAAW;gBACpB,OAAO,EAAE,IAAI;aACd;YACD;gBACE,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,WAAW;gBACpB,IAAI,EAAE,GAAG;aACV;YACD;gBACE,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,iBAAiB;gBAC1B,OAAO,EAAE,IAAI;aACd;YACD;gBACE,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,wBAAwB;gBAC9B,OAAO,EAAE,2BAA2B;gBACpC,OAAO,EAAE,KAAK;aACf;SACF,CAAC,CAAC;QAEH,kBAAkB;QAClB,IAAI,CAAC,GAAG,CAAC,eAAK,CAAC,MAAM,CAAC,kCAAkC,CAAC,CAAC,CAAC;QAC3D,MAAM,QAAQ,GAAG;YACf,QAAQ,EAAE,WAAoB;YAC9B,IAAI,EAAE,SAAS,CAAC,MAAM;YACtB,QAAQ,EAAE,SAAS,CAAC,QAAQ;YAC5B,IAAI,EAAE,SAAS,CAAC,IAAI;YACpB,QAAQ,EAAE,SAAS,CAAC,QAAQ;YAC5B,OAAO,EAAE,SAAS,CAAC,OAAO;YAC1B,sBAAsB,EAAE,SAAS,CAAC,sBAAsB;SACzD,CAAC;QACF,MAAM,EAAE,GAAG,IAAI,gCAAkB,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,UAAU,GAAG,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;QACnC,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC;QAEjB,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACxB,IAAI,CAAC,KAAK,CAAC,sBAAsB,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;YACvD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,qCAAqC,CAAC,CAAC,CAAC;QAE7D,mBAAmB;QACnB,MAAM,SAAS,GAAG,MAAM,kBAAQ,CAAC,MAAM,CAAC;YACtC;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,cAAc;gBACvB,OAAO,EAAE,CAAC,QAAQ,EAAE,WAAW,EAAE,MAAM,CAAC;aACzC;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,aAAa;gBACtB,OAAO,EAAE,CAAC,OAAY,EAAE,EAAE;oBACxB,IAAI,OAAO,CAAC,QAAQ,KAAK,QAAQ;wBAAE,OAAO,qBAAqB,CAAC;oBAChE,IAAI,OAAO,CAAC,QAAQ,KAAK,WAAW;wBAAE,OAAO,wBAAwB,CAAC;oBACtE,IAAI,OAAO,CAAC,QAAQ,KAAK,MAAM;wBAAE,OAAO,oBAAoB,CAAC;oBAC7D,OAAO,EAAE,CAAC;gBACZ,CAAC;aACF;YACD;gBACE,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,UAAU;gBACnB,IAAI,EAAE,GAAG;gBACT,QAAQ,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,qBAAqB;aACvE;SACF,CAAC,CAAC;QAEH,wBAAwB;QACxB,MAAM,cAAc,GAAG,MAAM,kBAAQ,CAAC,MAAM,CAAC;YAC3C;gBACE,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,YAAY;gBAClB,OAAO,EAAE,uCAAuC;gBAChD,OAAO,EAAE,IAAI;aACd;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,gBAAgB;gBACtB,OAAO,EAAE,yDAAyD;gBAClE,IAAI,EAAE,CAAC,OAAY,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU;aAC3C;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,iBAAiB;gBACvB,OAAO,EAAE,wEAAwE;gBACjF,IAAI,EAAE,CAAC,OAAY,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU;aAC3C;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,iBAAiB;gBACvB,OAAO,EAAE,mDAAmD;gBAC5D,IAAI,EAAE,CAAC,OAAY,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU;aAC3C;SACF,CAAC,CAAC;QAEH,uBAAuB;QACvB,MAAM,MAAM,GAAG,+BAAY,CAAC,aAAa,EAAE,CAAC;QAE5C,0BAA0B;QAC1B,MAAM,CAAC,QAAQ,GAAG;YAChB,MAAM,EAAE,SAAS,CAAC,MAAM;YACxB,IAAI,EAAE,IAAI;YACV,QAAQ,EAAE,SAAS,CAAC,QAAQ;YAC5B,IAAI,EAAE,SAAS,CAAC,IAAI;YACpB,QAAQ,EAAE,SAAS,CAAC,QAAQ;YAC5B,OAAO,EAAE,SAAS,CAAC,OAAO;YAC1B,sBAAsB,EAAE,SAAS,CAAC,sBAAsB;YACxD,iBAAiB,EAAE,KAAK;SACzB,CAAC;QAEF,MAAM,CAAC,EAAE,GAAG;YACV,QAAQ,EAAE,SAAS,CAAC,QAAe;YACnC,KAAK,EAAE,SAAS,CAAC,KAAK;YACtB,MAAM,EAAE,SAAS,CAAC,MAAM;YACxB,WAAW,EAAE,GAAG;YAChB,SAAS,EAAE,IAAI;SAChB,CAAC;QAEF,IAAI,cAAc,CAAC,UAAU,EAAE,CAAC;YAC7B,MAA6C,CAAC,aAAa,CAAC,GAAG;gBAC9D,cAAc,EAAE,cAAc,CAAC,cAAc,IAAI,SAAS;gBAC1D,eAAe,EAAE,cAAc,CAAC,eAAe;oBAC7C,CAAC,CAAC,cAAc,CAAC,eAAe,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBACxE,CAAC,CAAC,SAAS;gBACb,eAAe,EAAE,cAAc,CAAC,eAAe,IAAI,SAAS;aAC7D,CAAC;QACJ,CAAC;QAED,qBAAqB;QACrB,MAAM,+BAAY,CAAC,IAAI,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;QAEjD,IAAI,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC,CAAC;QAChE,IAAI,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC;QACtC,IAAI,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;QAC1C,IAAI,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;IAC5D,CAAC;;AArKM,gBAAW,GAAG,oCAAoC,CAAC;AAEnD,aAAQ,GAAG,CAAC,oBAAoB,CAAC,CAAC;kBAHtB,IAAI"}
+{"version":3,"file":"init.js","sourceRoot":"","sources":["../../src/commands/init.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;AAEH,sCAAsC;AACtC,wDAAgC;AAChC,kDAA0B;AAC1B,gEAAyD;AACzD,yDAA6D;AAE7D,MAAqB,IAAK,SAAQ,cAAO;IAKvC,KAAK,CAAC,GAAG;QACP,IAAI,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC,CAAC;QAEnD,yBAAyB;QACzB,MAAM,SAAS,GAAG,MAAM,kBAAQ,CAAC,MAAM,CAAC;YACtC;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,kBAAkB;gBAC3B,OAAO,EAAE,WAAW;aACrB;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,gBAAgB;gBACzB,QAAQ,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,2BAA2B;aAC7E;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,WAAW;gBACpB,OAAO,EAAE,IAAI;aACd;YACD;gBACE,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,WAAW;gBACpB,IAAI,EAAE,GAAG;aACV;YACD;gBACE,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,iBAAiB;gBAC1B,OAAO,EAAE,IAAI;aACd;YACD;gBACE,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,wBAAwB;gBAC9B,OAAO,EAAE,2BAA2B;gBACpC,OAAO,EAAE,KAAK;aACf;SACF,CAAC,CAAC;QAEH,kBAAkB;QAClB,IAAI,CAAC,GAAG,CAAC,eAAK,CAAC,MAAM,CAAC,kCAAkC,CAAC,CAAC,CAAC;QAC3D,MAAM,QAAQ,GAAG;YACf,QAAQ,EAAE,WAAoB;YAC9B,IAAI,EAAE,SAAS,CAAC,MAAM;YACtB,QAAQ,EAAE,SAAS,CAAC,QAAQ;YAC5B,IAAI,EAAE,SAAS,CAAC,IAAI;YACpB,QAAQ,EAAE,SAAS,CAAC,QAAQ;YAC5B,OAAO,EAAE,SAAS,CAAC,OAAO;YAC1B,sBAAsB,EAAE,SAAS,CAAC,sBAAsB;SACzD,CAAC;QACF,MAAM,EAAE,GAAG,IAAI,gCAAkB,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,UAAU,GAAG,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;QACnC,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC;QAEjB,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACxB,IAAI,CAAC,KAAK,CAAC,sBAAsB,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;YACvD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,qCAAqC,CAAC,CAAC,CAAC;QAE7D,mBAAmB;QACnB,MAAM,SAAS,GAAG,MAAM,kBAAQ,CAAC,MAAM,CAAC;YACtC;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,cAAc;gBACvB,OAAO,EAAE,CAAC,QAAQ,EAAE,WAAW,EAAE,MAAM,CAAC;aACzC;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,aAAa;gBACtB,OAAO,EAAE,CAAC,OAAY,EAAE,EAAE;oBACxB,IAAI,OAAO,CAAC,QAAQ,KAAK,QAAQ;wBAAE,OAAO,qBAAqB,CAAC;oBAChE,IAAI,OAAO,CAAC,QAAQ,KAAK,WAAW;wBAAE,OAAO,wBAAwB,CAAC;oBACtE,IAAI,OAAO,CAAC,QAAQ,KAAK,MAAM;wBAAE,OAAO,oBAAoB,CAAC;oBAC7D,OAAO,EAAE,CAAC;gBACZ,CAAC;aACF;YACD;gBACE,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,UAAU;gBACnB,IAAI,EAAE,GAAG;gBACT,QAAQ,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,qBAAqB;aACvE;SACF,CAAC,CAAC;QAEH,wBAAwB;QACxB,MAAM,cAAc,GAAG,MAAM,kBAAQ,CAAC,MAAM,CAAC;YAC3C;gBACE,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,YAAY;gBAClB,OAAO,EAAE,uCAAuC;gBAChD,OAAO,EAAE,IAAI;aACd;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,gBAAgB;gBACtB,OAAO,EAAE,yDAAyD;gBAClE,IAAI,EAAE,CAAC,OAAY,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU;aAC3C;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,iBAAiB;gBACvB,OAAO,EAAE,wEAAwE;gBACjF,IAAI,EAAE,CAAC,OAAY,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU;aAC3C;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,iBAAiB;gBACvB,OAAO,EAAE,mDAAmD;gBAC5D,IAAI,EAAE,CAAC,OAAY,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU;aAC3C;SACF,CAAC,CAAC;QAEH,kCAAkC;QAClC,MAAM,YAAY,GAAG,MAAM,kBAAQ,CAAC,MAAM,CAAC;YACzC;gBACE,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,qBAAqB;gBAC3B,OAAO,EAAE,yCAAyC;gBAClD,OAAO,EAAE,KAAK;aACf;YACD;gBACE,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,iBAAiB;gBACvB,OAAO,EAAE,0CAA0C;gBACnD,OAAO,EAAE,CAAC;gBACV,IAAI,EAAE,CAAC,OAAY,EAAE,EAAE,CAAC,OAAO,CAAC,mBAAmB;gBACnD,QAAQ,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,IAAI,wBAAwB;aACnE;YACD;gBACE,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,gEAAgE;gBACzE,OAAO,EAAE,EAAE;gBACX,IAAI,EAAE,CAAC,OAAY,EAAE,EAAE,CAAC,OAAO,CAAC,mBAAmB;gBACnD,QAAQ,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,KAAK,IAAI,CAAC,IAAI,sBAAsB;aAClE;YACD;gBACE,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,aAAa;gBACnB,OAAO,EAAE,oDAAoD;gBAC7D,OAAO,EAAE,MAAM;gBACf,IAAI,EAAE,CAAC,OAAY,EAAE,EAAE,CAAC,OAAO,CAAC,mBAAmB;gBACnD,QAAQ,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,KAAK,IAAI,CAAC,IAAI,sBAAsB;aAClE;YACD;gBACE,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,+CAA+C;gBACxD,OAAO,EAAE,KAAK;gBACd,IAAI,EAAE,CAAC,OAAY,EAAE,EAAE,CAAC,OAAO,CAAC,mBAAmB;aACpD;SACF,CAAC,CAAC;QAEH,uBAAuB;QACvB,MAAM,MAAM,GAAG,+BAAY,CAAC,aAAa,EAAE,CAAC;QAE5C,0BAA0B;QAC1B,MAAM,CAAC,QAAQ,GAAG;YAChB,MAAM,EAAE,SAAS,CAAC,MAAM;YACxB,IAAI,EAAE,IAAI;YACV,QAAQ,EAAE,SAAS,CAAC,QAAQ;YAC5B,IAAI,EAAE,SAAS,CAAC,IAAI;YACpB,QAAQ,EAAE,SAAS,CAAC,QAAQ;YAC5B,OAAO,EAAE,SAAS,CAAC,OAAO;YAC1B,sBAAsB,EAAE,SAAS,CAAC,sBAAsB;YACxD,iBAAiB,EAAE,KAAK;SACzB,CAAC;QAEF,MAAM,CAAC,EAAE,GAAG;YACV,QAAQ,EAAE,SAAS,CAAC,QAAe;YACnC,KAAK,EAAE,SAAS,CAAC,KAAK;YACtB,MAAM,EAAE,SAAS,CAAC,MAAM;YACxB,WAAW,EAAE,GAAG;YAChB,SAAS,EAAE,IAAI;SAChB,CAAC;QAEF,IAAI,cAAc,CAAC,UAAU,EAAE,CAAC;YAC7B,MAA6C,CAAC,aAAa,CAAC,GAAG;gBAC9D,cAAc,EAAE,cAAc,CAAC,cAAc,IAAI,SAAS;gBAC1D,eAAe,EAAE,cAAc,CAAC,eAAe;oBAC7C,CAAC,CAAC,cAAc,CAAC,eAAe,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBACxE,CAAC,CAAC,SAAS;gBACb,eAAe,EAAE,cAAc,CAAC,eAAe,IAAI,SAAS;aAC7D,CAAC;QACJ,CAAC;QAED,gDAAgD;QAChD,IAAI,YAAY,CAAC,mBAAmB,EAAE,CAAC;YACrC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,GAAG;gBACtC,OAAO,EAAE,IAAI;gBACb,eAAe,EAAE,YAAY,CAAC,eAAe,IAAI,CAAC;gBAClD,gBAAgB,EAAE,YAAY,CAAC,gBAAgB,IAAI,KAAK;gBACxD,yBAAyB,EAAE,IAAI;gBAC/B,iBAAiB,EAAE,IAAI;gBACvB,WAAW,EAAE,YAAY,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM;gBACvF,eAAe,EAAE,EAAE;gBACnB,SAAS,EAAE,YAAY,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE;aAC9E,CAAC;QACJ,CAAC;QAED,qBAAqB;QACrB,MAAM,+BAAY,CAAC,IAAI,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;QAEjD,IAAI,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC,CAAC;QAChE,IAAI,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC;QACtC,IAAI,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;QAC1C,IAAI,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;IAC5D,CAAC;;AA5NM,gBAAW,GAAG,oCAAoC,CAAC;AAEnD,aAAQ,GAAG,CAAC,oBAAoB,CAAC,CAAC;kBAHtB,IAAI"}

package/dist/core/AnalysisOrchestrator.d.ts

@@ -33,5 +33,9 @@ export declare class AnalysisOrchestrator {
      * Create run-numbered folder for this analysis run
      */
     private createRunFolder;
+    /**
+     * Generate sample queries for AI agents (like Skip)
+     */
+    private generateSampleQueries;
 }
 //# sourceMappingURL=AnalysisOrchestrator.d.ts.map

package/dist/core/AnalysisOrchestrator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AnalysisOrchestrator.d.ts","sourceRoot":"","sources":["../../src/core/AnalysisOrchestrator.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAaH,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,qBAAqB,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAIvE,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,eAAe,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,GAAG,KAAK,IAAI,CAAC;CACpD;AAED,MAAM,WAAW,kBAAkB;IACjC,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,EAAE,qBAAqB,CAAC;IAC7B,GAAG,EAAE,WAAW,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,eAAe,CAAC,CAAS;IACjC,OAAO,CAAC,UAAU,CAAwC;gBAE9C,OAAO,EAAE,eAAe;IAMpC;;OAEG;IACU,OAAO,IAAI,OAAO,CAAC,kBAAkB,CAAC;IAqQnD;;OAEG;YACW,iBAAiB;IAa/B;;OAEG;YACW,eAAe;CAoB9B"}
+{"version":3,"file":"AnalysisOrchestrator.d.ts","sourceRoot":"","sources":["../../src/core/AnalysisOrchestrator.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAcH,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,qBAAqB,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAIvE,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,eAAe,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,GAAG,KAAK,IAAI,CAAC;CACpD;AAED,MAAM,WAAW,kBAAkB;IACjC,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,EAAE,qBAAqB,CAAC;IAC7B,GAAG,EAAE,WAAW,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,eAAe,CAAC,CAAS;IACjC,OAAO,CAAC,UAAU,CAAwC;gBAE9C,OAAO,EAAE,eAAe;IAMpC;;OAEG;IACU,OAAO,IAAI,OAAO,CAAC,kBAAkB,CAAC;IA6QnD;;OAEG;YACW,iBAAiB;IAa/B;;OAEG;YACW,eAAe;IAqB7B;;OAEG;YACW,qBAAqB;CA4FpC"}

package/dist/core/AnalysisOrchestrator.js

@@ -38,6 +38,7 @@ const PromptEngine_js_1 = require("../prompts/PromptEngine.js");
 const AnalysisEngine_js_1 = require("./AnalysisEngine.js");
 const SQLGenerator_js_1 = require("../generators/SQLGenerator.js");
 const MarkdownGenerator_js_1 = require("../generators/MarkdownGenerator.js");
+const SampleQueryGenerator_js_1 = require("../generators/SampleQueryGenerator.js");
 const DiscoveryTriggerAnalyzer_js_1 = require("../discovery/DiscoveryTriggerAnalyzer.js");
 const DiscoveryEngine_js_1 = require("../discovery/DiscoveryEngine.js");
 class AnalysisOrchestrator {
@@ -223,6 +224,13 @@ class AnalysisOrchestrator {
             if (!converged) {
                 iterationTracker.completeRun(run, false, 'Max iterations reached');
             }
+            // Sample Query Generation (if enabled)
+            if (this.config.analysis.sampleQueryGeneration?.enabled) {
+                this.onProgress('Generating sample queries');
+                await this.generateSampleQueries(state, promptEngine, db.getDriver(), runFolder);
+                stateManager.updateSummary(state);
+                await stateManager.save(state);
+            }
             // Final state update
             stateManager.updateSummary(state);
             await stateManager.save(state);
@@ -289,6 +297,81 @@ class AnalysisOrchestrator {
         await fs.mkdir(runFolder, { recursive: true });
         return runFolder;
     }
+    /**
+     * Generate sample queries for AI agents (like Skip)
+     */
+    async generateSampleQueries(state, promptEngine, driver, runFolder) {
+        const config = this.config.analysis.sampleQueryGeneration;
+        if (!config)
+            return;
+        // Use main AI config model, effortLevel, and maxTokens
+        const model = this.config.ai.model;
+        const effortLevel = this.config.ai.effortLevel || 75;
+        const maxTokens = this.config.ai.maxTokens || 16000;
+        // Determine output file paths for incremental writes
+        const queriesPath = path.join(runFolder, 'sample-queries.json');
+        const summaryPath = path.join(runFolder, 'sample-queries-summary.json');
+        const generator = new SampleQueryGenerator_js_1.SampleQueryGenerator(config, promptEngine, driver, model, effortLevel, maxTokens, queriesPath, summaryPath);
+        try {
+            const result = await generator.generateQueries(state.schemas);
+            if (result.success) {
+                state.phases.sampleQueryGeneration = {
+                    startedAt: new Date().toISOString(),
+                    completedAt: new Date().toISOString(),
+                    status: 'completed',
+                    queries: result.queries,
+                    summary: result.summary
+                };
+                // Save queries and summary (final write to ensure we have the complete set)
+                // Note: Both were already written incrementally during generation
+                await fs.writeFile(queriesPath, JSON.stringify(result.queries, null, 2), 'utf-8');
+                await fs.writeFile(summaryPath, JSON.stringify(result.summary, null, 2), 'utf-8');
+                this.onProgress('Sample queries generated', {
+                    total: result.summary.totalQueriesGenerated,
+                    validated: result.summary.queriesValidated,
+                    tokens: result.summary.tokensUsed,
+                    cost: result.summary.estimatedCost
+                });
+            }
+            else {
+                state.phases.sampleQueryGeneration = {
+                    startedAt: new Date().toISOString(),
+                    completedAt: new Date().toISOString(),
+                    status: 'failed',
+                    queries: [],
+                    summary: result.summary,
+                    errorMessage: result.errorMessage
+                };
+                this.onProgress('Sample query generation failed', {
+                    error: result.errorMessage
+                });
+            }
+        }
+        catch (error) {
+            state.phases.sampleQueryGeneration = {
+                startedAt: new Date().toISOString(),
+                completedAt: new Date().toISOString(),
+                status: 'failed',
+                queries: [],
+                summary: {
+                    totalQueriesGenerated: 0,
+                    queriesValidated: 0,
+                    queriesFailed: 0,
+                    totalExecutionTime: 0,
+                    tokensUsed: 0,
+                    estimatedCost: 0,
+                    averageConfidence: 0,
+                    queriesByType: {},
+                    queriesByPattern: {},
+                    queriesByComplexity: {}
+                },
+                errorMessage: error.message
+            };
+            this.onProgress('Sample query generation error', {
+                error: error.message
+            });
+        }
+    }
 }
 exports.AnalysisOrchestrator = AnalysisOrchestrator;
 //# sourceMappingURL=AnalysisOrchestrator.js.map