@memberjunction/query-gen 0.0.1 → 2.126.1

package/dist/core/QueryDatabaseWriter.js

@@ -0,0 +1,152 @@
+"use strict";
+/**
+ * QueryDatabaseWriter - Write validated queries directly to the database
+ *
+ * Creates Query entities in the database. QueryFields and QueryParameters
+ * are automatically extracted by QueryEntity.server.ts using AI analysis
+ * of the SQL template during the Save() operation.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QueryDatabaseWriter = void 0;
+const core_1 = require("@memberjunction/core");
+const error_handlers_1 = require("../utils/error-handlers");
+const query_helpers_1 = require("../utils/query-helpers");
+/**
+ * QueryDatabaseWriter class
+ * Writes validated queries directly to the database
+ */
+class QueryDatabaseWriter {
+    categoryCache = new Map();
+    constructor() {
+        // No config needed - category info comes from ValidatedQuery
+    }
+    /**
+     * Write validated queries to the database
+     *
+     * Creates Query entities. QueryFields and QueryParameters are automatically
+     * extracted by QueryEntity.server.ts using AI analysis of the SQL template.
+     * This happens asynchronously during the Save() operation.
+     *
+     * Errors for individual queries are logged but don't stop the batch process.
+     *
+     * @param validatedQueries - Array of validated queries to write
+     * @param contextUser - User context for entity operations
+     * @returns Write result with success status and per-query results
+     */
+    async writeQueriesToDatabase(validatedQueries, contextUser) {
+        const md = new core_1.Metadata();
+        const results = [];
+        // Process each query
+        for (const vq of validatedQueries) {
+            try {
+                // Get or create the category for this query (with caching)
+                const categoryId = await this.getCategoryId(vq.category, contextUser);
+                // Create Query entity ONLY (NO manual fields/params creation)
+                // QueryEntity.server.ts will automatically:
+                // - Detect Nunjucks syntax
+                // - Extract parameters using AI
+                // - Create QueryParameter records
+                // - Create QueryField records
+                // - Set UsesTemplate flag
+                const query = await md.GetEntityObject('Queries', contextUser);
+                query.NewRecord();
+                query.Name = (0, query_helpers_1.generateQueryName)(vq.businessQuestion);
+                query.CategoryID = categoryId;
+                query.UserQuestion = vq.businessQuestion.userQuestion;
+                query.Description = vq.businessQuestion.description;
+                query.TechnicalDescription = vq.businessQuestion.technicalDescription;
+                query.SQL = vq.query.sql;
+                query.OriginalSQL = vq.query.sql;
+                query.Status = 'Pending';
+                const saved = await query.Save();
+                if (!saved) {
+                    throw new Error(`Failed to save query: ${query.LatestResult?.Message}`);
+                }
+                results.push(`✓ ${query.Name} (ID: ${query.ID}) - AI extraction queued`);
+            }
+            catch (error) {
+                results.push(`✗ ${vq.businessQuestion.userQuestion}: ${(0, error_handlers_1.extractErrorMessage)(error, 'Database Write')}`);
+            }
+        }
+        return {
+            success: true,
+            results
+        };
+    }
+    /**
+     * Get or create category ID from QueryCategoryInfo
+     * Uses caching to avoid repeated lookups/creations
+     * Ensures parent categories exist before creating children
+     */
+    async getCategoryId(category, contextUser) {
+        // Check cache first (by path for uniqueness)
+        if (this.categoryCache.has(category.path)) {
+            return this.categoryCache.get(category.path);
+        }
+        // If this category has a parent, ensure parent exists first
+        let parentId = null;
+        if (category.parentName) {
+            // Create parent category info (it's a root category)
+            const parentCategory = {
+                name: category.parentName,
+                parentName: null,
+                description: 'Automatically generated queries from query-gen tool',
+                path: category.parentName
+            };
+            parentId = await this.getCategoryId(parentCategory, contextUser);
+        }
+        // Find or create this category
+        const categoryId = await this.findOrCreateCategory(category.name, parentId, category.description, contextUser);
+        // Cache for future use
+        this.categoryCache.set(category.path, categoryId);
+        return categoryId;
+    }
+    /**
+     * Find or create a query category
+     *
+     * Searches for an existing category with the given name and parent, or creates it if not found.
+     *
+     * @param categoryName - Name of the category to find or create
+     * @param parentCategoryId - Parent category ID (null for root categories)
+     * @param description - Description for new categories
+     * @param contextUser - User context for entity operations
+     * @returns Category ID
+     */
+    async findOrCreateCategory(categoryName, parentCategoryId, description, contextUser) {
+        const rv = new core_1.RunView();
+        // Build filter to match both name and parent
+        let filter = `Name='${categoryName.replace(/'/g, "''")}'`;
+        if (parentCategoryId) {
+            filter += ` AND ParentID='${parentCategoryId}'`;
+        }
+        else {
+            filter += ` AND ParentID IS NULL`;
+        }
+        const result = await rv.RunView({
+            EntityName: 'Query Categories',
+            ExtraFilter: filter,
+            ResultType: 'entity_object'
+        }, contextUser);
+        if (!result.Success) {
+            throw new Error(`Failed to search for category: ${result.ErrorMessage}`);
+        }
+        // If found, return existing category ID
+        if (result.Results && result.Results.length > 0) {
+            return result.Results[0].ID;
+        }
+        // Category doesn't exist, create it
+        const md = new core_1.Metadata();
+        const category = await md.GetEntityObject('Query Categories', contextUser);
+        category.NewRecord();
+        category.Name = categoryName;
+        category.ParentID = parentCategoryId;
+        category.Description = description;
+        const saved = await category.Save();
+        if (!saved) {
+            throw new Error(`Failed to create category: ${category.LatestResult?.Message}`);
+        }
+        return category.ID;
+    }
+}
+exports.QueryDatabaseWriter = QueryDatabaseWriter;
+//# sourceMappingURL=QueryDatabaseWriter.js.map

package/dist/core/QueryDatabaseWriter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryDatabaseWriter.js","sourceRoot":"","sources":["../../src/core/QueryDatabaseWriter.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;AAEH,+CAAmE;AAMnE,4DAA8D;AAC9D,0DAA2D;AAE3D;;;GAGG;AACH,MAAa,mBAAmB;IACtB,aAAa,GAAwB,IAAI,GAAG,EAAE,CAAC;IAEvD;QACE,6DAA6D;IAC/D,CAAC;IACD;;;;;;;;;;;;OAYG;IACH,KAAK,CAAC,sBAAsB,CAC1B,gBAAkC,EAClC,WAAqB;QAErB,MAAM,EAAE,GAAG,IAAI,eAAQ,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAa,EAAE,CAAC;QAE7B,qBAAqB;QACrB,KAAK,MAAM,EAAE,IAAI,gBAAgB,EAAE,CAAC;YAClC,IAAI,CAAC;gBACH,2DAA2D;gBAC3D,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;gBAEtE,8DAA8D;gBAC9D,4CAA4C;gBAC5C,2BAA2B;gBAC3B,gCAAgC;gBAChC,kCAAkC;gBAClC,8BAA8B;gBAC9B,0BAA0B;gBAC1B,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,eAAe,CAAc,SAAS,EAAE,WAAW,CAAC,CAAC;gBAC5E,KAAK,CAAC,SAAS,EAAE,CAAC;gBAClB,KAAK,CAAC,IAAI,GAAG,IAAA,iCAAiB,EAAC,EAAE,CAAC,gBAAgB,CAAC,CAAC;gBACpD,KAAK,CAAC,UAAU,GAAG,UAAU,CAAC;gBAC9B,KAAK,CAAC,YAAY,GAAG,EAAE,CAAC,gBAAgB,CAAC,YAAY,CAAC;gBACtD,KAAK,CAAC,WAAW,GAAG,EAAE,CAAC,gBAAgB,CAAC,WAAW,CAAC;gBACpD,KAAK,CAAC,oBAAoB,GAAG,EAAE,CAAC,gBAAgB,CAAC,oBAAoB,CAAC;gBACtE,KAAK,CAAC,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC;gBACzB,KAAK,CAAC,WAAW,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC;gBACjC,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC;gBAEzB,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;gBACjC,IAAI,CAAC,KAAK,EAAE,CAAC;oBACX,MAAM,IAAI,KAAK,CAAC,yBAAyB,KAAK,CAAC,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC1E,CAAC;gBAED,OAAO,CAAC,IAAI,CAAC,KAAK,KAAK,CAAC,IAAI,SAAS,KAAK,CAAC,EAAE,0BAA0B,CAAC,CAAC;YAE3E,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACxB,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,gBAAgB,CAAC,YAAY,KAAK,IAAA,oCAAmB,EAAC,KAAK,EAAE,gBAAgB,CAAC,EAAE,CAAC,CAAC;YACzG,CAAC;QACH,CAAC;QAED,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO;SACR,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACK,KAAK,CAAC,aAAa,CACzB,QAAwF,EACxF,WAAqB;QAErB,6CAA6C;QAC7C,IAAI,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YAC1C,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAE,CAAC;QAChD,CAAC;QAED,4DAA4D;QAC5D,IAAI,QAAQ,GAAkB,IAAI,CAAC;QACnC,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;YACxB,qDAAqD;YACrD,MAAM,cAAc,GAAG;gBACrB,IAAI,EAAE,QAAQ,CAAC,UAAU;gBACzB,UAAU,EAAE,IAAI;gBAChB,WAAW,EAAE,qDAAqD;gBAClE,IAAI,EAAE,QAAQ,CAAC,UAAU;aAC1B,CAAC;YACF,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;QACnE,CAAC;QAED,+BAA+B;QAC/B,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAChD,QAAQ,CAAC,IAAI,EACb,QAAQ,EACR,QAAQ,CAAC,WAAW,EACpB,WAAW,CACZ,CAAC;QAEF,uBAAuB;QACvB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAClD,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;;;;;;;;;OAUG;IACK,KAAK,CAAC,oBAAoB,CAChC,YAAoB,EACpB,gBAA+B,EAC/B,WAAmB,EACnB,WAAqB;QAErB,MAAM,EAAE,GAAG,IAAI,cAAO,EAAE,CAAC;QAEzB,6CAA6C;QAC7C,IAAI,MAAM,GAAG,SAAS,YAAY,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC;QAC1D,IAAI,gBAAgB,EAAE,CAAC;YACrB,MAAM,IAAI,kBAAkB,gBAAgB,GAAG,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,uBAAuB,CAAC;QACpC,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,OAAO,CAAsB;YACnD,UAAU,EAAE,kBAAkB;YAC9B,WAAW,EAAE,MAAM;YACnB,UAAU,EAAE,eAAe;SAC5B,EAAE,WAAW,CAAC,CAAC;QAEhB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,kCAAkC,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;QAC3E,CAAC;QAED,wCAAwC;QACxC,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChD,OAAO,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9B,CAAC;QAED,oCAAoC;QACpC,MAAM,EAAE,GAAG,IAAI,eAAQ,EAAE,CAAC;QAC1B,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,eAAe,CAAsB,kBAAkB,EAAE,WAAW,CAAC,CAAC;QAChG,QAAQ,CAAC,SAAS,EAAE,CAAC;QACrB,QAAQ,CAAC,IAAI,GAAG,YAAY,CAAC;QAC7B,QAAQ,CAAC,QAAQ,GAAG,gBAAgB,CAAC;QACrC,QAAQ,CAAC,WAAW,GAAG,WAAW,CAAC;QAEnC,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACpC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,8BAA8B,QAAQ,CAAC,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC;QAClF,CAAC;QAED,OAAO,QAAQ,CAAC,EAAE,CAAC;IACrB,CAAC;CACF;AArKD,kDAqKC"}

package/dist/core/QueryFixer.d.ts

@@ -0,0 +1,48 @@
+/**
+ * QueryFixer - Fixes SQL queries that fail execution
+ *
+ * Uses the SQL Query Fixer AI prompt to analyze errors and generate
+ * corrected SQL queries with updated metadata.
+ */
+import { UserInfo } from '@memberjunction/core';
+import { GeneratedQuery, EntityMetadataForPrompt, BusinessQuestion } from '../data/schema';
+import { QueryGenConfig } from '../cli/config';
+/**
+ * QueryFixer class
+ * Fixes SQL queries that fail to execute
+ */
+export declare class QueryFixer {
+    private contextUser;
+    private config;
+    constructor(contextUser: UserInfo, config: QueryGenConfig);
+    /**
+     * Fix a SQL query that failed to execute
+     * Uses AI to analyze the error and generate a corrected query
+     *
+     * @param query - The query that failed
+     * @param errorMessage - The error message from execution
+     * @param entityMetadata - Entity metadata for context
+     * @param businessQuestion - Original business question for context
+     * @returns Corrected SQL query with updated metadata
+     */
+    fixQuery(query: GeneratedQuery, errorMessage: string, entityMetadata: EntityMetadataForPrompt[], businessQuestion: BusinessQuestion): Promise<GeneratedQuery>;
+    /**
+     * Find prompt by name in AIEngine cache
+     * Throws if prompt not found
+     */
+    private findPromptByName;
+    /**
+     * Execute the SQL Query Fixer AI prompt
+     * Parses JSON response and validates structure
+     */
+    private executePrompt;
+    /**
+     * Validate fixed query structure
+     * Ensures query has proper metadata
+     *
+     * @param query - Fixed query to validate
+     * @throws Error if query structure is invalid
+     */
+    private validateFixedQuery;
+}
+//# sourceMappingURL=QueryFixer.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"QueryFixer.d.ts","sourceRoot":"","sources":["../../src/core/QueryFixer.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH,OAAO,EAAE,QAAQ,EAAa,MAAM,sBAAsB,CAAC;AAE3D,OAAO,EACL,cAAc,EACd,uBAAuB,EACvB,gBAAgB,EACjB,MAAM,gBAAgB,CAAC;AAExB,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAG/C;;;GAGG;AACH,qBAAa,UAAU;IAEnB,OAAO,CAAC,WAAW;IACnB,OAAO,CAAC,MAAM;gBADN,WAAW,EAAE,QAAQ,EACrB,MAAM,EAAE,cAAc;IAGhC;;;;;;;;;OASG;IACG,QAAQ,CACZ,KAAK,EAAE,cAAc,EACrB,YAAY,EAAE,MAAM,EACpB,cAAc,EAAE,uBAAuB,EAAE,EACzC,gBAAgB,EAAE,gBAAgB,GACjC,OAAO,CAAC,cAAc,CAAC;IA+B1B;;;OAGG;IACH,OAAO,CAAC,gBAAgB;IAWxB;;;OAGG;YACW,aAAa;IAuC3B;;;;;;OAMG;IACH,OAAO,CAAC,kBAAkB;CAW3B"}

package/dist/core/QueryFixer.js

@@ -0,0 +1,115 @@
+"use strict";
+/**
+ * QueryFixer - Fixes SQL queries that fail execution
+ *
+ * Uses the SQL Query Fixer AI prompt to analyze errors and generate
+ * corrected SQL queries with updated metadata.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QueryFixer = 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");
+/**
+ * QueryFixer class
+ * Fixes SQL queries that fail to execute
+ */
+class QueryFixer {
+    contextUser;
+    config;
+    constructor(contextUser, config) {
+        this.contextUser = contextUser;
+        this.config = config;
+    }
+    /**
+     * Fix a SQL query that failed to execute
+     * Uses AI to analyze the error and generate a corrected query
+     *
+     * @param query - The query that failed
+     * @param errorMessage - The error message from execution
+     * @param entityMetadata - Entity metadata for context
+     * @param businessQuestion - Original business question for context
+     * @returns Corrected SQL query with updated metadata
+     */
+    async fixQuery(query, errorMessage, entityMetadata, businessQuestion) {
+        try {
+            // Ensure AIEngine is configured
+            const aiEngine = aiengine_1.AIEngine.Instance;
+            await aiEngine.Config(false, this.contextUser);
+            // Find the SQL Query Fixer prompt
+            const prompt = this.findPromptByName(aiEngine, PromptNames_1.PROMPT_SQL_QUERY_FIXER);
+            // Prepare prompt data
+            const promptData = {
+                originalSQL: query.sql,
+                errorMessage,
+                parameters: query.parameters,
+                entityMetadata,
+                userQuestion: businessQuestion.userQuestion,
+                description: businessQuestion.description,
+            };
+            // Execute AI prompt to fix the query
+            const fixedQuery = await this.executePrompt(prompt, promptData);
+            // Validate the fixed query structure
+            this.validateFixedQuery(fixedQuery);
+            return fixedQuery;
+        }
+        catch (error) {
+            throw new Error((0, error_handlers_1.extractErrorMessage)(error, 'QueryFixer.fixQuery'));
+        }
+    }
+    /**
+     * 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 the SQL Query Fixer AI prompt
+     * Parses JSON response and validates structure
+     */
+    async executePrompt(prompt, promptData) {
+        // The SQL Query Fixer template returns { newSQL, reasoning }
+        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');
+        }
+        // Log the reasoning
+        if (this.config.verbose && result.result.reasoning) {
+            (0, core_1.LogStatus)(`Query fix reasoning: ${result.result.reasoning}`);
+        }
+        // Return GeneratedQuery format, preserving original parameters
+        return {
+            sql: result.result.newSQL,
+            parameters: promptData.parameters,
+        };
+    }
+    /**
+     * Validate fixed query structure
+     * Ensures query has proper metadata
+     *
+     * @param query - Fixed query to validate
+     * @throws Error if query structure is invalid
+     */
+    validateFixedQuery(query) {
+        // Validate SQL is present
+        if (!query.sql || query.sql.trim().length === 0) {
+            throw new Error('Fixed query has empty SQL');
+        }
+        // Validate parameters array
+        if (!Array.isArray(query.parameters)) {
+            throw new Error('Fixed query parameters must be an array');
+        }
+    }
+}
+exports.QueryFixer = QueryFixer;
+//# sourceMappingURL=QueryFixer.js.map

package/dist/core/QueryFixer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryFixer.js","sourceRoot":"","sources":["../../src/core/QueryFixer.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,uDAAoD;AAEpD,+CAA2D;AAC3D,4DAA8D;AAM9D,wDAAgE;AAEhE,4DAAqE;AAErE;;;GAGG;AACH,MAAa,UAAU;IAEX;IACA;IAFV,YACU,WAAqB,EACrB,MAAsB;QADtB,gBAAW,GAAX,WAAW,CAAU;QACrB,WAAM,GAAN,MAAM,CAAgB;IAC7B,CAAC;IAEJ;;;;;;;;;OASG;IACH,KAAK,CAAC,QAAQ,CACZ,KAAqB,EACrB,YAAoB,EACpB,cAAyC,EACzC,gBAAkC;QAElC,IAAI,CAAC;YACH,gCAAgC;YAChC,MAAM,QAAQ,GAAG,mBAAQ,CAAC,QAAQ,CAAC;YACnC,MAAM,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;YAE/C,kCAAkC;YAClC,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,oCAAsB,CAAC,CAAC;YAEvE,sBAAsB;YACtB,MAAM,UAAU,GAAG;gBACjB,WAAW,EAAE,KAAK,CAAC,GAAG;gBACtB,YAAY;gBACZ,UAAU,EAAE,KAAK,CAAC,UAAU;gBAC5B,cAAc;gBACd,YAAY,EAAE,gBAAgB,CAAC,YAAY;gBAC3C,WAAW,EAAE,gBAAgB,CAAC,WAAW;aAC1C,CAAC;YAEF,qCAAqC;YACrC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YAEhE,qCAAqC;YACrC,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAEpC,OAAO,UAAU,CAAC;QACpB,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,IAAA,oCAAmB,EAAC,KAAK,EAAE,qBAAqB,CAAC,CAAC,CAAC;QACrE,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,UAOC;QAED,6DAA6D;QAC7D,MAAM,MAAM,GAAG,MAAM,IAAA,2CAA0B,EAG5C,MAAM,EAAE,UAAU,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAEtD,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,oBAAoB;QACpB,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YACnD,IAAA,gBAAS,EAAC,wBAAwB,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;QAC/D,CAAC;QAED,+DAA+D;QAC/D,OAAO;YACL,GAAG,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;YACzB,UAAU,EAAE,UAAU,CAAC,UAAU;SAClC,CAAC;IACJ,CAAC;IAED;;;;;;OAMG;IACK,kBAAkB,CAAC,KAAqB;QAC9C,0BAA0B;QAC1B,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChD,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC/C,CAAC;QAED,4BAA4B;QAC5B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;YACrC,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;CACF;AAhID,gCAgIC"}

package/dist/core/QueryRefiner.d.ts

@@ -0,0 +1,94 @@
+/**
+ * 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.
+ */
+import { UserInfo } from '@memberjunction/core';
+import { GeneratedQuery, BusinessQuestion, EntityMetadataForPrompt, RefinedQuery } from '../data/schema';
+import { QueryTester } from './QueryTester';
+import { QueryGenConfig } from '../cli/config';
+/**
+ * QueryRefiner class
+ * Iteratively refines queries based on evaluation feedback
+ */
+export declare class QueryRefiner {
+    private tester;
+    private contextUser;
+    private config;
+    private entityMetadata;
+    constructor(tester: QueryTester, contextUser: UserInfo, config: QueryGenConfig);
+    /**
+     * 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
+     */
+    refineQuery(query: GeneratedQuery, businessQuestion: BusinessQuestion, entityMetadata: EntityMetadataForPrompt[], maxRefinements?: number): Promise<RefinedQuery>;
+    /**
+     * Configure AIEngine for prompt execution
+     * Ensures engine is ready before running prompts
+     */
+    private configureAIEngine;
+    /**
+     * Test current query using QueryTester
+     * Throws if query testing fails
+     */
+    private testCurrentQuery;
+    /**
+     * Determine if refinement should stop based on evaluation
+     * Stops if query answers question and doesn't need refinement
+     */
+    private shouldStopRefining;
+    /**
+     * Build success result when refinement loop completes successfully
+     */
+    private buildSuccessResult;
+    /**
+     * Build final result when max refinements reached
+     * Re-tests and re-evaluates final query
+     */
+    private buildFinalResult;
+    /**
+     * 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
+     */
+    private 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
+     */
+    private performRefinement;
+    /**
+     * Find prompt by name in AIEngine cache
+     * Throws if prompt not found
+     */
+    private findPromptByName;
+    /**
+     * Execute AI prompt and parse result
+     * Generic method for any prompt type
+     */
+    private executePrompt;
+    /**
+     * Log evaluation results for debugging
+     */
+    private logEvaluation;
+}
+//# sourceMappingURL=QueryRefiner.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"QueryRefiner.d.ts","sourceRoot":"","sources":["../../src/core/QueryRefiner.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH,OAAO,EAAE,QAAQ,EAAa,MAAM,sBAAsB,CAAC;AAE3D,OAAO,EACL,cAAc,EACd,gBAAgB,EAChB,uBAAuB,EACvB,YAAY,EAGb,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAG/C;;;GAGG;AACH,qBAAa,YAAY;IAIrB,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,WAAW;IACnB,OAAO,CAAC,MAAM;IALhB,OAAO,CAAC,cAAc,CAAiC;gBAG7C,MAAM,EAAE,WAAW,EACnB,WAAW,EAAE,QAAQ,EACrB,MAAM,EAAE,cAAc;IAGhC;;;;;;;;;;;OAWG;IACG,WAAW,CACf,KAAK,EAAE,cAAc,EACrB,gBAAgB,EAAE,gBAAgB,EAClC,cAAc,EAAE,uBAAuB,EAAE,EACzC,cAAc,GAAE,MAAU,GACzB,OAAO,CAAC,YAAY,CAAC;IA+FxB;;;OAGG;YACW,iBAAiB;IAS/B;;;OAGG;YACW,gBAAgB;IAY9B;;;OAGG;IACH,OAAO,CAAC,kBAAkB;IAI1B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAc1B;;;OAGG;YACW,gBAAgB;IAqB9B;;;;;;;;OAQG;YACW,aAAa;IAkC3B;;;;;;;;;OASG;YACW,iBAAiB;IAqC/B;;;OAGG;IACH,OAAO,CAAC,gBAAgB;IAWxB;;;OAGG;YACW,aAAa;IAwB3B;;OAEG;IACH,OAAO,CAAC,aAAa;CAiBtB"}