fix (tasks): user id retrieval issue

Author: itsskofficial

src/client/app/chat/page.js

@@ -559,23 +559,19 @@ export default function ChatPage() {
 						setDisplayedMessages((prev) =>
 							prev.map((msg) => {
 								if (msg.id === assistantMessageId) {
-									// --- CHANGED --- Handle the final 'done' event with parsed data
+									// Only update the message when the stream is done
 									if (parsed.done) {
 										return {
 											...msg,
-											content:
-												parsed.final_content ||
-												msg.content, // Replace content with clean version
-											turn_steps:
-												parsed.turn_steps ||
-												msg.turn_steps // Populate turn_steps
+											content: parsed.final_content || "", // Replace content with clean final version
+											turn_steps: parsed.turn_steps || [], // Populate turn_steps
+											tools: parsed.tools || [] // Update tools on final event
 										}
 									}
-									// For intermediate chunks, just append the token
+									// For intermediate chunks, we don't update the content to prevent streaming.
+									// We can still update tools if they arrive early.
 									return {
 										...msg,
-										content:
-											msg.content + (parsed.token || ""),
 										tools: parsed.tools || msg.tools
 									}
 								}

src/client/components/tasks/TaskCardList.js

@@ -71,7 +71,7 @@ const SubTaskItem = ({ task, onSelectTask }) => {
 			}}
 			className="flex items-center justify-between p-2 md:mx-2 rounded-md hover:bg-neutral-800 cursor-pointer transition-colors group"
 		>
-			<div className="flex flex-1 items-center gap-2 min-w-0">
+			<div className="flex flex-1 items-center gap-2 min-w-0 overflow-hidden">
 				<IconSubtask
 					size={14}
 					className="text-neutral-500 group-hover:text-neutral-300 flex-shrink-0"
@@ -201,26 +201,26 @@ const TaskCardList = ({ task, onSelectTask }) => {
 						</button>
 					)}
 				</div>
+				<AnimatePresence>
+					{isExpanded && hasSubTasks && (
+						<motion.div
+							initial={{ opacity: 0, height: 0 }}
+							animate={{ opacity: 1, height: "auto" }}
+							exit={{ opacity: 0, height: 0 }}
+							transition={{ duration: 0.2 }}
+							className="pb-2"
+						>
+							{subTasks.map((subTask) => (
+								<SubTaskItem
+									key={subTask.task_id}
+									task={subTask}
+									onSelectTask={onSelectTask}
+								/>
+							))}
+						</motion.div>
+					)}
+				</AnimatePresence>
 			</div>
-			<AnimatePresence>
-				{isExpanded && hasSubTasks && (
-					<motion.div
-						initial={{ opacity: 0, height: 0 }}
-						animate={{ opacity: 1, height: "auto" }}
-						exit={{ opacity: 0, height: 0 }}
-						transition={{ duration: 0.2 }}
-						className="pb-2"
-					>
-						{subTasks.map((subTask) => (
-							<SubTaskItem
-								key={subTask.task_id}
-								task={subTask}
-								onSelectTask={onSelectTask}
-							/>
-						))}
-					</motion.div>
-				)}
-			</AnimatePresence>
 		</motion.div>
 	)
 }

src/server/mcp_hub/tasks/main.py

@@ -6,10 +6,10 @@
 from json_extractor import JsonExtractor
 from datetime import datetime
 from zoneinfo import ZoneInfo, ZoneInfoNotFoundError
-
+from main.llm import run_agent as run_main_agent
 from . import auth, prompts
 from main.dependencies import mongo_manager # This is the main server's mongo manager
-from main.llm import run_agent, LLMProviderDownError
+from main.llm import LLMProviderDownError
 from workers.long_form_tasks import start_long_form_task
 from workers.tasks import generate_plan_from_context
 from workers.utils.text_utils import clean_llm_output

src/server/workers/tasks.py

@@ -325,7 +325,7 @@ async def async_refine_and_plan_ai_task(task_id: str, user_id: str):
         original_schedule = task.get("schedule")
 
         user_id = task["user_id"]
-        user_profile = await db_manager.user_profiles_collection.find_one({"user_id": user_id})
+        user_profile = await db_manager.get_user_profile(user_id)
         personal_info = user_profile.get("userData", {}).get("personalInfo", {}) if user_profile else {}
         user_name = personal_info.get("name", "User")
         user_timezone_str = personal_info.get("timezone", "UTC")