fix(scheduled-tasks): preserve @-mention contexts on edit; sync editor valueRef on input

Author: waleedlatif1

apps/sim/app/workspace/[workspaceId]/home/components/user-input/components/prompt-editor/use-prompt-editor.ts

@@ -563,6 +563,9 @@ export function usePromptEditor({
       }
 
       const caret = e.target.selectionStart ?? finalValue.length
+      // Keep the ref in lockstep with state so synchronous readers (getValue /
+      // getPlainValue) never see pre-keystroke text before the next render.
+      valueRef.current = finalValue
       setValueState(finalValue)
       syncMentionState(e.target, finalValue, caret)
       syncSlashState(e.target, finalValue, caret)

apps/sim/app/workspace/[workspaceId]/scheduled-tasks/components/task-modal/task-modal.tsx

@@ -1,6 +1,6 @@
 'use client'
 
-import { useState } from 'react'
+import { useEffect, useState } from 'react'
 import { format } from 'date-fns'
 import { useParams } from 'next/navigation'
 import {
@@ -49,6 +49,8 @@ export interface TaskDraft {
 export interface TaskEditSeed {
   scheduleId: string
   prompt: string
+  /** Stored `@`-mention contexts, re-registered so editing preserves them. */
+  contexts?: ChatContext[]
   launchDate: string
   launchTime: string
   recurrence: Recurrence
@@ -114,6 +116,15 @@ function TaskModalContent({
 }: Omit<TaskModalProps, 'open'>) {
   const { workspaceId } = useParams<{ workspaceId: string }>()
   const editor = usePromptEditor({ workspaceId, initialValue: edit?.prompt })
+
+  // Re-register the stored mentions once on open so saving an edit preserves
+  // them — the editor seeds from `initialValue` text only, not its contexts.
+  const setContexts = editor.setContexts
+  useEffect(() => {
+    if (edit?.contexts && edit.contexts.length > 0) setContexts(edit.contexts)
+    // Runs once per open; content remounts each time the dialog opens.
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [])
   const [launchDate, setLaunchDate] = useState(
     () => edit?.launchDate ?? format(slot?.date ?? new Date(), 'yyyy-MM-dd')
   )

apps/sim/app/workspace/[workspaceId]/scheduled-tasks/hooks/use-scheduled-tasks.ts

@@ -139,6 +139,7 @@ export function useScheduledTasks({
       return {
         scheduleId: schedule.id,
         prompt: schedule.prompt ?? '',
+        contexts: task.contexts,
         launchDate: format(task.runAt, 'yyyy-MM-dd'),
         launchTime,
         recurrence,