Inline Assist - Setting section & Support all providers (#1396)

* feat: Ghost Service Settings Section and Provider functionallity

* feat: Add Translations and changeset

* Update src/shared/WebviewMessage.ts

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* Update src/shared/ExtensionMessage.ts

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* Update src/core/webview/ClineProvider.ts

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* Update src/core/webview/ClineProvider.ts

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* test: ghost provider performance test (#1432)

* Update .changeset/orange-hats-like.md

Co-authored-by: Chris Hasson <hassoncs@gmail.com>

* Update webview-ui/src/i18n/locales/en/kilocode.json

Co-authored-by: Chris Hasson <hassoncs@gmail.com>

* refactor: update translations

---------

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Co-authored-by: Chris Hasson <hassoncs@gmail.com>

Author: 3 people

.changeset/orange-hats-like.md

@@ -0,0 +1,7 @@
+---
+"kilo-code": patch
+---
+
+Adds new Settings page for Inline Assist
+
+You can now select the provider you'd like to use for `Inline Assist` commands

packages/types/src/global-settings.ts

@@ -14,6 +14,7 @@ import { telemetrySettingsSchema } from "./telemetry.js"
 import { modeConfigSchema } from "./mode.js"
 import { customModePromptsSchema, customSupportPromptsSchema } from "./mode.js"
 import { languagesSchema } from "./vscode.js"
+import { ghostServiceSettingsSchema } from "./kilocode.js" // kilocode_change
 
 /**
  * GlobalSettings
@@ -116,6 +117,7 @@ export const globalSettingsSchema = z.object({
 	enhancementApiConfigId: z.string().optional(),
 	autocompleteApiConfigId: z.string().optional(), // kilocode_change
 	commitMessageApiConfigId: z.string().optional(), // kilocode_change
+	ghostServiceSettings: ghostServiceSettingsSchema, // kilocode_change
 	historyPreviewCollapsed: z.boolean().optional(),
 	profileThresholds: z.record(z.string(), z.number()).optional(),
 	hasOpenedModeSelector: z.boolean().optional(),
@@ -222,6 +224,7 @@ export const EVALS_SETTINGS: RooCodeSettings = {
 	soundEnabled: false,
 	soundVolume: 0.5,
 	systemNotificationsEnabled: true, // kilocode_change
+	ghostServiceSettings: {}, // kilocode_change
 
 	terminalOutputLineLimit: 500,
 	terminalShellIntegrationTimeout: 30000,

packages/types/src/index.ts

@@ -20,4 +20,5 @@ export * from "./terminal.js"
 export * from "./tool.js"
 export * from "./type-fu.js"
 export * from "./vscode.js"
+export * from "./kilocode.js" // kilocode_change
 export * from "./todo.js"

packages/types/src/kilocode.ts

@@ -0,0 +1,9 @@
+import { z } from "zod"
+
+export const ghostServiceSettingsSchema = z
+	.object({
+		apiConfigId: z.string().optional(),
+	})
+	.optional()
+
+export type GhostServiceSettings = z.infer<typeof ghostServiceSettingsSchema>

src/__mocks__/vscode.js

@@ -10,8 +10,8 @@ const mockDisposable = {
 }
 
 const mockUri = {
-	file: (path) => ({ fsPath: path, path, scheme: "file" }),
-	parse: (path) => ({ fsPath: path, path, scheme: "file" }),
+	file: (path) => ({ fsPath: path, path, scheme: "file", toString: () => path }),
+	parse: (path) => ({ fsPath: path, path, scheme: "file", toString: () => path }),
 }
 
 const mockRange = class {

src/core/webview/ClineProvider.ts

@@ -1468,6 +1468,7 @@ export class ClineProvider
 			maxConcurrentFileReads,
 			allowVeryLargeReads, // kilocode_change
 			autocompleteApiConfigId, // kilocode_change
+			ghostServiceSettings, // kilocode_changes
 			condensingApiConfigId,
 			customCondensingPrompt,
 			codebaseIndexConfig,
@@ -1579,6 +1580,7 @@ export class ClineProvider
 			sharingEnabled: sharingEnabled ?? false,
 			organizationAllowList,
 			autocompleteApiConfigId, // kilocode_change
+			ghostServiceSettings: ghostServiceSettings ?? {}, // kilocode_change
 			condensingApiConfigId,
 			customCondensingPrompt,
 			codebaseIndexModels: codebaseIndexModels ?? EMBEDDING_MODEL_PROFILES,
@@ -1723,6 +1725,7 @@ export class ClineProvider
 			enhancementApiConfigId: stateValues.enhancementApiConfigId,
 			commitMessageApiConfigId: stateValues.commitMessageApiConfigId, // kilocode_change
 			autocompleteApiConfigId: stateValues.autocompleteApiConfigId, // kilocode_change
+			ghostServiceSettings: stateValues.ghostServiceSettings ?? {}, // kilocode_change
 			experiments: stateValues.experiments ?? experimentDefault,
 			autoApprovalEnabled: stateValues.autoApprovalEnabled ?? true,
 			customModes,

src/core/webview/webviewMessageHandler.ts

@@ -13,6 +13,7 @@ import {
 	type GlobalState,
 	type ClineMessage,
 	TelemetryEventName,
+	ghostServiceSettingsSchema, // kilocode_change
 } from "@roo-code/types"
 import { CloudService } from "@roo-code/cloud"
 import { TelemetryService } from "@roo-code/telemetry"
@@ -1341,6 +1342,16 @@ export const webviewMessageHandler = async (
 			await updateGlobalState("autocompleteApiConfigId", message.text)
 			await provider.postStateToWebview()
 			break
+		case "ghostServiceSettings":
+			if (!message.values) {
+				return
+			}
+			// Validate ghostServiceSettings structure
+			const ghostServiceSettings = ghostServiceSettingsSchema.parse(message.values)
+			await updateGlobalState("ghostServiceSettings", ghostServiceSettings)
+			await provider.postStateToWebview()
+			vscode.commands.executeCommand("kilo-code.ghost.reload")
+			break
 		// kilocode_change end
 		case "condensingApiConfigId":
 			await updateGlobalState("condensingApiConfigId", message.text)

src/services/autocomplete/__tests__/MockTextDocument.test.ts

@@ -1,33 +1,58 @@
-import { describe, it, expect } from "vitest"
+import { describe, it, expect, vi, beforeEach } from "vitest"
 import * as vscode from "vscode"
 import { MockTextDocument } from "./MockTextDocument"
 
+vi.mock("vscode", () => ({
+	EndOfLine: { LF: 1, CRLF: 2 },
+	Uri: {
+		parse: vi.fn((str) => ({
+			fsPath: str,
+			path: str,
+			scheme: "file",
+		})),
+	},
+	Position: class Position {
+		constructor(
+			public line: number,
+			public character: number,
+		) {}
+	},
+	Range: class Range {
+		constructor(
+			public start: vscode.Position,
+			public end: vscode.Position,
+		) {}
+	},
+}))
+
 describe("MockTextDocument", () => {
+	const testUri = vscode.Uri.parse("file:///test.ts")
+
 	describe("constructor and basic properties", () => {
 		it("should create document from single line content", () => {
-			const doc = new MockTextDocument("const x = 1")
+			const doc = new MockTextDocument(testUri, "const x = 1")
 
 			expect(doc.lineCount).toBe(1)
 			expect(doc.getText()).toBe("const x = 1")
 		})
 
 		it("should create document from multi-line content", () => {
 			const content = "function test() {\n    return true\n}"
-			const doc = new MockTextDocument(content)
+			const doc = new MockTextDocument(testUri, content)
 
 			expect(doc.lineCount).toBe(3)
 			expect(doc.getText()).toBe(content)
 		})
 
 		it("should handle empty content", () => {
-			const doc = new MockTextDocument("")
+			const doc = new MockTextDocument(testUri, "")
 
 			expect(doc.lineCount).toBe(1)
 			expect(doc.getText()).toBe("")
 		})
 
 		it("should handle content with only newlines", () => {
-			const doc = new MockTextDocument("\n\n\n")
+			const doc = new MockTextDocument(testUri, "\n\n\n")
 
 			expect(doc.lineCount).toBe(4)
 			expect(doc.getText()).toBe("\n\n\n")
@@ -39,7 +64,7 @@ describe("MockTextDocument", () => {
 		let doc: MockTextDocument
 
 		beforeEach(() => {
-			doc = new MockTextDocument(multiLineContent)
+			doc = new MockTextDocument(testUri, multiLineContent)
 		})
 
 		it("should return full text when no range provided", () => {
@@ -94,7 +119,7 @@ describe("MockTextDocument", () => {
 		let doc: MockTextDocument
 
 		beforeEach(() => {
-			doc = new MockTextDocument(content)
+			doc = new MockTextDocument(testUri, content)
 		})
 
 		it("should return correct line information for first line", () => {
@@ -116,7 +141,7 @@ describe("MockTextDocument", () => {
 		})
 
 		it("should return correct line information for whitespace-only line", () => {
-			const docWithWhitespace = new MockTextDocument("line1\n    \nline3")
+			const docWithWhitespace = new MockTextDocument(testUri, "line1\n    \nline3")
 			const line = docWithWhitespace.lineAt(1)
 
 			expect(line.text).toBe("    ")
@@ -154,7 +179,7 @@ describe("MockTextDocument", () => {
 
 	describe("edge cases and special characters", () => {
 		it("should handle tabs correctly", () => {
-			const doc = new MockTextDocument("\tfunction test() {\n\t\treturn true\n\t}")
+			const doc = new MockTextDocument(testUri, "\tfunction test() {\n\t\treturn true\n\t}")
 
 			expect(doc.lineCount).toBe(3)
 
@@ -168,7 +193,7 @@ describe("MockTextDocument", () => {
 		})
 
 		it("should handle mixed whitespace", () => {
-			const doc = new MockTextDocument("  \t  const x = 1")
+			const doc = new MockTextDocument(testUri, "  \t  const x = 1")
 			const line = doc.lineAt(0)
 
 			expect(line.text).toBe("  \t  const x = 1")
@@ -177,15 +202,15 @@ describe("MockTextDocument", () => {
 		})
 
 		it("should handle unicode characters", () => {
-			const doc = new MockTextDocument("const 🚀 = 'rocket'\nconst 中文 = 'chinese'")
+			const doc = new MockTextDocument(testUri, "const 🚀 = 'rocket'\nconst 中文 = 'chinese'")
 
 			expect(doc.lineCount).toBe(2)
 			expect(doc.lineAt(0).text).toBe("const 🚀 = 'rocket'")
 			expect(doc.lineAt(1).text).toBe("const 中文 = 'chinese'")
 		})
 
 		it("should handle Windows line endings (CRLF)", () => {
-			const doc = new MockTextDocument("line1\r\nline2\r\nline3")
+			const doc = new MockTextDocument(testUri, "line1\r\nline2\r\nline3")
 
 			// Note: split("\n") will still work but will include \r in the text
 			expect(doc.lineCount).toBe(3)
@@ -197,7 +222,7 @@ describe("MockTextDocument", () => {
 
 	describe("integration with vscode types", () => {
 		it("should work with vscode.Range for getText", () => {
-			const doc = new MockTextDocument("function test() {\n    return 42\n}")
+			const doc = new MockTextDocument(testUri, "function test() {\n    return 42\n}")
 
 			// Create a range using vscode constructors
 			const start = new vscode.Position(0, 9)
@@ -208,7 +233,7 @@ describe("MockTextDocument", () => {
 		})
 
 		it("should return TextLine compatible with vscode interface", () => {
-			const doc = new MockTextDocument("    const value = 'test'")
+			const doc = new MockTextDocument(testUri, "    const value = 'test'")
 			const line = doc.lineAt(0)
 
 			// Verify it has all required TextLine properties

src/services/autocomplete/__tests__/MockTextDocument.ts

@@ -3,15 +3,33 @@ import * as vscode from "vscode"
 /**
  * A simulated vscode TextDocument for testing.
  */
-export class MockTextDocument {
+export class MockTextDocument implements vscode.TextDocument {
 	private contentLines: string[]
+	public uri: vscode.Uri
+	public fileName: string
+	public isUntitled: boolean = false
+	public languageId: string = "typescript"
+	public version: number = 1
+	public isDirty: boolean = false
+	public isClosed: boolean = false
+	public eol: vscode.EndOfLine = 1 // vscode.EndOfLine.LF
+	public get encoding(): "utf8" {
+		return "utf8"
+	}
+	public get notebook(): undefined {
+		return undefined
+	}
 
-	constructor(content: string) {
+	constructor(uri: vscode.Uri, content: string) {
+		this.uri = uri
+		this.fileName = uri.fsPath
 		this.contentLines = content.split("\n")
 	}
 
 	updateContent(newContent: string): void {
 		this.contentLines = newContent.split("\n")
+		this.version++
+		this.isDirty = true
 	}
 
 	getText(range?: vscode.Range): string {
@@ -49,21 +67,81 @@ export class MockTextDocument {
 	 * @param lineNumber The zero-based line number
 	 * @returns A simplified TextLine object containing the text and position information
 	 */
-	lineAt(lineNumber: number): vscode.TextLine {
-		if (lineNumber < 0 || lineNumber >= this.contentLines.length) {
-			throw new Error(`Invalid line number: ${lineNumber}`)
+	lineAt(positionOrLine: number | vscode.Position): vscode.TextLine {
+		const line = typeof positionOrLine === "number" ? positionOrLine : positionOrLine.line
+		if (line < 0 || line >= this.contentLines.length) {
+			throw new Error(`Invalid line number: ${line}`)
 		}
 
-		const text = this.contentLines[lineNumber]
-		const range = new vscode.Range(new vscode.Position(lineNumber, 0), new vscode.Position(lineNumber, text.length))
+		const text = this.contentLines[line]
+		const range = new vscode.Range(new vscode.Position(line, 0), new vscode.Position(line, text.length))
 
 		return {
 			text,
 			range,
-			lineNumber,
-			rangeIncludingLineBreak: range,
+			lineNumber: line,
+			rangeIncludingLineBreak: range, // Simplified for mock
 			firstNonWhitespaceCharacterIndex: text.search(/\S|$/),
 			isEmptyOrWhitespace: !/\S/.test(text),
-		} as vscode.TextLine
+		}
+	}
+
+	// Add other required methods with mock implementations
+	offsetAt(position: vscode.Position): number {
+		let offset = 0
+		for (let i = 0; i < position.line; i++) {
+			if (i < this.contentLines.length) {
+				offset += this.contentLines[i].length + 1 // +1 for newline
+			}
+		}
+		offset += position.character
+		return offset
+	}
+
+	positionAt(offset: number): vscode.Position {
+		let currentOffset = 0
+		for (let i = 0; i < this.contentLines.length; i++) {
+			const lineLength = this.contentLines[i].length + 1
+			if (currentOffset + lineLength > offset) {
+				return new vscode.Position(i, offset - currentOffset)
+			}
+			currentOffset += lineLength
+		}
+		// If offset is beyond the end of the document
+		const lastLine = this.contentLines.length - 1
+		const lastLineLength = this.contentLines[lastLine]?.length || 0
+		return new vscode.Position(lastLine, lastLineLength)
+	}
+
+	save(): Promise<boolean> {
+		this.isDirty = false
+		return Promise.resolve(true)
+	}
+
+	getWordRangeAtPosition(position: vscode.Position, regex?: RegExp): vscode.Range | undefined {
+		const line = this.lineAt(position.line)
+		const text = line.text
+		const wordRegex =
+			regex || /(-?\d*\.\d\w*)|([^\`\~\!\@\#\$\%\^\&\*\(\)\-\=\+\[\{\]\}\\\|\;\:\'\"\,\.\<\>\/\?\s]+)/g
+
+		let match
+		while ((match = wordRegex.exec(text)) !== null) {
+			const start = match.index
+			const end = start + match[0].length
+			if (position.character >= start && position.character <= end) {
+				return new vscode.Range(position.line, start, position.line, end)
+			}
+		}
+		return undefined
+	}
+
+	validateRange(range: vscode.Range): vscode.Range {
+		// Simplified validation
+		return range
+	}
+
+	validatePosition(position: vscode.Position): vscode.Position {
+		// Simplified validation
+		return position
 	}
 }

src/services/autocomplete/__tests__/MockTextEditor.ts

@@ -25,14 +25,18 @@ export class MockTextEditor {
 
 		if (cursorOffset === -1) {
 			// No cursor marker found - default to position (0,0)
-			this.document = new MockTextDocument(content) as unknown as vscode.TextDocument
+			const uri = vscode.Uri.parse("untitled:mockEditor")
+			this.document = new MockTextDocument(uri, content) as unknown as vscode.TextDocument
 			const defaultPosition = new vscode.Position(0, 0)
 			this.selection = new vscode.Selection(defaultPosition, defaultPosition)
 		} else {
 			// Cursor marker found - remove it and calculate position
 			const cleanContent =
 				content.substring(0, cursorOffset) + content.substring(cursorOffset + CURSOR_MARKER.length)
-			this.document = new MockTextDocument(cleanContent) as unknown as vscode.TextDocument
+
+			const uri = vscode.Uri.parse("untitled:mockEditor")
+			// Create document without cursor marker
+			this.document = new MockTextDocument(uri, cleanContent) as unknown as vscode.TextDocument
 
 			// Calculate line and character for cursor position
 			const beforeCursor = content.substring(0, cursorOffset)