fix: HAI Tasks state issue (#212)

* feat: implemented hai tasks

* feat: added mark as completed feature for hai tasks execution and enhanced speciai-mcp description

* fix: fixed the html issue when performing action during fuzzy search

* doc: added changeset

* chore: updated description

* chore: removed console logs

* fix: maintaining hai task in workspace level state instaed of global

---------

Co-authored-by: Magesh <mageshmscss@gmail.com>

Author: magesh-presidio

src/core/controller/ui/initializeWebview.ts

@@ -223,6 +223,22 @@ export async function initializeWebview(controller: Controller, _request: EmptyR
 		// Silently refresh MCP marketplace catalog
 		controller.silentlyRefreshMcpMarketplace()
 
+		// TAG:HAI - Restore HAI tasks if available
+		const haiConfig = controller.cacheService.getWorkspaceStateKey("haiConfig")
+		if (haiConfig?.folder) {
+			// Import loadHaiTasks to restore tasks
+			const { loadHaiTasks } = await import("./loadHaiTasks")
+			try {
+				await loadHaiTasks(controller, {
+					metadata: {},
+					folderPath: haiConfig.folder,
+					loadDefault: true,
+				})
+			} catch (error) {
+				console.error("Failed to restore HAI tasks on initialization:", error)
+			}
+		}
+
 		// Initialize telemetry service with user's current setting
 		controller.getStateToPostToWebview().then((state) => {
 			const { telemetrySetting } = state

src/core/controller/ui/loadHaiTasks.ts

@@ -23,18 +23,12 @@ function getFormattedDateTime(): string {
  */
 export async function loadHaiTasks(controller: Controller, request: HaiTasksLoadRequest): Promise<Empty> {
 	try {
-		const { folderPath, loadDefault } = request
+		const { folderPath } = request
+		let selectedFolderPath: string | undefined
 
-		if (loadDefault && folderPath) {
-			// Load from the provided path (refresh case)
-			const ts = getFormattedDateTime()
-			await fetchTaskFromSelectedFolder(controller, folderPath, ts)
-			await controller.context.globalState.update("haiConfig", { folder: folderPath, ts })
-		} else if (folderPath) {
-			// Load from specific path
-			const ts = getFormattedDateTime()
-			await fetchTaskFromSelectedFolder(controller, folderPath, ts)
-			await controller.context.globalState.update("haiConfig", { folder: folderPath, ts })
+		if (folderPath) {
+			// Load from the provided path (either refresh case or specific path)
+			selectedFolderPath = folderPath
 		} else {
 			// Show folder picker
 			const options: vscode.OpenDialogOptions = {
@@ -46,12 +40,16 @@ export async function loadHaiTasks(controller: Controller, request: HaiTasksLoad
 
 			const fileUri = await vscode.window.showOpenDialog(options)
 			if (fileUri && fileUri[0]) {
-				const ts = getFormattedDateTime()
-				await fetchTaskFromSelectedFolder(controller, fileUri[0].fsPath, ts)
-				await controller.context.globalState.update("haiConfig", { folder: fileUri[0].fsPath, ts })
+				selectedFolderPath = fileUri[0].fsPath
 			}
 		}
 
+		if (selectedFolderPath) {
+			const ts = getFormattedDateTime()
+			await fetchTaskFromSelectedFolder(controller, selectedFolderPath, ts)
+			controller.cacheService.setWorkspaceState("haiConfig", { folder: selectedFolderPath, ts })
+		}
+
 		return Empty.create({})
 	} catch (error) {
 		console.error(`Failed to load HAI tasks: ${error}`)

src/core/controller/ui/resetHaiTasks.ts

@@ -11,8 +11,8 @@ import { sendHaiTaskDataUpdate } from "./subscribeToHaiTaskData"
  */
 export async function resetHaiTasks(controller: Controller, _request: EmptyRequest): Promise<Empty> {
 	try {
-		// Clear the HAI config
-		await controller.context.globalState.update("haiConfig", {})
+		// Clear the HAI config using workspace state
+		controller.cacheService.setWorkspaceState("haiConfig", undefined)
 
 		// Send empty task data to all subscribed clients
 		sendHaiTaskDataUpdate({

src/core/storage/CacheService.ts

@@ -1095,6 +1095,7 @@ export class CacheService {
 			// TAG:HAI - HAI-specific workspace state fields
 			buildContextOptions: buildContextOptions,
 			buildIndexProgress: buildIndexProgress,
+			haiConfig: state.haiConfig,
 		} satisfies Partial<LocalState>
 
 		Object.assign(this.workspaceStateCache, workspaceStateFields)

src/core/storage/state-keys.ts

@@ -213,6 +213,7 @@ export type LocalStateKey =
 	| "workflowToggles"
 	| "buildContextOptions"
 	| "buildIndexProgress"
+	| "haiConfig"
 
 export interface GlobalState {
 	awsRegion: string | undefined
@@ -418,4 +419,5 @@ export interface LocalState {
 	// TAG:HAI
 	buildContextOptions: HaiBuildContextOptions
 	buildIndexProgress: HaiBuildIndexProgress
+	haiConfig: { folder?: string; ts?: string } | undefined
 }

src/core/storage/utils/state-helpers.ts

@@ -189,6 +189,7 @@ export async function readStateFromDisk(context: ExtensionContext) {
 	// TAG:HAI - Get HAI-specific workspace state values
 	const buildContextOptions = context.workspaceState.get("buildContextOptions") as any | undefined
 	const buildIndexProgress = context.workspaceState.get("buildIndexProgress") as any | undefined
+	const haiConfig = context.workspaceState.get("haiConfig") as { folder?: string; ts?: string } | undefined
 
 	// Get mode-related configurations
 	const mode = context.globalState.get("mode") as Mode | undefined
@@ -465,6 +466,7 @@ export async function readStateFromDisk(context: ExtensionContext) {
 			expertName,
 			isDeepCrawlEnabled,
 		},
+		haiConfig,
 
 		// Embedding configuration
 		embeddingConfiguration: {