update

src/main/orchestration/heartbeat.ts

@@ -1,12 +1,14 @@
 import { randomUUID } from "node:crypto";
 import type { AppDatabase } from "../database";
+import type { AgentRecord, ConnectorRecord } from "@shared/types";
 import type { OrchestrationDependencies } from "./types";
+import type { RunDispatchParams } from "./turn-context";
 
 export function createHeartbeat(
   deps: OrchestrationDependencies,
   resolveWorkspace: (task: { workspaceId: string | null; projectId?: string | null }, agent: { workspaceId: string | null }, companyId: string) => { workspace: ReturnType<AppDatabase["getWorkspace"]>; lockMode: "exclusive" | "shared" | "none"; source: string },
-  getRunEligibilityFailure: (agent: any, connector: any, projectId?: string | null) => string | null,
-  dispatchRunToWorker: (params: any) => void,
+  getRunEligibilityFailure: (agent: AgentRecord, connector: ConnectorRecord, projectId?: string | null) => string | null,
+  dispatchRunToWorker: (params: RunDispatchParams) => void,
   hasActiveRun: (filter: { taskId?: string; agentId?: string; companyId: string }) => boolean,
 ) {
   const {
@@ -224,12 +226,14 @@ export function createHeartbeat(
         logger.warn(`[wake] Refusing cross-company wake for agent ${agentId}: ${agent.companyId} !== ${companyId}`);
         return null;
       }
-    } catch {
+    } catch (error) {
+      logger.warn(`[wake] Failed to load agent ${agentId}:`, error);
       return null;
     }
     try {
       ensureAgentHeartbeatEnabled(agentId);
-    } catch {
+    } catch (error) {
+      logger.warn(`[wake] Failed to ensure heartbeat for agent ${agentId}:`, error);
       return null;
     }
 

src/main/orchestration/scheduler.ts

@@ -1,5 +1,6 @@
 import type { OrchestrationDependencies } from "./types";
 import { sweepStaleReviews } from "../task-quality-gates";
+import type { SweepStaleReviewsDeps } from "../task-quality-gates";
 import { resolveApprovalDecision } from "../approval-policy";
 import { dispatchAgentMessage } from "../message-dispatch";
 
@@ -134,12 +135,12 @@ export function createScheduler(
 
             // TASK-04: Stale review escalation sweep
             try {
-              const escalated = sweepStaleReviews(company.id, {
-                db: db as unknown as import("../task-quality-gates").SweepStaleReviewsDeps["db"],
+              const sweepDeps: SweepStaleReviewsDeps = {
+                db,
+                dispatchAgentMessage,
+                resolveApprovalDecision,
+                notify,
                 getCompanyExecutionMode: (cId: string) => db.getCompanyExecutionMode(cId),
-                dispatchAgentMessage: dispatchAgentMessage as unknown as import("../task-quality-gates").SweepStaleReviewsDeps["dispatchAgentMessage"],
-                resolveApprovalDecision: resolveApprovalDecision as unknown as import("../task-quality-gates").SweepStaleReviewsDeps["resolveApprovalDecision"],
-                notify: notify as unknown as import("../task-quality-gates").SweepStaleReviewsDeps["notify"],
                 handleTaskStatusChange: (taskId: string, cId: string, previousStatus: string, newStatus: string) => {
                   deps.dispatchAutomation("task_status_changed", { companyId: cId, taskId, previousStatus, newStatus });
                   publishDomainChanged();
@@ -150,11 +151,11 @@ export function createScheduler(
                 },
                 emitEvent: deps.emitEvent,
                 publishDomainChanged,
-                wakeAgentIfPossible: (agentId: string, cId: string, trigger: string) => {
-                  wakeAgentIfPossible(agentId, cId, trigger);
-                },
+                wakeAgentIfPossible: (agentId: string | null | undefined, cId: string, trigger: string) =>
+                  agentId != null ? wakeAgentIfPossible(agentId, cId, trigger) ?? null : null,
                 logger,
-              });
+              };
+              const escalated = sweepStaleReviews(company.id, sweepDeps);
               if (escalated > 0) {
                 logger.info(`[heartbeat] Auto-escalated ${escalated} stale review(s) for company ${company.name}`);
                 publishDomainChanged();

src/main/orchestration/task-runner.ts

@@ -1,13 +1,15 @@
 import { randomUUID } from "node:crypto";
 import type { AppDatabase } from "../database";
+import type { AgentRecord, ConnectorRecord } from "@shared/types";
 import type { DesktopResult } from "@shared/contracts";
 import type { OrchestrationDependencies } from "./types";
+import type { RunDispatchParams } from "./turn-context";
 
 export function createTaskRunner(
   deps: OrchestrationDependencies,
   resolveWorkspace: (task: { workspaceId: string | null; projectId?: string | null }, agent: { workspaceId: string | null }, companyId: string) => { workspace: ReturnType<AppDatabase["getWorkspace"]>; lockMode: "exclusive" | "shared" | "none"; source: string },
-  getRunEligibilityFailure: (agent: any, connector: any, projectId?: string | null) => string | null,
-  dispatchRunToWorker: (params: any) => void,
+  getRunEligibilityFailure: (agent: AgentRecord, connector: ConnectorRecord, projectId?: string | null) => string | null,
+  dispatchRunToWorker: (params: RunDispatchParams) => void,
   hasActiveRun: (filter: { taskId?: string; agentId?: string; companyId: string }) => boolean,
 ) {
   const {

src/main/orchestration/types.ts

@@ -1,4 +1,5 @@
 import type { AppDatabase } from "../database";
+import type { DesktopEvent } from "@shared/types";
 import type { NotificationOptions } from "../notification-service";
 import type { createBudgetEnforcementService } from "../budget-enforcement-service";
 import type { RuntimeSkillEntry } from "../skill-types";
@@ -54,7 +55,7 @@ export interface OrchestrationDependencies {
     error: (...args: unknown[]) => void;
   };
   publishDomainChanged: () => void;
-  emitEvent: (event: any) => void;
+  emitEvent: (event: DesktopEvent) => void;
   notify: (options: NotificationOptions) => void;
   createRunToken: (agentId: string, companyId: string, runId: string) => string;
   getApiPort: () => number;