fix(code-index): port fork reliability fixes (#34, #67, #102, #110)

.gitattributes

@@ -2,6 +2,9 @@ demo.gif filter=lfs diff=lfs merge=lfs -text
 assets/docs/demo.gif filter=lfs diff=lfs merge=lfs -text
 src/assets/docs/demo.gif filter=lfs diff=lfs merge=lfs -text
 
+# pnpm hashes patch bytes; keep LF on every checkout.
+patches/*.patch text eol=lf
+
 # Test snapshot files - mark as linguist-generated to exclude from GitHub language statistics
 *.snap linguist-generated=true
 

knip.json

@@ -22,7 +22,6 @@
 				"!**/__tests__/**!",
 				"!vitest.setup.ts!",
 				"!utils/vitest-verbosity.ts!",
-				"!utils/TaskSemaphore.ts!",
 				"!utils/WorkspacePathResolver.ts!",
 				"!test-utils/**!",
 				"!utils/versionToken.ts!",

package.json

@@ -128,6 +128,9 @@
 			"zod": "4.6.7",
 			"@lmstudio/sdk>zod": "3.25.76",
 			"katex": "0.19.1"
+		},
+		"patchedDependencies": {
+			"@qdrant/js-client-rest@1.20.0": "patches/@qdrant__js-client-rest@1.20.0.patch"
 		}
 	}
 }

packages/types/src/vscode-extension-host.ts

@@ -857,6 +857,9 @@ export type CheckpointRestorePayload = z.infer<typeof checkoutRestorePayloadSche
 export interface IndexingStatusPayload {
 	state: "Standby" | "Indexing" | "Indexed" | "Error" | "Stopping"
 	message: string
+	failedFiles?: FailedFileStatus[]
+	failedFileSummary?: FailedFileSummary
+	providerWaits?: ProviderWaitStatus[]
 }
 
 export interface IndexClearedPayload {
@@ -873,6 +876,23 @@ export type WebViewMessagePayload =
 	| EditQueuedMessagePayload
 	| { item: MarketplaceItem; parameters?: Record<string, string> }
 
+export interface FailedFileStatus {
+	path: string
+	activity: "read" | "parse" | "embed" | "upsert" | "delete"
+	reason: string
+	state: "retrying" | "exhausted"
+}
+
+export interface FailedFileSummary extends FailedFileStatus {
+	count: number
+}
+
+export interface ProviderWaitStatus {
+	service: string
+	resumeAt?: number
+	waitingFiles: string[]
+}
+
 export interface IndexingStatus {
 	systemStatus: string
 	message?: string
@@ -882,6 +902,9 @@ export interface IndexingStatus {
 	workspacePath?: string
 	workspaceEnabled?: boolean
 	autoEnableDefault?: boolean
+	failedFiles?: FailedFileStatus[]
+	failedFileSummary?: FailedFileSummary
+	providerWaits?: ProviderWaitStatus[]
 }
 
 export interface IndexingStatusUpdateMessage {

patches/@qdrant__js-client-rest@1.20.0.patch

@@ -0,0 +1,56 @@
+diff --git a/dist/browser/index.cjs b/dist/browser/index.cjs
+--- a/dist/browser/index.cjs
++++ b/dist/browser/index.cjs
+@@ -991,8 +991,8 @@
+         }
+         catch (error) {
+             if (error instanceof ApiError && error.status === 429) {
+-                const retryAfterHeader = error.headers.get('retry-after')?.[0];
+-                if (retryAfterHeader) {
++                const retryAfterHeader = error.headers.get('retry-after');
++                if (retryAfterHeader && !isNaN(Number(retryAfterHeader))) {
+                     throw new QdrantClientResourceExhaustedError(error.message, retryAfterHeader);
+                 }
+             }
+diff --git a/dist/browser/index.js b/dist/browser/index.js
+--- a/dist/browser/index.js
++++ b/dist/browser/index.js
+@@ -989,8 +989,8 @@
+         }
+         catch (error) {
+             if (error instanceof ApiError && error.status === 429) {
+-                const retryAfterHeader = error.headers.get('retry-after')?.[0];
+-                if (retryAfterHeader) {
++                const retryAfterHeader = error.headers.get('retry-after');
++                if (retryAfterHeader && !isNaN(Number(retryAfterHeader))) {
+                     throw new QdrantClientResourceExhaustedError(error.message, retryAfterHeader);
+                 }
+             }
+diff --git a/dist/cjs/api-client.js b/dist/cjs/api-client.js
+--- a/dist/cjs/api-client.js
++++ b/dist/cjs/api-client.js
+@@ -51,8 +51,8 @@
+         }
+         catch (error) {
+             if (error instanceof openapi_typescript_fetch_1.ApiError && error.status === 429) {
+-                const retryAfterHeader = error.headers.get('retry-after')?.[0];
+-                if (retryAfterHeader) {
++                const retryAfterHeader = error.headers.get('retry-after');
++                if (retryAfterHeader && !isNaN(Number(retryAfterHeader))) {
+                     throw new errors_js_1.QdrantClientResourceExhaustedError(error.message, retryAfterHeader);
+                 }
+             }
+diff --git a/dist/esm/api-client.js b/dist/esm/api-client.js
+--- a/dist/esm/api-client.js
++++ b/dist/esm/api-client.js
+@@ -47,8 +47,8 @@
+         }
+         catch (error) {
+             if (error instanceof ApiError && error.status === 429) {
+-                const retryAfterHeader = error.headers.get('retry-after')?.[0];
+-                if (retryAfterHeader) {
++                const retryAfterHeader = error.headers.get('retry-after');
++                if (retryAfterHeader && !isNaN(Number(retryAfterHeader))) {
+                     throw new QdrantClientResourceExhaustedError(error.message, retryAfterHeader);
+                 }
+             }

src/__tests__/extension.spec.ts

@@ -388,6 +388,9 @@ describe("extension.ts", () => {
 				workspacePath: "/workspace",
 				workspaceEnabled: true,
 				autoEnableDefault: true,
+				failedFiles: [],
+				failedFileSummary: undefined,
+				providerWaits: [],
 			}
 			scope["statusManager"]!["publishStatus"](status)
 			await Promise.resolve()
@@ -434,6 +437,9 @@ describe("extension.ts", () => {
 					workspacePath: "/workspace",
 					workspaceEnabled: true,
 					autoEnableDefault: true,
+					failedFiles: [],
+					failedFileSummary: undefined,
+					providerWaits: [],
 				}
 				scope["statusManager"]!["publishStatus"](status)
 				await Promise.resolve()

src/core/tools/CodebaseSearchTool.ts

@@ -4,7 +4,6 @@ import { Task } from "../task/Task"
 import { CodeIndexManagerRegistry } from "../../services/code-index/code-index-manager-registry"
 import { getWorkspacePath } from "../../utils/path"
 import { formatResponse } from "../prompts/responses"
-import { VectorStoreSearchResult } from "../../services/code-index/interfaces"
 import type { ToolUse } from "../../shared/tools"
 
 import { BaseTool, ToolCallbacks } from "./BaseTool"
@@ -78,18 +77,36 @@ export class CodebaseSearchTool extends BaseTool<"codebase_search"> {
 				throw new Error("Code Indexing is not initialized for this workspace.")
 			}
 
-			const searchResults: VectorStoreSearchResult[] = await manager.searchIndex(query, directoryPrefix)
-
-			if (!searchResults || searchResults.length === 0) {
-				pushToolResult(`No relevant code snippets found for the query: "${query}"`)
-				return
-			}
+			const {
+				results: searchResults,
+				incomplete,
+				failedPaths,
+				waitingPaths,
+			} = await manager.searchIndex(query, directoryPrefix)
+			const incompleteNotice = incomplete
+				? [
+						"Partial index: results may be incomplete.",
+						...(failedPaths.length ? ["Failed files:", ...failedPaths.map((file) => `- ${file}`)] : []),
+						...(waitingPaths.length
+							? [
+									"Waiting files (indexing resumes automatically):",
+									...waitingPaths.map((file) => `- ${file}`),
+								]
+							: []),
+					].join("\n") + "\n\n"
+				: ""
 
 			const jsonResult = {
 				query,
 				results: [],
+				incomplete,
+				failedPaths,
+				waitingPaths,
 			} as {
 				query: string
+				incomplete: boolean
+				failedPaths: string[]
+				waitingPaths: string[]
 				results: Array<{
 					filePath: string
 					score: number
@@ -117,7 +134,12 @@ export class CodebaseSearchTool extends BaseTool<"codebase_search"> {
 			const payload = { tool: "codebaseSearch", content: jsonResult }
 			await task.say("codebase_search_result", JSON.stringify(payload))
 
-			const output = `Query: ${query}
+			if (jsonResult.results.length === 0) {
+				pushToolResult(`${incompleteNotice}No relevant code snippets found for the query: "${query}"`)
+				return
+			}
+
+			const output = `${incompleteNotice}Query: ${query}
 Results:
 
 ${jsonResult.results
@@ -131,7 +153,7 @@ Code Chunk: ${result.codeChunk}
 	.join("\n")}`
 
 			pushToolResult(output)
-		} catch (error: any) {
+		} catch (error) {
 			await handleError("codebase_search", error)
 		}
 	}