@memberjunction/query-gen 0.0.1 → 2.126.1

package/dist/core/QueryRefiner.js

@@ -0,0 +1,267 @@
+"use strict";
+/**
+ * QueryRefiner - Iteratively improves queries based on evaluation feedback
+ *
+ * Uses evaluation and refinement AI prompts to assess if queries answer
+ * the business question correctly and improve them through iterations.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QueryRefiner = void 0;
+const aiengine_1 = require("@memberjunction/aiengine");
+const core_1 = require("@memberjunction/core");
+const error_handlers_1 = require("../utils/error-handlers");
+const PromptNames_1 = require("../prompts/PromptNames");
+const prompt_helpers_1 = require("../utils/prompt-helpers");
+/**
+ * QueryRefiner class
+ * Iteratively refines queries based on evaluation feedback
+ */
+class QueryRefiner {
+    tester;
+    contextUser;
+    config;
+    entityMetadata = [];
+    constructor(tester, contextUser, config) {
+        this.tester = tester;
+        this.contextUser = contextUser;
+        this.config = config;
+    }
+    /**
+     * Refine a query through evaluation and improvement iterations
+     *
+     * Loops up to maxRefinements times, testing and evaluating each iteration.
+     * Returns when query passes evaluation or max refinements reached.
+     *
+     * @param query - Initial generated query
+     * @param businessQuestion - Original business question
+     * @param entityMetadata - Entity metadata for refinement
+     * @param maxRefinements - Maximum refinement iterations (default: 3)
+     * @returns Refined query with test results and evaluation
+     */
+    async refineQuery(query, businessQuestion, entityMetadata, maxRefinements = 3) {
+        // Store entity metadata for use in evaluation
+        this.entityMetadata = entityMetadata;
+        let currentQuery = query;
+        let refinementCount = 0;
+        // Track the last successfully tested query and its results
+        let lastWorkingQuery = query;
+        let lastWorkingTestResult = null;
+        let lastWorkingEvaluation = null;
+        // Ensure AIEngine is configured
+        await this.configureAIEngine();
+        while (refinementCount < maxRefinements) {
+            // 1. Test the current query
+            let testResult;
+            try {
+                testResult = await this.testCurrentQuery(currentQuery);
+                // Success! Save this as our last working version
+                lastWorkingQuery = currentQuery;
+                lastWorkingTestResult = testResult;
+            }
+            catch (error) {
+                // Query broke during refinement - revert to last working version
+                if (this.config.verbose) {
+                    (0, core_1.LogStatus)(`Refinement produced broken query: ${(0, error_handlers_1.extractErrorMessage)(error, 'Refinement Test')}. Reverting to last working version.`);
+                }
+                // If we have a previous working version, use that
+                if (lastWorkingTestResult && lastWorkingEvaluation) {
+                    return this.buildSuccessResult(lastWorkingQuery, lastWorkingTestResult, lastWorkingEvaluation, refinementCount);
+                }
+                // No previous working version - throw the error
+                throw error;
+            }
+            // 2. Evaluate if it answers the question
+            const evaluation = await this.evaluateQuery(currentQuery, businessQuestion, testResult);
+            lastWorkingEvaluation = evaluation;
+            // 3. If evaluation passes, we're done!
+            if (this.shouldStopRefining(evaluation)) {
+                return this.buildSuccessResult(currentQuery, testResult, evaluation, refinementCount);
+            }
+            // 4. Refine the query based on suggestions
+            refinementCount++;
+            if (this.config.verbose) {
+                (0, core_1.LogStatus)(`Refinement iteration ${refinementCount}/${maxRefinements}`);
+            }
+            currentQuery = await this.performRefinement(currentQuery, businessQuestion, evaluation, entityMetadata);
+        }
+        // Reached max refinements - return best attempt
+        // Use the last successfully tested query
+        if (lastWorkingTestResult && lastWorkingEvaluation) {
+            return {
+                query: lastWorkingQuery,
+                testResult: lastWorkingTestResult,
+                evaluation: lastWorkingEvaluation,
+                refinementCount,
+                reachedMaxRefinements: true,
+            };
+        }
+        // Fallback: try to build final result with current query
+        return await this.buildFinalResult(currentQuery, businessQuestion, refinementCount);
+    }
+    /**
+     * Configure AIEngine for prompt execution
+     * Ensures engine is ready before running prompts
+     */
+    async configureAIEngine() {
+        try {
+            const aiEngine = aiengine_1.AIEngine.Instance;
+            await aiEngine.Config(false, this.contextUser);
+        }
+        catch (error) {
+            throw new Error((0, error_handlers_1.extractErrorMessage)(error, 'AIEngine Configuration'));
+        }
+    }
+    /**
+     * Test current query using QueryTester
+     * Throws if query testing fails
+     */
+    async testCurrentQuery(query) {
+        const testResult = await this.tester.testQuery(query);
+        if (!testResult.success) {
+            throw new Error(`Query testing failed after ${testResult.attempts} attempts: ${testResult.error}`);
+        }
+        return testResult;
+    }
+    /**
+     * Determine if refinement should stop based on evaluation
+     * Stops if query answers question and doesn't need refinement
+     */
+    shouldStopRefining(evaluation) {
+        return evaluation.answersQuestion && !evaluation.needsRefinement;
+    }
+    /**
+     * Build success result when refinement loop completes successfully
+     */
+    buildSuccessResult(query, testResult, evaluation, refinementCount) {
+        return {
+            query,
+            testResult,
+            evaluation,
+            refinementCount,
+        };
+    }
+    /**
+     * Build final result when max refinements reached
+     * Re-tests and re-evaluates final query
+     */
+    async buildFinalResult(query, businessQuestion, refinementCount) {
+        const testResult = await this.tester.testQuery(query);
+        const evaluation = await this.evaluateQuery(query, businessQuestion, testResult);
+        return {
+            query,
+            testResult,
+            evaluation,
+            refinementCount,
+            reachedMaxRefinements: true,
+        };
+    }
+    /**
+     * Evaluate if query answers the business question correctly
+     * Uses Query Result Evaluator AI prompt
+     *
+     * @param query - Query to evaluate
+     * @param businessQuestion - Original business question
+     * @param testResult - Test execution results with sample data
+     * @returns Evaluation with confidence and suggestions
+     */
+    async evaluateQuery(query, businessQuestion, testResult) {
+        try {
+            const aiEngine = aiengine_1.AIEngine.Instance;
+            const prompt = this.findPromptByName(aiEngine, PromptNames_1.PROMPT_QUERY_EVALUATOR);
+            // Limit sample results to first 10 rows for efficiency
+            const sampleResults = testResult.sampleRows?.slice(0, 10) || [];
+            const promptData = {
+                userQuestion: businessQuestion.userQuestion,
+                description: businessQuestion.description,
+                technicalDescription: businessQuestion.technicalDescription,
+                entityMetadata: this.entityMetadata,
+                generatedSQL: query.sql,
+                parameters: query.parameters,
+                sampleResults,
+            };
+            const evaluation = await this.executePrompt(prompt, promptData);
+            this.logEvaluation(evaluation);
+            return evaluation;
+        }
+        catch (error) {
+            throw new Error((0, error_handlers_1.extractErrorMessage)(error, 'QueryRefiner.evaluateQuery'));
+        }
+    }
+    /**
+     * Refine query based on evaluation feedback
+     * Uses Query Refiner AI prompt
+     *
+     * @param query - Current query to refine
+     * @param businessQuestion - Original business question
+     * @param evaluation - Evaluation feedback
+     * @param entityMetadata - Entity metadata for refinement
+     * @returns Refined query with improvements
+     */
+    async performRefinement(query, businessQuestion, evaluation, entityMetadata) {
+        try {
+            const aiEngine = aiengine_1.AIEngine.Instance;
+            const prompt = this.findPromptByName(aiEngine, PromptNames_1.PROMPT_QUERY_REFINER);
+            const promptData = {
+                userQuestion: businessQuestion.userQuestion,
+                description: businessQuestion.description,
+                currentSQL: query.sql,
+                evaluationFeedback: evaluation,
+                entityMetadata,
+            };
+            const refinedQuery = await this.executePrompt(prompt, promptData);
+            if (this.config.verbose) {
+                (0, core_1.LogStatus)(`Refinements applied: ${refinedQuery.improvementsSummary}`);
+            }
+            return {
+                sql: refinedQuery.sql,
+                parameters: refinedQuery.parameters,
+            };
+        }
+        catch (error) {
+            throw new Error((0, error_handlers_1.extractErrorMessage)(error, 'QueryRefiner.performRefinement'));
+        }
+    }
+    /**
+     * Find prompt by name in AIEngine cache
+     * Throws if prompt not found
+     */
+    findPromptByName(aiEngine, promptName) {
+        const prompt = aiEngine.Prompts.find((p) => p.Name === promptName);
+        if (!prompt) {
+            throw new Error(`Prompt '${promptName}' not found in AIEngine cache`);
+        }
+        return prompt;
+    }
+    /**
+     * Execute AI prompt and parse result
+     * Generic method for any prompt type
+     */
+    async executePrompt(prompt, promptData) {
+        const result = await (0, prompt_helpers_1.executePromptWithOverrides)(prompt, promptData, this.contextUser, this.config);
+        if (!result || !result.success) {
+            throw new Error(`AI prompt execution failed: ${result?.errorMessage || 'Unknown error'}`);
+        }
+        if (!result.result) {
+            throw new Error('AI prompt returned no result');
+        }
+        return result.result;
+    }
+    /**
+     * Log evaluation results for debugging
+     */
+    logEvaluation(evaluation) {
+        if (!this.config.verbose)
+            return;
+        (0, core_1.LogStatus)(`Evaluation: answersQuestion=${evaluation.answersQuestion}, ` +
+            `confidence=${evaluation.confidence}, ` +
+            `needsRefinement=${evaluation.needsRefinement}`);
+        if (evaluation.reasoning) {
+            (0, core_1.LogStatus)(`Reasoning: ${evaluation.reasoning}`);
+        }
+        if (evaluation.suggestions.length > 0) {
+            (0, core_1.LogStatus)(`Suggestions: ${evaluation.suggestions.join('; ')}`);
+        }
+    }
+}
+exports.QueryRefiner = QueryRefiner;
+//# sourceMappingURL=QueryRefiner.js.map

package/dist/core/QueryRefiner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryRefiner.js","sourceRoot":"","sources":["../../src/core/QueryRefiner.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,uDAAoD;AAEpD,+CAA2D;AAC3D,4DAA8D;AAU9D,wDAAsF;AAEtF,4DAAqE;AAErE;;;GAGG;AACH,MAAa,YAAY;IAIb;IACA;IACA;IALF,cAAc,GAA8B,EAAE,CAAC;IAEvD,YACU,MAAmB,EACnB,WAAqB,EACrB,MAAsB;QAFtB,WAAM,GAAN,MAAM,CAAa;QACnB,gBAAW,GAAX,WAAW,CAAU;QACrB,WAAM,GAAN,MAAM,CAAgB;IAC7B,CAAC;IAEJ;;;;;;;;;;;OAWG;IACH,KAAK,CAAC,WAAW,CACf,KAAqB,EACrB,gBAAkC,EAClC,cAAyC,EACzC,iBAAyB,CAAC;QAE1B,8CAA8C;QAC9C,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QAErC,IAAI,YAAY,GAAG,KAAK,CAAC;QACzB,IAAI,eAAe,GAAG,CAAC,CAAC;QAExB,2DAA2D;QAC3D,IAAI,gBAAgB,GAAG,KAAK,CAAC;QAC7B,IAAI,qBAAqB,GAA2B,IAAI,CAAC;QACzD,IAAI,qBAAqB,GAA2B,IAAI,CAAC;QAEzD,gCAAgC;QAChC,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,OAAO,eAAe,GAAG,cAAc,EAAE,CAAC;YACxC,4BAA4B;YAC5B,IAAI,UAA2B,CAAC;YAChC,IAAI,CAAC;gBACH,UAAU,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;gBACvD,iDAAiD;gBACjD,gBAAgB,GAAG,YAAY,CAAC;gBAChC,qBAAqB,GAAG,UAAU,CAAC;YACrC,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACxB,iEAAiE;gBACjE,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACxB,IAAA,gBAAS,EAAC,qCAAqC,IAAA,oCAAmB,EAAC,KAAK,EAAE,iBAAiB,CAAC,sCAAsC,CAAC,CAAC;gBACtI,CAAC;gBAED,kDAAkD;gBAClD,IAAI,qBAAqB,IAAI,qBAAqB,EAAE,CAAC;oBACnD,OAAO,IAAI,CAAC,kBAAkB,CAC5B,gBAAgB,EAChB,qBAAqB,EACrB,qBAAqB,EACrB,eAAe,CAChB,CAAC;gBACJ,CAAC;gBAED,gDAAgD;gBAChD,MAAM,KAAK,CAAC;YACd,CAAC;YAED,yCAAyC;YACzC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,CACzC,YAAY,EACZ,gBAAgB,EAChB,UAAU,CACX,CAAC;YACF,qBAAqB,GAAG,UAAU,CAAC;YAEnC,uCAAuC;YACvC,IAAI,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC,EAAE,CAAC;gBACxC,OAAO,IAAI,CAAC,kBAAkB,CAC5B,YAAY,EACZ,UAAU,EACV,UAAU,EACV,eAAe,CAChB,CAAC;YACJ,CAAC;YAED,2CAA2C;YAC3C,eAAe,EAAE,CAAC;YAClB,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACxB,IAAA,gBAAS,EAAC,wBAAwB,eAAe,IAAI,cAAc,EAAE,CAAC,CAAC;YACzE,CAAC;YAED,YAAY,GAAG,MAAM,IAAI,CAAC,iBAAiB,CACzC,YAAY,EACZ,gBAAgB,EAChB,UAAU,EACV,cAAc,CACf,CAAC;QACJ,CAAC;QAED,gDAAgD;QAChD,yCAAyC;QACzC,IAAI,qBAAqB,IAAI,qBAAqB,EAAE,CAAC;YACnD,OAAO;gBACL,KAAK,EAAE,gBAAgB;gBACvB,UAAU,EAAE,qBAAqB;gBACjC,UAAU,EAAE,qBAAqB;gBACjC,eAAe;gBACf,qBAAqB,EAAE,IAAI;aAC5B,CAAC;QACJ,CAAC;QAED,yDAAyD;QACzD,OAAO,MAAM,IAAI,CAAC,gBAAgB,CAChC,YAAY,EACZ,gBAAgB,EAChB,eAAe,CAChB,CAAC;IACJ,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,iBAAiB;QAC7B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,mBAAQ,CAAC,QAAQ,CAAC;YACnC,MAAM,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QACjD,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,IAAA,oCAAmB,EAAC,KAAK,EAAE,wBAAwB,CAAC,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,gBAAgB,CAAC,KAAqB;QAClD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAEtD,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CACb,8BAA8B,UAAU,CAAC,QAAQ,cAAc,UAAU,CAAC,KAAK,EAAE,CAClF,CAAC;QACJ,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;;OAGG;IACK,kBAAkB,CAAC,UAA2B;QACpD,OAAO,UAAU,CAAC,eAAe,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC;IACnE,CAAC;IAED;;OAEG;IACK,kBAAkB,CACxB,KAAqB,EACrB,UAA2B,EAC3B,UAA2B,EAC3B,eAAuB;QAEvB,OAAO;YACL,KAAK;YACL,UAAU;YACV,UAAU;YACV,eAAe;SAChB,CAAC;IACJ,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,gBAAgB,CAC5B,KAAqB,EACrB,gBAAkC,EAClC,eAAuB;QAEvB,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,CACzC,KAAK,EACL,gBAAgB,EAChB,UAAU,CACX,CAAC;QAEF,OAAO;YACL,KAAK;YACL,UAAU;YACV,UAAU;YACV,eAAe;YACf,qBAAqB,EAAE,IAAI;SAC5B,CAAC;IACJ,CAAC;IAED;;;;;;;;OAQG;IACK,KAAK,CAAC,aAAa,CACzB,KAAqB,EACrB,gBAAkC,EAClC,UAA2B;QAE3B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,mBAAQ,CAAC,QAAQ,CAAC;YACnC,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,oCAAsB,CAAC,CAAC;YAEvE,uDAAuD;YACvD,MAAM,aAAa,GAAG,UAAU,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;YAEhE,MAAM,UAAU,GAAG;gBACjB,YAAY,EAAE,gBAAgB,CAAC,YAAY;gBAC3C,WAAW,EAAE,gBAAgB,CAAC,WAAW;gBACzC,oBAAoB,EAAE,gBAAgB,CAAC,oBAAoB;gBAC3D,cAAc,EAAE,IAAI,CAAC,cAAc;gBACnC,YAAY,EAAE,KAAK,CAAC,GAAG;gBACvB,UAAU,EAAE,KAAK,CAAC,UAAU;gBAC5B,aAAa;aACd,CAAC;YAEF,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,CACzC,MAAM,EACN,UAAU,CACX,CAAC;YAEF,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YAC/B,OAAO,UAAU,CAAC;QACpB,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,IAAA,oCAAmB,EAAC,KAAK,EAAE,4BAA4B,CAAC,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;IAED;;;;;;;;;OASG;IACK,KAAK,CAAC,iBAAiB,CAC7B,KAAqB,EACrB,gBAAkC,EAClC,UAA2B,EAC3B,cAAyC;QAEzC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,mBAAQ,CAAC,QAAQ,CAAC;YACnC,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,kCAAoB,CAAC,CAAC;YAErE,MAAM,UAAU,GAAG;gBACjB,YAAY,EAAE,gBAAgB,CAAC,YAAY;gBAC3C,WAAW,EAAE,gBAAgB,CAAC,WAAW;gBACzC,UAAU,EAAE,KAAK,CAAC,GAAG;gBACrB,kBAAkB,EAAE,UAAU;gBAC9B,cAAc;aACf,CAAC;YAEF,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,aAAa,CAE3C,MAAM,EAAE,UAAU,CAAC,CAAC;YAEtB,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACxB,IAAA,gBAAS,EAAC,wBAAwB,YAAY,CAAC,mBAAmB,EAAE,CAAC,CAAC;YACxE,CAAC;YAED,OAAO;gBACL,GAAG,EAAE,YAAY,CAAC,GAAG;gBACrB,UAAU,EAAE,YAAY,CAAC,UAAU;aACpC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CACb,IAAA,oCAAmB,EAAC,KAAK,EAAE,gCAAgC,CAAC,CAC7D,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,gBAAgB,CACtB,QAAkB,EAClB,UAAkB;QAElB,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;QACnE,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,WAAW,UAAU,+BAA+B,CAAC,CAAC;QACxE,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,aAAa,CACzB,MAA8B,EAC9B,UAAmC;QAEnC,MAAM,MAAM,GAAG,MAAM,IAAA,2CAA0B,EAC7C,MAAM,EACN,UAAU,EACV,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,MAAM,CACZ,CAAC;QAEF,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CACb,+BAA+B,MAAM,EAAE,YAAY,IAAI,eAAe,EAAE,CACzE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAClD,CAAC;QAED,OAAO,MAAM,CAAC,MAAM,CAAC;IACvB,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,UAA2B;QAC/C,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO;QAEjC,IAAA,gBAAS,EACP,+BAA+B,UAAU,CAAC,eAAe,IAAI;YAC3D,cAAc,UAAU,CAAC,UAAU,IAAI;YACvC,mBAAmB,UAAU,CAAC,eAAe,EAAE,CAClD,CAAC;QAEF,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YACzB,IAAA,gBAAS,EAAC,cAAc,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,IAAI,UAAU,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtC,IAAA,gBAAS,EAAC,gBAAgB,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;CACF;AAjWD,oCAiWC"}

package/dist/core/QueryTester.d.ts

@@ -0,0 +1,70 @@
+/**
+ * QueryTester - Tests and validates SQL queries
+ *
+ * Renders Nunjucks templates with sample parameter values and executes
+ * queries against the database. Handles error fixing with retry loop.
+ */
+import { DatabaseProviderBase, UserInfo } from '@memberjunction/core';
+import { GeneratedQuery, QueryTestResult, EntityMetadataForPrompt, BusinessQuestion } from '../data/schema';
+import { QueryGenConfig } from '../cli/config';
+/**
+ * QueryTester class
+ * Tests SQL queries by rendering templates and executing against database
+ */
+export declare class QueryTester {
+    private dataProvider;
+    private entityMetadata;
+    private businessQuestion;
+    private contextUser;
+    private config;
+    private nunjucksEnv;
+    constructor(dataProvider: DatabaseProviderBase, entityMetadata: EntityMetadataForPrompt[], businessQuestion: BusinessQuestion, contextUser: UserInfo, config: QueryGenConfig);
+    /**
+     * Test a query by rendering template with sample values and executing it
+     * Retries up to maxAttempts times, calling QueryFixer on failures
+     *
+     * @param query - Generated query to test
+     * @param maxAttempts - Maximum number of retry attempts (default: 5)
+     * @returns Test result with success status, SQL, and sample data
+     */
+    testQuery(query: GeneratedQuery, maxAttempts?: number): Promise<QueryTestResult>;
+    /**
+     * Render query template with sample parameter values
+     * Uses QueryParameterProcessor for proper type handling
+     *
+     * @param query - Generated query with parameters
+     * @returns Rendered SQL string ready for execution
+     */
+    private renderQueryTemplate;
+    /**
+     * Processes a raw parameter value based on its type, handling special cases like arrays.
+     * Follows Skip-Brain pattern for parameter processing.
+     * For array types, this function will:
+     * - Parse JSON arrays if the value is a JSON string
+     * - Split comma-separated strings as a fallback
+     * - Return as-is for sqlIn filter to handle
+     *
+     * @param rawValue - The raw parameter value (from sampleValue)
+     * @param paramType - The parameter type ('string', 'number', 'date', 'boolean', 'array')
+     * @returns Processed value ready for use in Nunjucks template
+     */
+    private processParameterValue;
+    /**
+     * Execute SQL query against database
+     * Uses DataProvider to run query with contextUser
+     *
+     * @param sql - Rendered SQL query
+     * @returns Array of result rows
+     */
+    private executeSQLQuery;
+    /**
+     * Fix a query that failed to execute
+     * Uses QueryFixer with AI to analyze error and generate correction
+     *
+     * @param query - Query that failed
+     * @param errorMessage - Error message from execution
+     * @returns Corrected query
+     */
+    private fixQuery;
+}
+//# sourceMappingURL=QueryTester.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"QueryTester.d.ts","sourceRoot":"","sources":["../../src/core/QueryTester.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EACL,oBAAoB,EAEpB,QAAQ,EAET,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EACL,cAAc,EACd,eAAe,EACf,uBAAuB,EACvB,gBAAgB,EACjB,MAAM,gBAAgB,CAAC;AAExB,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAE/C;;;GAGG;AACH,qBAAa,WAAW;IAIpB,OAAO,CAAC,YAAY;IACpB,OAAO,CAAC,cAAc;IACtB,OAAO,CAAC,gBAAgB;IACxB,OAAO,CAAC,WAAW;IACnB,OAAO,CAAC,MAAM;IAPhB,OAAO,CAAC,WAAW,CAAuB;gBAGhC,YAAY,EAAE,oBAAoB,EAClC,cAAc,EAAE,uBAAuB,EAAE,EACzC,gBAAgB,EAAE,gBAAgB,EAClC,WAAW,EAAE,QAAQ,EACrB,MAAM,EAAE,cAAc;IAqBhC;;;;;;;OAOG;IACG,SAAS,CACb,KAAK,EAAE,cAAc,EACrB,WAAW,GAAE,MAAU,GACtB,OAAO,CAAC,eAAe,CAAC;IAgD3B;;;;;;OAMG;IACH,OAAO,CAAC,mBAAmB;IAsB3B;;;;;;;;;;;OAWG;IACH,OAAO,CAAC,qBAAqB;IA0D7B;;;;;;OAMG;YACW,eAAe;IAqB7B;;;;;;;OAOG;YACW,QAAQ;CAYvB"}

package/dist/core/QueryTester.js

@@ -0,0 +1,243 @@
+"use strict";
+/**
+ * QueryTester - Tests and validates SQL queries
+ *
+ * Renders Nunjucks templates with sample parameter values and executes
+ * queries against the database. Handles error fixing with retry loop.
+ */
+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;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QueryTester = void 0;
+const nunjucks = __importStar(require("nunjucks"));
+const core_1 = require("@memberjunction/core");
+const error_handlers_1 = require("../utils/error-handlers");
+const QueryFixer_1 = require("./QueryFixer");
+/**
+ * QueryTester class
+ * Tests SQL queries by rendering templates and executing against database
+ */
+class QueryTester {
+    dataProvider;
+    entityMetadata;
+    businessQuestion;
+    contextUser;
+    config;
+    nunjucksEnv;
+    constructor(dataProvider, entityMetadata, businessQuestion, contextUser, config) {
+        this.dataProvider = dataProvider;
+        this.entityMetadata = entityMetadata;
+        this.businessQuestion = businessQuestion;
+        this.contextUser = contextUser;
+        this.config = config;
+        // Initialize Nunjucks environment with SQL-safe filters
+        this.nunjucksEnv = new nunjucks.Environment(null, {
+            autoescape: false,
+            throwOnUndefined: true,
+            trimBlocks: true,
+            lstripBlocks: true,
+        });
+        // Add custom SQL-safe filters from RunQuerySQLFilterManager
+        const filterManager = core_1.RunQuerySQLFilterManager.Instance;
+        const filters = filterManager.getAllFilters();
+        for (const filter of filters) {
+            if (filter.implementation) {
+                this.nunjucksEnv.addFilter(filter.name, filter.implementation);
+            }
+        }
+    }
+    /**
+     * Test a query by rendering template with sample values and executing it
+     * Retries up to maxAttempts times, calling QueryFixer on failures
+     *
+     * @param query - Generated query to test
+     * @param maxAttempts - Maximum number of retry attempts (default: 5)
+     * @returns Test result with success status, SQL, and sample data
+     */
+    async testQuery(query, maxAttempts = 5) {
+        let attempt = 0;
+        let lastError;
+        let currentQuery = query;
+        while (attempt < maxAttempts) {
+            attempt++;
+            try {
+                // 1. Render template with sample parameter values
+                const renderedSQL = this.renderQueryTemplate(currentQuery);
+                // 2. Execute SQL on database
+                const results = await this.executeSQLQuery(renderedSQL);
+                // 3. Success! (Empty results are valid - query executed without errors)
+                // Note: We don't validate rowCount because:
+                // - Empty results may indicate no data in database (not a query error)
+                // - Query structure can be correct even with zero rows returned
+                // - Testing should focus on SQL syntax/execution, not data presence
+                return {
+                    success: true,
+                    renderedSQL,
+                    rowCount: results.length,
+                    sampleRows: results.slice(0, 10), // Return first 10 rows
+                    attempts: attempt,
+                };
+            }
+            catch (error) {
+                lastError = (0, error_handlers_1.extractErrorMessage)(error, 'Query Testing');
+                if (this.config.verbose) {
+                    (0, core_1.LogError)(`Attempt ${attempt}/${maxAttempts} failed: ${lastError}`);
+                }
+                // 5. If not last attempt, try to fix the query
+                if (attempt < maxAttempts) {
+                    currentQuery = await this.fixQuery(currentQuery, lastError);
+                }
+            }
+        }
+        // Failed after max attempts
+        return {
+            success: false,
+            error: lastError,
+            attempts: maxAttempts,
+        };
+    }
+    /**
+     * Render query template with sample parameter values
+     * Uses QueryParameterProcessor for proper type handling
+     *
+     * @param query - Generated query with parameters
+     * @returns Rendered SQL string ready for execution
+     */
+    renderQueryTemplate(query) {
+        // Build parameter values object
+        const paramValues = {};
+        for (const param of query.parameters) {
+            const rawValue = param.sampleValue;
+            if (rawValue !== undefined && rawValue !== null) {
+                paramValues[param.name] = this.processParameterValue(rawValue, param.type);
+            }
+        }
+        try {
+            // Render template using Nunjucks with SQL-safe filters
+            const renderedSQL = this.nunjucksEnv.renderString(query.sql, paramValues);
+            return renderedSQL;
+        }
+        catch (error) {
+            throw new Error(`Template rendering failed: ${(0, error_handlers_1.extractErrorMessage)(error, 'Nunjucks')}`);
+        }
+    }
+    /**
+     * Processes a raw parameter value based on its type, handling special cases like arrays.
+     * Follows Skip-Brain pattern for parameter processing.
+     * For array types, this function will:
+     * - Parse JSON arrays if the value is a JSON string
+     * - Split comma-separated strings as a fallback
+     * - Return as-is for sqlIn filter to handle
+     *
+     * @param rawValue - The raw parameter value (from sampleValue)
+     * @param paramType - The parameter type ('string', 'number', 'date', 'boolean', 'array')
+     * @returns Processed value ready for use in Nunjucks template
+     */
+    processParameterValue(rawValue, paramType) {
+        if (rawValue === undefined || rawValue === null) {
+            return rawValue;
+        }
+        // For array type parameters, ensure value is compatible with sqlIn filter
+        if (paramType === 'array' && typeof rawValue === 'string') {
+            try {
+                // Try to parse as JSON array
+                const parsed = JSON.parse(rawValue);
+                if (Array.isArray(parsed)) {
+                    return parsed;
+                }
+            }
+            catch {
+                // Not valid JSON - return as-is for sqlIn filter to handle comma-separated strings
+            }
+            // Return comma-separated string as-is - sqlIn filter handles this
+            return rawValue;
+        }
+        // For non-array types, convert as needed
+        switch (paramType) {
+            case 'number':
+                if (typeof rawValue === 'string') {
+                    const num = Number(rawValue);
+                    if (isNaN(num)) {
+                        throw new Error(`Invalid number sample value: ${rawValue}`);
+                    }
+                    return num;
+                }
+                return rawValue;
+            case 'boolean':
+                if (typeof rawValue === 'string') {
+                    const lower = rawValue.toLowerCase();
+                    if (lower !== 'true' && lower !== 'false') {
+                        throw new Error(`Invalid boolean sample value: ${rawValue}`);
+                    }
+                    return lower === 'true';
+                }
+                return rawValue;
+            case 'date':
+                if (typeof rawValue === 'string') {
+                    const date = new Date(rawValue);
+                    if (isNaN(date.getTime())) {
+                        throw new Error(`Invalid date sample value: ${rawValue}`);
+                    }
+                    return date;
+                }
+                return rawValue;
+            case 'string':
+            default:
+                return rawValue;
+        }
+    }
+    /**
+     * Execute SQL query against database
+     * Uses DataProvider to run query with contextUser
+     *
+     * @param sql - Rendered SQL query
+     * @returns Array of result rows
+     */
+    async executeSQLQuery(sql) {
+        try {
+            const result = await this.dataProvider.ExecuteSQL(sql, undefined, undefined, this.contextUser);
+            if (!result || !Array.isArray(result)) {
+                throw new Error('ExecuteSQL returned invalid result format');
+            }
+            return result;
+        }
+        catch (error) {
+            throw new Error(`SQL execution failed: ${(0, error_handlers_1.extractErrorMessage)(error, 'Database')}`);
+        }
+    }
+    /**
+     * Fix a query that failed to execute
+     * Uses QueryFixer with AI to analyze error and generate correction
+     *
+     * @param query - Query that failed
+     * @param errorMessage - Error message from execution
+     * @returns Corrected query
+     */
+    async fixQuery(query, errorMessage) {
+        const fixer = new QueryFixer_1.QueryFixer(this.contextUser, this.config);
+        return await fixer.fixQuery(query, errorMessage, this.entityMetadata, this.businessQuestion);
+    }
+}
+exports.QueryTester = QueryTester;
+//# sourceMappingURL=QueryTester.js.map

package/dist/core/QueryTester.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryTester.js","sourceRoot":"","sources":["../../src/core/QueryTester.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,mDAAqC;AACrC,+CAK8B;AAC9B,4DAA8D;AAO9D,6CAA0C;AAG1C;;;GAGG;AACH,MAAa,WAAW;IAIZ;IACA;IACA;IACA;IACA;IAPF,WAAW,CAAuB;IAE1C,YACU,YAAkC,EAClC,cAAyC,EACzC,gBAAkC,EAClC,WAAqB,EACrB,MAAsB;QAJtB,iBAAY,GAAZ,YAAY,CAAsB;QAClC,mBAAc,GAAd,cAAc,CAA2B;QACzC,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,gBAAW,GAAX,WAAW,CAAU;QACrB,WAAM,GAAN,MAAM,CAAgB;QAE9B,wDAAwD;QACxD,IAAI,CAAC,WAAW,GAAG,IAAI,QAAQ,CAAC,WAAW,CAAC,IAAI,EAAE;YAChD,UAAU,EAAE,KAAK;YACjB,gBAAgB,EAAE,IAAI;YACtB,UAAU,EAAE,IAAI;YAChB,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;QAEH,4DAA4D;QAC5D,MAAM,aAAa,GAAG,+BAAwB,CAAC,QAAQ,CAAC;QACxD,MAAM,OAAO,GAAG,aAAa,CAAC,aAAa,EAAE,CAAC;QAE9C,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;gBAC1B,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,cAAc,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,SAAS,CACb,KAAqB,EACrB,cAAsB,CAAC;QAEvB,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,IAAI,SAA6B,CAAC;QAClC,IAAI,YAAY,GAAG,KAAK,CAAC;QAEzB,OAAO,OAAO,GAAG,WAAW,EAAE,CAAC;YAC7B,OAAO,EAAE,CAAC;YAEV,IAAI,CAAC;gBACH,kDAAkD;gBAClD,MAAM,WAAW,GAAG,IAAI,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;gBAE3D,6BAA6B;gBAC7B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;gBAExD,wEAAwE;gBACxE,4CAA4C;gBAC5C,uEAAuE;gBACvE,gEAAgE;gBAChE,oEAAoE;gBACpE,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,WAAW;oBACX,QAAQ,EAAE,OAAO,CAAC,MAAM;oBACxB,UAAU,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,uBAAuB;oBACzD,QAAQ,EAAE,OAAO;iBAClB,CAAC;YACJ,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACxB,SAAS,GAAG,IAAA,oCAAmB,EAAC,KAAK,EAAE,eAAe,CAAC,CAAC;gBACxD,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACxB,IAAA,eAAQ,EAAC,WAAW,OAAO,IAAI,WAAW,YAAY,SAAS,EAAE,CAAC,CAAC;gBACrE,CAAC;gBAED,+CAA+C;gBAC/C,IAAI,OAAO,GAAG,WAAW,EAAE,CAAC;oBAC1B,YAAY,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;gBAC9D,CAAC;YACH,CAAC;QACH,CAAC;QAED,4BAA4B;QAC5B,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,SAAS;YAChB,QAAQ,EAAE,WAAW;SACtB,CAAC;IACJ,CAAC;IAED;;;;;;OAMG;IACK,mBAAmB,CAAC,KAAqB;QAC/C,gCAAgC;QAChC,MAAM,WAAW,GAA4B,EAAE,CAAC;QAEhD,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;YACrC,MAAM,QAAQ,GAAG,KAAK,CAAC,WAAW,CAAC;YACnC,IAAI,QAAQ,KAAK,SAAS,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;gBAChD,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,qBAAqB,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YAC7E,CAAC;QACH,CAAC;QAED,IAAI,CAAC;YACH,uDAAuD;YACvD,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;YAC1E,OAAO,WAAW,CAAC;QACrB,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CACb,8BAA8B,IAAA,oCAAmB,EAAC,KAAK,EAAE,UAAU,CAAC,EAAE,CACvE,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;;;;;;;;;OAWG;IACK,qBAAqB,CAAC,QAAiB,EAAE,SAAiB;QAChE,IAAI,QAAQ,KAAK,SAAS,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;YAChD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,0EAA0E;QAC1E,IAAI,SAAS,KAAK,OAAO,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1D,IAAI,CAAC;gBACH,6BAA6B;gBAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;gBACpC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC1B,OAAO,MAAM,CAAC;gBAChB,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,mFAAmF;YACrF,CAAC;YACD,kEAAkE;YAClE,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,yCAAyC;QACzC,QAAQ,SAAS,EAAE,CAAC;YAClB,KAAK,QAAQ;gBACX,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;oBACjC,MAAM,GAAG,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;oBAC7B,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;wBACf,MAAM,IAAI,KAAK,CAAC,gCAAgC,QAAQ,EAAE,CAAC,CAAC;oBAC9D,CAAC;oBACD,OAAO,GAAG,CAAC;gBACb,CAAC;gBACD,OAAO,QAAQ,CAAC;YAElB,KAAK,SAAS;gBACZ,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;oBACjC,MAAM,KAAK,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;oBACrC,IAAI,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;wBAC1C,MAAM,IAAI,KAAK,CAAC,iCAAiC,QAAQ,EAAE,CAAC,CAAC;oBAC/D,CAAC;oBACD,OAAO,KAAK,KAAK,MAAM,CAAC;gBAC1B,CAAC;gBACD,OAAO,QAAQ,CAAC;YAElB,KAAK,MAAM;gBACT,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;oBACjC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,QAAQ,CAAC,CAAC;oBAChC,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;wBAC1B,MAAM,IAAI,KAAK,CAAC,8BAA8B,QAAQ,EAAE,CAAC,CAAC;oBAC5D,CAAC;oBACD,OAAO,IAAI,CAAC;gBACd,CAAC;gBACD,OAAO,QAAQ,CAAC;YAElB,KAAK,QAAQ,CAAC;YACd;gBACE,OAAO,QAAQ,CAAC;QACpB,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACK,KAAK,CAAC,eAAe,CAAC,GAAW;QACvC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,CAC/C,GAAG,EACH,SAAS,EACT,SAAS,EACT,IAAI,CAAC,WAAW,CACjB,CAAC;YAEF,IAAI,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBACtC,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;YAC/D,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CACb,yBAAyB,IAAA,oCAAmB,EAAC,KAAK,EAAE,UAAU,CAAC,EAAE,CAClE,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;;;;;OAOG;IACK,KAAK,CAAC,QAAQ,CACpB,KAAqB,EACrB,YAAoB;QAEpB,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5D,OAAO,MAAM,KAAK,CAAC,QAAQ,CACzB,KAAK,EACL,YAAY,EACZ,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,gBAAgB,CACtB,CAAC;IACJ,CAAC;CACF;AA3OD,kCA2OC"}

package/dist/core/QueryWriter.d.ts

@@ -0,0 +1,57 @@
+/**
+ * QueryWriter - Generates SQL query templates using AI with few-shot learning
+ *
+ * Uses the SQL Query Writer AI prompt to generate Nunjucks SQL templates
+ * based on business questions and similar golden query examples.
+ */
+import { UserInfo } from '@memberjunction/core';
+import { QueryGenConfig } from '../cli/config';
+import { BusinessQuestion, GeneratedQuery, EntityMetadataForPrompt, GoldenQuery } from '../data/schema';
+/**
+ * QueryWriter class
+ * Generates Nunjucks SQL query templates using AI with few-shot learning
+ */
+export declare class QueryWriter {
+    private contextUser;
+    private config;
+    constructor(contextUser: UserInfo, config: QueryGenConfig);
+    /**
+     * Generate SQL query template for a business question
+     * Uses few-shot learning with similar golden query examples
+     *
+     * @param businessQuestion - Business question to answer with SQL
+     * @param entityMetadata - Available entity metadata for query
+     * @param fewShotExamples - Similar golden queries for few-shot learning
+     * @returns Generated SQL query with parameters and output schema
+     */
+    generateQuery(businessQuestion: BusinessQuestion, entityMetadata: EntityMetadataForPrompt[], fewShotExamples: GoldenQuery[]): Promise<GeneratedQuery>;
+    /**
+     * Find prompt by name in AIEngine cache
+     * Throws if prompt not found
+     */
+    private findPromptByName;
+    /**
+     * Execute the SQL Query Writer AI prompt with retry logic for validation failures
+     * Parses JSON response and validates structure, retrying with feedback if validation fails
+     */
+    private executePrompt;
+    /**
+     * Validate generated query structure
+     * Ensures query has proper SQL template syntax and valid metadata
+     *
+     * @param query - Generated query to validate
+     * @throws Error if query structure is invalid
+     */
+    private validateGeneratedQuery;
+    /**
+     * Check if SQL uses base views (vw* pattern)
+     * Basic heuristic: looks for view names in FROM and JOIN clauses
+     */
+    private usesBaseViews;
+    /**
+     * Validate a single query parameter
+     * Ensures all required fields are present and valid
+     */
+    private validateParameter;
+}
+//# sourceMappingURL=QueryWriter.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"QueryWriter.d.ts","sourceRoot":"","sources":["../../src/core/QueryWriter.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH,OAAO,EAAE,QAAQ,EAAa,MAAM,sBAAsB,CAAC;AAC3D,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAG/C,OAAO,EAAE,gBAAgB,EAAE,cAAc,EAAE,uBAAuB,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAGxG;;;GAGG;AACH,qBAAa,WAAW;IAEpB,OAAO,CAAC,WAAW;IACnB,OAAO,CAAC,MAAM;gBADN,WAAW,EAAE,QAAQ,EACrB,MAAM,EAAE,cAAc;IAGhC;;;;;;;;OAQG;IACG,aAAa,CACjB,gBAAgB,EAAE,gBAAgB,EAClC,cAAc,EAAE,uBAAuB,EAAE,EACzC,eAAe,EAAE,WAAW,EAAE,GAC7B,OAAO,CAAC,cAAc,CAAC;IAgC1B;;;OAGG;IACH,OAAO,CAAC,gBAAgB;IAWxB;;;OAGG;YACW,aAAa;IAwE3B;;;;;;OAMG;IACH,OAAO,CAAC,sBAAsB;IAsB9B;;;OAGG;IACH,OAAO,CAAC,aAAa;IAMrB;;;OAGG;IACH,OAAO,CAAC,iBAAiB;CAiC1B"}