fix(scheduled-tasks): open on edit/create

Author: Sg312

apps/sim/lib/copilot/resources/extraction.test.ts

@@ -2,7 +2,7 @@
  * @vitest-environment node
  */
 import { describe, expect, it } from 'vitest'
-import { extractResourcesFromToolResult } from './extraction'
+import { extractDeletedResourcesFromToolResult, extractResourcesFromToolResult } from './extraction'
 
 describe('extractResourcesFromToolResult', () => {
   it('extracts file resources from create_file results', () => {
@@ -141,4 +141,66 @@ describe('extractResourcesFromToolResult', () => {
 
     expect(resources).toEqual([])
   })
+
+  it('auto-opens a scheduledtask resource from manage_scheduled_task create results', () => {
+    const resources = extractResourcesFromToolResult(
+      'manage_scheduled_task',
+      { operation: 'create', args: { title: 'Daily Report' } },
+      { jobId: 'sched_123', title: 'Daily Report', message: 'Job created successfully.' }
+    )
+
+    expect(resources).toEqual([{ type: 'scheduledtask', id: 'sched_123', title: 'Daily Report' }])
+  })
+
+  it('auto-opens a scheduledtask resource on update, falling back to the args title', () => {
+    const resources = extractResourcesFromToolResult(
+      'manage_scheduled_task',
+      { operation: 'update', args: { jobId: 'sched_123', title: 'Renamed Task' } },
+      { jobId: 'sched_123', updated: ['title'], message: 'Job updated successfully' }
+    )
+
+    expect(resources).toEqual([{ type: 'scheduledtask', id: 'sched_123', title: 'Renamed Task' }])
+  })
+
+  it('does not auto-open for read-only manage_scheduled_task operations', () => {
+    expect(
+      extractResourcesFromToolResult(
+        'manage_scheduled_task',
+        { operation: 'list' },
+        { jobs: [], count: 0 }
+      )
+    ).toEqual([])
+    expect(
+      extractResourcesFromToolResult(
+        'manage_scheduled_task',
+        { operation: 'get', args: { jobId: 'sched_123' } },
+        { id: 'sched_123', title: 'Daily Report' }
+      )
+    ).toEqual([])
+  })
+})
+
+describe('extractDeletedResourcesFromToolResult', () => {
+  it('removes scheduledtask resources on manage_scheduled_task delete', () => {
+    const resources = extractDeletedResourcesFromToolResult(
+      'manage_scheduled_task',
+      { operation: 'delete', args: { jobIds: ['sched_1', 'sched_2'] } },
+      { deleted: ['sched_1', 'sched_2'], notFound: [] }
+    )
+
+    expect(resources).toEqual([
+      { type: 'scheduledtask', id: 'sched_1', title: 'Scheduled Task' },
+      { type: 'scheduledtask', id: 'sched_2', title: 'Scheduled Task' },
+    ])
+  })
+
+  it('does not remove anything for non-delete manage_scheduled_task ops', () => {
+    expect(
+      extractDeletedResourcesFromToolResult(
+        'manage_scheduled_task',
+        { operation: 'update', args: { jobId: 'sched_1' } },
+        { jobId: 'sched_1', updated: ['title'] }
+      )
+    ).toEqual([])
+  })
 })

apps/sim/lib/copilot/resources/extraction.ts

@@ -11,6 +11,7 @@ import {
   GenerateVideo,
   Knowledge,
   KnowledgeBase,
+  ManageScheduledTask,
   UserTable,
   WorkspaceFile,
 } from '@/lib/copilot/generated/tool-catalog-v1'
@@ -29,6 +30,7 @@ const RESOURCE_TOOL_NAMES: Set<string> = new Set([
   FunctionExecute.id,
   KnowledgeBase.id,
   Knowledge.id,
+  ManageScheduledTask.id,
   GenerateImage.id,
   GenerateVideo.id,
   GenerateAudio.id,
@@ -219,6 +221,19 @@ export function extractResourcesFromToolResult(
       return resources
     }
 
+    case ManageScheduledTask.id: {
+      // Read-only ops never auto-open; only create/update surface the task.
+      const op = getOperation(params)
+      if (op === 'list' || op === 'get') return []
+      const jobId = (result.jobId as string) ?? (data.jobId as string)
+      if (jobId) {
+        const args = asRecord(params?.args)
+        const title = (result.title as string) ?? (args.title as string) ?? 'Scheduled Task'
+        return [{ type: 'scheduledtask', id: jobId, title }]
+      }
+      return []
+    }
+
     default:
       return []
   }
@@ -229,6 +244,7 @@ const DELETE_CAPABLE_TOOL_RESOURCE_TYPE: Record<string, ResourceType> = {
   [WorkspaceFile.id]: 'file',
   [UserTable.id]: 'table',
   [KnowledgeBase.id]: 'knowledgebase',
+  [ManageScheduledTask.id]: 'scheduledtask',
 }
 
 export function hasDeleteCapability(toolName: string): boolean {
@@ -292,6 +308,12 @@ export function extractDeletedResourcesFromToolResult(
       return []
     }
 
+    case ManageScheduledTask.id: {
+      if (operation !== 'delete') return []
+      const deletedIds = Array.isArray(result.deleted) ? (result.deleted as string[]) : []
+      return deletedIds.map((id) => ({ type: resourceType, id, title: 'Scheduled Task' }))
+    }
+
     default:
       return []
   }