@vellumai/assistant 0.4.43 → 0.4.45

package/src/daemon/mcp-reload-service.ts

@@ -0,0 +1,123 @@
+/**
+ * Shared MCP reload business logic.
+ *
+ * Used by the HTTP route (`runtime/routes/mcp-routes.ts`) so the reload
+ * behaviour is defined in exactly one place.
+ */
+
+import { getConfig, invalidateConfigCache } from "../config/loader.js";
+import { getMcpServerManager } from "../mcp/manager.js";
+import { createMcpToolsFromServer } from "../tools/mcp/mcp-tool-factory.js";
+import { registerMcpTools, unregisterAllMcpTools } from "../tools/registry.js";
+import { getLogger } from "../util/logger.js";
+
+const log = getLogger("mcp-reload-service");
+
+/** Per-server reload result. */
+export interface McpReloadServerResult {
+  id: string;
+  connected: boolean;
+  /** True when the server is explicitly disabled in config. */
+  disabled?: boolean;
+  toolCount: number;
+  tools: string[];
+}
+
+export interface McpReloadResult {
+  success: boolean;
+  serverCount?: number;
+  toolCount?: number;
+  servers?: McpReloadServerResult[];
+  error?: string;
+}
+
+let reloadInProgress: Promise<McpReloadResult> | null = null;
+
+/**
+ * Stop all MCP servers, reload configuration from disk, and restart
+ * servers with the updated config. Returns a summary of the reload.
+ *
+ * Concurrent calls are serialized — if a reload is already in progress
+ * the caller receives the same promise instead of starting a second one.
+ */
+export function reloadMcpServers(): Promise<McpReloadResult> {
+  if (reloadInProgress) {
+    log.info("MCP reload already in progress, awaiting existing operation");
+    return reloadInProgress;
+  }
+  reloadInProgress = doReload().finally(() => {
+    reloadInProgress = null;
+  });
+  return reloadInProgress;
+}
+
+async function doReload(): Promise<McpReloadResult> {
+  try {
+    const manager = getMcpServerManager();
+
+    // 1. Validate new config before tearing down existing servers.
+    //    If the config is broken we abort early, preserving the current
+    //    working MCP setup instead of leaving zero servers.
+    invalidateConfigCache();
+    const config = getConfig();
+
+    // 2. Stop existing MCP servers + unregister their tools
+    await manager.stop();
+    unregisterAllMcpTools();
+    const serverIds = config.mcp?.servers
+      ? Object.keys(config.mcp.servers)
+      : [];
+
+    // 3. Restart MCP servers
+    let serverCount = 0;
+    let toolCount = 0;
+    const servers: McpReloadServerResult[] = [];
+
+    if (config.mcp?.servers && Object.keys(config.mcp.servers).length > 0) {
+      const serverToolInfos = await manager.start(config.mcp);
+      for (const { serverId, serverConfig, tools } of serverToolInfos) {
+        const mcpTools = createMcpToolsFromServer(
+          tools,
+          serverId,
+          serverConfig,
+          manager,
+        );
+        const accepted = registerMcpTools(mcpTools);
+        const acceptedNames = accepted.map((t) => t.name);
+        toolCount += accepted.length;
+        servers.push({
+          id: serverId,
+          connected: true,
+          toolCount: accepted.length,
+          tools: acceptedNames,
+        });
+      }
+      // Include servers that were configured but failed to connect or are disabled
+      for (const id of serverIds) {
+        if (!servers.some((s) => s.id === id)) {
+          const serverConfig = config.mcp!.servers![id];
+          const isDisabled = serverConfig?.enabled === false;
+          servers.push({
+            id,
+            connected: false,
+            disabled: isDisabled || undefined,
+            toolCount: 0,
+            tools: [],
+          });
+        }
+      }
+      serverCount = servers.length;
+    }
+
+    // Sessions pick up new MCP tools automatically on their next turn
+    // via the dynamic resolver in createResolveToolsCallback — no need
+    // to evict sessions.
+
+    log.info({ serverCount, toolCount }, "MCP servers reloaded");
+    return { success: true, serverCount, toolCount, servers };
+  } catch (err) {
+    const error = err instanceof Error ? err.message : String(err);
+    log.error({ err }, "MCP reload failed");
+    return { success: false, error };
+  }
+}

package/src/daemon/message-protocol.ts

@@ -20,6 +20,8 @@ export * from "./message-types/contacts.js";
 export * from "./message-types/diagnostics.js";
 export * from "./message-types/documents.js";
 export * from "./message-types/guardian-actions.js";
+export * from "./message-types/host-bash.js";
+export * from "./message-types/host-file.js";
 export * from "./message-types/inbox.js";
 export * from "./message-types/integrations.js";
 export * from "./message-types/memory.js";
@@ -66,6 +68,8 @@ import type {
   _GuardianActionsClientMessages,
   _GuardianActionsServerMessages,
 } from "./message-types/guardian-actions.js";
+import type { _HostBashServerMessages } from "./message-types/host-bash.js";
+import type { _HostFileServerMessages } from "./message-types/host-file.js";
 import type {
   _InboxClientMessages,
   _InboxServerMessages,
@@ -175,6 +179,8 @@ export type ServerMessage =
   | _SubagentsServerMessages
   | _DocumentsServerMessages
   | _GuardianActionsServerMessages
+  | _HostBashServerMessages
+  | _HostFileServerMessages
   | _MemoryServerMessages
   | _WorkspaceServerMessages
   | _SchedulesServerMessages

package/src/daemon/message-types/apps.ts

@@ -18,12 +18,6 @@ export interface AppsListRequest {
   type: "apps_list";
 }
 
-export interface HomeBaseGetRequest {
-  type: "home_base_get";
-  /** If true, daemon ensures a durable Home Base link exists before responding. */
-  ensureLinked?: boolean;
-}
-
 export interface AppOpenRequest {
   type: "app_open_request";
   appId: string;
@@ -176,23 +170,6 @@ export interface AppsListResponse {
   }>;
 }
 
-export interface HomeBaseGetResponse {
-  type: "home_base_get_response";
-  homeBase: {
-    appId: string;
-    source: string;
-    starterTasks: string[];
-    onboardingTasks: string[];
-    preview: {
-      title: string;
-      subtitle: string;
-      description: string;
-      icon: string;
-      metrics: Array<{ label: string; value: string }>;
-    };
-  } | null;
-}
-
 export interface SharedAppsListResponse {
   type: "shared_apps_list_response";
   apps: Array<{
@@ -361,7 +338,6 @@ export interface AppFilesChanged {
 export type _AppsClientMessages =
   | AppDataRequest
   | AppsListRequest
-  | HomeBaseGetRequest
   | AppOpenRequest
   | SharedAppsListRequest
   | AppDeleteRequest
@@ -386,7 +362,6 @@ export type _AppsClientMessages =
 export type _AppsServerMessages =
   | AppDataResponse
   | AppsListResponse
-  | HomeBaseGetResponse
   | SharedAppsListResponse
   | AppDeleteResponse
   | SharedAppDeleteResponse

package/src/daemon/message-types/browser.ts

@@ -1,5 +1,5 @@
 // Browser interaction types.
-// CDP request/response IPC was removed — Playwright's connectOverCDP is broken
+// CDP request/response messaging was removed — Playwright's connectOverCDP is broken
 // under Bun's runtime. Browser is now launched directly via Playwright.
 
 // --- Domain-level union aliases (consumed by the barrel file) ---

package/src/daemon/message-types/computer-use.ts

@@ -1,9 +1,6 @@
 // Computer use, task routing, ride shotgun, and watch observation types.
 
-import type {
-  CommandIntent,
-  UserMessageAttachment,
-} from "./shared.js";
+import type { CommandIntent, UserMessageAttachment } from "./shared.js";
 
 // === Client → Server ===
 

package/src/daemon/message-types/guardian-actions.ts

@@ -1,4 +1,4 @@
-// Guardian action decision IPC types.
+// Guardian action decision message types.
 // Enables desktop clients to fetch pending guardian prompts and submit
 // button decisions deterministically (without text parsing).
 

package/src/daemon/message-types/host-bash.ts

@@ -0,0 +1,18 @@
+// Host bash proxy types.
+// Enables proxying shell commands to the desktop client (host machine)
+// when running as a managed assistant.
+
+// === Server → Client ===
+
+export interface HostBashRequest {
+  type: "host_bash_request";
+  requestId: string;
+  sessionId: string;
+  command: string;
+  working_dir?: string;
+  timeout_seconds?: number;
+}
+
+// --- Domain-level union aliases (consumed by the barrel file) ---
+
+export type _HostBashServerMessages = HostBashRequest;

package/src/daemon/message-types/host-file.ts

@@ -0,0 +1,44 @@
+// Host file proxy types.
+// Enables proxying file operations to the desktop client (host machine)
+// when running as a managed assistant.
+
+// === Server → Client ===
+
+export interface HostFileReadRequest {
+  type: "host_file_request";
+  requestId: string;
+  sessionId: string;
+  operation: "read";
+  path: string;
+  offset?: number;
+  limit?: number;
+}
+
+export interface HostFileWriteRequest {
+  type: "host_file_request";
+  requestId: string;
+  sessionId: string;
+  operation: "write";
+  path: string;
+  content: string;
+}
+
+export interface HostFileEditRequest {
+  type: "host_file_request";
+  requestId: string;
+  sessionId: string;
+  operation: "edit";
+  path: string;
+  old_string: string;
+  new_string: string;
+  replace_all?: boolean;
+}
+
+export type HostFileRequest =
+  | HostFileReadRequest
+  | HostFileWriteRequest
+  | HostFileEditRequest;
+
+// --- Domain-level union aliases (consumed by the barrel file) ---
+
+export type _HostFileServerMessages = HostFileRequest;

package/src/daemon/message-types/integrations.ts

@@ -1,4 +1,4 @@
-// External service integrations: Slack, Telegram, Twitter, Vercel, ingress, guardian.
+// External service integrations: Slack, Telegram, Vercel, ingress, guardian.
 
 import type { ChannelId } from "../../channels/types.js";
 
@@ -29,22 +29,6 @@ export interface VercelApiConfigRequest {
   apiToken?: string;
 }
 
-export interface TwitterIntegrationConfigRequest {
-  type: "twitter_integration_config";
-  action:
-    | "get"
-    | "set_mode"
-    | "set_local_client"
-    | "clear_local_client"
-    | "disconnect"
-    | "get_strategy"
-    | "set_strategy";
-  mode?: "local_byo" | "managed";
-  clientId?: string;
-  clientSecret?: string;
-  strategy?: string;
-}
-
 export interface TelegramConfigRequest {
   type: "telegram_config";
   action: "get" | "set" | "clear" | "set_commands" | "setup";
@@ -73,14 +57,6 @@ export interface ChannelVerificationSessionRequest {
   contactChannelId?: string;
 }
 
-export interface TwitterAuthStartRequest {
-  type: "twitter_auth_start";
-}
-
-export interface TwitterAuthStatusRequest {
-  type: "twitter_auth_status";
-}
-
 export interface IntegrationListRequest {
   type: "integration_list";
 }
@@ -140,31 +116,6 @@ export interface VercelApiConfigResponse {
   error?: string;
 }
 
-export interface ManagedPrerequisites {
-  /** Whether twitter.integrationMode is set to "managed" in config. */
-  integrationModeManaged: boolean;
-  /** Whether the assistant API key credential exists in secure storage. */
-  assistantApiKeyPresent: boolean;
-  /** Whether the platform base URL is configured (platform registration resolvable). */
-  platformAssistantIdResolvable: boolean;
-}
-
-export interface TwitterIntegrationConfigResponse {
-  type: "twitter_integration_config_response";
-  success: boolean;
-  mode?: "local_byo" | "managed";
-  managedAvailable: boolean;
-  /** Detailed prerequisite status for managed Twitter availability. */
-  managedPrerequisites?: ManagedPrerequisites;
-  localClientConfigured: boolean;
-  connected: boolean;
-  accountInfo?: string;
-  strategy?: "oauth" | "browser" | "auto";
-  /** Whether the user has explicitly set a strategy (vs. relying on the default 'auto'). */
-  strategyConfigured?: boolean;
-  error?: string;
-}
-
 export interface TelegramConfigResponse {
   type: "telegram_config_response";
   success: boolean;
@@ -217,23 +168,6 @@ export interface ChannelVerificationSessionResponse {
   pendingBootstrap?: boolean;
 }
 
-export interface TwitterAuthResult {
-  type: "twitter_auth_result";
-  success: boolean;
-  accountInfo?: string;
-  /** Machine-readable error code for programmatic handling (e.g. "managed_missing_api_key", "managed_auth_via_platform"). */
-  errorCode?: string;
-  error?: string;
-}
-
-export interface TwitterAuthStatusResponse {
-  type: "twitter_auth_status_response";
-  connected: boolean;
-  accountInfo?: string;
-  mode?: "local_byo" | "managed";
-  error?: string;
-}
-
 export interface IntegrationListResponse {
   type: "integration_list_response";
   integrations: Array<{
@@ -283,11 +217,8 @@ export type _IntegrationsClientMessages =
   | IngressConfigRequest
   | PlatformConfigRequest
   | VercelApiConfigRequest
-  | TwitterIntegrationConfigRequest
   | TelegramConfigRequest
   | ChannelVerificationSessionRequest
-  | TwitterAuthStartRequest
-  | TwitterAuthStatusRequest
   | IntegrationListRequest
   | IntegrationConnectRequest
   | IntegrationDisconnectRequest
@@ -299,11 +230,8 @@ export type _IntegrationsServerMessages =
   | IngressConfigResponse
   | PlatformConfigResponse
   | VercelApiConfigResponse
-  | TwitterIntegrationConfigResponse
   | TelegramConfigResponse
   | ChannelVerificationSessionResponse
-  | TwitterAuthResult
-  | TwitterAuthStatusResponse
   | IntegrationListResponse
   | IntegrationConnectResult
   | OAuthConnectResultResponse

package/src/daemon/message-types/messages.ts

@@ -95,11 +95,20 @@ export interface ToolOutputChunk {
   subToolId?: string;
 }
 
+export interface ToolUsePreviewStart {
+  type: "tool_use_preview_start";
+  toolUseId: string;
+  toolName: string;
+  sessionId?: string;
+}
+
 export interface ToolInputDelta {
   type: "tool_input_delta";
   toolName: string;
   content: string;
   sessionId?: string;
+  /** The tool_use block ID for client-side correlation. */
+  toolUseId?: string;
 }
 
 export interface ToolResult {
@@ -117,6 +126,8 @@ export interface ToolResult {
   sessionId?: string;
   /** Base64-encoded image data extracted from contentBlocks (e.g. browser_screenshot). */
   imageData?: string;
+  /** The tool_use block ID for client-side correlation. */
+  toolUseId?: string;
 }
 
 export interface ConfirmationRequest {
@@ -169,6 +180,8 @@ export interface MessageComplete {
   type: "message_complete";
   sessionId?: string;
   attachments?: UserMessageAttachment[];
+  /** Database ID of the persisted assistant message, if any. */
+  messageId?: string;
 }
 
 export interface ErrorMessage {
@@ -270,6 +283,7 @@ export interface AssistantActivityState {
     | "thinking_delta"
     | "first_text_delta"
     | "tool_use_start"
+    | "preview_start"
     | "tool_result_received"
     | "confirmation_requested"
     | "confirmation_resolved"
@@ -325,6 +339,7 @@ export type _MessagesServerMessages =
   | AssistantTextDelta
   | AssistantThinkingDelta
   | ToolUseStart
+  | ToolUsePreviewStart
   | ToolOutputChunk
   | ToolInputDelta
   | ToolResult

package/src/daemon/message-types/schedules.ts

@@ -1,4 +1,4 @@
-// Schedule, reminder, watcher, and heartbeat types.
+// Schedule, watcher, and heartbeat types.
 
 // === Client → Server ===
 
@@ -6,15 +6,6 @@ export interface SchedulesList {
   type: "schedules_list";
 }
 
-export interface RemindersList {
-  type: "reminders_list";
-}
-
-export interface ReminderCancel {
-  type: "reminder_cancel";
-  id: string;
-}
-
 export interface ScheduleToggle {
   type: "schedule_toggle";
   id: string;
@@ -26,6 +17,11 @@ export interface ScheduleRemove {
   id: string;
 }
 
+export interface ScheduleCancel {
+  type: "schedule_cancel";
+  id: string;
+}
+
 export interface ScheduleRunNow {
   type: "schedule_run_now";
   id: string;
@@ -67,28 +63,18 @@ export interface SchedulesListResponse {
     name: string;
     enabled: boolean;
     syntax: string;
-    expression: string;
-    cronExpression: string;
+    expression: string | null;
+    cronExpression: string | null;
     timezone: string | null;
     message: string;
     nextRunAt: number;
     lastRunAt: number | null;
     lastStatus: string | null;
     description: string;
-  }>;
-}
-
-export interface RemindersListResponse {
-  type: "reminders_list_response";
-  reminders: Array<{
-    id: string;
-    label: string;
-    message: string;
-    fireAt: number;
     mode: string;
     status: string;
-    firedAt: number | null;
-    createdAt: number;
+    routingIntent: string;
+    isOneShot: boolean;
   }>;
 }
 
@@ -144,9 +130,8 @@ export type _SchedulesClientMessages =
   | SchedulesList
   | ScheduleToggle
   | ScheduleRemove
+  | ScheduleCancel
   | ScheduleRunNow
-  | RemindersList
-  | ReminderCancel
   | HeartbeatConfig
   | HeartbeatRunsList
   | HeartbeatRunNow
@@ -155,7 +140,6 @@ export type _SchedulesClientMessages =
 
 export type _SchedulesServerMessages =
   | SchedulesListResponse
-  | RemindersListResponse
   | HeartbeatAlert
   | HeartbeatConfigResponse
   | HeartbeatRunsListResponse

package/src/daemon/message-types/sessions.ts

@@ -273,6 +273,8 @@ export interface GenerationHandoff {
   requestId?: string;
   queuedCount: number;
   attachments?: UserMessageAttachment[];
+  /** Database ID of the persisted assistant message, if any. */
+  messageId?: string;
 }
 
 export interface ModelInfo {
@@ -304,7 +306,12 @@ export interface HistoryResponseSurface {
   surfaceType: string;
   title?: string;
   data: Record<string, unknown>;
-  actions?: Array<{ id: string; label: string; style?: string }>;
+  actions?: Array<{
+    id: string;
+    label: string;
+    style?: string;
+    data?: Record<string, unknown>;
+  }>;
   display?: string;
 }
 
@@ -388,7 +395,6 @@ export type SessionErrorCode =
   | "PROVIDER_API"
   | "PROVIDER_BILLING"
   | "CONTEXT_TOO_LARGE"
-  | "QUEUE_FULL"
   | "SESSION_ABORTED"
   | "SESSION_PROCESSING_FAILED"
   | "REGENERATE_FAILED"

package/src/daemon/message-types/settings.ts

@@ -2,7 +2,7 @@
 
 // === Client → Server ===
 
-/** Request from a session or IPC client to change the voice activation key. */
+/** Request from a session or client to change the voice activation key. */
 export interface VoiceConfigUpdateRequest {
   type: "voice_config_update";
   /** The desired activation key (enum value or natural-language name). */

package/src/daemon/message-types/shared.ts

@@ -1,4 +1,4 @@
-// Shared types used across multiple IPC domains.
+// Shared types used across multiple message domains.
 
 export type ThreadType = "standard" | "private";
 

package/src/daemon/message-types/surfaces.ts

@@ -23,6 +23,8 @@ export interface SurfaceAction {
   id: string;
   label: string;
   style?: "primary" | "secondary" | "destructive";
+  /** Optional data payload returned to the daemon when this action is clicked. */
+  data?: Record<string, unknown>;
 }
 
 export interface CardSurfaceData {