wip

Author: brianc

src/services/autocomplete/AutocompleteModel.ts

@@ -86,6 +86,11 @@ export class AutocompleteModel {
 	 * @returns ILLM instance or null if configuration is invalid
 	 */
 	public createILLMFromProfile(profile: ProviderSettings, provider: AutocompleteProviderKey): ILLM | null {
+		const model = "codestral-latest"
+		const apiKey = 'TODO_FILL_THIS_IN'
+		const apiBase = 'https://api.mistral.ai/v1'
+		const overrideProvider = 'mistral'
+
 		try {
 			// Extract provider-specific configuration
 			const config = this.extractProviderConfig(profile, provider)
@@ -96,26 +101,26 @@ export class AutocompleteModel {
 
 			// Build LLM options
 			const llmOptions: LLMOptions = {
-				model: config.model,
-				apiKey: config.apiKey,
-				apiBase: config.apiBase,
+				model: model || config.model,
+				apiKey: apiKey || config.apiKey,
+				apiBase: apiBase || config.apiBase,
 				contextLength: 32000, // Default for Codestral models
 				completionOptions: {
-					model: config.model,
+					model: model || config.model,
 					temperature: 0.2, // Lower temperature for more deterministic autocomplete
 					maxTokens: 256, // Reasonable limit for code completions
 				},
 				autocompleteOptions: {
 					...DEFAULT_AUTOCOMPLETE_OPTS,
 					useCache: false, // Disable caching for autocomplete
 				},
-				uniqueId: `autocomplete-${provider}-${Date.now()}`,
+				uniqueId: `autocomplete-${overrideProvider || provider}-${Date.now()}`,
 			}
 
 			// Create appropriate LLM instance based on provider
-			return this.createLLMInstance(provider, llmOptions)
+			return this.createLLMInstance(overrideProvider || provider, llmOptions)
 		} catch (error) {
-			console.error(`[AutocompleteModel] Error creating ILLM for provider ${provider}:`, error)
+			console.error(`[AutocompleteModel] Error creating ILLM for provider ${overrideProvider || provider}:`, error)
 			return null
 		}
 	}

src/services/continuedev/core/autocomplete/CompletionProvider.ts

@@ -55,6 +55,7 @@ export class CompletionProvider {
 			return undefined
 		}
 
+
 		// Temporary fix for JetBrains autocomplete bug as described in https://github.com/continuedev/continue/pull/3022
 		if (llm.model === undefined && llm.completionOptions?.model !== undefined) {
 			llm.model = llm.completionOptions.model
@@ -75,6 +76,7 @@ export class CompletionProvider {
 			llm.useLegacyCompletionsEndpoint = true
 		}
 
+		console.log('using LLM', llm)
 		return llm
 	}
 
@@ -136,6 +138,7 @@ export class CompletionProvider {
 			// Create abort signal if not given
 			if (!token) {
 				const controller = this.loggingService.createAbortController(input.completionId)
+				console.log('creating abort token because none given')
 				token = controller.signal
 			}
 			const startTime = Date.now()
@@ -211,9 +214,12 @@ export class CompletionProvider {
 
 				// Don't postprocess if aborted
 				if (token.aborted) {
+					console.log('aborted')
 					return undefined
 				}
 
+				console.log('raw completion', completion)
+
 				const processedCompletion = helper.options.transform
 					? postprocessCompletion({
 							completion,

src/services/continuedev/core/util/logger.ts

@@ -0,0 +1,32 @@
+function getIconFromLevel(level: string): string {
+  switch (level) {
+    case 'debug':
+      return '🔵';
+    case 'info':
+      return '🟢';
+    case 'warn':
+      return '🟡';
+    case 'error':
+      return '🔴';
+  }
+  return 'X'
+}
+
+export class Logger {
+  constructor(private filename: string, private includeFilename = false) {}
+  #formatMessage(level: string, message: string): string {
+    return `${getIconFromLevel(level)} ${this.includeFilename ? `[${this.filename}] ` : ''}${message}`;
+  }
+  debug(message: string, ...args: any[]) {
+    console.debug(this.#formatMessage('debug', message), ...args);
+  }
+  info(message: string, ...args: any[]) {
+    console.info(this.#formatMessage('info', message), ...args);
+  }
+  warn(message: string, ...args: any[]) {
+    console.info(this.#formatMessage('warn', message), ...args);
+  }
+  error(message: string, ...args: any[]) {
+    console.info(this.#formatMessage('error', message), ...args);
+  }
+}

src/services/continuedev/core/util/treeSitter.ts

@@ -119,8 +119,11 @@ export const IGNORE_PATH_PATTERNS: Partial<Record<LanguageName, RegExp[]>> = {
 }
 
 export async function getParserForFile(filepath: string) {
+	console.log('getting parser for filePath', filepath)
 	try {
+		console.log('calling parser init')
 		await Parser.init()
+		console.log('parse init worked')
 		const parser = new Parser()
 
 		const language = await getLanguageForFile(filepath)
@@ -238,12 +241,16 @@ async function loadLanguageForFileExt(fileExtension: string): Promise<Language>
 
 	for (const p of candidatePaths) {
 		if (fs.existsSync(p)) {
+			console.log('trying to load', p)
 			return await Language.load(p)
+		} else {
+			console.log('not found', p)
 		}
 	}
 
 	// Fallback (will throw with a clear path in error if still missing)
 	const fallback = path.join(candidateRoots[0]!, "node_modules", "tree-sitter-wasms", "out", filename)
+	console.log('loading fallback path', fallback)
 	return await Language.load(fallback)
 }