@memberjunction/server 2.50.0 → 2.51.0

package/dist/resolvers/RunAIAgentResolver.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"RunAIAgentResolver.d.ts","sourceRoot":"","sources":["../../src/resolvers/RunAIAgentResolver.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAI1C,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAE1D,qBACa,gBAAgB;IAEzB,OAAO,EAAE,OAAO,CAAC;IAGjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAGhB,YAAY,CAAC,EAAE,MAAM,CAAC;IAGtB,KAAK,CAAC,EAAE,MAAM,CAAC;IAGf,eAAe,CAAC,EAAE,MAAM,CAAC;IAGzB,UAAU,CAAC,EAAE,MAAM,CAAC;IAGpB,UAAU,CAAC,EAAE,MAAM,CAAC;IAGpB,SAAS,CAAC,EAAE,MAAM,CAAC;IAGnB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,qBACa,kBAAmB,SAAQ,YAAY;IAE1C,UAAU,CACI,OAAO,EAAE,MAAM,EACxB,EAAE,WAAW,EAAE,EAAE;QAAE,WAAW,EAAE,WAAW,CAAA;KAAE,EACnC,YAAY,EAAE,MAAM,EACJ,IAAI,CAAC,EAAE,MAAM,EACL,YAAY,CAAC,EAAE,MAAM,GAC/D,OAAO,CAAC,gBAAgB,CAAC;CA+H/B"}
+{"version":3,"file":"RunAIAgentResolver.d.ts","sourceRoot":"","sources":["../../src/resolvers/RunAIAgentResolver.ts"],"names":[],"mappings":"AAAA,OAAO,EAA2D,YAAY,EAA8C,MAAM,cAAc,CAAC;AACjJ,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAM1C,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAG1D,qBACa,gBAAgB;IAEzB,OAAO,EAAE,OAAO,CAAC;IAGjB,YAAY,CAAC,EAAE,MAAM,CAAC;IAGtB,eAAe,CAAC,EAAE,MAAM,CAAC;IAGzB,OAAO,EAAE,MAAM,CAAC;CACnB;AAED,qBACa,sBAAsB;IAE/B,WAAW,EAAE,MAAM,CAAC;IAGpB,UAAU,EAAE,MAAM,CAAC;IAGnB,OAAO,EAAE,MAAM,CAAC;IAGhB,SAAS,CAAC,EAAE,MAAM,CAAC;IAGnB,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,qBACa,qBAAqB;IAE9B,OAAO,EAAE,MAAM,CAAC;IAGhB,SAAS,EAAE,OAAO,CAAC;IAGnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAGlB,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,qBACa,yBAAyB;IAElC,MAAM,EAAE,MAAM,CAAC;IAGf,QAAQ,EAAE,MAAM,CAAC;IAGjB,SAAS,CAAC,EAAE,MAAM,CAAC;IAGnB,SAAS,CAAC,EAAE,MAAM,CAAC;IAGnB,SAAS,EAAE,IAAI,CAAC;IAGhB,OAAO,CAAC,EAAE,IAAI,CAAC;IAGf,MAAM,EAAE,MAAM,CAAC;IAGf,MAAM,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,qBACa,kBAAkB;IAE3B,WAAW,EAAE,MAAM,CAAC;IAGpB,aAAa,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,qBACa,2BAA2B;IAEpC,SAAS,EAAE,MAAM,CAAC;IAGlB,UAAU,EAAE,MAAM,CAAC;IAGnB,IAAI,EAAE,UAAU,GAAG,WAAW,GAAG,gBAAgB,GAAG,UAAU,CAAC;IAG/D,QAAQ,CAAC,EAAE,sBAAsB,CAAC;IAGlC,SAAS,CAAC,EAAE,qBAAqB,CAAC;IAGlC,aAAa,CAAC,EAAE,kBAAkB,CAAC;IAGnC,SAAS,EAAE,IAAI,CAAC;CACnB;AAMD,qBACa,kBAAmB,SAAQ,YAAY;IAKhD,OAAO,CAAC,mBAAmB;IAqC3B,OAAO,CAAC,qBAAqB;IAoC7B,OAAO,CAAC,cAAc;YAaR,aAAa;IAsB3B,OAAO,CAAC,sBAAsB;IAmC9B,OAAO,CAAC,qBAAqB;IAa7B,OAAO,CAAC,sBAAsB;IAe9B,OAAO,CAAC,uBAAuB;IA4BzB,UAAU,CACI,OAAO,EAAE,MAAM,EACxB,EAAE,WAAW,EAAE,EAAE;QAAE,WAAW,EAAE,WAAW,CAAA;KAAE,EACnC,YAAY,EAAE,MAAM,EACnB,SAAS,EAAE,MAAM,EACzB,MAAM,EAAE,YAAY,EACG,IAAI,CAAC,EAAE,MAAM,EACL,YAAY,CAAC,EAAE,MAAM,GAC/D,OAAO,CAAC,gBAAgB,CAAC;IA6F5B,OAAO,CAAC,kBAAkB;CA2C7B"}

package/dist/resolvers/RunAIAgentResolver.js

@@ -10,20 +10,17 @@ var __metadata = (this && this.__metadata) || function (k, v) {
 var __param = (this && this.__param) || function (paramIndex, decorator) {
     return function (target, key) { decorator(target, key, paramIndex); }
 };
-import { Resolver, Mutation, Arg, Ctx, ObjectType, Field } from 'type-graphql';
-import { LogError, LogStatus, Metadata } from '@memberjunction/core';
+import { Resolver, Mutation, Arg, Ctx, ObjectType, Field, PubSub, PubSubEngine, ID } from 'type-graphql';
+import { LogError, LogStatus } from '@memberjunction/core';
 import { AgentRunner } from '@memberjunction/ai-agents';
+import { AIEngine } from '@memberjunction/aiengine';
 import { ResolverBase } from '../generic/ResolverBase.js';
+import { PUSH_STATUS_UPDATES_TOPIC } from '../generic/PushStatusResolver.js';
 let AIAgentRunResult = class AIAgentRunResult {
     success;
-    output;
-    parsedResult;
-    error;
+    errorMessage;
     executionTimeMs;
-    tokensUsed;
-    agentRunId;
-    rawResult;
-    nextStep;
+    payload;
 };
 __decorate([
     Field(),
@@ -32,148 +29,432 @@ __decorate([
 __decorate([
     Field({ nullable: true }),
     __metadata("design:type", String)
-], AIAgentRunResult.prototype, "output", void 0);
+], AIAgentRunResult.prototype, "errorMessage", void 0);
 __decorate([
     Field({ nullable: true }),
+    __metadata("design:type", Number)
+], AIAgentRunResult.prototype, "executionTimeMs", void 0);
+__decorate([
+    Field(),
+    __metadata("design:type", String)
+], AIAgentRunResult.prototype, "payload", void 0);
+AIAgentRunResult = __decorate([
+    ObjectType()
+], AIAgentRunResult);
+export { AIAgentRunResult };
+let AgentExecutionProgress = class AgentExecutionProgress {
+    currentStep;
+    percentage;
+    message;
+    agentName;
+    agentType;
+};
+__decorate([
+    Field(),
     __metadata("design:type", String)
-], AIAgentRunResult.prototype, "parsedResult", void 0);
+], AgentExecutionProgress.prototype, "currentStep", void 0);
+__decorate([
+    Field(),
+    __metadata("design:type", Number)
+], AgentExecutionProgress.prototype, "percentage", void 0);
+__decorate([
+    Field(),
+    __metadata("design:type", String)
+], AgentExecutionProgress.prototype, "message", void 0);
 __decorate([
     Field({ nullable: true }),
     __metadata("design:type", String)
-], AIAgentRunResult.prototype, "error", void 0);
+], AgentExecutionProgress.prototype, "agentName", void 0);
 __decorate([
     Field({ nullable: true }),
-    __metadata("design:type", Number)
-], AIAgentRunResult.prototype, "executionTimeMs", void 0);
+    __metadata("design:type", String)
+], AgentExecutionProgress.prototype, "agentType", void 0);
+AgentExecutionProgress = __decorate([
+    ObjectType()
+], AgentExecutionProgress);
+export { AgentExecutionProgress };
+let AgentStreamingContent = class AgentStreamingContent {
+    content;
+    isPartial;
+    stepName;
+    agentName;
+};
+__decorate([
+    Field(),
+    __metadata("design:type", String)
+], AgentStreamingContent.prototype, "content", void 0);
+__decorate([
+    Field(),
+    __metadata("design:type", Boolean)
+], AgentStreamingContent.prototype, "isPartial", void 0);
 __decorate([
     Field({ nullable: true }),
-    __metadata("design:type", Number)
-], AIAgentRunResult.prototype, "tokensUsed", void 0);
+    __metadata("design:type", String)
+], AgentStreamingContent.prototype, "stepName", void 0);
 __decorate([
     Field({ nullable: true }),
     __metadata("design:type", String)
-], AIAgentRunResult.prototype, "agentRunId", void 0);
+], AgentStreamingContent.prototype, "agentName", void 0);
+AgentStreamingContent = __decorate([
+    ObjectType()
+], AgentStreamingContent);
+export { AgentStreamingContent };
+let AgentExecutionStepSummary = class AgentExecutionStepSummary {
+    stepId;
+    stepName;
+    agentName;
+    agentType;
+    startTime;
+    endTime;
+    status;
+    result;
+};
+__decorate([
+    Field(),
+    __metadata("design:type", String)
+], AgentExecutionStepSummary.prototype, "stepId", void 0);
+__decorate([
+    Field(),
+    __metadata("design:type", String)
+], AgentExecutionStepSummary.prototype, "stepName", void 0);
 __decorate([
     Field({ nullable: true }),
     __metadata("design:type", String)
-], AIAgentRunResult.prototype, "rawResult", void 0);
+], AgentExecutionStepSummary.prototype, "agentName", void 0);
 __decorate([
     Field({ nullable: true }),
     __metadata("design:type", String)
-], AIAgentRunResult.prototype, "nextStep", void 0);
-AIAgentRunResult = __decorate([
+], AgentExecutionStepSummary.prototype, "agentType", void 0);
+__decorate([
+    Field(),
+    __metadata("design:type", Date)
+], AgentExecutionStepSummary.prototype, "startTime", void 0);
+__decorate([
+    Field({ nullable: true }),
+    __metadata("design:type", Date)
+], AgentExecutionStepSummary.prototype, "endTime", void 0);
+__decorate([
+    Field(),
+    __metadata("design:type", String)
+], AgentExecutionStepSummary.prototype, "status", void 0);
+__decorate([
+    Field({ nullable: true }),
+    __metadata("design:type", String)
+], AgentExecutionStepSummary.prototype, "result", void 0);
+AgentExecutionStepSummary = __decorate([
     ObjectType()
-], AIAgentRunResult);
-export { AIAgentRunResult };
+], AgentExecutionStepSummary);
+export { AgentExecutionStepSummary };
+let AgentPartialResult = class AgentPartialResult {
+    currentStep;
+    partialOutput;
+};
+__decorate([
+    Field(),
+    __metadata("design:type", String)
+], AgentPartialResult.prototype, "currentStep", void 0);
+__decorate([
+    Field({ nullable: true }),
+    __metadata("design:type", String)
+], AgentPartialResult.prototype, "partialOutput", void 0);
+AgentPartialResult = __decorate([
+    ObjectType()
+], AgentPartialResult);
+export { AgentPartialResult };
+let AgentExecutionStreamMessage = class AgentExecutionStreamMessage {
+    sessionId;
+    agentRunId;
+    type;
+    progress;
+    streaming;
+    partialResult;
+    timestamp;
+};
+__decorate([
+    Field(() => ID),
+    __metadata("design:type", String)
+], AgentExecutionStreamMessage.prototype, "sessionId", void 0);
+__decorate([
+    Field(() => ID),
+    __metadata("design:type", String)
+], AgentExecutionStreamMessage.prototype, "agentRunId", void 0);
+__decorate([
+    Field(),
+    __metadata("design:type", String)
+], AgentExecutionStreamMessage.prototype, "type", void 0);
+__decorate([
+    Field({ nullable: true }),
+    __metadata("design:type", AgentExecutionProgress)
+], AgentExecutionStreamMessage.prototype, "progress", void 0);
+__decorate([
+    Field({ nullable: true }),
+    __metadata("design:type", AgentStreamingContent)
+], AgentExecutionStreamMessage.prototype, "streaming", void 0);
+__decorate([
+    Field({ nullable: true }),
+    __metadata("design:type", AgentPartialResult)
+], AgentExecutionStreamMessage.prototype, "partialResult", void 0);
+__decorate([
+    Field(),
+    __metadata("design:type", Date)
+], AgentExecutionStreamMessage.prototype, "timestamp", void 0);
+AgentExecutionStreamMessage = __decorate([
+    ObjectType()
+], AgentExecutionStreamMessage);
+export { AgentExecutionStreamMessage };
 let RunAIAgentResolver = class RunAIAgentResolver extends ResolverBase {
-    async RunAIAgent(agentId, { userPayload }, messagesJson, data, templateData) {
+    sanitizeAgentResult(result) {
+        const sanitized = {
+            success: result.success,
+            returnValue: result.returnValue,
+            errorMessage: result.errorMessage,
+            finalStep: result.finalStep,
+            cancelled: result.cancelled,
+            cancellationReason: result.cancellationReason
+        };
+        if (result.agentRun) {
+            sanitized.agentRun = {
+                ID: result.agentRun.ID,
+                Status: result.agentRun.Status,
+                StartedAt: result.agentRun.StartedAt,
+                CompletedAt: result.agentRun.CompletedAt,
+                Success: result.agentRun.Success,
+                ErrorMessage: result.agentRun.ErrorMessage,
+                AgentID: result.agentRun.AgentID,
+                Result: result.agentRun.Result,
+                TotalTokensUsed: result.agentRun.TotalTokensUsed,
+                TotalCost: result.agentRun.TotalCost
+            };
+        }
+        if (result.executionTree && Array.isArray(result.executionTree)) {
+            sanitized.executionTree = this.sanitizeExecutionTree(result.executionTree);
+        }
+        return sanitized;
+    }
+    sanitizeExecutionTree(nodes) {
+        return nodes.map(node => ({
+            step: node.step ? {
+                ID: node.step.ID,
+                AgentRunID: node.step.AgentRunID,
+                StepNumber: node.step.StepNumber,
+                StepType: node.step.StepType,
+                StepName: node.step.StepName,
+                TargetID: node.step.TargetID,
+                Status: node.step.Status,
+                StartedAt: node.step.StartedAt,
+                CompletedAt: node.step.CompletedAt,
+                Success: node.step.Success,
+                ErrorMessage: node.step.ErrorMessage,
+                InputData: node.step.InputData,
+                OutputData: node.step.OutputData
+            } : null,
+            inputData: node.inputData,
+            outputData: node.outputData,
+            executionType: node.executionType,
+            startTime: node.startTime,
+            endTime: node.endTime,
+            durationMs: node.durationMs,
+            nextStepDecision: node.nextStepDecision,
+            children: node.children && node.children.length > 0
+                ? this.sanitizeExecutionTree(node.children)
+                : [],
+            depth: node.depth,
+            parentStepId: node.parentStepId,
+            agentHierarchy: node.agentHierarchy
+        }));
+    }
+    parseJsonInput(jsonString, fieldName) {
+        if (!jsonString)
+            return {};
+        try {
+            return JSON.parse(jsonString);
+        }
+        catch (parseError) {
+            throw new Error(`Invalid JSON in ${fieldName}: ${parseError.message}`);
+        }
+    }
+    async validateAgent(agentId, currentUser) {
+        await AIEngine.Instance.Config(false, currentUser);
+        const agentEntity = AIEngine.Instance.Agents.find((a) => a.ID === agentId);
+        if (!agentEntity) {
+            throw new Error(`AI Agent with ID ${agentId} not found`);
+        }
+        if (agentEntity.Status !== 'Active') {
+            throw new Error(`AI Agent "${agentEntity.Name}" is not active (Status: ${agentEntity.Status})`);
+        }
+        return agentEntity;
+    }
+    createProgressCallback(pubSub, sessionId, userPayload, agentRunIdRef) {
+        return (progress) => {
+            const significantSteps = ['prompt_execution', 'action_execution', 'subagent_execution', 'decision_processing'];
+            if (!significantSteps.includes(progress.step)) {
+                console.log(`🔇 Skipping noise progress: ${progress.step}`);
+                return;
+            }
+            console.log('📡 Publishing progress update:', {
+                step: progress.step,
+                percentage: progress.percentage,
+                message: progress.message,
+                sessionId,
+                agentRunId: agentRunIdRef.id
+            });
+            const progressMsg = {
+                sessionId,
+                agentRunId: agentRunIdRef.id,
+                type: 'progress',
+                progress: {
+                    currentStep: progress.step,
+                    percentage: progress.percentage,
+                    message: progress.message,
+                    agentName: progress.metadata?.agentName || undefined,
+                    agentType: progress.metadata?.agentType || undefined
+                },
+                timestamp: new Date()
+            };
+            this.PublishProgressUpdate(pubSub, progressMsg, userPayload);
+        };
+    }
+    PublishProgressUpdate(pubSub, data, userPayload) {
+        pubSub.publish(PUSH_STATUS_UPDATES_TOPIC, {
+            message: JSON.stringify({
+                resolver: 'RunAIAgentResolver',
+                type: 'ExecutionProgress',
+                status: 'ok',
+                data,
+            }),
+            sessionId: userPayload.sessionId,
+        });
+    }
+    PublishStreamingUpdate(pubSub, data, userPayload) {
+        pubSub.publish(PUSH_STATUS_UPDATES_TOPIC, {
+            message: JSON.stringify({
+                resolver: 'RunAIAgentResolver',
+                type: 'StreamingContent',
+                status: 'ok',
+                data,
+            }),
+            sessionId: userPayload.sessionId,
+        });
+    }
+    createStreamingCallback(pubSub, sessionId, userPayload, agentRunIdRef) {
+        return (chunk) => {
+            console.log('💬 Publishing streaming content:', {
+                content: chunk.content.substring(0, 50) + '...',
+                isComplete: chunk.isComplete,
+                stepType: chunk.stepType,
+                sessionId,
+                agentRunId: agentRunIdRef.id
+            });
+            const streamMsg = {
+                sessionId,
+                agentRunId: agentRunIdRef.id,
+                type: 'streaming',
+                streaming: {
+                    content: chunk.content,
+                    isPartial: !chunk.isComplete,
+                    stepName: chunk.stepType,
+                    agentName: chunk.modelName
+                },
+                timestamp: new Date()
+            };
+            this.PublishStreamingUpdate(pubSub, streamMsg, userPayload);
+        };
+    }
+    async RunAIAgent(agentId, { userPayload }, messagesJson, sessionId, pubSub, data, templateData) {
         const startTime = Date.now();
         try {
             LogStatus(`=== RUNNING AI AGENT FOR ID: ${agentId} ===`);
-            let parsedMessages;
-            try {
-                parsedMessages = JSON.parse(messagesJson);
-                if (!Array.isArray(parsedMessages)) {
-                    throw new Error('Messages must be an array');
-                }
-            }
-            catch (parseError) {
-                return {
-                    success: false,
-                    error: `Invalid JSON in messages: ${parseError.message}`,
-                    executionTimeMs: Date.now() - startTime
-                };
-            }
-            let parsedData = {};
-            let parsedTemplateData = {};
-            if (data) {
-                try {
-                    parsedData = JSON.parse(data);
-                }
-                catch (parseError) {
-                    return {
-                        success: false,
-                        error: `Invalid JSON in data: ${parseError.message}`,
-                        executionTimeMs: Date.now() - startTime
-                    };
-                }
-            }
-            if (templateData) {
-                try {
-                    parsedTemplateData = JSON.parse(templateData);
-                }
-                catch (parseError) {
-                    return {
-                        success: false,
-                        error: `Invalid JSON in template data: ${parseError.message}`,
-                        executionTimeMs: Date.now() - startTime
-                    };
-                }
+            const parsedMessages = this.parseJsonInput(messagesJson, 'messages');
+            if (!Array.isArray(parsedMessages)) {
+                throw new Error('Messages must be an array');
             }
+            const parsedData = this.parseJsonInput(data, 'data');
+            const parsedTemplateData = this.parseJsonInput(templateData, 'templateData');
             const currentUser = this.GetUserFromPayload(userPayload);
             if (!currentUser) {
-                return {
-                    success: false,
-                    error: 'Unable to determine current user',
-                    executionTimeMs: Date.now() - startTime
-                };
-            }
-            const md = new Metadata();
-            const agentEntity = await md.GetEntityObject('AI Agents', currentUser);
-            await agentEntity.Load(agentId);
-            if (!agentEntity.IsSaved) {
-                return {
-                    success: false,
-                    error: `AI Agent with ID ${agentId} not found`,
-                    executionTimeMs: Date.now() - startTime
-                };
-            }
-            if (agentEntity.Status !== 'Active') {
-                return {
-                    success: false,
-                    error: `AI Agent "${agentEntity.Name}" is not active (Status: ${agentEntity.Status})`,
-                    executionTimeMs: Date.now() - startTime
-                };
+                throw new Error('Unable to determine current user');
             }
+            const agentEntity = await this.validateAgent(agentId, currentUser);
             const agentRunner = new AgentRunner();
+            const agentRunIdRef = { id: 'pending' };
+            console.log(`🚀 Starting agent execution with sessionId: ${sessionId}`);
             const result = await agentRunner.RunAgent({
                 agent: agentEntity,
                 conversationMessages: parsedMessages,
                 contextUser: currentUser,
-                ...(Object.keys(parsedData).length > 0 && { data: parsedData }),
-                ...(Object.keys(parsedTemplateData).length > 0 && { templateData: parsedTemplateData })
+                onProgress: this.createProgressCallback(pubSub, sessionId, userPayload, agentRunIdRef),
+                onStreaming: this.createStreamingCallback(pubSub, sessionId, userPayload, agentRunIdRef)
             });
+            if (result.agentRun) {
+                agentRunIdRef.id = result.agentRun.ID;
+            }
             const executionTime = Date.now() - startTime;
-            if (result.nextStep !== 'failed') {
+            this.publishFinalEvents(pubSub, sessionId, userPayload, result);
+            const sanitizedResult = this.sanitizeAgentResult(result);
+            const payload = JSON.stringify(sanitizedResult);
+            if (result.success) {
                 LogStatus(`=== AI AGENT RUN COMPLETED FOR: ${agentEntity.Name} (${executionTime}ms) ===`);
-                return {
-                    success: true,
-                    output: result.rawResult,
-                    parsedResult: typeof result.returnValue === 'string' ? result.returnValue : JSON.stringify(result.returnValue),
-                    rawResult: result.rawResult,
-                    executionTimeMs: executionTime,
-                    nextStep: result.nextStep
-                };
             }
             else {
                 LogError(`AI Agent run failed for ${agentEntity.Name}: ${result.errorMessage}`);
-                return {
-                    success: false,
-                    error: result.errorMessage,
-                    executionTimeMs: executionTime,
-                    nextStep: result.nextStep
-                };
             }
+            return {
+                success: result.success,
+                errorMessage: result.errorMessage || undefined,
+                executionTimeMs: executionTime,
+                payload
+            };
         }
         catch (error) {
             const executionTime = Date.now() - startTime;
             LogError(`AI Agent run failed:`, undefined, error);
-            return {
+            const errorResult = {
                 success: false,
-                error: error.message || 'Unknown error occurred',
+                errorMessage: error.message || 'Unknown error occurred',
                 executionTimeMs: executionTime
             };
+            return {
+                success: false,
+                errorMessage: errorResult.errorMessage,
+                executionTimeMs: executionTime,
+                payload: JSON.stringify(errorResult)
+            };
+        }
+    }
+    publishFinalEvents(pubSub, sessionId, userPayload, result) {
+        if (result.agentRun) {
+            let lastStep = 'Completed';
+            if (result.executionTree && result.executionTree.length > 0) {
+                const findLastStep = (nodes) => {
+                    let last = nodes[nodes.length - 1];
+                    if (last.children && last.children.length > 0) {
+                        return findLastStep(last.children);
+                    }
+                    return last;
+                };
+                const lastNode = findLastStep(result.executionTree);
+                lastStep = lastNode.step?.StepName || 'Completed';
+            }
+            const partialResult = {
+                currentStep: lastStep,
+                partialOutput: result.returnValue || undefined
+            };
+            const partialMsg = {
+                sessionId,
+                agentRunId: result.agentRun.ID,
+                type: 'partial_result',
+                partialResult,
+                timestamp: new Date()
+            };
+            this.PublishStreamingUpdate(pubSub, partialMsg, userPayload);
         }
+        const completeMsg = {
+            sessionId,
+            agentRunId: result.agentRun?.ID || 'unknown',
+            type: 'complete',
+            timestamp: new Date()
+        };
+        this.PublishStreamingUpdate(pubSub, completeMsg, userPayload);
     }
 };
 __decorate([
@@ -181,10 +462,12 @@ __decorate([
     __param(0, Arg('agentId')),
     __param(1, Ctx()),
     __param(2, Arg('messages')),
-    __param(3, Arg('data', { nullable: true })),
-    __param(4, Arg('templateData', { nullable: true })),
+    __param(3, Arg('sessionId')),
+    __param(4, PubSub()),
+    __param(5, Arg('data', { nullable: true })),
+    __param(6, Arg('templateData', { nullable: true })),
     __metadata("design:type", Function),
-    __metadata("design:paramtypes", [String, Object, String, String, String]),
+    __metadata("design:paramtypes", [String, Object, String, String, PubSubEngine, String, String]),
     __metadata("design:returntype", Promise)
 ], RunAIAgentResolver.prototype, "RunAIAgent", null);
 RunAIAgentResolver = __decorate([

package/dist/resolvers/RunAIAgentResolver.js.map

@@ -1 +1 @@
-{"version":3,"file":"RunAIAgentResolver.js","sourceRoot":"","sources":["../../src/resolvers/RunAIAgentResolver.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,cAAc,CAAC;AAE/E,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAErE,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAGnD,IAAM,gBAAgB,GAAtB,MAAM,gBAAgB;IAEzB,OAAO,CAAU;IAGjB,MAAM,CAAU;IAGhB,YAAY,CAAU;IAGtB,KAAK,CAAU;IAGf,eAAe,CAAU;IAGzB,UAAU,CAAU;IAGpB,UAAU,CAAU;IAGpB,SAAS,CAAU;IAGnB,QAAQ,CAAU;CACrB,CAAA;AAzBG;IADC,KAAK,EAAE;;iDACS;AAGjB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;gDACV;AAGhB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;sDACJ;AAGtB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;+CACX;AAGf;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;yDACD;AAGzB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;oDACN;AAGpB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;oDACN;AAGpB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;mDACP;AAGnB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;kDACR;AA1BT,gBAAgB;IAD5B,UAAU,EAAE;GACA,gBAAgB,CA2B5B;;AAGM,IAAM,kBAAkB,GAAxB,MAAM,kBAAmB,SAAQ,YAAY;IAE1C,AAAN,KAAK,CAAC,UAAU,CACI,OAAe,EACxB,EAAE,WAAW,EAAgC,EACnC,YAAoB,EACJ,IAAa,EACL,YAAqB;QAE9D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YACD,SAAS,CAAC,gCAAgC,OAAO,MAAM,CAAC,CAAC;YAGzD,IAAI,cAAc,CAAC;YACnB,IAAI,CAAC;gBACD,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;gBAC1C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;oBACjC,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;gBACjD,CAAC;YACL,CAAC;YAAC,OAAO,UAAU,EAAE,CAAC;gBAClB,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,6BAA8B,UAAoB,CAAC,OAAO,EAAE;oBACnE,eAAe,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;iBAC1C,CAAC;YACN,CAAC;YAGD,IAAI,UAAU,GAAG,EAAE,CAAC;YACpB,IAAI,kBAAkB,GAAG,EAAE,CAAC;YAE5B,IAAI,IAAI,EAAE,CAAC;gBACP,IAAI,CAAC;oBACD,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAClC,CAAC;gBAAC,OAAO,UAAU,EAAE,CAAC;oBAClB,OAAO;wBACH,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,yBAA0B,UAAoB,CAAC,OAAO,EAAE;wBAC/D,eAAe,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;qBAC1C,CAAC;gBACN,CAAC;YACL,CAAC;YAED,IAAI,YAAY,EAAE,CAAC;gBACf,IAAI,CAAC;oBACD,kBAAkB,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;gBAClD,CAAC;gBAAC,OAAO,UAAU,EAAE,CAAC;oBAClB,OAAO;wBACH,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,kCAAmC,UAAoB,CAAC,OAAO,EAAE;wBACxE,eAAe,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;qBAC1C,CAAC;gBACN,CAAC;YACL,CAAC;YAGD,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC;YACzD,IAAI,CAAC,WAAW,EAAE,CAAC;gBACf,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,kCAAkC;oBACzC,eAAe,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;iBAC1C,CAAC;YACN,CAAC;YAED,MAAM,EAAE,GAAG,IAAI,QAAQ,EAAE,CAAC;YAG1B,MAAM,WAAW,GAAG,MAAM,EAAE,CAAC,eAAe,CAAgB,WAAW,EAAE,WAAW,CAAC,CAAC;YACtF,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAEhC,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;gBACvB,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,oBAAoB,OAAO,YAAY;oBAC9C,eAAe,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;iBAC1C,CAAC;YACN,CAAC;YAGD,IAAI,WAAW,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAClC,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,aAAa,WAAW,CAAC,IAAI,4BAA4B,WAAW,CAAC,MAAM,GAAG;oBACrF,eAAe,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;iBAC1C,CAAC;YACN,CAAC;YAGD,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;YAGtC,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC;gBACtC,KAAK,EAAE,WAAW;gBAClB,oBAAoB,EAAE,cAAc;gBACpC,WAAW,EAAE,WAAW;gBACxB,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;gBAC/D,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,YAAY,EAAE,kBAAkB,EAAE,CAAC;aAC1F,CAAC,CAAC;YAEH,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAE7C,IAAI,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBAC/B,SAAS,CAAC,mCAAmC,WAAW,CAAC,IAAI,KAAK,aAAa,SAAS,CAAC,CAAC;gBAE1F,OAAO;oBACH,OAAO,EAAE,IAAI;oBACb,MAAM,EAAE,MAAM,CAAC,SAAS;oBACxB,YAAY,EAAE,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,WAAW,CAAC;oBAC9G,SAAS,EAAE,MAAM,CAAC,SAAS;oBAC3B,eAAe,EAAE,aAAa;oBAC9B,QAAQ,EAAE,MAAM,CAAC,QAAQ;iBAC5B,CAAC;YACN,CAAC;iBAAM,CAAC;gBACJ,QAAQ,CAAC,2BAA2B,WAAW,CAAC,IAAI,KAAK,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;gBAChF,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,MAAM,CAAC,YAAY;oBAC1B,eAAe,EAAE,aAAa;oBAC9B,QAAQ,EAAE,MAAM,CAAC,QAAQ;iBAC5B,CAAC;YACN,CAAC;QAEL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC7C,QAAQ,CAAC,sBAAsB,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;YACnD,OAAO;gBACH,OAAO,EAAE,KAAK;gBACd,KAAK,EAAG,KAAe,CAAC,OAAO,IAAI,wBAAwB;gBAC3D,eAAe,EAAE,aAAa;aACjC,CAAC;QACN,CAAC;IACL,CAAC;CACJ,CAAA;AArIS;IADL,QAAQ,CAAC,GAAG,EAAE,CAAC,gBAAgB,CAAC;IAE5B,WAAA,GAAG,CAAC,SAAS,CAAC,CAAA;IACd,WAAA,GAAG,EAAE,CAAA;IACL,WAAA,GAAG,CAAC,UAAU,CAAC,CAAA;IACf,WAAA,GAAG,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAA;IAC/B,WAAA,GAAG,CAAC,cAAc,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAA;;;;oDA+H3C;AAtIQ,kBAAkB;IAD9B,QAAQ,EAAE;GACE,kBAAkB,CAuI9B"}
+{"version":3,"file":"RunAIAgentResolver.js","sourceRoot":"","sources":["../../src/resolvers/RunAIAgentResolver.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,YAAY,EAA0C,EAAE,EAAE,MAAM,cAAc,CAAC;AAEjJ,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAE3D,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAExD,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAE,yBAAyB,EAAE,MAAM,kCAAkC,CAAC;AAGtE,IAAM,gBAAgB,GAAtB,MAAM,gBAAgB;IAEzB,OAAO,CAAU;IAGjB,YAAY,CAAU;IAGtB,eAAe,CAAU;IAGzB,OAAO,CAAS;CACnB,CAAA;AAVG;IADC,KAAK,EAAE;;iDACS;AAGjB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;sDACJ;AAGtB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;yDACD;AAGzB;IADC,KAAK,EAAE;;iDACQ;AAXP,gBAAgB;IAD5B,UAAU,EAAE;GACA,gBAAgB,CAY5B;;AAGM,IAAM,sBAAsB,GAA5B,MAAM,sBAAsB;IAE/B,WAAW,CAAS;IAGpB,UAAU,CAAS;IAGnB,OAAO,CAAS;IAGhB,SAAS,CAAU;IAGnB,SAAS,CAAU;CACtB,CAAA;AAbG;IADC,KAAK,EAAE;;2DACY;AAGpB;IADC,KAAK,EAAE;;0DACW;AAGnB;IADC,KAAK,EAAE;;uDACQ;AAGhB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;yDACP;AAGnB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;yDACP;AAdV,sBAAsB;IADlC,UAAU,EAAE;GACA,sBAAsB,CAelC;;AAGM,IAAM,qBAAqB,GAA3B,MAAM,qBAAqB;IAE9B,OAAO,CAAS;IAGhB,SAAS,CAAU;IAGnB,QAAQ,CAAU;IAGlB,SAAS,CAAU;CACtB,CAAA;AAVG;IADC,KAAK,EAAE;;sDACQ;AAGhB;IADC,KAAK,EAAE;;wDACW;AAGnB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;uDACR;AAGlB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;wDACP;AAXV,qBAAqB;IADjC,UAAU,EAAE;GACA,qBAAqB,CAYjC;;AAGM,IAAM,yBAAyB,GAA/B,MAAM,yBAAyB;IAElC,MAAM,CAAS;IAGf,QAAQ,CAAS;IAGjB,SAAS,CAAU;IAGnB,SAAS,CAAU;IAGnB,SAAS,CAAO;IAGhB,OAAO,CAAQ;IAGf,MAAM,CAAS;IAGf,MAAM,CAAU;CACnB,CAAA;AAtBG;IADC,KAAK,EAAE;;yDACO;AAGf;IADC,KAAK,EAAE;;2DACS;AAGjB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;4DACP;AAGnB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;4DACP;AAGnB;IADC,KAAK,EAAE;8BACG,IAAI;4DAAC;AAGhB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;8BAChB,IAAI;0DAAC;AAGf;IADC,KAAK,EAAE;;yDACO;AAGf;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;yDACV;AAvBP,yBAAyB;IADrC,UAAU,EAAE;GACA,yBAAyB,CAwBrC;;AAGM,IAAM,kBAAkB,GAAxB,MAAM,kBAAkB;IAE3B,WAAW,CAAS;IAGpB,aAAa,CAAU;CAC1B,CAAA;AAJG;IADC,KAAK,EAAE;;uDACY;AAGpB;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;yDACH;AALd,kBAAkB;IAD9B,UAAU,EAAE;GACA,kBAAkB,CAM9B;;AAGM,IAAM,2BAA2B,GAAjC,MAAM,2BAA2B;IAEpC,SAAS,CAAS;IAGlB,UAAU,CAAS;IAGnB,IAAI,CAA2D;IAG/D,QAAQ,CAA0B;IAGlC,SAAS,CAAyB;IAGlC,aAAa,CAAsB;IAGnC,SAAS,CAAO;CACnB,CAAA;AAnBG;IADC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;;8DACE;AAGlB;IADC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;;+DACG;AAGnB;IADC,KAAK,EAAE;;yDACuD;AAG/D;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;8BACf,sBAAsB;6DAAC;AAGlC;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;8BACd,qBAAqB;8DAAC;AAGlC;IADC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;8BACV,kBAAkB;kEAAC;AAGnC;IADC,KAAK,EAAE;8BACG,IAAI;8DAAC;AApBP,2BAA2B;IADvC,UAAU,EAAE;GACA,2BAA2B,CAqBvC;;AAOM,IAAM,kBAAkB,GAAxB,MAAM,kBAAmB,SAAQ,YAAY;IAKxC,mBAAmB,CAAC,MAA0B;QAClD,MAAM,SAAS,GAAQ;YACnB,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;SAChD,CAAC;QAGF,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YAClB,SAAS,CAAC,QAAQ,GAAG;gBACjB,EAAE,EAAE,MAAM,CAAC,QAAQ,CAAC,EAAE;gBACtB,MAAM,EAAE,MAAM,CAAC,QAAQ,CAAC,MAAM;gBAC9B,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,SAAS;gBACpC,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,WAAW;gBACxC,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,OAAO;gBAChC,YAAY,EAAE,MAAM,CAAC,QAAQ,CAAC,YAAY;gBAC1C,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,OAAO;gBAChC,MAAM,EAAE,MAAM,CAAC,QAAQ,CAAC,MAAM;gBAC9B,eAAe,EAAE,MAAM,CAAC,QAAQ,CAAC,eAAe;gBAChD,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,SAAS;aACvC,CAAC;QACN,CAAC;QAGD,IAAI,MAAM,CAAC,aAAa,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;YAC9D,SAAS,CAAC,aAAa,GAAG,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAC/E,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;IAKO,qBAAqB,CAAC,KAAY;QACtC,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACtB,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;gBACd,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,EAAE;gBAChB,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;gBAChC,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;gBAChC,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,QAAQ;gBAC5B,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,QAAQ;gBAC5B,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,QAAQ;gBAC5B,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM;gBACxB,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS;gBAC9B,WAAW,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW;gBAClC,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO;gBAC1B,YAAY,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY;gBACpC,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS;gBAC9B,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;aACnC,CAAC,CAAC,CAAC,IAAI;YACR,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC;gBAC/C,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAC3C,CAAC,CAAC,EAAE;YACR,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,cAAc,EAAE,IAAI,CAAC,cAAc;SACtC,CAAC,CAAC,CAAC;IACR,CAAC;IAKO,cAAc,CAAC,UAA8B,EAAE,SAAiB;QACpE,IAAI,CAAC,UAAU;YAAE,OAAO,EAAE,CAAC;QAE3B,IAAI,CAAC;YACD,OAAO,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAClC,CAAC;QAAC,OAAO,UAAU,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,mBAAmB,SAAS,KAAM,UAAoB,CAAC,OAAO,EAAE,CAAC,CAAC;QACtF,CAAC;IACL,CAAC;IAKO,KAAK,CAAC,aAAa,CAAC,OAAe,EAAE,WAAgB;QAEzD,MAAM,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;QAGnD,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAgB,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CAAC,CAAC;QAE1F,IAAI,CAAC,WAAW,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,oBAAoB,OAAO,YAAY,CAAC,CAAC;QAC7D,CAAC;QAGD,IAAI,WAAW,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,aAAa,WAAW,CAAC,IAAI,4BAA4B,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC;QACpG,CAAC;QAED,OAAO,WAAW,CAAC;IACvB,CAAC;IAKO,sBAAsB,CAAC,MAAoB,EAAE,SAAiB,EAAE,WAAwB,EAAE,aAA6B;QAC3H,OAAO,CAAC,QAAa,EAAE,EAAE;YAErB,MAAM,gBAAgB,GAAG,CAAC,kBAAkB,EAAE,kBAAkB,EAAE,oBAAoB,EAAE,qBAAqB,CAAC,CAAC;YAC/G,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5C,OAAO,CAAC,GAAG,CAAC,+BAA+B,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC5D,OAAO;YACX,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,gCAAgC,EAAE;gBAC1C,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,SAAS;gBACT,UAAU,EAAE,aAAa,CAAC,EAAE;aAC/B,CAAC,CAAC;YAGH,MAAM,WAAW,GAAgC;gBAC7C,SAAS;gBACT,UAAU,EAAE,aAAa,CAAC,EAAE;gBAC5B,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACN,WAAW,EAAE,QAAQ,CAAC,IAAI;oBAC1B,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,OAAO,EAAE,QAAQ,CAAC,OAAO;oBACzB,SAAS,EAAG,QAAQ,CAAC,QAAgB,EAAE,SAAS,IAAI,SAAS;oBAC7D,SAAS,EAAG,QAAQ,CAAC,QAAgB,EAAE,SAAS,IAAI,SAAS;iBAChE;gBACD,SAAS,EAAE,IAAI,IAAI,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;QACjE,CAAC,CAAC;IACN,CAAC;IAEO,qBAAqB,CAAC,MAAoB,EAAE,IAAS,EAAE,WAAwB;QACnF,MAAM,CAAC,OAAO,CAAC,yBAAyB,EAAE;YACtC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;gBACpB,QAAQ,EAAE,oBAAoB;gBAC9B,IAAI,EAAE,mBAAmB;gBACzB,MAAM,EAAE,IAAI;gBACZ,IAAI;aACP,CAAC;YACF,SAAS,EAAE,WAAW,CAAC,SAAS;SACnC,CAAC,CAAC;IACP,CAAC;IAGO,sBAAsB,CAAC,MAAoB,EAAE,IAAS,EAAE,WAAwB;QACpF,MAAM,CAAC,OAAO,CAAC,yBAAyB,EAAE;YACtC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;gBACpB,QAAQ,EAAE,oBAAoB;gBAC9B,IAAI,EAAE,kBAAkB;gBACxB,MAAM,EAAE,IAAI;gBACZ,IAAI;aACP,CAAC;YACF,SAAS,EAAE,WAAW,CAAC,SAAS;SACnC,CAAC,CAAC;IACP,CAAC;IAKO,uBAAuB,CAAC,MAAoB,EAAE,SAAiB,EAAE,WAAwB,EAAE,aAA6B;QAC5H,OAAO,CAAC,KAAU,EAAE,EAAE;YAClB,OAAO,CAAC,GAAG,CAAC,kCAAkC,EAAE;gBAC5C,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK;gBAC/C,UAAU,EAAE,KAAK,CAAC,UAAU;gBAC5B,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,SAAS;gBACT,UAAU,EAAE,aAAa,CAAC,EAAE;aAC/B,CAAC,CAAC;YAGH,MAAM,SAAS,GAAgC;gBAC3C,SAAS;gBACT,UAAU,EAAE,aAAa,CAAC,EAAE;gBAC5B,IAAI,EAAE,WAAW;gBACjB,SAAS,EAAE;oBACP,OAAO,EAAE,KAAK,CAAC,OAAO;oBACtB,SAAS,EAAE,CAAC,KAAK,CAAC,UAAU;oBAC5B,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,SAAS,EAAE,KAAK,CAAC,SAAS;iBAC7B;gBACD,SAAS,EAAE,IAAI,IAAI,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,sBAAsB,CAAC,MAAM,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;QAChE,CAAC,CAAC;IACN,CAAC;IAGK,AAAN,KAAK,CAAC,UAAU,CACI,OAAe,EACxB,EAAE,WAAW,EAAgC,EACnC,YAAoB,EACnB,SAAiB,EACzB,MAAoB,EACG,IAAa,EACL,YAAqB;QAE9D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YACD,SAAS,CAAC,gCAAgC,OAAO,MAAM,CAAC,CAAC;YAGzD,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;YACrE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;gBACjC,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YACjD,CAAC;YAGD,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACrD,MAAM,kBAAkB,GAAG,IAAI,CAAC,cAAc,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC;YAG7E,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC;YACzD,IAAI,CAAC,WAAW,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACxD,CAAC;YAGD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;YAGnE,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;YAGtC,MAAM,aAAa,GAAG,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC;YAExC,OAAO,CAAC,GAAG,CAAC,+CAA+C,SAAS,EAAE,CAAC,CAAC;YAGxE,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC;gBACtC,KAAK,EAAE,WAAW;gBAClB,oBAAoB,EAAE,cAAc;gBACpC,WAAW,EAAE,WAAW;gBACxB,UAAU,EAAE,IAAI,CAAC,sBAAsB,CAAC,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,aAAa,CAAC;gBACtF,WAAW,EAAE,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,aAAa,CAAC;aAC3F,CAAC,CAAC;YAGH,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBAClB,aAAa,CAAC,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1C,CAAC;YAED,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAG7C,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;YAGhE,MAAM,eAAe,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;YACzD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YAGhD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACjB,SAAS,CAAC,mCAAmC,WAAW,CAAC,IAAI,KAAK,aAAa,SAAS,CAAC,CAAC;YAC9F,CAAC;iBAAM,CAAC;gBACJ,QAAQ,CAAC,2BAA2B,WAAW,CAAC,IAAI,KAAK,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;YACpF,CAAC;YAED,OAAO;gBACH,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,SAAS;gBAC9C,eAAe,EAAE,aAAa;gBAC9B,OAAO;aACV,CAAC;QAEN,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC7C,QAAQ,CAAC,sBAAsB,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;YAGnD,MAAM,WAAW,GAAG;gBAChB,OAAO,EAAE,KAAK;gBACd,YAAY,EAAG,KAAe,CAAC,OAAO,IAAI,wBAAwB;gBAClE,eAAe,EAAE,aAAa;aACjC,CAAC;YAEF,OAAO;gBACH,OAAO,EAAE,KAAK;gBACd,YAAY,EAAE,WAAW,CAAC,YAAY;gBACtC,eAAe,EAAE,aAAa;gBAC9B,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;aACvC,CAAC;QACN,CAAC;IACL,CAAC;IAKO,kBAAkB,CAAC,MAAoB,EAAE,SAAiB,EAAE,WAAwB,EAAE,MAA0B;QACpH,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YAElB,IAAI,QAAQ,GAAG,WAAW,CAAC;YAC3B,IAAI,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAE1D,MAAM,YAAY,GAAG,CAAC,KAAY,EAAO,EAAE;oBACvC,IAAI,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBACnC,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAC5C,OAAO,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;oBACvC,CAAC;oBACD,OAAO,IAAI,CAAC;gBAChB,CAAC,CAAC;gBACF,MAAM,QAAQ,GAAG,YAAY,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;gBACpD,QAAQ,GAAG,QAAQ,CAAC,IAAI,EAAE,QAAQ,IAAI,WAAW,CAAC;YACtD,CAAC;YAGD,MAAM,aAAa,GAAuB;gBACtC,WAAW,EAAE,QAAQ;gBACrB,aAAa,EAAE,MAAM,CAAC,WAAW,IAAI,SAAS;aACjD,CAAC;YAEF,MAAM,UAAU,GAAgC;gBAC5C,SAAS;gBACT,UAAU,EAAE,MAAM,CAAC,QAAQ,CAAC,EAAE;gBAC9B,IAAI,EAAE,gBAAgB;gBACtB,aAAa;gBACb,SAAS,EAAE,IAAI,IAAI,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,sBAAsB,CAAC,MAAM,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;QACjE,CAAC;QAGD,MAAM,WAAW,GAAgC;YAC7C,SAAS;YACT,UAAU,EAAE,MAAM,CAAC,QAAQ,EAAE,EAAE,IAAI,SAAS;YAC5C,IAAI,EAAE,UAAU;YAChB,SAAS,EAAE,IAAI,IAAI,EAAE;SACxB,CAAC;QACF,IAAI,CAAC,sBAAsB,CAAC,MAAM,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;IAClE,CAAC;CAEJ,CAAA;AAhJS;IADL,QAAQ,CAAC,GAAG,EAAE,CAAC,gBAAgB,CAAC;IAE5B,WAAA,GAAG,CAAC,SAAS,CAAC,CAAA;IACd,WAAA,GAAG,EAAE,CAAA;IACL,WAAA,GAAG,CAAC,UAAU,CAAC,CAAA;IACf,WAAA,GAAG,CAAC,WAAW,CAAC,CAAA;IAChB,WAAA,MAAM,EAAE,CAAA;IACR,WAAA,GAAG,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAA;IAC/B,WAAA,GAAG,CAAC,cAAc,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAA;;qEAFtB,YAAY;;oDA2FjC;AA5SQ,kBAAkB;IAD9B,QAAQ,EAAE;GACE,kBAAkB,CA4V9B"}

package/dist/resolvers/SqlLoggingConfigResolver.d.ts

@@ -0,0 +1,61 @@
+import { AppContext } from '../types.js';
+import { ResolverBase } from '../generic/ResolverBase.js';
+export declare class SqlLoggingOptions {
+    formatAsMigration?: boolean;
+    description?: string;
+    statementTypes?: 'queries' | 'mutations' | 'both';
+    batchSeparator?: string;
+    prettyPrint?: boolean;
+    logRecordChangeMetadata?: boolean;
+    retainEmptyLogFiles?: boolean;
+    filterByUserId?: string;
+    sessionName?: string;
+}
+export declare class SqlLoggingSession {
+    id: string;
+    filePath: string;
+    startTime: Date;
+    statementCount: number;
+    options: SqlLoggingOptions;
+    sessionName?: string;
+    filterByUserId?: string;
+}
+export declare class SqlLoggingOptionsInput {
+    formatAsMigration?: boolean;
+    description?: string;
+    statementTypes?: 'queries' | 'mutations' | 'both';
+    batchSeparator?: string;
+    prettyPrint?: boolean;
+    logRecordChangeMetadata?: boolean;
+    retainEmptyLogFiles?: boolean;
+    filterByUserId?: string;
+    sessionName?: string;
+}
+export declare class StartSqlLoggingInput {
+    fileName?: string;
+    options?: SqlLoggingOptionsInput;
+    filterToCurrentUser?: boolean;
+}
+export declare class SqlLoggingConfig {
+    enabled: boolean;
+    defaultOptions: SqlLoggingOptions;
+    allowedLogDirectory: string;
+    maxActiveSessions: number;
+    autoCleanupEmptyFiles: boolean;
+    sessionTimeout: number;
+    activeSessionCount: number;
+}
+export declare class SqlLoggingConfigResolver extends ResolverBase {
+    private static readonly LOG_FILE_PREFIX;
+    private checkOwnerAccess;
+    sqlLoggingConfig(context: AppContext): Promise<SqlLoggingConfig>;
+    activeSqlLoggingSessions(context: AppContext): Promise<SqlLoggingSession[]>;
+    startSqlLogging(input: StartSqlLoggingInput, context: AppContext): Promise<SqlLoggingSession>;
+    stopSqlLogging(sessionId: string, context: AppContext): Promise<boolean>;
+    stopAllSqlLogging(context: AppContext): Promise<boolean>;
+    updateSqlLoggingDefaults(options: SqlLoggingOptionsInput, context: AppContext): Promise<SqlLoggingOptions>;
+    readSqlLogFile(sessionId: string, maxLines: number | null, context: AppContext): Promise<string>;
+    debugCurrentUserEmail(context: AppContext): Promise<string>;
+    private ensureDirectoryExists;
+}
+//# sourceMappingURL=SqlLoggingConfigResolver.d.ts.map