ff

Author: Sg312

apps/sim/lib/copilot/tools/handlers/function-execute.ts

@@ -2,6 +2,7 @@ import { createLogger } from '@sim/logger'
 import { decodeVfsPathSegments, encodeVfsPathSegments } from '@/lib/copilot/vfs/path-utils'
 import { resolveWorkflowAliasForWorkspace } from '@/lib/copilot/vfs/workflow-alias-resolver'
 import { isPlanAliasPath, workflowAliasSandboxPath } from '@/lib/copilot/vfs/workflow-aliases'
+import { isMothershipBetaFeaturesEnabled } from '@/lib/core/config/feature-flags'
 import { getTableById, listTables, queryRows } from '@/lib/table/service'
 import { listWorkspaceFileFolders } from '@/lib/uploads/contexts/workspace/workspace-file-folder-manager'
 import {
@@ -71,7 +72,9 @@ async function resolveInputFiles(
   let totalSize = 0
 
   if (inputFiles?.length && workspaceId) {
-    const allFiles = await listWorkspaceFiles(workspaceId, { includeReservedSystemFiles: true })
+    const allFiles = await listWorkspaceFiles(workspaceId, {
+      includeReservedSystemFiles: isMothershipBetaFeaturesEnabled,
+    })
     for (const fileRef of inputFiles) {
       const filePath =
         typeof fileRef === 'string'
@@ -124,11 +127,11 @@ async function resolveInputFiles(
 
   if (inputDirectories?.length && workspaceId) {
     const folders = await listWorkspaceFileFolders(workspaceId, {
-      includeReservedSystemFolders: true,
+      includeReservedSystemFolders: isMothershipBetaFeaturesEnabled,
     })
     const allFiles = await listWorkspaceFiles(workspaceId, {
       folders,
-      includeReservedSystemFiles: true,
+      includeReservedSystemFiles: isMothershipBetaFeaturesEnabled,
     })
     for (const dirRef of inputDirectories) {
       const dirPath =

apps/sim/lib/copilot/tools/server/files/touch-plan.test.ts

@@ -18,6 +18,10 @@ vi.mock('@/lib/copilot/vfs/resource-writer', () => ({
   writeWorkspaceFileByPath: mocks.writeWorkspaceFileByPath,
 }))
 
+vi.mock('@/lib/core/config/feature-flags', () => ({
+  isMothershipBetaFeaturesEnabled: true,
+}))
+
 import { touchPlanServerTool } from './touch-plan'
 
 describe('touch_plan server tool', () => {

apps/sim/lib/copilot/tools/server/files/touch-plan.ts

@@ -12,6 +12,7 @@ import {
 } from '@/lib/copilot/vfs/path-utils'
 import { writeWorkspaceFileByPath } from '@/lib/copilot/vfs/resource-writer'
 import { resolveWorkflowAliasForWorkspace } from '@/lib/copilot/vfs/workflow-alias-resolver'
+import { isMothershipBetaFeaturesEnabled } from '@/lib/core/config/feature-flags'
 
 const logger = createLogger('TouchPlanServerTool')
 const TOUCH_PLAN_TOOL_ID = 'touch_plan'
@@ -64,6 +65,9 @@ function normalizePlanRelativePath(name: string): string {
 export const touchPlanServerTool: BaseServerTool<TouchPlanArgs, TouchPlanResult> = {
   name: TOUCH_PLAN_TOOL_ID,
   async execute(params: TouchPlanArgs, context?: ServerToolContext): Promise<TouchPlanResult> {
+    if (!isMothershipBetaFeaturesEnabled) {
+      return { success: false, message: 'touch_plan is not available' }
+    }
     if (!context?.userId) {
       throw new Error('Authentication required')
     }

apps/sim/lib/copilot/tools/server/router.ts

@@ -25,6 +25,7 @@ import {
   type BaseServerTool,
   type ServerToolContext,
 } from '@/lib/copilot/tools/server/base-tool'
+import { isMothershipBetaFeaturesEnabled } from '@/lib/core/config/feature-flags'
 import { getBlocksMetadataServerTool } from '@/lib/copilot/tools/server/blocks/get-blocks-metadata-tool'
 import { getTriggerBlocksServerTool } from '@/lib/copilot/tools/server/blocks/get-trigger-blocks'
 import { searchDocumentationServerTool } from '@/lib/copilot/tools/server/docs/search-documentation'
@@ -130,7 +131,7 @@ function isWriteAction(toolName: string, action: string | undefined): boolean {
 }
 
 /** Registry of all server tools. Tools self-declare their validation schemas. */
-const serverToolRegistry: Record<string, BaseServerTool> = {
+const baseServerToolRegistry: Record<string, BaseServerTool> = {
   [getBlocksMetadataServerTool.name]: getBlocksMetadataServerTool,
   [getTriggerBlocksServerTool.name]: getTriggerBlocksServerTool,
   [editWorkflowServerTool.name]: editWorkflowServerTool,
@@ -159,16 +160,25 @@ const serverToolRegistry: Record<string, BaseServerTool> = {
   [generateImageServerTool.name]: generateImageServerTool,
 }
 
+function getServerToolRegistry(): Record<string, BaseServerTool> {
+  if (isMothershipBetaFeaturesEnabled) {
+    return baseServerToolRegistry
+  }
+  const registry = { ...baseServerToolRegistry }
+  delete registry[touchPlanServerTool.name]
+  return registry
+}
+
 export function getRegisteredServerToolNames(): string[] {
-  return Object.keys(serverToolRegistry)
+  return Object.keys(getServerToolRegistry())
 }
 
 export async function routeExecution(
   toolName: string,
   payload: unknown,
   context?: ServerToolContext
 ): Promise<unknown> {
-  const tool = serverToolRegistry[toolName]
+  const tool = getServerToolRegistry()[toolName]
   if (!tool) {
     throw new Error(`Unknown server tool: ${toolName}`)
   }

apps/sim/lib/copilot/vfs/workflow-alias-resolver.ts

@@ -8,12 +8,14 @@ import {
   resolveWorkflowAliasPath,
   type WorkflowAliasTarget,
 } from '@/lib/copilot/vfs/workflow-aliases'
+import { isMothershipBetaFeaturesEnabled } from '@/lib/core/config/feature-flags'
 import { canonicalizeVfsPath } from './path-utils'
 
 export async function resolveWorkflowAliasForWorkspace(args: {
   workspaceId: string
   path: string
 }): Promise<WorkflowAliasTarget | null> {
+  if (!isMothershipBetaFeaturesEnabled) return null
   if (!isPlanAliasPath(args.path)) return null
 
   let canonicalPath: string