@memberjunction/ai-agents 5.2.0 → 5.3.1

package/dist/index.d.ts

@@ -21,5 +21,6 @@ export * from './types/payload-operations.js';
 export * from './AgentDataPreloader.js';
 export * from './agent-context-injector.js';
 export * from './memory-manager-agent.js';
+export * from './query-builder-agent.js';
 export { RerankerService, RerankerConfiguration, parseRerankerConfiguration, RerankServiceResult, RerankObservabilityOptions, LLMReranker } from '@memberjunction/ai-reranker';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,cAAc,+BAA+B,CAAC;AAC9C,cAAc,wCAAwC,CAAC;AACvD,cAAc,wCAAwC,CAAC;AACvD,cAAc,cAAc,CAAC;AAC7B,cAAc,eAAe,CAAC;AAC9B,cAAc,eAAe,CAAC;AAC9B,cAAc,kBAAkB,CAAC;AACjC,cAAc,yBAAyB,CAAC;AACxC,cAAc,0BAA0B,CAAC;AACzC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,sBAAsB,CAAC;AACrC,cAAc,0BAA0B,CAAC;AACzC,cAAc,wBAAwB,CAAC;AAGvC,OAAO,EACH,eAAe,EACf,qBAAqB,EACrB,0BAA0B,EAC1B,mBAAmB,EACnB,0BAA0B,EAC1B,WAAW,EACd,MAAM,6BAA6B,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,cAAc,+BAA+B,CAAC;AAC9C,cAAc,wCAAwC,CAAC;AACvD,cAAc,wCAAwC,CAAC;AACvD,cAAc,cAAc,CAAC;AAC7B,cAAc,eAAe,CAAC;AAC9B,cAAc,eAAe,CAAC;AAC9B,cAAc,kBAAkB,CAAC;AACjC,cAAc,yBAAyB,CAAC;AACxC,cAAc,0BAA0B,CAAC;AACzC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,sBAAsB,CAAC;AACrC,cAAc,0BAA0B,CAAC;AACzC,cAAc,wBAAwB,CAAC;AACvC,cAAc,uBAAuB,CAAC;AAGtC,OAAO,EACH,eAAe,EACf,qBAAqB,EACrB,0BAA0B,EAC1B,mBAAmB,EACnB,0BAA0B,EAC1B,WAAW,EACd,MAAM,6BAA6B,CAAC"}

package/dist/index.js

@@ -21,6 +21,7 @@ export * from './types/payload-operations.js';
 export * from './AgentDataPreloader.js';
 export * from './agent-context-injector.js';
 export * from './memory-manager-agent.js';
+export * from './query-builder-agent.js';
 // Re-export from ai-reranker for backward compatibility
 export { RerankerService, parseRerankerConfiguration, LLMReranker } from '@memberjunction/ai-reranker';
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,cAAc,+BAA+B,CAAC;AAC9C,cAAc,wCAAwC,CAAC;AACvD,cAAc,wCAAwC,CAAC;AACvD,cAAc,cAAc,CAAC;AAC7B,cAAc,eAAe,CAAC;AAC9B,cAAc,eAAe,CAAC;AAC9B,cAAc,kBAAkB,CAAC;AACjC,cAAc,yBAAyB,CAAC;AACxC,cAAc,0BAA0B,CAAC;AACzC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,sBAAsB,CAAC;AACrC,cAAc,0BAA0B,CAAC;AACzC,cAAc,wBAAwB,CAAC;AAEvC,wDAAwD;AACxD,OAAO,EACH,eAAe,EAEf,0BAA0B,EAG1B,WAAW,EACd,MAAM,6BAA6B,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,cAAc,+BAA+B,CAAC;AAC9C,cAAc,wCAAwC,CAAC;AACvD,cAAc,wCAAwC,CAAC;AACvD,cAAc,cAAc,CAAC;AAC7B,cAAc,eAAe,CAAC;AAC9B,cAAc,eAAe,CAAC;AAC9B,cAAc,kBAAkB,CAAC;AACjC,cAAc,yBAAyB,CAAC;AACxC,cAAc,0BAA0B,CAAC;AACzC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,sBAAsB,CAAC;AACrC,cAAc,0BAA0B,CAAC;AACzC,cAAc,wBAAwB,CAAC;AACvC,cAAc,uBAAuB,CAAC;AAEtC,wDAAwD;AACxD,OAAO,EACH,eAAe,EAEf,0BAA0B,EAG1B,WAAW,EACd,MAAM,6BAA6B,CAAC"}

package/dist/memory-manager-agent.js

@@ -242,13 +242,13 @@ let MemoryManagerAgent = class MemoryManagerAgent extends BaseAgent {
             ID IN (
                 SELECT DISTINCT ar.ID
                 FROM __mj.AIAgentRun ar
-                INNER JOIN __mj.Conversation c ON ar.ConversationID = c.ID
-                INNER JOIN __mj.ConversationDetail cd ON cd.ConversationID = c.ID
-                INNER JOIN __mj.ConversationDetailArtifact cda ON cda.ConversationDetailID = cd.ID
-                INNER JOIN __mj.ArtifactVersion av ON av.ID = cda.ArtifactVersionID
+                INNER JOIN __mj.vwConversations c ON ar.ConversationID = c.ID
+                INNER JOIN __mj.vwConversationDetails cd ON cd.ConversationID = c.ID
+                INNER JOIN __mj.vwConversationDetailArtifacts cda ON cda.ConversationDetailID = cd.ID
+                INNER JOIN __mj.vwArtifactVersions av ON av.ID = cda.ArtifactVersionID
                 WHERE EXISTS (
                     SELECT 1
-                    FROM __mj.ArtifactUse au
+                    FROM __mj.vwArtifactUses au
                     WHERE au.ArtifactVersionID = av.ID
                     ${sinceFilter}
                     GROUP BY au.ArtifactVersionID

package/dist/query-builder-agent.d.ts

@@ -0,0 +1,27 @@
+import { BaseAgent } from './base-agent.js';
+import { ExecuteAgentParams, BaseAgentNextStep, MJAIAgentRunEntityExtended, MJAIAgentRunStepEntityExtended } from '@memberjunction/ai-core-plus';
+/**
+ * Query Builder Agent - Orchestrator for building data queries.
+ *
+ * Extends BaseAgent to add post-processing when the agent completes:
+ *  - Formats SQL in `metadata.sql` using sql-formatter for consistent,
+ *    readable output regardless of how the LLM emitted it.
+ */
+export declare class QueryBuilderAgent extends BaseAgent {
+    /**
+     * After the base validation passes, format the SQL in the payload
+     * so it is always stored with proper indentation and line breaks.
+     */
+    protected validateSuccessNextStep<P>(params: ExecuteAgentParams, nextStep: BaseAgentNextStep<P>, currentPayload: P, agentRun: MJAIAgentRunEntityExtended, currentStep: MJAIAgentRunStepEntityExtended): Promise<BaseAgentNextStep<P>>;
+    /**
+     * Applies payload change request (if any) to produce the actual payload
+     * that will be persisted.
+     */
+    private resolveEffectivePayload;
+    /**
+     * Looks for `metadata.sql` in the payload and formats it with sql-formatter.
+     * Returns a new payload with the formatted SQL, or null if no formatting was needed.
+     */
+    private formatPayloadSql;
+}
+//# sourceMappingURL=query-builder-agent.d.ts.map

package/dist/query-builder-agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-builder-agent.d.ts","sourceRoot":"","sources":["../src/query-builder-agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAEzC,OAAO,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,0BAA0B,EAAE,8BAA8B,EAAE,MAAM,8BAA8B,CAAC;AAIjJ;;;;;;GAMG;AACH,qBACa,iBAAkB,SAAQ,SAAS;IAE5C;;;OAGG;cACsB,uBAAuB,CAAC,CAAC,EAC9C,MAAM,EAAE,kBAAkB,EAC1B,QAAQ,EAAE,iBAAiB,CAAC,CAAC,CAAC,EAC9B,cAAc,EAAE,CAAC,EACjB,QAAQ,EAAE,0BAA0B,EACpC,WAAW,EAAE,8BAA8B,GAC5C,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC;IAsBhC;;;OAGG;IACH,OAAO,CAAC,uBAAuB;IAY/B;;;OAGG;IACH,OAAO,CAAC,gBAAgB;CAmC3B"}

package/dist/query-builder-agent.js

@@ -0,0 +1,97 @@
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+import { BaseAgent } from './base-agent.js';
+import { PayloadManager } from './PayloadManager.js';
+import { RegisterClass } from '@memberjunction/global';
+import { format as formatSQL } from 'sql-formatter';
+/**
+ * Query Builder Agent - Orchestrator for building data queries.
+ *
+ * Extends BaseAgent to add post-processing when the agent completes:
+ *  - Formats SQL in `metadata.sql` using sql-formatter for consistent,
+ *    readable output regardless of how the LLM emitted it.
+ */
+let QueryBuilderAgent = class QueryBuilderAgent extends BaseAgent {
+    /**
+     * After the base validation passes, format the SQL in the payload
+     * so it is always stored with proper indentation and line breaks.
+     */
+    async validateSuccessNextStep(params, nextStep, currentPayload, agentRun, currentStep) {
+        // Run base validation first (min-execution checks, payload schema, etc.)
+        const baseResult = await super.validateSuccessNextStep(params, nextStep, currentPayload, agentRun, currentStep);
+        if (baseResult.step === 'Retry') {
+            return baseResult;
+        }
+        // Resolve the effective payload: apply any pending change request
+        const effectivePayload = this.resolveEffectivePayload(currentPayload, baseResult);
+        // Format SQL if present
+        const formatted = this.formatPayloadSql(effectivePayload);
+        if (formatted) {
+            return {
+                ...baseResult,
+                newPayload: formatted
+            };
+        }
+        return baseResult;
+    }
+    /**
+     * Applies payload change request (if any) to produce the actual payload
+     * that will be persisted.
+     */
+    resolveEffectivePayload(currentPayload, nextStep) {
+        if (nextStep.payloadChangeRequest) {
+            const pm = new PayloadManager();
+            const result = pm.applyAgentChangeRequest(currentPayload, nextStep.payloadChangeRequest);
+            return result.result || currentPayload;
+        }
+        if (nextStep.newPayload) {
+            return nextStep.newPayload;
+        }
+        return currentPayload;
+    }
+    /**
+     * Looks for `metadata.sql` in the payload and formats it with sql-formatter.
+     * Returns a new payload with the formatted SQL, or null if no formatting was needed.
+     */
+    formatPayloadSql(payload) {
+        const p = payload;
+        const metadata = p?.metadata;
+        const rawSql = metadata?.sql;
+        if (typeof rawSql !== 'string' || rawSql.trim().length === 0) {
+            return null;
+        }
+        try {
+            const formatted = formatSQL(rawSql, {
+                language: 'tsql',
+                tabWidth: 4,
+                keywordCase: 'upper',
+                linesBetweenQueries: 2
+            });
+            // Only update if formatting actually changed something
+            if (formatted === rawSql) {
+                return null;
+            }
+            return {
+                ...p,
+                metadata: {
+                    ...metadata,
+                    sql: formatted
+                }
+            };
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            console.warn(`QueryBuilderAgent: sql-formatter failed, keeping original SQL. Error: ${message}`);
+            return null;
+        }
+    }
+};
+QueryBuilderAgent = __decorate([
+    RegisterClass(BaseAgent, 'QueryBuilderAgent')
+], QueryBuilderAgent);
+export { QueryBuilderAgent };
+//# sourceMappingURL=query-builder-agent.js.map

package/dist/query-builder-agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-builder-agent.js","sourceRoot":"","sources":["../src/query-builder-agent.ts"],"names":[],"mappings":";;;;;;AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAElD,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,EAAE,MAAM,IAAI,SAAS,EAAE,MAAM,eAAe,CAAC;AAEpD;;;;;;GAMG;AAEI,IAAM,iBAAiB,GAAvB,MAAM,iBAAkB,SAAQ,SAAS;IAE5C;;;OAGG;IACgB,KAAK,CAAC,uBAAuB,CAC5C,MAA0B,EAC1B,QAA8B,EAC9B,cAAiB,EACjB,QAAoC,EACpC,WAA2C;QAE3C,yEAAyE;QACzE,MAAM,UAAU,GAAG,MAAM,KAAK,CAAC,uBAAuB,CAAC,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC;QAChH,IAAI,UAAU,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC9B,OAAO,UAAU,CAAC;QACtB,CAAC;QAED,kEAAkE;QAClE,MAAM,gBAAgB,GAAG,IAAI,CAAC,uBAAuB,CAAI,cAAc,EAAE,UAAU,CAAC,CAAC;QAErF,wBAAwB;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAI,gBAAgB,CAAC,CAAC;QAC7D,IAAI,SAAS,EAAE,CAAC;YACZ,OAAO;gBACH,GAAG,UAAU;gBACb,UAAU,EAAE,SAAS;aACxB,CAAC;QACN,CAAC;QAED,OAAO,UAAU,CAAC;IACtB,CAAC;IAED;;;OAGG;IACK,uBAAuB,CAAI,cAAiB,EAAE,QAA8B;QAChF,IAAI,QAAQ,CAAC,oBAAoB,EAAE,CAAC;YAChC,MAAM,EAAE,GAAG,IAAI,cAAc,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,EAAE,CAAC,uBAAuB,CAAI,cAAc,EAAE,QAAQ,CAAC,oBAAoB,CAAC,CAAC;YAC5F,OAAO,MAAM,CAAC,MAAM,IAAI,cAAc,CAAC;QAC3C,CAAC;QACD,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;YACtB,OAAO,QAAQ,CAAC,UAAU,CAAC;QAC/B,CAAC;QACD,OAAO,cAAc,CAAC;IAC1B,CAAC;IAED;;;OAGG;IACK,gBAAgB,CAAI,OAAU;QAClC,MAAM,CAAC,GAAG,OAAkC,CAAC;QAC7C,MAAM,QAAQ,GAAG,CAAC,EAAE,QAA+C,CAAC;QACpE,MAAM,MAAM,GAAG,QAAQ,EAAE,GAAG,CAAC;QAE7B,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3D,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,IAAI,CAAC;YACD,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,EAAE;gBAChC,QAAQ,EAAE,MAAM;gBAChB,QAAQ,EAAE,CAAC;gBACX,WAAW,EAAE,OAAO;gBACpB,mBAAmB,EAAE,CAAC;aACzB,CAAC,CAAC;YAEH,uDAAuD;YACvD,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;gBACvB,OAAO,IAAI,CAAC;YAChB,CAAC;YAED,OAAO;gBACH,GAAG,CAAC;gBACJ,QAAQ,EAAE;oBACN,GAAG,QAAQ;oBACX,GAAG,EAAE,SAAS;iBACjB;aACC,CAAC;QACX,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACpB,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjE,OAAO,CAAC,IAAI,CAAC,yEAAyE,OAAO,EAAE,CAAC,CAAC;YACjG,OAAO,IAAI,CAAC;QAChB,CAAC;IACL,CAAC;CACJ,CAAA;AAzFY,iBAAiB;IAD7B,aAAa,CAAC,SAAS,EAAE,mBAAmB,CAAC;GACjC,iBAAiB,CAyF7B"}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@memberjunction/ai-agents",
   "type": "module",
-  "version": "5.2.0",
+  "version": "5.3.1",
   "description": "MemberJunction: AI Agent Execution and Management",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -17,20 +17,21 @@
   "author": "MemberJunction.com",
   "license": "ISC",
   "dependencies": {
-    "@memberjunction/actions": "5.2.0",
-    "@memberjunction/actions-base": "5.2.0",
-    "@memberjunction/ai": "5.2.0",
-    "@memberjunction/ai-core-plus": "5.2.0",
-    "@memberjunction/ai-engine-base": "5.2.0",
-    "@memberjunction/ai-prompts": "5.2.0",
-    "@memberjunction/ai-reranker": "5.2.0",
-    "@memberjunction/aiengine": "5.2.0",
-    "@memberjunction/core": "5.2.0",
-    "@memberjunction/core-entities": "5.2.0",
-    "@memberjunction/global": "5.2.0",
+    "@memberjunction/actions": "5.3.1",
+    "@memberjunction/actions-base": "5.3.1",
+    "@memberjunction/ai": "5.3.1",
+    "@memberjunction/ai-core-plus": "5.3.1",
+    "@memberjunction/ai-engine-base": "5.3.1",
+    "@memberjunction/ai-prompts": "5.3.1",
+    "@memberjunction/ai-reranker": "5.3.1",
+    "@memberjunction/aiengine": "5.3.1",
+    "@memberjunction/core": "5.3.1",
+    "@memberjunction/core-entities": "5.3.1",
+    "@memberjunction/global": "5.3.1",
     "dotenv": "^17.2.4",
     "lodash": "^4.17.23",
-    "rxjs": "^7.8.2"
+    "rxjs": "^7.8.2",
+    "sql-formatter": "^15.7.0"
   },
   "devDependencies": {
     "@types/lodash": "^4.17.23",