Release v5.4.1

.github/ISSUE_TEMPLATE/config.yml

@@ -7,7 +7,7 @@ contact_links:
       url: https://github.com/MatterAIOrg/AxonCode/discussions/categories/2-design-improvements
       about: Suggestions for better design (where the current UI/UX is not clear).
     - name: Leave a Review
-      url: https://marketplace.visualstudio.com/items?itemName=kilocode.Kilo-Code&ssr=false#review-details
+      url: https://marketplace.visualstudio.com/items?itemName=matterai.axon-code&ssr=false#review-details
       about: Enjoying Axon Code? Leave a review here!
     - name: Join our Discord
       url: https://discord.gg/fJU5DvanU3

apps/kilocode-docs/.kilocode/rules/memory-bank/tech.md

@@ -110,7 +110,7 @@ npx docusaurus build
 
 ### VS Code Marketplace
 
-- **Extension URL**: https://marketplace.visualstudio.com/items?itemName=kilocode.kilo-code
+- **Extension URL**: https://marketplace.visualstudio.com/items?itemName=matterai.axon-code
 - **Open VSX**: https://open-vsx.org/extension/kilocode/kilo-code
 
 ## Deployment Architecture

apps/kilocode-docs/docs/getting-started/installing.md

@@ -15,7 +15,7 @@ Axon Code is a VS Code extension that brings AI-powered coding assistance direct
 
 :::tip
 
-If you already have VS Code installed: [Click here to install Axon Code](vscode:extension/kilocode.Kilo-Code)
+If you already have VS Code installed: [Click here to install Axon Code](vscode:extension/matterai.axon-code)
 
 :::
 
@@ -36,7 +36,7 @@ After installation, find the Axon Code icon (<img src="/docs/img/matterai-ic.svg
 
 :::tip
 
-If you already have Cursor installed: [Click here to install Axon Code](cursor:extension/kilocode.Kilo-Code)
+If you already have Cursor installed: [Click here to install Axon Code](cursor:extension/matterai.axon-code)
 
 :::
 

apps/kilocode-docs/docs/seats/getting-started.md

@@ -49,7 +49,7 @@ seats for Kilo in the Teams or Enterprise subscription brings transparent AI cod
 Team members receive invitation emails with these steps:
 
 1. Accept the team invitation
-2. Install Axon Code from [VS Code Marketplace](vscode:extension/kilocode.kilo-code)
+2. Install Axon Code from [VS Code Marketplace](vscode:extension/matterai.axon-code)
 3. Sign in with their invited email
 4. Start coding with AI assistance
 

apps/kilocode-docs/i18n/zh-CN/docusaurus-plugin-content-docs/current/getting-started/installing.md

@@ -15,7 +15,7 @@ Kilo Code是一款VS Code扩展，可将AI编程助手直接集成到编辑器
 
 :::tip
 
-若已安装VS Code：[点击此处直接安装Kilo Code](vscode:extension/kilocode.Kilo-Code)
+若已安装VS Code：[点击此处直接安装Kilo Code](vscode:extension/matterai.axon-code)
 
 :::
 
@@ -36,7 +36,7 @@ Kilo Code是一款VS Code扩展，可将AI编程助手直接集成到编辑器
 
 :::tip
 
-若已安装Cursor：[点击此处直接安装Kilo Code](cursor:extension/kilocode.Kilo-Code)
+若已安装Cursor：[点击此处直接安装Kilo Code](cursor:extension/matterai.axon-code)
 
 :::
 

apps/kilocode-docs/i18n/zh-CN/docusaurus-theme-classic/footer.json

@@ -37,7 +37,7 @@
 	},
 	"link.item.label.VS Code Marketplace": {
 		"message": "VS Code 扩展市场",
-		"description": "The label of footer link with label=VS Code Marketplace linking to https://marketplace.visualstudio.com/items?itemName=kilocode.kilo-code"
+		"description": "The label of footer link with label=VS Code Marketplace linking to https://marketplace.visualstudio.com/items?itemName=matterai.axon-code"
 	},
 	"link.item.label.Open VSX Registry": {
 		"message": "Open VSX Registry",

apps/kilocode-docs/src/constants.ts

@@ -19,7 +19,7 @@ export const GITHUB_ISSUES_MAIN_URL = `${GITHUB_MAIN_REPO_URL}/issues`
 export const GITHUB_FEATURES_URL = `${GITHUB_MAIN_REPO_URL}/discussions/categories/ideas`
 
 // Download links
-export const VSCODE_MARKETPLACE_URL = "https://marketplace.visualstudio.com/items?itemName=kilocode.kilo-code"
+export const VSCODE_MARKETPLACE_URL = "https://marketplace.visualstudio.com/items?itemName=matterai.axon-code"
 export const OPEN_VSX_URL = "https://open-vsx.org/extension/kilocode/kilo-code"
 
 // Company links

apps/playwright-e2e/helpers/webview-helpers.ts

@@ -11,7 +11,7 @@ const defaultPlaywrightApiConfig = {
 
 export async function findWebview(workbox: Page): Promise<FrameLocator> {
 	const webviewFrameEl = workbox.frameLocator(
-		'iframe[src*="extensionId=kilocode.kilo-code"][src*="purpose=webviewView"]',
+		'iframe[src*="extensionId=matterai.axon-code"][src*="purpose=webviewView"]',
 	)
 	await webviewFrameEl.locator("#active-frame")
 	return webviewFrameEl.frameLocator("#active-frame")

apps/playwright-e2e/tests/playwright-base-test.ts

@@ -69,7 +69,7 @@ export const test = base.extend<TestFixtures>({
 				"--disable-crash-reporter",
 				"--enable-logging",
 				"--log-level=0",
-				"--disable-extensions-except=kilocode.kilo-code",
+				"--disable-extensions-except=matterai.axon-code",
 				"--disable-extension-recommendations",
 				"--disable-extension-update-check",
 				"--disable-default-apps",
@@ -79,7 +79,7 @@ export const test = base.extend<TestFixtures>({
 				`--extensionDevelopmentPath=${path.resolve(__dirname, "..", "..", "..", "src")}`,
 				`--extensions-dir=${path.join(defaultCachePath, "extensions")}`,
 				`--user-data-dir=${userDataDir}`,
-				"--enable-proposed-api=kilocode.kilo-code",
+				"--enable-proposed-api=matterai.axon-code",
 				await createProject(),
 			],
 		})

apps/vscode-e2e/.vscode-test.mjs

@@ -12,5 +12,5 @@ export default defineConfig({
 		ui: "tdd",
 		timeout: 60000,
 	},
-	launchArgs: ["--enable-proposed-api=kilocode.Kilo-Code", "--disable-extensions"],
+	launchArgs: ["--enable-proposed-api=matterai.axon-code", "--disable-extensions"],
 })

apps/vscode-e2e/src/suite/index.ts

@@ -8,7 +8,7 @@ import type { RooCodeAPI } from "@roo-code/types"
 import { waitFor } from "./utils"
 
 export async function run() {
-	const extension = vscode.extensions.getExtension<RooCodeAPI>("kilocode.kilo-code")
+	const extension = vscode.extensions.getExtension<RooCodeAPI>("matterai.axon-code")
 
 	if (!extension) {
 		throw new Error("Extension not found")

benchmark/src/runExercise.ts

@@ -28,7 +28,7 @@ export async function run() {
 	 * Activate the extension.
 	 */
 
-	const extension = vscode.extensions.getExtension<RooCodeAPI>("kilocode.Kilo-Code")
+	const extension = vscode.extensions.getExtension<RooCodeAPI>("matterai.axon-code")
 
 	if (!extension) {
 		throw new Error("Extension not found.")

cli/src/host/VSCode.ts

@@ -2068,7 +2068,7 @@ export function createVSCodeAPIMock(extensionRootPath: string, workspacePath: st
 			all: [],
 			getExtension: (extensionId: string) => {
 				// Mock the extension object with extensionUri for theme loading
-				if (extensionId === "kilocode.kilo-code") {
+				if (extensionId === "matterai.axon-code") {
 					return {
 						id: extensionId,
 						extensionUri: context.extensionUri,

packages/types/src/__tests__/kilocode.test.ts

@@ -258,8 +258,8 @@ describe("URL functions", () => {
 			expect(getKiloUrlFromToken("https://api.matterai.so/organizations/123/defaults", prodToken)).toBe(
 				"https://api.matterai.so/organizations/123/defaults",
 			)
-			expect(getKiloUrlFromToken("https://api.matterai.so/v1/web/", prodToken)).toBe(
-				"https://api.matterai.so/v1/web/",
+			expect(getKiloUrlFromToken("https://api2.matterai.so/v1/web/", prodToken)).toBe(
+				"https://api2.matterai.so/v1/web/",
 			)
 			expect(getKiloUrlFromToken("https://api.matterai.so/users/notifications", prodToken)).toBe(
 				"https://api.matterai.so/users/notifications",

packages/types/src/codebase-index.ts

@@ -5,8 +5,8 @@ import { z } from "zod"
  */
 export const CODEBASE_INDEX_DEFAULTS = {
 	MIN_SEARCH_RESULTS: 10,
-	MAX_SEARCH_RESULTS: 5,
-	DEFAULT_SEARCH_RESULTS: 5,
+	MAX_SEARCH_RESULTS: 2,
+	DEFAULT_SEARCH_RESULTS: 2,
 	SEARCH_RESULTS_STEP: 10,
 	MIN_SEARCH_SCORE: 0.5,
 	MAX_SEARCH_SCORE: 1,

scripts/reset-kilocode-state.sh

@@ -5,12 +5,12 @@ echo "Kilocode state is being reset.  This probably doesn't work while VS Code i
 # Reset the secrets:
 sqlite3 ~/Library/Application\ Support/Code/User/globalStorage/state.vscdb \
 "DELETE FROM ItemTable WHERE \
-    key = 'kilocode.kilo-code' OR \
+    key = 'matterai.axon-code' OR \
     key LIKE 'workbench.view.extension.kilo-code%' OR \
-    key LIKE 'secret://{\"extensionId\":\"kilocode.kilo-code\",%';"
+    key LIKE 'secret://{\"extensionId\":\"matterai.axon-code\",%';"
 
 # delete all kilocode state files:
-rm -rf ~/Library/Application\ Support/Code/User/globalStorage/kilocode.kilo-code/
+rm -rf ~/Library/Application\ Support/Code/User/globalStorage/matterai.axon-code/
 
 # clear some of the vscode cache that I've observed contains kilocode related entries:
 rm -f ~/Library/Application\ Support/Code/CachedProfilesData/__default__profile__/extensions.user.cache

src/api/providers/kilocode-models.ts

@@ -27,10 +27,10 @@ export type KiloCodeModel = {
 }
 
 export const KILO_CODE_MODELS: Record<string, KiloCodeModel> = {
-	"axon-code": {
-		id: "axon-code",
-		name: "Axon Code",
-		description: "Axon Code is super intelligent LLM model for coding tasks",
+	"axon-auto": {
+		id: "axon-auto",
+		name: "Auto",
+		description: "Auto is a model that automatically selects the best model for the task",
 		input_modalities: ["text"],
 		context_length: 200000,
 		max_output_length: 32768,
@@ -53,22 +53,21 @@ export const KILO_CODE_MODELS: Record<string, KiloCodeModel> = {
 		created: 1750426201,
 		owned_by: "matterai",
 		pricing: {
-			prompt: "0.000001",
-			completion: "0.000004",
+			prompt: "0.0",
+			completion: "0.0",
 			image: "0",
 			request: "0",
 			input_cache_reads: "0",
 			input_cache_writes: "0",
 		},
 	},
-	"axon-mini": {
-		id: "axon-mini",
-		name: "Axon Mini",
-		description:
-			"Axon Mini is an general purpose super intelligent LLM coding model for low-effort day-to-day tasks",
+	"axon-code": {
+		id: "axon-code",
+		name: "Axon Code 1.3",
+		description: "Axon Code is super intelligent LLM model for coding tasks",
 		input_modalities: ["text"],
 		context_length: 200000,
-		max_output_length: 16384,
+		max_output_length: 32768,
 		output_modalities: ["text"],
 		supported_sampling_parameters: [
 			"temperature",
@@ -88,8 +87,8 @@ export const KILO_CODE_MODELS: Record<string, KiloCodeModel> = {
 		created: 1750426201,
 		owned_by: "matterai",
 		pricing: {
-			prompt: "2.5e-7",
-			completion: "0.000001",
+			prompt: "0.0",
+			completion: "0.0",
 			image: "0",
 			request: "0",
 			input_cache_reads: "0",
@@ -98,9 +97,9 @@ export const KILO_CODE_MODELS: Record<string, KiloCodeModel> = {
 	},
 	"axon-code-2": {
 		id: "axon-code-2",
-		name: "Axon Code 2",
+		name: "Axon Code 2.1",
 		description:
-			"Axon Code 2 is the next-generation of Axon Code for coding tasks, currently in experimental stage.",
+			"Axon Code 2.1 is the next-generation of Axon Code for coding tasks, currently in experimental stage.",
 		input_modalities: ["text"],
 		context_length: 200000,
 		max_output_length: 64000,
@@ -133,9 +132,9 @@ export const KILO_CODE_MODELS: Record<string, KiloCodeModel> = {
 	},
 	"axon-code-2-pro": {
 		id: "axon-code-2-pro",
-		name: "Axon Code 2 Pro",
+		name: "Axon Code 2.1 Pro",
 		description:
-			"Axon Code 2 Pro is the next-generation of Axon Code for coding tasks, currently in experimental stage.",
+			"Axon Code 2.1 Pro is the next-generation of Axon Code for coding tasks, currently in experimental stage.",
 		input_modalities: ["text"],
 		context_length: 200000,
 		max_output_length: 64000,
@@ -166,73 +165,4 @@ export const KILO_CODE_MODELS: Record<string, KiloCodeModel> = {
 			input_cache_writes: "0",
 		},
 	},
-	// "gemini-3-flash-preview": {
-	// 	id: "gemini-3-flash-preview",
-	// 	name: "Gemini 3 Flash Preview",
-	// 	description:
-	// 		"Gemini 3 Flash Preview model is built for speed, combining frontier intelligence with superior search and grounding.",
-	// 	input_modalities: ["text"],
-	// 	context_length: 256000,
-	// 	max_output_length: 32768,
-	// 	output_modalities: ["text"],
-	// 	supported_sampling_parameters: [
-	// 		"temperature",
-	// 		"top_p",
-	// 		"top_k",
-	// 		"repetition_penalty",
-	// 		"frequency_penalty",
-	// 		"presence_penalty",
-	// 		"seed",
-	// 		"stop",
-	// 	],
-	// 	supported_features: ["tools", "structured_outputs", "web_search"],
-	// 	openrouter: {
-	// 		slug: "google/gemini-3-flash-preview",
-	// 	},
-	// 	datacenters: [{ country_code: "US" }],
-	// 	created: 1750426201,
-	// 	owned_by: "google",
-	// 	pricing: {
-	// 		prompt: "0.000001",
-	// 		completion: "0.000004",
-	// 		image: "0",
-	// 		request: "0",
-	// 		input_cache_reads: "0",
-	// 		input_cache_writes: "0",
-	// 	},
-	// },
-	// "gemini-3-pro-preview": {
-	// 	id: "gemini-3-pro-preview",
-	// 	name: "Gemini 3 Pro Preview",
-	// 	description: "Gemini 3 Pro Preview model is an agentic and vibe-coding model by Google",
-	// 	input_modalities: ["text"],
-	// 	context_length: 256000,
-	// 	max_output_length: 32768,
-	// 	output_modalities: ["text"],
-	// 	supported_sampling_parameters: [
-	// 		"temperature",
-	// 		"top_p",
-	// 		"top_k",
-	// 		"repetition_penalty",
-	// 		"frequency_penalty",
-	// 		"presence_penalty",
-	// 		"seed",
-	// 		"stop",
-	// 	],
-	// 	supported_features: ["tools", "structured_outputs", "web_search"],
-	// 	openrouter: {
-	// 		slug: "google/gemini-3-pro-preview",
-	// 	},
-	// 	datacenters: [{ country_code: "US" }],
-	// 	created: 1750426201,
-	// 	owned_by: "google",
-	// 	pricing: {
-	// 		prompt: "0.000001",
-	// 		completion: "0.000004",
-	// 		image: "0",
-	// 		request: "0",
-	// 		input_cache_reads: "0",
-	// 		input_cache_writes: "0",
-	// 	},
-	// },
 }

src/api/providers/kilocode-openrouter.ts

@@ -30,7 +30,7 @@ export class KilocodeOpenrouterHandler extends OpenRouterHandler {
 	constructor(options: ApiHandlerOptions) {
 		options = {
 			...options,
-			openRouterBaseUrl: getKiloUrlFromToken("https://api.matterai.so/v1/web/", options.kilocodeToken ?? ""),
+			openRouterBaseUrl: getKiloUrlFromToken("https://api2.matterai.so/v1/web/", options.kilocodeToken ?? ""),
 			openRouterApiKey: options.kilocodeToken,
 		}
 

src/api/providers/openrouter.ts

@@ -228,22 +228,19 @@ export class OpenRouterHandler extends BaseProvider implements SingleCompletionH
 				this.customRequestOptions(metadata), // kilocode_change
 			)
 		} catch (error) {
-			// kilocode_change start
 			if (this.providerName == "KiloCode" && isAnyRecognizedKiloCodeError(error)) {
 				throw error
 			}
 			const err = new Error(makeOpenRouterErrorReadable(error)) as any
 			err.status = error?.status || error?.code
 			throw err
-			// kilocode_change end
 		}
 
 		let lastUsage: CompletionUsage | undefined = undefined
 		let inferenceProvider: string | undefined // kilocode_change
 
 		try {
 			let fullContent = ""
-			let fullReasoning = "" // kilocode_change: variable kept for structural integrity if needed, but unused logs removed
 
 			let isThinking = false
 

src/core/tools/reportBugTool.ts

@@ -49,7 +49,7 @@ export async function reportBugTool(
 			// Derive system information values algorithmically
 			const operatingSystem = os.platform() + " " + os.release()
 			const kilocodeVersion =
-				vscode.extensions.getExtension("kilocode.kilo-code")?.packageJSON.version || "Unknown"
+				vscode.extensions.getExtension("matterai.axon-code")?.packageJSON.version || "Unknown"
 			const systemInfo = `VSCode: ${vscode.version}, Node.js: ${process.version}, Architecture: ${os.arch()}`
 			const providerAndModel = `${(await cline.providerRef.deref()?.contextProxy.getGlobalState("apiProvider")) as string} / ${cline.api.getModel().id}`
 

src/extension.ts

@@ -256,7 +256,7 @@ export async function activate(context: vscode.ExtensionContext) {
 			// https://discord.com/channels/1349288496988160052/1395865796026040470
 			await vscode.commands.executeCommand(
 				"workbench.action.openWalkthrough",
-				"kilocode.kilo-code#kiloCodeWalkthrough",
+				"matterai.axon-code#kiloCodeWalkthrough",
 				false,
 			)
 		} catch (error) {