fix (tasks): minor bug fixes

Author: itsskofficial

src/client/app/tasks/page.js

@@ -105,6 +105,21 @@ function TasksPageContent() {
 	const [groupBy, setGroupBy] = useState("status")
 	const [isAddingNewTask, setIsAddingNewTask] = useState(false)
 
+	// Keep selectedTask in sync with the main tasks list
+	useEffect(() => {
+		if (selectedTask) {
+			const updatedSelectedTask = tasks.find(
+				(t) => t.task_id === selectedTask.task_id
+			)
+			if (updatedSelectedTask) {
+				setSelectedTask(updatedSelectedTask)
+			} else {
+				// If task is not in the list anymore (e.g., deleted), close the panel.
+				setSelectedTask(null)
+			}
+		}
+	}, [tasks, selectedTask])
+
 	// Check for query param to auto-open demo on first visit from onboarding
 	useEffect(() => {
 		const showDemo = searchParams.get("show_demo")
@@ -167,6 +182,17 @@ function TasksPageContent() {
 		// eslint-disable-next-line react-hooks/exhaustive-deps
 	}, [])
 
+	// Listen for custom event from LayoutWrapper to refresh tasks
+	useEffect(() => {
+		const handleBackendUpdate = () => {
+			console.log("Received tasksUpdatedFromBackend event, fetching tasks...")
+			toast.success("Task list updated from backend.")
+			fetchTasks()
+		}
+		window.addEventListener("tasksUpdatedFromBackend", handleBackendUpdate)
+		return () => window.removeEventListener("tasksUpdatedFromBackend", handleBackendUpdate)
+	}, [fetchTasks])
+
 	// Listen for real-time progress updates from WebSocket
 	useEffect(() => {
 		const handleProgressUpdate = (event) => {
@@ -349,6 +375,17 @@ function TasksPageContent() {
 		)
 	}
 
+	const handleSendTaskChatMessage = (taskId, message) =>
+		handleAction(
+			() =>
+				fetch("/api/tasks/chat", {
+					method: "POST",
+					headers: { "Content-Type": "application/json" },
+					body: JSON.stringify({ taskId, message })
+				}),
+			"Message sent. Re-planning task..."
+		)
+
 	const handleUpdateTask = async (updatedTask) => {
 		await handleAction(
 			() =>
@@ -450,18 +487,13 @@ function TasksPageContent() {
 						handleRerunTask(taskId)
 						setSelectedTask(null)
 					}}
-					onAnswerClarifications={(taskId, answers) => {
-						handleAnswerClarifications(taskId, answers)
-						setSelectedTask(null)
-					}}
+					onAnswerClarifications={handleAnswerClarifications}
 					onArchiveTask={(taskId) => {
 						handleArchiveTask(taskId)
 						setSelectedTask(null)
 					}}
-					onMarkComplete={(taskId) => {
-						handleMarkComplete(taskId)
-						setSelectedTask(null)
-					}}
+					onMarkComplete={handleMarkComplete}
+					onSendChatMessage={handleSendTaskChatMessage}
 				/>
 			</div>
 

src/client/components/LayoutWrapper.js

@@ -91,6 +91,12 @@ export default function LayoutWrapper({ children }) {
 								detail: data.payload
 							})
 						)
+					} else if (data.type === "task_list_updated") {
+						// This is a generic event telling the app that tasks have changed
+						// on the backend and the UI should refetch them.
+						window.dispatchEvent(
+							new CustomEvent("tasksUpdatedFromBackend")
+						)
 					}
 				}
 				ws.onclose = () => {

src/client/components/NotificationsOverlay.js

@@ -11,11 +11,11 @@ import {
 	IconX,
 	IconArrowRight
 } from "@tabler/icons-react"
-import { formatDistanceToNow } from "date-fns"
+import { formatDistanceToNow, parseISO } from "date-fns"
 
 const NotificationItem = ({ notification, onDelete, onClick }) => {
 	const formattedTimestamp = formatDistanceToNow(
-		new Date(notification.timestamp),
+		parseISO(notification.timestamp),
 		{ addSuffix: true }
 	)
 

src/client/components/PostHogProvider.jsx

@@ -12,9 +12,8 @@ export function PostHogProvider({ children }) {
 				ui_host:
 					process.env.NEXT_PUBLIC_POSTHOG_HOST ||
 					"https://us.posthog.com",
-				defaults: "2025-05-24",
 				capture_exceptions: true,
-				debug: process.env.NODE_ENV === "development"
+				autocapture: false // Disable autocapture to reduce event volume
 			})
 		}
 	}, [])

src/client/components/tasks/ScheduleEditor.js

@@ -17,6 +17,29 @@ const ScheduleEditor = ({ schedule, setSchedule }) => {
 		setSchedule(baseSchedule)
 	}
 
+	const handleRunAtChange = (e) => {
+		const localDateTimeString = e.target.value
+		if (localDateTimeString) {
+			// The input value is a string like "2024-07-26T09:00".
+			// new Date() will parse this as 9 AM in the browser's local timezone.
+			const localDate = new Date(localDateTimeString)
+			// .toISOString() converts it to a UTC string, which is what we want to store.
+			setSchedule({ ...schedule, run_at: localDate.toISOString() })
+		} else {
+			setSchedule({ ...schedule, run_at: null })
+		}
+	}
+
+	const getLocalDateTimeString = (isoString) => {
+		if (!isoString) return ""
+		const date = new Date(isoString)
+		const tzOffset = date.getTimezoneOffset() * 60000 // offset in milliseconds
+		const localISOTime = new Date(date.getTime() - tzOffset)
+			.toISOString()
+			.slice(0, 16)
+		return localISOTime
+	}
+
 	const handleDayToggle = (day) => {
 		const currentDays = schedule.days || []
 		const newDays = currentDays.includes(day)
@@ -55,16 +78,8 @@ const ScheduleEditor = ({ schedule, setSchedule }) => {
 					</label>
 					<input
 						type="datetime-local"
-						value={
-							schedule.run_at
-								? new Date(schedule.run_at)
-										.toISOString()
-										.slice(0, 16)
-								: ""
-						}
-						onChange={(e) =>
-							setSchedule({ ...schedule, run_at: e.target.value })
-						}
+						value={getLocalDateTimeString(schedule.run_at)}
+						onChange={handleRunAtChange}
 						className="w-full p-2 bg-neutral-700 border border-neutral-600 rounded-md focus:border-blue-500"
 					/>
 					<p className="text-xs text-gray-500 mt-1">
@@ -94,18 +109,9 @@ const ScheduleEditor = ({ schedule, setSchedule }) => {
 						</select>
 					</div>
 					<div>
-						<label
-							className="text-xs text-gray-400 block mb-1"
-							data-tooltip-id="schedule-tooltip"
-							data-tooltip-content="Tasks are scheduled in Coordinated Universal Time (UTC) to ensure consistency across timezones."
-						>
-							Time (UTC)
+						<label className="text-xs text-gray-400 block mb-1">
+							Time (Local)
 						</label>
-						<Tooltip
-							place="top"
-							id="schedule-tooltip"
-							style={{ zIndex: 99999 }}
-						/>
 						<input
 							type="time"
 							value={schedule.time || "09:00"}

src/client/components/tasks/TaskDetailsContent.js

@@ -1,6 +1,6 @@
 "use client"
 
-import React, { useState } from "react"
+import React, { useState, useEffect, useRef } from "react"
 import toast from "react-hot-toast"
 import { cn } from "@utils/cn"
 import { taskStatusColors, priorityMap } from "./constants"
@@ -10,10 +10,59 @@ import {
 	IconSparkles,
 	IconUser,
 	IconX,
-	IconLoader
+	IconLoader,
+	IconSend
 } from "@tabler/icons-react"
 import ScheduleEditor from "@components/tasks/ScheduleEditor"
 import ExecutionUpdate from "./ExecutionUpdate"
+import ChatBubble from "@components/ChatBubble"
+
+const TaskChatSection = ({ task, onSendChatMessage }) => {
+	const [message, setMessage] = useState("")
+	const chatEndRef = React.useRef(null)
+
+	useEffect(() => {
+		chatEndRef.current?.scrollIntoView({ behavior: "smooth" })
+	}, [task.chat_history])
+
+	const handleSend = () => {
+		if (message.trim()) {
+			onSendChatMessage(task.task_id, message)
+			setMessage("")
+		}
+	}
+
+	return (
+		<div className="mt-6 pt-6 border-t border-neutral-800">
+			<h4 className="font-semibold text-neutral-300 mb-4">
+				Task Conversation
+			</h4>
+			<div className="space-y-4 max-h-64 overflow-y-auto custom-scrollbar pr-2">
+				{(task.chat_history || []).map((msg, index) => (
+					<ChatBubble key={index} message={msg.content} isUser={msg.role === 'user'} />
+				))}
+				<div ref={chatEndRef} />
+			</div>
+			<div className="mt-4 flex items-center gap-2">
+				<input
+					type="text"
+					value={message}
+					onChange={(e) => setMessage(e.target.value)}
+					onKeyDown={(e) => e.key === "Enter" && handleSend()}
+					placeholder="Ask for changes or follow-ups..."
+					className="flex-grow p-2 bg-neutral-800 border border-neutral-700 rounded-lg text-sm"
+				/>
+				<button
+					onClick={handleSend}
+					className="p-2 bg-blue-600 rounded-lg text-white hover:bg-blue-500 disabled:opacity-50"
+					disabled={!message.trim()}
+				>
+					<IconSend size={16} />
+				</button>
+			</div>
+		</div>
+	)
+}
 
 // New component for handling clarification questions
 const ClarificationInputSection = ({ run, task, onAnswerClarifications }) => {
@@ -98,7 +147,8 @@ const TaskDetailsContent = ({
 	handleStepChange,
 	allTools,
 	integrations,
-	onAnswerClarifications
+	onAnswerClarifications,
+	onSendChatMessage
 }) => {
 	if (!task) {
 		return null
@@ -432,6 +482,10 @@ const TaskDetailsContent = ({
 					</div>
 				))
 			)}
+
+			{(task.status === "completed" || (task.chat_history && task.chat_history.length > 0)) && (
+				<TaskChatSection task={task} onSendChatMessage={onSendChatMessage} />
+			)}
 		</div>
 	)
 }

src/client/components/tasks/TaskDetailsPanel.js

@@ -28,7 +28,8 @@ const TaskDetailsPanel = ({
 	onAnswerClarifications,
 	onArchiveTask,
 	onMarkComplete,
-	className
+	className,
+	onSendChatMessage
 }) => {
 	const [isEditing, setIsEditing] = useState(false)
 	const [editableTask, setEditableTask] = useState(task)
@@ -142,6 +143,7 @@ const TaskDetailsPanel = ({
 							allTools={allTools}
 							integrations={integrations}
 							onAnswerClarifications={onAnswerClarifications}
+							onSendChatMessage={onSendChatMessage}
 						/>
 					</main>
 

src/server/main/tasks/prompts.py

@@ -12,10 +12,10 @@
     - `0`: High priority (urgent, important, deadlines).
     - `1`: Medium priority (standard tasks, default).
     - `2`: Low priority (can be done anytime, not urgent).
-3.  **Schedule:** Analyze the prompt for any scheduling information (dates, times, recurrence).
-    - If no specific time is mentioned, set the schedule to `null`.
-    - If a specific date and time is mentioned for a one-time task, use the `once` type. The `run_at` value MUST be in `YYYY-MM-DDTHH:MM` format for local datetime inputs.
-    - If the task is recurring, use the `recurring` type.
+3.  **Schedule:** Analyze the prompt for any scheduling information (dates, times, recurrence). Decipher whether the task is a one-time event or recurring, and format the schedule accordingly:
+    - If no specific time is mentioned, the task has to be performed right now.
+    - If a specific date and time is mentioned and its a one-time task, use the `once` type. The `run_at` value MUST be in `YYYY-MM-DDTHH:MM` format for local datetime inputs.
+    - If the task is found to be recurring, use the `recurring` type.
         - `frequency` can be "daily" or "weekly".
         - `time` MUST be in "HH:MM" 24-hour format.
         - For "weekly" frequency, `days` MUST be a list of full day names (e.g., ["Monday", "Wednesday"]).
@@ -61,7 +61,10 @@
 {{
   "description": "Organize my downloads folder",
   "priority": 2,
-  "schedule": null
+  "schedule": {{
+    "type": "once",
+    "run_at": "CURRENT_DATE_TIME_IN_USER_TIMEZONE"
+  }}
 }}
 ```
-"""
+"""

src/server/main/tasks/routes.py

@@ -139,6 +139,13 @@ async def update_task(
     update_data = request.dict(exclude_unset=True)
     update_data.pop("taskId", None)
 
+    # If the schedule is being updated, inject the user's timezone for recurring tasks.
+    if 'schedule' in update_data and update_data['schedule']:
+        if update_data['schedule'].get('type') == 'recurring':
+            user_profile = await mongo_manager.get_user_profile(user_id)
+            user_timezone_str = user_profile.get("userData", {}).get("personalInfo", {}).get("timezone", "UTC")
+            update_data['schedule']['timezone'] = user_timezone_str
+
     # If assignee is changed to 'ai' and task is in a non-planned state, trigger planning
     if 'assignee' in update_data and update_data['assignee'] == 'ai':
         if task.get('status') == 'pending': # Assuming 'pending' is the status for user-assigned tasks
@@ -246,6 +253,7 @@ async def approve_task(
     schedule_data = task_doc.get("schedule") or {}
 
     if schedule_data.get("type") == "recurring":
+        # The schedule object should contain the user's timezone, added when the task was created/updated.
         next_run = calculate_next_run(schedule_data)
         if next_run:
             update_data["next_execution_at"] = next_run
@@ -256,7 +264,11 @@ async def approve_task(
             update_data["error"] = "Could not calculate next run time for recurring task."
     elif schedule_data.get("type") == "once" and schedule_data.get("run_at"):
         run_at_time_str = schedule_data.get("run_at")
-        run_at_time = datetime.fromisoformat(run_at_time_str).replace(tzinfo=timezone.utc)
+        run_at_time = datetime.fromisoformat(run_at_time_str)
+        # Ensure datetime is timezone-aware for comparison. New data will have 'Z' and be aware.
+        if run_at_time.tzinfo is None:
+            # This branch is for backward compatibility. It ASSUMES the naive datetime was stored as UTC.
+            run_at_time = run_at_time.replace(tzinfo=timezone.utc)
         if run_at_time > datetime.now(timezone.utc):
             update_data["next_execution_at"] = run_at_time
             update_data["status"] = "pending"
@@ -348,4 +360,22 @@ async def internal_progress_update(request: ProgressUpdateRequest):
     except Exception as e:
         logger.error(f"Failed to push progress update via websocket for task {request.task_id}: {e}", exc_info=True)
         # Don't fail the worker, just log it.
-        return {"status": "error", "detail": str(e)}
+        return {"status": "error", "detail": str(e)}
+
+@router.post("/internal/task-update-push", include_in_schema=False)
+async def internal_task_update_push(request: ProgressUpdateRequest): # Reusing model for convenience
+    """
+    Internal endpoint for workers to tell the main server to push a generic
+    'tasks have changed' notification to the client via WebSocket.
+    """
+    logger.info(f"Received internal request to push task list update for user {request.user_id}")
+    try:
+        await websocket_manager.send_personal_json_message(
+            {"type": "task_list_updated"},
+            request.user_id,
+            connection_type="notifications"
+        )
+        return {"status": "success"}
+    except Exception as e:
+        logger.error(f"Failed to push task list update via websocket for user {request.user_id}: {e}", exc_info=True)
+        return {"status": "error", "detail": str(e)}