@sema-agent/server 1.246.0 → 1.248.0

package/dist/main.js

@@ -5,7 +5,7 @@ import { homedir } from "node:os";
 import { join } from "node:path";
 import { loadRemoteExec } from "@sema-agent/registry-core/node";
 import { skillContentHash } from "@sema-agent/registry-core";
-import { Runner, TtlSessionStore, uuidv7, FileRosterStore, defaultTaskRegistry, combinePolicies, createDurableQuestionPolicy, createAllowDenyPolicy, tightenTaskSpec, isThinkingLevel, QUESTION_AWAITS_RESUME, FileWorkflowRunStore, InMemoryWorkflowRunStore, workflowsCapability, NodeLspManager, createFileWorkflowScriptStore, expandTiers, NodeExecutionEnv, CenterPromptSource, FilePromptArtifactStore, FilePromptSourceStateStore, MemoryPromptArtifactStore, MemoryPromptSourceStateStore, verifyPromptArtifact } from "@sema-agent/core";
+import { Runner, TtlSessionStore, uuidv7, FileRosterStore, FileBackgroundAgentStore, defaultTaskRegistry, combinePolicies, createDurableQuestionPolicy, createAllowDenyPolicy, tightenTaskSpec, isThinkingLevel, QUESTION_AWAITS_RESUME, FileWorkflowRunStore, InMemoryWorkflowRunStore, workflowsCapability, NodeLspManager, createFileWorkflowScriptStore, expandTiers, NodeExecutionEnv, CenterPromptSource, FilePromptArtifactStore, FilePromptSourceStateStore, MemoryPromptArtifactStore, MemoryPromptSourceStateStore, verifyPromptArtifact } from "@sema-agent/core";
 import { PgMemoryEngineBackend, ensurePgMemoryEngineSchema } from "./plugins/memory-engine-pg.js";
 import { TiDBMemoryEngineBackend, ensureTiDBMemoryEngineSchema } from "./plugins/memory-engine-tidb.js";
 import { PgMemoryHistoryStore, ensurePgMemoryHistorySchema, PgMemorySyncStore, ensurePgMemorySyncSchema } from "./plugins/memory-sync-store-pg.js";
@@ -78,6 +78,7 @@ import { createConfigProvider, raceBootFetch, BOOT_FETCH_DEFERRED } from "./conf
 import { validatePromptsDomain, centerPromptProvider, centerIdentityAssembled, applyCatalogToSource, CORE_ENGINE_VERSION, withPromptArtifactBackfill } from "./capabilities/center-prompts.js";
 import { defaultLkgPath, defaultSkillCacheDir, saveLkg, loadLkg } from "./config-lkg.js";
 import { TiDBRosterStore, PgRosterStore, ensureTiDBRosterSchema, ensurePgRosterSchema } from "./plugins/roster-store-sql.js";
+import { TiDBBackgroundAgentStore, PgBackgroundAgentStore, ensureTiDBBackgroundAgentSchema, ensurePgBackgroundAgentSchema } from "./plugins/background-agent-store-sql.js";
 import { createPrincipalCapsClient, gateExecutionLane, scopedTokenNeedsWorker, applyObserverEnvOptIn } from "./runtime-caps-resolver.js";
 import { applyRuntimeGovernance, stripDelegationTools } from "./runtime-governance.js";
 import { parseTaskSettings, applyTaskSettings, coercePermissionMode, withPermissionMode, effectiveThinking, acceptAppendSystemPrompt, hasConstitutionAnchors, providerDropsAppend, MAX_SETTINGS_OUTPUT_STYLE_CHARS } from "./task-settings.js";
@@ -514,6 +515,28 @@ async function main() {
         if (rosterStore)
             logger.info("roster_store_enabled", { backend: pgPool ? "pg" : mysqlPool ? "tidb" : "file" });
     }
+    let backgroundAgentStore;
+    {
+        const mysqlPool = backend?.mysqlPool?.();
+        const pgPool = backend?.pgPool?.();
+        if (pgPool) {
+            const q = async (text, params) => { const r = await pgPool.query(text, params); return { rows: r.rows }; };
+            await ensurePgBackgroundAgentSchema(q);
+            backgroundAgentStore = new PgBackgroundAgentStore(q);
+        }
+        else if (mysqlPool) {
+            await ensureTiDBBackgroundAgentSchema(mysqlPool);
+            backgroundAgentStore = new TiDBBackgroundAgentStore(mysqlPool);
+        }
+        else if (backend?.kind === "local") {
+            backgroundAgentStore = new FileBackgroundAgentStore(config.localDataRoot ?? localRoot);
+            if (config.requirePrincipal === true) {
+                logger.warn("background_agent_store_local_multitenant", { note: "file-store reap only sweeps scope 'default'; principal-scoped durable rows are never reaped on the local backend (scope enumeration = core ask)" });
+            }
+        }
+        if (backgroundAgentStore)
+            logger.info("background_agent_store_enabled", { backend: pgPool ? "pg" : mysqlPool ? "tidb" : "file" });
+    }
     const memoryExportBackend = config.memoryEngineBackend !== "file" ? memoryEngine?.backend : undefined;
     const memoryLane = memoryEngineRemoteLanePosture(config);
     if (config.memoryEngineBackend === "file" || !config.memoryEngineEnabled) {
@@ -910,6 +933,7 @@ async function main() {
     const runnerDeps = {
         promptSource,
         ...(rosterStore ? { rosterStore } : {}),
+        ...(backgroundAgentStore ? { backgroundAgentStore } : {}),
         brain,
         models: config.models,
         roles: config.roles,
@@ -1098,7 +1122,7 @@ async function main() {
     if (sessionTitler)
         logger.info("session_auto_title_enabled", {});
     const workflowsCapable = workflowsCapability(runnerDeps);
-    const subRunner = new Runner({ brain, models: config.models, roles: config.roles, pricing, tracer, promptSource, sessionStore: new ForkRoutingSessionStore(sessionStore, new TtlSessionStore({ defaultTtlDays: 1 / 24 })), ...(executionEnvFactory ? { executionEnvFactory } : {}), ...(lspManager ? { lspManager } : {}) });
+    const subRunner = new Runner({ brain, models: config.models, roles: config.roles, pricing, tracer, promptSource, sessionStore: new ForkRoutingSessionStore(sessionStore, new TtlSessionStore({ defaultTtlDays: 1 / 24 })), ...(executionEnvFactory ? { executionEnvFactory } : {}), ...(lspManager ? { lspManager } : {}), ...(backgroundAgentStore ? { backgroundAgentStore } : {}) });
     const authorize = createAuthorizer(config, sessionStore);
     const runStore = backend ? backend.run() : undefined;
     const resumeAnchorStore = backend ? backend.resumeAnchor() : undefined;
@@ -1198,6 +1222,7 @@ async function main() {
     const scenarioDeps = {
         runner, subRunner, model: "default", skills, repoClient, requirePrincipal: config.requirePrincipal,
         ...(webSearch ? { webSearch } : {}), metrics, logger,
+        ...(backgroundAgentStore ? { backgroundAgentStore } : {}),
         oaApiBaseUrl: config.oaApiBaseUrl, oaServiceToken: config.oaServiceToken, oaIssue: config.oaIssue,
         brandIdentity: config.configProvider === "local",
         ...(sendUserFileToolSpec
@@ -1228,6 +1253,27 @@ async function main() {
         }
     };
     let wfRunReapInFlight = false;
+    let bgAgentReapInFlight = false;
+    const reapBgAgents = backgroundAgentStore
+        ? async () => {
+            const store = backgroundAgentStore;
+            const scopes = await (store.listReapScopes?.() ?? Promise.resolve(["default"]));
+            let rows = 0, sessions = 0, skipped = 0;
+            for (const scope of scopes) {
+                const r = await defaultTaskRegistry.reapDurableAgents(scope, { store, sessions: subRunner.sessions }, {
+                    maxAgeMs: config.backgroundAgentRetentionMs,
+                    staleRunningMaxAgeMs: config.backgroundAgentStaleRunningMs,
+                });
+                rows += r.rowsReaped;
+                sessions += r.sessionsReleased;
+                skipped += r.skippedNoSessions;
+            }
+            if (rows > 0 || sessions > 0) {
+                metrics.inc("bg_agents_reaped_total", {}, rows);
+                logger.info("reaper_swept", { metric: "bg_agents_reaped_total", count: rows, sessionsReleased: sessions, skippedNoSessions: skipped });
+            }
+        }
+        : undefined;
     const probeClockSkew = async () => {
         try {
             const dbMs = await backend?.dbNowMs?.();
@@ -1273,6 +1319,10 @@ async function main() {
                 ?.reapAcked?.(Date.now() - config.workflowRunRetentionMs)
                 .catch(() => undefined);
         }
+        if (reapBgAgents && !bgAgentReapInFlight) {
+            bgAgentReapInFlight = true;
+            void reapBgAgents().catch(() => undefined).finally(() => (bgAgentReapInFlight = false));
+        }
         if (config.scratchpadSweepTtlMs > 0) {
             void sweepStaleScratchpads(config.localDataRoot ?? localRoot, { olderThanMs: config.scratchpadSweepTtlMs })
                 .then((removed) => {