actoviq-agent-sdk 0.4.21 → 0.4.22

package/dist/src/runtime/agentClient.js

@@ -12,7 +12,10 @@ import { createExaSearchContribution, createRequestRetryPolicyContribution, crea
 import { HadamardContributionHost } from '../contrib/contributionHost.js';
 import { InMemoryContributionToolRegistry, contributionRetryPolicyKey, contributionToolRegistryKey, } from '../contrib/contributionServices.js';
 import { conversationExtensionsFactoryKey, createBuiltInConversationExtensionsContribution, } from './conversationExtensions.js';
-import { createBuiltInToolPolicyContribution, toolPolicyFactoryKey } from './toolPolicyPipeline.js';
+import { createBuiltInToolPolicyContribution, InMemoryToolPolicyListenerRegistry, toolPolicyFactoryKey, toolPolicyListenerRegistryKey, } from './toolPolicyPipeline.js';
+import { BuiltInExtensionToggles, createBuiltInExtensionsApi, resolveBuiltInExtensionStates, } from '../extensions/builtInExtensions.js';
+import { createSecurityPolicyContribution } from '../extensions/securityPolicy.js';
+import { createFilterOutputPolicyContribution } from '../extensions/filterOutputPolicy.js';
 import { codeDispatchFormatterKey, createCodeDispatchTranscriptContribution, defaultCodeDispatchTranscriptFormatter } from './codeDispatchTranscript.js';
 import { resolveHadamardSettingsStore } from '../config/hadamardSettingsStore.js';
 import { resolveRuntimeConfig } from '../config/resolveRuntimeConfig.js';
@@ -64,7 +67,7 @@ import { TeammateStore } from '../storage/teammateStore.js';
 import { createCheckpointTools, FileChangeJournal, FileCheckpointService, } from '../checkpoint/index.js';
 import { SandboxExecutor } from '../sandbox/sandboxExecutor.js';
 import { resolveSandboxPolicy } from '../sandbox/policyResolver.js';
-import { CodeIntelligenceService, createCodeIntelligenceTools, LanguageServerRegistry, } from '../codeIntel/index.js';
+import { CodeIntelligenceService, createCodeIntelligenceTools, LanguageServerRegistry, resolveLanguageServerDefinitions, } from '../codeIntel/index.js';
 import { HookRunner } from '../hooks/hookRunner.js';
 import { createPromptHookHandler } from '../hooks/handlers/promptHook.js';
 import { HadamardSwarmApi } from '../swarm/hadamardSwarm.js';
@@ -430,6 +433,8 @@ export class HadamardAgentClient {
     codeIntelligence;
     /** Persistent, project-scoped root/child Agent execution graph. */
     executions;
+    /** Built-in extension states (security guard, output filter, UI stubs) with persisted toggling. */
+    builtInExtensions;
     sessionManager;
     sessionTurnCoordinator = new SessionTurnCoordinator();
     agentDefinitions;
@@ -449,6 +454,8 @@ export class HadamardAgentClient {
     codeActServices = new Map();
     sessionRuntimeOverrides = new Map();
     backgroundTaskManager;
+    /** User-facing listeners fired when any background task settles (best-effort). */
+    backgroundTaskSettledListeners = new Set();
     defaultPermissionMode;
     defaultPermissions;
     defaultClassifier;
@@ -468,7 +475,7 @@ export class HadamardAgentClient {
         managedSkills: [],
         disableDefaultSkills: false,
         loadDefaultSkillDirectories: true,
-    }, defaultPermissionMode, defaultPermissions, defaultClassifier, defaultApprover, sessionManagerConfig, maxSubagentDepth = 1, maxSubagentFanout = 8, taskWorktreeCoordinator, externalAgentRunner, runtimeClosers = [], contributionHost) {
+    }, defaultPermissionMode, defaultPermissions, defaultClassifier, defaultApprover, sessionManagerConfig, maxSubagentDepth = 1, maxSubagentFanout = 8, taskWorktreeCoordinator, externalAgentRunner, runtimeClosers = [], contributionHost, extensionToggles) {
         this.config = config;
         this.store = store;
         this.backgroundTaskStore = backgroundTaskStore;
@@ -492,6 +499,7 @@ export class HadamardAgentClient {
         this.approvalPolicy = new ApprovalPolicy(path.join(this.config.homeDir, 'policy', 'approvals.json'));
         this.auditLog = new AuditLog(path.join(this.config.homeDir, 'policy', 'audit.ndjson'));
         this.executions = new HadamardAgentExecutionsApi(executionStore);
+        this.builtInExtensions = createBuiltInExtensionsApi(extensionToggles ?? new BuiltInExtensionToggles(resolveBuiltInExtensionStates(undefined)), this.config.homeDir);
         this.checkpoints = new FileCheckpointService({
             storageRoot: path.join(this.config.sessionDirectory, 'file-checkpoints'),
             workspaceRoot: this.config.workDir,
@@ -702,7 +710,10 @@ export class HadamardAgentClient {
         if (this.defaultTools.some(tool => tool.name === BASH_TOOL_NAME)) {
             this.replaceDefaultTool(createBashTool({
                 backgroundTaskManager: this.backgroundTaskManager,
-                onBackgroundTaskSettled: task => this.enqueueTaskNotification(task),
+                onBackgroundTaskSettled: task => {
+                    this.enqueueTaskNotification(task);
+                    this.notifyBackgroundTaskSettled(task);
+                },
             }));
         }
         this.replaceDefaultTool(this.createSendMessageTool());
@@ -764,16 +775,29 @@ export class HadamardAgentClient {
         const contributionHost = new HadamardContributionHost();
         const contributionTools = new InMemoryContributionToolRegistry();
         contributionHost.registerService(contributionToolRegistryKey, contributionTools);
+        // Registered before any contribution apply runs so the tool-policy factory
+        // and the extension contributions can both resolve it.
+        contributionHost.registerService(toolPolicyListenerRegistryKey, new InMemoryToolPolicyListenerRegistry());
         let managedSettings = {};
         if (options.managedPlugins !== false) {
             managedSettings = isRecord(options.managedPlugins)
                 ? options.managedPlugins
                 : (await resolveHadamardSettingsStore({ homeDir: config.homeDir })).raw;
         }
+        // Built-in extension states read from the same settings payload; with
+        // managedPlugins disabled the settings store is read once for extensions.
+        const extensionSettingsRaw = isRecord(options.managedPlugins)
+            ? options.managedPlugins
+            : options.managedPlugins !== false
+                ? managedSettings
+                : (await resolveHadamardSettingsStore({ homeDir: config.homeDir })).raw;
+        const extensionToggles = new BuiltInExtensionToggles(resolveBuiltInExtensionStates(extensionSettingsRaw, options.extensions));
         await contributionHost.loadMany([
             createRequestRetryPolicyContribution(config.retryPolicy),
             createBuiltInConversationExtensionsContribution(),
             createBuiltInToolPolicyContribution(),
+            createSecurityPolicyContribution(extensionToggles),
+            createFilterOutputPolicyContribution(extensionToggles),
             createCodeDispatchTranscriptContribution(),
             createTavilySearchContribution(managedSettings),
             createExaSearchContribution(managedSettings),
@@ -951,10 +975,16 @@ export class HadamardAgentClient {
                 definition.name,
                 definition,
             ])).values()];
-        client = new HadamardAgentClient(config, store, executionStore, backgroundTaskStore, mailboxStore, teammateStore, modelApi, mcpManager, deduplicatedTools, defaultMcpServers, options.hooks, agentDefinitions, deduplicatedSkills, skillComposition, policyPermissionMode(config.effectivePolicy) ?? options.permissionMode, [
+        // Code intelligence: merge explicit language servers with auto-detected
+        // presets (PATH probe), so LSP tools work out of the box. Async, so it
+        // happens here in create() rather than in the synchronous constructor.
+        const languageServers = await resolveLanguageServerDefinitions(config.languageServers, {
+            autoDetect: config.autoDetectLanguageServers,
+        });
+        client = new HadamardAgentClient({ ...config, languageServers }, store, executionStore, backgroundTaskStore, mailboxStore, teammateStore, modelApi, mcpManager, deduplicatedTools, defaultMcpServers, options.hooks, agentDefinitions, deduplicatedSkills, skillComposition, policyPermissionMode(config.effectivePolicy) ?? options.permissionMode, [
             ...policyPermissionRules(config.effectivePolicy),
             ...(options.permissions ?? []),
-        ], options.classifier, options.approver, options.sessionManager, options.maxSubagentDepth, options.maxSubagentFanout, taskWorktreeCoordinator, options.externalAgentRunner, runtimeClosers, contributionHost);
+        ], options.classifier, options.approver, options.sessionManager, options.maxSubagentDepth, options.maxSubagentFanout, taskWorktreeCoordinator, options.externalAgentRunner, runtimeClosers, contributionHost, extensionToggles);
         const interruptedTasks = await client.backgroundTaskManager.reconcileInterruptedTasks();
         await client.reconcileInterruptedAgentExecutions(interruptedTasks);
         return client;
@@ -3447,6 +3477,28 @@ export class HadamardAgentClient {
             this.pendingRuntimeNotifications.set(notificationKey, queue);
         }
     }
+    /**
+     * Subscribe to background task settlement (completed/failed/cancelled) for
+     * user-facing surfaces. Fires for every settled task — including ad-hoc
+     * tasks without a parent session — independently of the agent-facing
+     * `<task_notification>` enqueue path. Returns an unsubscribe function.
+     */
+    onBackgroundTaskSettled(listener) {
+        this.backgroundTaskSettledListeners.add(listener);
+        return () => {
+            this.backgroundTaskSettledListeners.delete(listener);
+        };
+    }
+    notifyBackgroundTaskSettled(task) {
+        for (const listener of this.backgroundTaskSettledListeners) {
+            try {
+                listener(task);
+            }
+            catch {
+                // Listener errors must never break the settle path.
+            }
+        }
+    }
     drainRuntimeNotifications(sessionId) {
         const queued = this.pendingRuntimeNotifications.get(sessionId) ?? [];
         this.pendingRuntimeNotifications.delete(sessionId);
@@ -3534,6 +3586,7 @@ export class HadamardAgentClient {
                 onSettled: async (task) => {
                     this.updatePendingDelegation(options.parentSessionId ?? options.parentRunId, task);
                     this.enqueueTaskNotification(task);
+                    this.notifyBackgroundTaskSettled(task);
                     if (task.status === 'completed') {
                         await this.tryCompleteExecutionEdge(execution.edge, task.text);
                     }
@@ -3607,6 +3660,7 @@ export class HadamardAgentClient {
                 onSettled: async (task) => {
                     this.updatePendingDelegation(notificationParent, task);
                     this.enqueueTaskNotification(task);
+                    this.notifyBackgroundTaskSettled(task);
                     if (task.status === 'completed' && hasCollaborationEdge) {
                         await this.tryCompleteExecutionEdge(execution.edge, task.text);
                     }