bopodev-api 0.1.32 → 0.1.34

package/src/services/heartbeat-service/heartbeat-run.ts

@@ -9,6 +9,7 @@ import {
   ControlPlaneRequestHeadersSchema,
   ControlPlaneRuntimeEnvSchema,
   ExecutionOutcomeSchema,
+  mergeBuiltinSkillIdsForInjection,
   type ExecutionOutcome,
   type RunArtifact,
   type RunCompletionReason,
@@ -59,6 +60,7 @@ import { createHeartbeatRunsRealtimeEvent, loadHeartbeatRunsRealtimeSnapshot } f
 import { publishAttentionSnapshot } from "../../realtime/attention";
 import { publishOfficeOccupantForAgent } from "../../realtime/office-space";
 import { appendProjectBudgetUsage, checkAgentBudget, checkProjectBudget } from "../budget-service";
+import { materializeLinkedSkillsForRuntime } from "../company-skill-file-service";
 import { appendDurableFact, loadAgentMemoryContext, persistHeartbeatMemory } from "../memory-file-service";
 import { calculateModelPricedUsdCost } from "../model-pricing";
 import { runPluginHook } from "../plugin-runtime";
@@ -527,6 +529,7 @@ export async function runHeartbeatForAgent(
   let transcriptLiveHighSignalCount = 0;
   let transcriptPersistFailureReported = false;
   let pluginFailureSummary: string[] = [];
+  let linkedSkillsMaterializationCleanup: (() => Promise<void>) | null = null;
   const seenResultMessages = new Set<string>();
   const runDigestSignals: RunDigestSignal[] = [];
 
@@ -706,13 +709,24 @@ export async function runHeartbeatForAgent(
     if (runMode === "redo") {
       state = clearResumeState(state);
     }
-    const heartbeatRuntimeEnv = buildHeartbeatRuntimeEnv({
-      companyId,
-      agentId: agent.id,
-      heartbeatRunId: runId,
-      canHireAgents: agent.canHireAgents,
-      wakeContext: options?.wakeContext
+    const mergedSkillIds = mergeRuntimeForExecution(
+      { enabledSkillIds: persistedRuntime.enabledSkillIds },
+      state.runtime
+    ).enabledSkillIds;
+    const linkedMaterialized = await materializeLinkedSkillsForRuntime(companyId, {
+      enabledSkillIds: mergedSkillIds
     });
+    linkedSkillsMaterializationCleanup = linkedMaterialized.cleanup;
+    const heartbeatRuntimeEnv: Record<string, string> = {
+      ...buildHeartbeatRuntimeEnv({
+        companyId,
+        agentId: agent.id,
+        heartbeatRunId: runId,
+        canHireAgents: agent.canHireAgents,
+        wakeContext: options?.wakeContext
+      }),
+      ...(linkedMaterialized.root ? { BOPODEV_MATERIALIZED_LINKED_SKILLS_ROOT: linkedMaterialized.root } : {})
+    };
     const runtimeFromConfig = {
       command: persistedRuntime.runtimeCommand,
       args: persistedRuntime.runtimeArgs,
@@ -745,15 +759,27 @@ export async function runHeartbeatForAgent(
       thinkingEffort: persistedRuntime.runtimeThinkingEffort,
       bootstrapPrompt: persistedRuntime.bootstrapPrompt,
       interruptGraceSec: persistedRuntime.interruptGraceSec,
-      runPolicy: persistedRuntime.runPolicy
+      runPolicy: persistedRuntime.runPolicy,
+      enabledSkillIds: persistedRuntime.enabledSkillIds
     };
     const mergedRuntime = mergeRuntimeForExecution(runtimeFromConfig, state.runtime);
+    const runtimeForWorkspace = {
+      ...mergedRuntime,
+      env: {
+        ...(mergedRuntime.env ?? {}),
+        ...(mergedRuntime.enabledSkillIds !== undefined
+          ? {
+              BOPODEV_ENABLED_SKILL_IDS: mergeBuiltinSkillIdsForInjection(mergedRuntime.enabledSkillIds)!.join(",")
+            }
+          : {})
+      }
+    };
     const workspaceResolution = await resolveRuntimeWorkspaceForWorkItems(
       db,
       companyId,
       agent.id,
       contextWorkItems,
-      mergedRuntime
+      runtimeForWorkspace
     );
     await mkdir(join(resolveAgentFallbackWorkspace(companyId, agent.id), "operating"), { recursive: true });
     state = {
@@ -1735,6 +1761,14 @@ export async function runHeartbeatForAgent(
       });
     }
   } finally {
+    if (linkedSkillsMaterializationCleanup) {
+      try {
+        await linkedSkillsMaterializationCleanup();
+      } catch {
+        // Best effort: linked skill temp dir cleanup should not mask heartbeat teardown.
+      }
+      linkedSkillsMaterializationCleanup = null;
+    }
     await transcriptWriteQueue;
     if (discardIdleNoWorkRunAfterFlush) {
       try {
@@ -3380,6 +3414,7 @@ function parseAgentState(stateBlob: string | null) {
           model?: string;
           thinkingEffort?: "auto" | "low" | "medium" | "high";
           bootstrapPrompt?: string;
+          enabledSkillIds?: string[];
           runPolicy?: {
             sandboxMode?: "workspace_write" | "full_access";
             allowWebSearch?: boolean;
@@ -3924,6 +3959,7 @@ function mergeRuntimeForExecution(
         model?: string;
         thinkingEffort?: "auto" | "low" | "medium" | "high";
         bootstrapPrompt?: string;
+        enabledSkillIds?: string[];
         runPolicy?: {
           sandboxMode?: "workspace_write" | "full_access";
           allowWebSearch?: boolean;
@@ -3943,6 +3979,7 @@ function mergeRuntimeForExecution(
         model?: string;
         thinkingEffort?: "auto" | "low" | "medium" | "high";
         bootstrapPrompt?: string;
+        enabledSkillIds?: string[];
         runPolicy?: {
           sandboxMode?: "workspace_write" | "full_access";
           allowWebSearch?: boolean;