@memberjunction/query-gen 0.0.1 → 2.126.1

package/dist/utils/graph-helpers.d.ts

@@ -0,0 +1,51 @@
+/**
+ * Graph visualization and entity metadata formatting utilities
+ */
+import { EntityInfo } from '@memberjunction/core';
+/**
+ * Entity metadata formatted for LLM prompts (concise version)
+ */
+export interface EntityMetadataForPrompt {
+    Name: string;
+    Description: string;
+    SchemaName: string;
+    FieldCount: number;
+    RelatedEntities: Array<{
+        name: string;
+        type: string;
+    }>;
+}
+/**
+ * Generates a simple text-based relationship graph for LLM prompts
+ *
+ * Output format:
+ * ```
+ * Customers: → Orders, → Addresses
+ * Orders: → OrderDetails, → Customers
+ * Products: → OrderDetails, → Categories
+ * ```
+ */
+export declare function generateRelationshipGraph(entities: EntityInfo[]): string;
+/**
+ * Generates a Mermaid diagram for richer visualization
+ *
+ * This can be used in prompts that support Mermaid syntax.
+ * Output format:
+ * ```mermaid
+ * graph LR
+ *   Customers[Customers] --> Orders[Orders]
+ *   Orders[Orders] --> OrderDetails[OrderDetails]
+ * ```
+ */
+export declare function generateMermaidDiagram(entities: EntityInfo[]): string;
+/**
+ * Formats entity metadata for LLM prompt (concise version with key info)
+ *
+ * Extracts only the essential information needed for entity grouping:
+ * - Entity name and description
+ * - Schema name
+ * - Field count (as a proxy for data richness)
+ * - Related entities with relationship types
+ */
+export declare function formatEntitiesForPrompt(entities: EntityInfo[]): EntityMetadataForPrompt[];
+//# sourceMappingURL=graph-helpers.d.ts.map

package/dist/utils/graph-helpers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"graph-helpers.d.ts","sourceRoot":"","sources":["../../src/utils/graph-helpers.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAElD;;GAEG;AACH,MAAM,WAAW,uBAAuB;IACtC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,eAAe,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACxD;AAED;;;;;;;;;GASG;AACH,wBAAgB,yBAAyB,CAAC,QAAQ,EAAE,UAAU,EAAE,GAAG,MAAM,CAiBxE;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,UAAU,EAAE,GAAG,MAAM,CAmBrE;AAED;;;;;;;;GAQG;AACH,wBAAgB,uBAAuB,CAAC,QAAQ,EAAE,UAAU,EAAE,GAAG,uBAAuB,EAAE,CAWzF"}

package/dist/utils/graph-helpers.js

@@ -0,0 +1,82 @@
+"use strict";
+/**
+ * Graph visualization and entity metadata formatting utilities
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.formatEntitiesForPrompt = exports.generateMermaidDiagram = exports.generateRelationshipGraph = void 0;
+/**
+ * Generates a simple text-based relationship graph for LLM prompts
+ *
+ * Output format:
+ * ```
+ * Customers: → Orders, → Addresses
+ * Orders: → OrderDetails, → Customers
+ * Products: → OrderDetails, → Categories
+ * ```
+ */
+function generateRelationshipGraph(entities) {
+    const lines = [];
+    for (const entity of entities) {
+        if (entity.RelatedEntities.length === 0) {
+            lines.push(`${entity.Name}: (no relationships)`);
+            continue;
+        }
+        const relations = entity.RelatedEntities
+            .map(rel => `→ ${rel.RelatedEntity}`)
+            .join(', ');
+        lines.push(`${entity.Name}: ${relations}`);
+    }
+    return lines.join('\n');
+}
+exports.generateRelationshipGraph = generateRelationshipGraph;
+/**
+ * Generates a Mermaid diagram for richer visualization
+ *
+ * This can be used in prompts that support Mermaid syntax.
+ * Output format:
+ * ```mermaid
+ * graph LR
+ *   Customers[Customers] --> Orders[Orders]
+ *   Orders[Orders] --> OrderDetails[OrderDetails]
+ * ```
+ */
+function generateMermaidDiagram(entities) {
+    const lines = ['graph LR'];
+    const processedPairs = new Set();
+    for (const entity of entities) {
+        const safeEntityName = entity.Name.replace(/\s/g, '_');
+        for (const rel of entity.RelatedEntities) {
+            const safeRelatedName = rel.RelatedEntity.replace(/\s/g, '_');
+            const pairKey = [safeEntityName, safeRelatedName].sort().join('|');
+            if (!processedPairs.has(pairKey)) {
+                lines.push(`  ${safeEntityName}[${entity.Name}] --> ${safeRelatedName}[${rel.RelatedEntity}]`);
+                processedPairs.add(pairKey);
+            }
+        }
+    }
+    return lines.join('\n');
+}
+exports.generateMermaidDiagram = generateMermaidDiagram;
+/**
+ * Formats entity metadata for LLM prompt (concise version with key info)
+ *
+ * Extracts only the essential information needed for entity grouping:
+ * - Entity name and description
+ * - Schema name
+ * - Field count (as a proxy for data richness)
+ * - Related entities with relationship types
+ */
+function formatEntitiesForPrompt(entities) {
+    return entities.map(entity => ({
+        Name: entity.Name,
+        Description: entity.Description || 'No description available',
+        SchemaName: entity.SchemaName || 'dbo',
+        FieldCount: entity.Fields.length,
+        RelatedEntities: entity.RelatedEntities.map(rel => ({
+            name: rel.RelatedEntity,
+            type: rel.Type
+        }))
+    }));
+}
+exports.formatEntitiesForPrompt = formatEntitiesForPrompt;
+//# sourceMappingURL=graph-helpers.js.map

package/dist/utils/graph-helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"graph-helpers.js","sourceRoot":"","sources":["../../src/utils/graph-helpers.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAeH;;;;;;;;;GASG;AACH,SAAgB,yBAAyB,CAAC,QAAsB;IAC9D,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;QAC9B,IAAI,MAAM,CAAC,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxC,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,sBAAsB,CAAC,CAAC;YACjD,SAAS;QACX,CAAC;QAED,MAAM,SAAS,GAAG,MAAM,CAAC,eAAe;aACrC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,GAAG,CAAC,aAAa,EAAE,CAAC;aACpC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAjBD,8DAiBC;AAED;;;;;;;;;;GAUG;AACH,SAAgB,sBAAsB,CAAC,QAAsB;IAC3D,MAAM,KAAK,GAAG,CAAC,UAAU,CAAC,CAAC;IAC3B,MAAM,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;IAEzC,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;QAC9B,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAEvD,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;YACzC,MAAM,eAAe,GAAG,GAAG,CAAC,aAAa,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;YAC9D,MAAM,OAAO,GAAG,CAAC,cAAc,EAAE,eAAe,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAEnE,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBACjC,KAAK,CAAC,IAAI,CAAC,KAAK,cAAc,IAAI,MAAM,CAAC,IAAI,SAAS,eAAe,IAAI,GAAG,CAAC,aAAa,GAAG,CAAC,CAAC;gBAC/F,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAnBD,wDAmBC;AAED;;;;;;;;GAQG;AACH,SAAgB,uBAAuB,CAAC,QAAsB;IAC5D,OAAO,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC7B,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,0BAA0B;QAC7D,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,KAAK;QACtC,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;QAChC,eAAe,EAAE,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAClD,IAAI,EAAE,GAAG,CAAC,aAAa;YACvB,IAAI,EAAE,GAAG,CAAC,IAAI;SACf,CAAC,CAAC;KACJ,CAAC,CAAC,CAAC;AACN,CAAC;AAXD,0DAWC"}

package/dist/utils/prompt-helpers.d.ts

@@ -0,0 +1,25 @@
+/**
+ * Prompt execution helpers with model/vendor override support
+ */
+import { UserInfo } from '@memberjunction/core';
+import { AIPromptEntityExtended } from '@memberjunction/core-entities';
+import { QueryGenConfig } from '../cli/config';
+/**
+ * Execute a prompt with optional model/vendor overrides from QueryGenConfig
+ *
+ * Uses AIPromptParams.override parameter to apply runtime model/vendor overrides.
+ * If config specifies modelOverride or vendorOverride, looks up their IDs from
+ * the AIEngine cache and passes them to the prompt execution.
+ *
+ * @param prompt - The AI prompt to execute (from AIEngine.Instance.Prompts)
+ * @param data - Data to pass to the prompt template
+ * @param contextUser - User context for server-side operations
+ * @param config - QueryGen configuration (for model/vendor overrides)
+ * @returns Promise resolving to the prompt result
+ */
+export declare function executePromptWithOverrides<T>(prompt: AIPromptEntityExtended, data: Record<string, unknown>, contextUser: UserInfo, config: QueryGenConfig): Promise<{
+    success: boolean;
+    result?: T;
+    errorMessage?: string;
+}>;
+//# sourceMappingURL=prompt-helpers.d.ts.map

package/dist/utils/prompt-helpers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-helpers.d.ts","sourceRoot":"","sources":["../../src/utils/prompt-helpers.ts"],"names":[],"mappings":"AAAA;;GAEG;AAIH,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AAEvE,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAE/C;;;;;;;;;;;;GAYG;AACH,wBAAsB,0BAA0B,CAAC,CAAC,EAChD,MAAM,EAAE,sBAAsB,EAC9B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,WAAW,EAAE,QAAQ,EACrB,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC;IAAE,OAAO,EAAE,OAAO,CAAC;IAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAAC,YAAY,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAiBlE"}

package/dist/utils/prompt-helpers.js

@@ -0,0 +1,66 @@
+"use strict";
+/**
+ * Prompt execution helpers with model/vendor override support
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.executePromptWithOverrides = void 0;
+const ai_core_plus_1 = require("@memberjunction/ai-core-plus");
+const ai_prompts_1 = require("@memberjunction/ai-prompts");
+const aiengine_1 = require("@memberjunction/aiengine");
+/**
+ * Execute a prompt with optional model/vendor overrides from QueryGenConfig
+ *
+ * Uses AIPromptParams.override parameter to apply runtime model/vendor overrides.
+ * If config specifies modelOverride or vendorOverride, looks up their IDs from
+ * the AIEngine cache and passes them to the prompt execution.
+ *
+ * @param prompt - The AI prompt to execute (from AIEngine.Instance.Prompts)
+ * @param data - Data to pass to the prompt template
+ * @param contextUser - User context for server-side operations
+ * @param config - QueryGen configuration (for model/vendor overrides)
+ * @returns Promise resolving to the prompt result
+ */
+async function executePromptWithOverrides(prompt, data, contextUser, config) {
+    const promptParams = new ai_core_plus_1.AIPromptParams();
+    promptParams.prompt = prompt;
+    promptParams.data = data;
+    promptParams.contextUser = contextUser;
+    promptParams.skipValidation = false;
+    // Apply model/vendor overrides using built-in AIPromptParams.override
+    if (config.modelOverride || config.vendorOverride) {
+        const overrideIds = resolveModelVendorOverrides(config);
+        if (overrideIds.modelId || overrideIds.vendorId) {
+            promptParams.override = overrideIds;
+        }
+    }
+    const runner = new ai_prompts_1.AIPromptRunner();
+    return await runner.ExecutePrompt(promptParams);
+}
+exports.executePromptWithOverrides = executePromptWithOverrides;
+/**
+ * Resolve model/vendor names to IDs for AIPromptParams.override
+ *
+ * Looks up model and vendor by name in the AIEngine cache (already loaded).
+ *
+ * @param config - QueryGen configuration with modelOverride/vendorOverride names
+ * @returns Object with modelId and/or vendorId, or empty object if none found
+ */
+function resolveModelVendorOverrides(config) {
+    const result = {};
+    // Look up model ID from AIEngine cache if modelOverride is set
+    if (config.modelOverride) {
+        const model = aiengine_1.AIEngine.Instance.Models.find(m => m.Name === config.modelOverride);
+        if (model && model.ID) {
+            result.modelId = model.ID;
+        }
+    }
+    // Look up vendor ID from AIEngine cache if vendorOverride is set
+    if (config.vendorOverride) {
+        const vendor = aiengine_1.AIEngine.Instance.Vendors.find(v => v.Name === config.vendorOverride);
+        if (vendor && vendor.ID) {
+            result.vendorId = vendor.ID;
+        }
+    }
+    return result;
+}
+//# sourceMappingURL=prompt-helpers.js.map

package/dist/utils/prompt-helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-helpers.js","sourceRoot":"","sources":["../../src/utils/prompt-helpers.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAEH,+DAA8D;AAC9D,2DAA4D;AAG5D,uDAAoD;AAGpD;;;;;;;;;;;;GAYG;AACI,KAAK,UAAU,0BAA0B,CAC9C,MAA8B,EAC9B,IAA6B,EAC7B,WAAqB,EACrB,MAAsB;IAEtB,MAAM,YAAY,GAAG,IAAI,6BAAc,EAAE,CAAC;IAC1C,YAAY,CAAC,MAAM,GAAG,MAAM,CAAC;IAC7B,YAAY,CAAC,IAAI,GAAG,IAAI,CAAC;IACzB,YAAY,CAAC,WAAW,GAAG,WAAW,CAAC;IACvC,YAAY,CAAC,cAAc,GAAG,KAAK,CAAC;IAEpC,sEAAsE;IACtE,IAAI,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;QAClD,MAAM,WAAW,GAAG,2BAA2B,CAAC,MAAM,CAAC,CAAC;QACxD,IAAI,WAAW,CAAC,OAAO,IAAI,WAAW,CAAC,QAAQ,EAAE,CAAC;YAChD,YAAY,CAAC,QAAQ,GAAG,WAAW,CAAC;QACtC,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,2BAAc,EAAE,CAAC;IACpC,OAAO,MAAM,MAAM,CAAC,aAAa,CAAI,YAAY,CAAC,CAAC;AACrD,CAAC;AAtBD,gEAsBC;AAED;;;;;;;GAOG;AACH,SAAS,2BAA2B,CAClC,MAAsB;IAEtB,MAAM,MAAM,GAA4C,EAAE,CAAC;IAE3D,+DAA+D;IAC/D,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,mBAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,aAAa,CAAC,CAAC;QAClF,IAAI,KAAK,IAAI,KAAK,CAAC,EAAE,EAAE,CAAC;YACtB,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,iEAAiE;IACjE,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;QAC1B,MAAM,MAAM,GAAG,mBAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,cAAc,CAAC,CAAC;QACrF,IAAI,MAAM,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;YACxB,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC,EAAE,CAAC;QAC9B,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/utils/query-helpers.d.ts

@@ -0,0 +1,23 @@
+/**
+ * Query generation utility functions
+ */
+import { BusinessQuestion } from '../data/schema';
+/**
+ * Generate a query name from a business question
+ *
+ * Removes question marks, filters short words, capitalizes each word,
+ * and limits to 5 words for a concise query name.
+ *
+ * @param question - The business question to convert
+ * @returns A formatted query name
+ *
+ * @example
+ * generateQueryName({ userQuestion: "What are the top customers by revenue?" })
+ * // Returns: "What Are Top Customers Revenue"
+ *
+ * @example
+ * generateQueryName({ userQuestion: "Show me all active users" })
+ * // Returns: "Show All Active Users"
+ */
+export declare function generateQueryName(question: BusinessQuestion): string;
+//# sourceMappingURL=query-helpers.d.ts.map

package/dist/utils/query-helpers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-helpers.d.ts","sourceRoot":"","sources":["../../src/utils/query-helpers.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAElD;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,gBAAgB,GAAG,MAAM,CAQpE"}

package/dist/utils/query-helpers.js

@@ -0,0 +1,34 @@
+"use strict";
+/**
+ * Query generation utility functions
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateQueryName = void 0;
+/**
+ * Generate a query name from a business question
+ *
+ * Removes question marks, filters short words, capitalizes each word,
+ * and limits to 5 words for a concise query name.
+ *
+ * @param question - The business question to convert
+ * @returns A formatted query name
+ *
+ * @example
+ * generateQueryName({ userQuestion: "What are the top customers by revenue?" })
+ * // Returns: "What Are Top Customers Revenue"
+ *
+ * @example
+ * generateQueryName({ userQuestion: "Show me all active users" })
+ * // Returns: "Show All Active Users"
+ */
+function generateQueryName(question) {
+    return question.userQuestion
+        .replace(/\?/g, '')
+        .split(' ')
+        .filter(word => word.length > 2)
+        .map(word => word.charAt(0).toUpperCase() + word.slice(1).toLowerCase())
+        .slice(0, 5)
+        .join(' ');
+}
+exports.generateQueryName = generateQueryName;
+//# sourceMappingURL=query-helpers.js.map

package/dist/utils/query-helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-helpers.js","sourceRoot":"","sources":["../../src/utils/query-helpers.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAIH;;;;;;;;;;;;;;;;GAgBG;AACH,SAAgB,iBAAiB,CAAC,QAA0B;IAC1D,OAAO,QAAQ,CAAC,YAAY;SACzB,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC;SAClB,KAAK,CAAC,GAAG,CAAC;SACV,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;SAC/B,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;SACvE,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;SACX,IAAI,CAAC,GAAG,CAAC,CAAC;AACf,CAAC;AARD,8CAQC"}

package/dist/utils/user-helpers.d.ts

@@ -0,0 +1,15 @@
+/**
+ * User helper utilities for QueryGen CLI operations
+ */
+import { UserInfo } from '@memberjunction/core';
+/**
+ * Get the system user from UserCache
+ *
+ * The System user is populated in the UserCache when the database provider is initialized.
+ * This user is used for CLI operations where no specific user context exists.
+ *
+ * @returns The System UserInfo object from the cache
+ * @throws Error if System user is not found in cache or doesn't have Developer role
+ */
+export declare function getSystemUser(): UserInfo;
+//# sourceMappingURL=user-helpers.d.ts.map

package/dist/utils/user-helpers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"user-helpers.d.ts","sourceRoot":"","sources":["../../src/utils/user-helpers.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAGhD;;;;;;;;GAQG;AACH,wBAAgB,aAAa,IAAI,QAAQ,CAsBxC"}

package/dist/utils/user-helpers.js

@@ -0,0 +1,32 @@
+"use strict";
+/**
+ * User helper utilities for QueryGen CLI operations
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getSystemUser = void 0;
+const sqlserver_dataprovider_1 = require("@memberjunction/sqlserver-dataprovider");
+/**
+ * Get the system user from UserCache
+ *
+ * The System user is populated in the UserCache when the database provider is initialized.
+ * This user is used for CLI operations where no specific user context exists.
+ *
+ * @returns The System UserInfo object from the cache
+ * @throws Error if System user is not found in cache or doesn't have Developer role
+ */
+function getSystemUser() {
+    const sysUser = sqlserver_dataprovider_1.UserCache.Instance.UserByName("System", false);
+    if (!sysUser) {
+        throw new Error("System user not found in cache. Ensure the database provider is initialized " +
+            "before running QueryGen commands (e.g., via 'mj querygen' which initializes the provider).");
+    }
+    // Check if the System user has the Developer role
+    const hasDeveloperRole = sysUser.UserRoles && sysUser.UserRoles.some(userRole => userRole.Role.trim().toLowerCase() === 'developer');
+    if (!hasDeveloperRole) {
+        throw new Error("System user does not have the 'Developer' role. " +
+            "The System user must have the Developer role to perform QueryGen operations.");
+    }
+    return sysUser;
+}
+exports.getSystemUser = getSystemUser;
+//# sourceMappingURL=user-helpers.js.map

package/dist/utils/user-helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"user-helpers.js","sourceRoot":"","sources":["../../src/utils/user-helpers.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAGH,mFAAmE;AAEnE;;;;;;;;GAQG;AACH,SAAgB,aAAa;IAC3B,MAAM,OAAO,GAAG,kCAAS,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC/D,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CACb,8EAA8E;YAC9E,4FAA4F,CAC7F,CAAC;IACJ,CAAC;IAED,kDAAkD;IAClD,MAAM,gBAAgB,GAAG,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,IAAI,CAClE,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,KAAK,WAAW,CAC/D,CAAC;IAEF,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CACb,kDAAkD;YAClD,8EAA8E,CAC/E,CAAC;IACJ,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAtBD,sCAsBC"}

package/dist/vectors/EmbeddingService.d.ts

@@ -0,0 +1,58 @@
+/**
+ * EmbeddingService - Generates embeddings for queries and golden queries
+ *
+ * Wraps AIEngine embedding functionality for QueryGen use cases.
+ * Generates embeddings for multiple fields (name, userQuestion, description, technicalDescription).
+ */
+import { QueryEmbeddings, GoldenQuery, EmbeddedGoldenQuery } from '../data/schema';
+/**
+ * Service for generating embeddings using AI Engine
+ */
+export declare class EmbeddingService {
+    private readonly modelName;
+    /**
+     * Create an EmbeddingService with the specified embedding model
+     *
+     * @param modelName - Name of the embedding model to use (default: 'text-embedding-3-small')
+     */
+    constructor(modelName?: string);
+    /**
+     * Embed a single query by generating embeddings for all its fields
+     *
+     * Generates separate embeddings for:
+     * - userQuestion: Natural language question
+     * - description: High-level description
+     * - technicalDescription: Technical implementation details
+     *
+     * @param query - Query to embed (partial structure with text fields)
+     * @returns Embeddings for all fields
+     */
+    embedQuery(query: {
+        userQuestion: string;
+        description: string;
+        technicalDescription: string;
+    }): Promise<QueryEmbeddings>;
+    /**
+     * Embed all golden queries for few-shot learning
+     *
+     * Golden queries are example queries that serve as few-shot learning examples.
+     * This method embeds all fields of all golden queries for similarity search.
+     * Note: name field is excluded from embeddings as it's not available during query generation.
+     *
+     * @param goldenQueries - Array of golden queries to embed
+     * @returns Array of golden queries with their embeddings
+     */
+    embedGoldenQueries(goldenQueries: GoldenQuery[]): Promise<EmbeddedGoldenQuery[]>;
+    /**
+     * Embed multiple queries in batch
+     *
+     * @param queries - Array of queries to embed
+     * @returns Array of embeddings corresponding to input queries
+     */
+    embedQueries(queries: Array<{
+        userQuestion: string;
+        description: string;
+        technicalDescription: string;
+    }>): Promise<QueryEmbeddings[]>;
+}
+//# sourceMappingURL=EmbeddingService.d.ts.map

package/dist/vectors/EmbeddingService.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"EmbeddingService.d.ts","sourceRoot":"","sources":["../../src/vectors/EmbeddingService.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,eAAe,EAAE,WAAW,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAEnF;;GAEG;AACH,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IAEnC;;;;OAIG;gBACS,SAAS,GAAE,MAAiC;IAIxD;;;;;;;;;;OAUG;IACG,UAAU,CAAC,KAAK,EAAE;QACtB,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;QACpB,oBAAoB,EAAE,MAAM,CAAC;KAC9B,GAAG,OAAO,CAAC,eAAe,CAAC;IAiB5B;;;;;;;;;OASG;IACG,kBAAkB,CAAC,aAAa,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAmBtF;;;;;OAKG;IACG,YAAY,CAChB,OAAO,EAAE,KAAK,CAAC;QACb,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;QACpB,oBAAoB,EAAE,MAAM,CAAC;KAC9B,CAAC,GACD,OAAO,CAAC,eAAe,EAAE,CAAC;CAU9B"}

package/dist/vectors/EmbeddingService.js

@@ -0,0 +1,90 @@
+"use strict";
+/**
+ * EmbeddingService - Generates embeddings for queries and golden queries
+ *
+ * Wraps AIEngine embedding functionality for QueryGen use cases.
+ * Generates embeddings for multiple fields (name, userQuestion, description, technicalDescription).
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EmbeddingService = void 0;
+const aiengine_1 = require("@memberjunction/aiengine");
+/**
+ * Service for generating embeddings using AI Engine
+ */
+class EmbeddingService {
+    modelName;
+    /**
+     * Create an EmbeddingService with the specified embedding model
+     *
+     * @param modelName - Name of the embedding model to use (default: 'text-embedding-3-small')
+     */
+    constructor(modelName = 'text-embedding-3-small') {
+        this.modelName = modelName;
+    }
+    /**
+     * Embed a single query by generating embeddings for all its fields
+     *
+     * Generates separate embeddings for:
+     * - userQuestion: Natural language question
+     * - description: High-level description
+     * - technicalDescription: Technical implementation details
+     *
+     * @param query - Query to embed (partial structure with text fields)
+     * @returns Embeddings for all fields
+     */
+    async embedQuery(query) {
+        const aiEngine = aiengine_1.AIEngine.Instance;
+        // Generate embeddings for each field in parallel
+        const [userQuestionResult, descResult, techDescResult] = await Promise.all([
+            aiEngine.EmbedTextLocal(query.userQuestion),
+            aiEngine.EmbedTextLocal(query.description),
+            aiEngine.EmbedTextLocal(query.technicalDescription),
+        ]);
+        return {
+            userQuestion: userQuestionResult.result.vector,
+            description: descResult.result.vector,
+            technicalDescription: techDescResult.result.vector,
+        };
+    }
+    /**
+     * Embed all golden queries for few-shot learning
+     *
+     * Golden queries are example queries that serve as few-shot learning examples.
+     * This method embeds all fields of all golden queries for similarity search.
+     * Note: name field is excluded from embeddings as it's not available during query generation.
+     *
+     * @param goldenQueries - Array of golden queries to embed
+     * @returns Array of golden queries with their embeddings
+     */
+    async embedGoldenQueries(goldenQueries) {
+        const embedded = [];
+        for (const query of goldenQueries) {
+            const embeddings = await this.embedQuery({
+                userQuestion: query.userQuestion,
+                description: query.description,
+                technicalDescription: query.technicalDescription,
+            });
+            embedded.push({
+                query,
+                embeddings,
+            });
+        }
+        return embedded;
+    }
+    /**
+     * Embed multiple queries in batch
+     *
+     * @param queries - Array of queries to embed
+     * @returns Array of embeddings corresponding to input queries
+     */
+    async embedQueries(queries) {
+        const embeddings = [];
+        for (const query of queries) {
+            const embedding = await this.embedQuery(query);
+            embeddings.push(embedding);
+        }
+        return embeddings;
+    }
+}
+exports.EmbeddingService = EmbeddingService;
+//# sourceMappingURL=EmbeddingService.js.map

package/dist/vectors/EmbeddingService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"EmbeddingService.js","sourceRoot":"","sources":["../../src/vectors/EmbeddingService.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,uDAAoD;AAGpD;;GAEG;AACH,MAAa,gBAAgB;IACV,SAAS,CAAS;IAEnC;;;;OAIG;IACH,YAAY,YAAoB,wBAAwB;QACtD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,UAAU,CAAC,KAIhB;QACC,MAAM,QAAQ,GAAG,mBAAQ,CAAC,QAAQ,CAAC;QAEnC,iDAAiD;QACjD,MAAM,CAAC,kBAAkB,EAAE,UAAU,EAAE,cAAc,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACzE,QAAQ,CAAC,cAAc,CAAC,KAAK,CAAC,YAAY,CAAC;YAC3C,QAAQ,CAAC,cAAc,CAAC,KAAK,CAAC,WAAW,CAAC;YAC1C,QAAQ,CAAC,cAAc,CAAC,KAAK,CAAC,oBAAoB,CAAC;SACpD,CAAC,CAAC;QAEH,OAAO;YACL,YAAY,EAAE,kBAAkB,CAAC,MAAM,CAAC,MAAM;YAC9C,WAAW,EAAE,UAAU,CAAC,MAAM,CAAC,MAAM;YACrC,oBAAoB,EAAE,cAAc,CAAC,MAAM,CAAC,MAAM;SACnD,CAAC;IACJ,CAAC;IAED;;;;;;;;;OASG;IACH,KAAK,CAAC,kBAAkB,CAAC,aAA4B;QACnD,MAAM,QAAQ,GAA0B,EAAE,CAAC;QAE3C,KAAK,MAAM,KAAK,IAAI,aAAa,EAAE,CAAC;YAClC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC;gBACvC,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,oBAAoB,EAAE,KAAK,CAAC,oBAAoB;aACjD,CAAC,CAAC;YAEH,QAAQ,CAAC,IAAI,CAAC;gBACZ,KAAK;gBACL,UAAU;aACX,CAAC,CAAC;QACL,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,YAAY,CAChB,OAIE;QAEF,MAAM,UAAU,GAAsB,EAAE,CAAC;QAEzC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAC/C,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC7B,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;CACF;AA/FD,4CA+FC"}

package/dist/vectors/SimilaritySearch.d.ts

@@ -0,0 +1,51 @@
+/**
+ * SimilaritySearch - Finds similar golden queries using weighted cosine similarity
+ *
+ * Implements weighted similarity search across multiple fields (name, userQuestion,
+ * description, technicalDescription) to find the most relevant few-shot examples.
+ */
+import { QueryEmbeddings, EmbeddedGoldenQuery, SimilarQuery } from '../data/schema';
+import { SimpleVectorService } from '@memberjunction/ai-vectors-memory';
+/**
+ * SimilaritySearch class
+ * Finds most similar golden queries using weighted cosine similarity across multiple fields
+ */
+export declare class SimilaritySearch extends SimpleVectorService {
+    /**
+     * Weights for each field in similarity calculation
+     * Total weights sum to 1.0
+     *
+     * Weight distribution prioritizes technical specifications:
+     * - userQuestion: 0.20 (less important - natural language is variable)
+     * - description: 0.40 (high-level business logic matching)
+     * - technicalDescription: 0.40 (technical implementation details)
+     */
+    private readonly weights;
+    /**
+     * Find similar golden queries using weighted cosine similarity
+     *
+     * Calculates similarity for each field separately, then combines with weights.
+     * ALWAYS returns topK results, even if below threshold (threshold is informational).
+     *
+     * @param queryEmbeddings - Embeddings for the user's query (one per field)
+     * @param goldenEmbeddings - Array of golden queries with their embeddings
+     * @param topK - Number of most similar queries to return (default: 5)
+     * @returns Array of top-K most similar golden queries with scores
+     */
+    findSimilarQueries(queryEmbeddings: QueryEmbeddings, goldenEmbeddings: EmbeddedGoldenQuery[], topK?: number): Promise<SimilarQuery[]>;
+    /**
+     * Calculate cosine similarity for each field separately
+     * Note: name field excluded as queries don't have names during generation
+     */
+    private calculateFieldSimilarities;
+    /**
+     * Calculate weighted sum of field similarities
+     * Note: name field excluded, weights adjusted accordingly
+     */
+    private calculateWeightedScore;
+    /**
+     * Select top K results sorted by similarity
+     */
+    private selectTopK;
+}
+//# sourceMappingURL=SimilaritySearch.d.ts.map

package/dist/vectors/SimilaritySearch.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"SimilaritySearch.d.ts","sourceRoot":"","sources":["../../src/vectors/SimilaritySearch.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,eAAe,EAAE,mBAAmB,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACpF,OAAO,EAAE,mBAAmB,EAAE,MAAM,mCAAmC,CAAC;AAYxE;;;GAGG;AACH,qBAAa,gBAAiB,SAAQ,mBAAmB;IACvD;;;;;;;;OAQG;IACH,OAAO,CAAC,QAAQ,CAAC,OAAO,CAItB;IAEF;;;;;;;;;;OAUG;IACG,kBAAkB,CACtB,eAAe,EAAE,eAAe,EAChC,gBAAgB,EAAE,mBAAmB,EAAE,EACvC,IAAI,GAAE,MAAU,GACf,OAAO,CAAC,YAAY,EAAE,CAAC;IAiB1B;;;OAGG;IACH,OAAO,CAAC,0BAA0B;IAWlC;;;OAGG;IACH,OAAO,CAAC,sBAAsB;IAQ9B;;OAEG;IACH,OAAO,CAAC,UAAU;CAKnB"}