@memberjunction/ai-agents 2.122.2 → 2.123.1

package/dist/base-agent.js

@@ -53,6 +53,7 @@ class BaseAgent {
         this._allProgressSteps = [];
         this._subAgentRuns = [];
         this._payloadManager = new PayloadManager_1.PayloadManager();
+        this._effectiveActions = [];
         this._validationRetryCount = 0;
         this._contextRecoveryAttempts = 0;
         this.MAX_RECOVERY_ATTEMPTS = 1;
@@ -566,7 +567,7 @@ class BaseAgent {
         const agentType = config.agentType;
         const systemPrompt = config.systemPrompt;
         const childPrompt = config.childPrompt;
-        const promptTemplateData = await this.gatherPromptTemplateData(params.agent, params.contextUser, params.data);
+        const promptTemplateData = await this.gatherPromptTemplateData(params.agent, params.contextUser, params.data, params.actionChanges);
         const promptParams = new ai_core_plus_1.AIPromptParams();
         if (systemPrompt) {
             promptParams.prompt = systemPrompt;
@@ -1605,7 +1606,7 @@ The context is now within limits. Please retry your request with the recovered c
             content: `Sub-agent '${subAgent.name}' result:\n${typeof result === 'string' ? result : JSON.stringify(result, null, 2)}`
         };
     }
-    async gatherPromptTemplateData(agent, _contextUser, extraData) {
+    async gatherPromptTemplateData(agent, _contextUser, extraData, actionChanges) {
         try {
             const engine = aiengine_1.AIEngine.Instance;
             const activeSubAgents = engine.Agents.filter(a => a.ParentID === agent.ID && a.Status === 'Active')
@@ -1616,15 +1617,20 @@ The context is now within limits. Please retry your request with the recovered c
             activeAgentRelationships.forEach(ar => uniqueActiveSubAgentIDs.add(ar.SubAgentID));
             const uniqueActiveSubAgents = Array.from(uniqueActiveSubAgentIDs).map(id => engine.Agents.find(a => a.ID === id));
             const agentActions = engine.AgentActions.filter(aa => aa.AgentID === agent.ID && aa.Status === 'Active');
-            const actions = actions_1.ActionEngineServer.Instance.Actions.filter(a => agentActions.some(aa => aa.ActionID === a.ID));
+            let actions = actions_1.ActionEngineServer.Instance.Actions.filter(a => agentActions.some(aa => aa.ActionID === a.ID));
+            if (actionChanges?.length) {
+                const isRoot = this._depth === 0;
+                actions = this.applyActionChanges(actions, actionChanges, agent.ID, isRoot);
+            }
             const activeActions = actions.filter(a => a.Status === 'Active');
+            this._effectiveActions = activeActions;
             const contextData = {
                 agentName: agent.Name,
                 agentDescription: agent.Description,
                 parentAgentName: agent.Parent ? agent.Parent.trim() : "",
                 subAgentCount: uniqueActiveSubAgents.length,
                 subAgentDetails: this.formatSubAgentDetails(uniqueActiveSubAgents),
-                actionCount: actions.length,
+                actionCount: activeActions.length,
                 actionDetails: this.formatActionDetails(activeActions),
             };
             if (extraData) {
@@ -1741,6 +1747,7 @@ The context is now within limits. Please retry your request with the recovered c
                 this.logStatus(`🎯 Propagating effort level ${params.effortLevel} to sub-agent '${subAgentRequest.name}'`, true, params);
             }
             const parentStepCountsToPass = [...this._parentStepCounts, stepCount + 1];
+            const subAgentActionChanges = this.filterActionChangesForSubAgent(params.actionChanges);
             const result = await runner.RunAgent({
                 agent: subAgent,
                 conversationMessages: subAgentMessages,
@@ -1761,6 +1768,7 @@ The context is now within limits. Please retry your request with the recovered c
                 },
                 context: params.context,
                 verbose: params.verbose,
+                actionChanges: subAgentActionChanges,
                 onAgentRunCreated: async (agentRunId) => {
                     stepEntity.TargetLogID = agentRunId;
                     await stepEntity.Save();
@@ -1876,6 +1884,67 @@ The context is now within limits. Please retry your request with the recovered c
         const agentType = aiengine_1.AIEngine.Instance.AgentTypes.find(at => at.ID === typeID);
         return agentType?.Name || 'Unknown';
     }
+    doesChangeScopeApply(scope, agentId, isRoot, agentIds) {
+        switch (scope) {
+            case 'global':
+                return true;
+            case 'root':
+                return isRoot;
+            case 'all-subagents':
+                return !isRoot;
+            case 'specific':
+                return agentIds?.includes(agentId) ?? false;
+            default:
+                return false;
+        }
+    }
+    applyActionChanges(baseActions, actionChanges, agentId, isRoot) {
+        let actions = [...baseActions];
+        for (const change of actionChanges) {
+            if (!this.doesChangeScopeApply(change.scope, agentId, isRoot, change.agentIds)) {
+                continue;
+            }
+            if (change.mode === 'add') {
+                for (const actionId of change.actionIds) {
+                    if (!actions.some(a => a.ID === actionId)) {
+                        const actionToAdd = actions_1.ActionEngineServer.Instance.Actions.find(a => a.ID === actionId);
+                        if (actionToAdd) {
+                            actions.push(actionToAdd);
+                        }
+                        else {
+                            (0, core_1.LogStatus)(`Action with ID '${actionId}' not found in ActionEngineServer - skipping add`);
+                        }
+                    }
+                }
+            }
+            else if (change.mode === 'remove') {
+                actions = actions.filter(a => !change.actionIds.includes(a.ID));
+            }
+        }
+        return actions;
+    }
+    filterActionChangesForSubAgent(actionChanges) {
+        if (!actionChanges?.length) {
+            return undefined;
+        }
+        const filtered = [];
+        for (const change of actionChanges) {
+            switch (change.scope) {
+                case 'root':
+                    continue;
+                case 'global':
+                    filtered.push(change);
+                    break;
+                case 'all-subagents':
+                    filtered.push({ ...change, scope: 'global' });
+                    break;
+                case 'specific':
+                    filtered.push(change);
+                    break;
+            }
+        }
+        return filtered.length > 0 ? filtered : undefined;
+    }
     getValueListItems(valueListID) {
         return null;
     }
@@ -3124,6 +3193,9 @@ The context is now within limits. Please retry your request with the recovered c
             }
             const actionEngine = actions_1.ActionEngineServer.Instance;
             const agentActions = aiengine_1.AIEngine.Instance.AgentActions.filter(aa => aa.AgentID === params.agent.ID);
+            const effectiveActions = this._effectiveActions.length > 0
+                ? this._effectiveActions
+                : actionEngine.Actions.filter(a => agentActions.some(aa => aa.ActionID === a.ID));
             try {
                 const currentPayload = previousDecision?.newPayload || previousDecision?.previousPayload || params.payload;
                 await this.AgentTypeInstance.PreProcessActionStep(actions, currentPayload, this.AgentTypeState, previousDecision, params);
@@ -3135,9 +3207,9 @@ The context is now within limits. Please retry your request with the recovered c
             const baseStepNumber = (this._agentRun.Steps?.length || 0) + 1;
             let lastStep = undefined;
             const actionPromises = actions.map(async (aa) => {
-                const actionEntity = actionEngine.Actions.find(a => a.Name === aa.name && agentActions.some(agentAction => agentAction.ActionID === a.ID));
+                const actionEntity = effectiveActions.find(a => a.Name === aa.name);
                 if (!actionEntity) {
-                    throw new Error(`Action "${aa.name}" Not Found for Agent "${params.agent.Name}"`);
+                    throw new Error(`Action "${aa.name}" Not Found for Agent "${params.agent.Name}". Available actions: ${effectiveActions.map(a => a.Name).join(', ')}`);
                 }
                 const actionInputData = {
                     actionName: aa.name,