Add additional context check to task cleanup

Author: TheodoreSpeaks

apps/sim/lib/cleanup/chat-cleanup.ts

@@ -11,17 +11,25 @@ const logger = createLogger('ChatCleanup')
 
 const COPILOT_CLEANUP_BATCH_SIZE = 1000
 
+/**
+ * Only storage in these contexts is tied to chat/task lifecycle. Workspace
+ * files, execution logs, knowledge bases, profile pictures, etc. are owned by
+ * other subsystems and must never be touched by chat cleanup — even if a row
+ * somehow ends up with `chatId` set through a future flow.
+ */
+const CHAT_SCOPED_CONTEXTS = ['copilot', 'mothership'] as const satisfies readonly StorageContext[]
+type ChatScopedContext = (typeof CHAT_SCOPED_CONTEXTS)[number]
+
 interface FileRef {
   key: string
-  context: StorageContext
+  context: ChatScopedContext
 }
 
 /**
  * Collect all file storage keys associated with the given chat IDs.
  * Two sources:
- * 1. workspaceFiles rows with chatId FK — includes the storage context
- * 2. fileAttachments[].key inside copilotChats.messages JSONB — these are
- *    uploaded with context='copilot'
+ * 1. workspaceFiles rows with chatId FK — filtered to chat-scoped contexts only
+ * 2. fileAttachments[].key inside copilotChats.messages JSONB — all copilot uploads
  */
 export async function collectChatFiles(chatIds: string[]): Promise<FileRef[]> {
   const files: FileRef[] = []
@@ -33,7 +41,13 @@ export async function collectChatFiles(chatIds: string[]): Promise<FileRef[]> {
     db
       .select({ key: workspaceFiles.key, context: workspaceFiles.context })
       .from(workspaceFiles)
-      .where(and(inArray(workspaceFiles.chatId, chatIds), isNull(workspaceFiles.deletedAt))),
+      .where(
+        and(
+          inArray(workspaceFiles.chatId, chatIds),
+          isNull(workspaceFiles.deletedAt),
+          inArray(workspaceFiles.context, [...CHAT_SCOPED_CONTEXTS])
+        )
+      ),
     db
       .select({ messages: copilotChats.messages })
       .from(copilotChats)
@@ -43,7 +57,7 @@ export async function collectChatFiles(chatIds: string[]): Promise<FileRef[]> {
   for (const f of linkedFiles) {
     if (!seen.has(f.key)) {
       seen.add(f.key)
-      files.push({ key: f.key, context: f.context as StorageContext })
+      files.push({ key: f.key, context: f.context as ChatScopedContext })
     }
   }