feat(config): add quick and interactive model configuration

Author: RajaRakoto

.taskmaster/config.json

@@ -14,7 +14,7 @@
     },
     "fallback": {
       "provider": "gemini-cli",
-      "modelId": "gemini-2.5-pro",
+      "modelId": "gemini-2.5-flash",
       "maxTokens": 65536,
       "temperature": 0.2
     }

src/constants/index.ts

@@ -61,7 +61,7 @@ export const TASKMASTER_INIT_MSG = "TaskMaster AI Core initialized";
 export const TASKS_FILE_WARN: (path: string) => string = (path: string) =>
 	`tasks.json not found at "${path}". Please generate tasks from PRD first.`;
 
-// Supported languages for TMAI responses
+// supported languages for TMAI responses
 export const LANG = [
 	"English",
 	"French",
@@ -72,3 +72,9 @@ export const LANG = [
 	"Chinese",
 	"Japanese",
 ] as const;
+
+// ai models configuration
+export const AI_MODELS = [
+	{ name: "Gemini 2.5 Pro", value: "gemini-2.5-pro" },
+	{ name: "Gemini 2.5 Flash", value: "gemini-2.5-flash" },
+];

src/core/taskmaster/TaskMaster.ts

@@ -27,9 +27,6 @@ import {
 	readJsonFileAsync,
 } from "@/utils/extras";
 
-/* asks */
-import { askLangAsync } from "@/core/taskmaster/asks";
-
 /* types */
 import type { T_PackageManager } from "@/@types/index";
 import type { I_Tasks, Status, Priority } from "@/@types/tasks";
@@ -326,21 +323,6 @@ export class TaskMaster {
 		}, oraOptions);
 	}
 
-	// TODO: done
-	/**
-	 * @description Sets the response language for TMAI
-	 * @param lang Language to set for TMAI responses
-	 */
-	public async setLangAsync(lang: string): Promise<void> {
-		await this.executeCommandAsync(
-			`Setting TMAI response language to ${chalk.bold(lang)}...`,
-			`Language set to ${lang} successfully!`,
-			`Failed to set language to ${lang}`,
-			this._mainCommand,
-			["lang", `--response=${lang}`],
-		);
-	}
-
 	// TODO: done
 	/**
 	 * @description Initializes the task-master AI by creating a PRD file
@@ -370,7 +352,7 @@ export class TaskMaster {
 	/**
 	 * @description Configures AI models for task-master by running the interactive setup
 	 */
-	public async configAsync(): Promise<void> {
+	public async interactiveConfigModelAsync(): Promise<void> {
 		await this.executeCommandAsync(
 			"Configuring AI models...",
 			"AI models configured successfully!",
@@ -380,6 +362,61 @@ export class TaskMaster {
 		);
 	}
 
+	// TODO: done
+	/**
+	 * @description Configures AI models with specified models
+	 * @param mainModel The main AI model to use
+	 * @param researchModel The research AI model to use
+	 * @param fallbackModel The fallback AI model to use
+	 */
+	public async configModelsAsync(
+		mainModel: string,
+		researchModel: string,
+		fallbackModel: string,
+	): Promise<void> {
+		const oraOptions = {
+			text: "Configuring AI models...",
+			successText: chalk.bgGreen("AI models configured successfully!"),
+			failText: chalk.bgRed("AI model configuration failed"),
+		};
+
+		await oraPromise(async () => {
+			await runCommandAsync(
+				this._mainCommand,
+				["models", "--set-main", mainModel],
+				false,
+				false,
+			);
+			await runCommandAsync(
+				this._mainCommand,
+				["models", "--set-research", researchModel],
+				false,
+				false,
+			);
+			await runCommandAsync(
+				this._mainCommand,
+				["models", "--set-fallback", fallbackModel],
+				false,
+				false,
+			);
+		}, oraOptions);
+	}
+
+	// TODO: done
+	/**
+	 * @description Sets the response language for TMAI
+	 * @param lang Language to set for TMAI responses
+	 */
+	public async setLangAsync(lang: string): Promise<void> {
+		await this.executeCommandAsync(
+			`Setting TMAI response language to ${chalk.bold(lang)}...`,
+			`Language set to ${lang} successfully!`,
+			`Failed to set language to ${lang}`,
+			this._mainCommand,
+			["lang", `--response=${lang}`],
+		);
+	}
+
 	// ==============================================
 	// Method for Generation and Decomposition
 	// ==============================================

src/core/taskmaster/asks.ts

@@ -21,6 +21,7 @@ import {
 	PRD_PATH,
 	TASKS_STATUSES,
 	TASKS_BCK_DEST_PATH,
+	AI_MODELS,
 } from "@/constants";
 
 /* utils */
@@ -479,3 +480,47 @@ export async function askLangAsync(): Promise<string> {
 	});
 	return lang;
 }
+
+/**
+ * @description Prompts the user to select AI models for main, research, and fallback
+ * Note: This list is designed for faster TMAI testing using free or generously quota'd models.
+ * For a wider selection, use the standard interactive configuration mode.
+ */
+export async function askModelsAsync(): Promise<{
+	mainModel: string;
+	researchModel: string;
+	fallbackModel: string;
+}> {
+	console.log(
+		chalk.blue(
+			"Note: This list is optimized for quicker TMAI testing with free or high-quota models. These models are fully suitable for real projects too. For more options, use the standard interactive configuration mode.",
+		),
+	);
+
+	const { mainModel, researchModel, fallbackModel } = await inquirer.prompt<{
+		mainModel: string;
+		researchModel: string;
+		fallbackModel: string;
+	}>([
+		{
+			type: "list",
+			name: "mainModel",
+			message: "Select the main AI model:",
+			choices: AI_MODELS,
+		},
+		{
+			type: "list",
+			name: "researchModel",
+			message: "Select the research AI model:",
+			choices: AI_MODELS,
+		},
+		{
+			type: "list",
+			name: "fallbackModel",
+			message: "Select the fallback AI model:",
+			choices: AI_MODELS,
+		},
+	]);
+
+	return { mainModel, researchModel, fallbackModel };
+}

src/core/taskmaster/exec.ts

@@ -31,6 +31,7 @@ import {
 	askHybridTaskIdAsync,
 	askMultipleTaskIdAsync,
 	askLangAsync,
+	askModelsAsync,
 } from "@/core/taskmaster/asks";
 
 import chalk from "chalk";
@@ -65,8 +66,11 @@ export async function tmaiInitAsync() {
 		await tmai.installAsync();
 	} else if (choice.tmaiInitMenu === "tmai-init") {
 		await tmai.initAsync();
+	} else if (choice.tmaiInitMenu === "tmai-interactive-config") {
+		await tmai.interactiveConfigModelAsync();
 	} else if (choice.tmaiInitMenu === "tmai-config") {
-		await tmai.configAsync();
+		const { mainModel, researchModel, fallbackModel } = await askModelsAsync();
+		await tmai.configModelsAsync(mainModel, researchModel, fallbackModel);
 	} else if (choice.tmaiInitMenu === "tmai-lang") {
 		const lang = await askLangAsync();
 		await tmai.setLangAsync(lang);

src/prompt.ts

@@ -88,11 +88,19 @@ export const tmaiInitMenu_prompt = [
 				value: "tmai-init",
 			},
 			{
-				name: `${emoji.get("wrench")} 3 - Configure AI models`,
+				name: `${emoji.get(
+					"wrench",
+				)} 3 - Configure AI models (interactive mode)`,
+				value: "tmai-interactive-config",
+			},
+			{
+				name: `${emoji.get("fast_forward")} 4 - Configure AI models (quickly)`,
 				value: "tmai-config",
 			},
 			{
-				name: `${emoji.get("globe_with_meridians")} 4 - Set the response language for AI-generated content in TMAI`,
+				name: `${emoji.get(
+					"globe_with_meridians",
+				)} 5 - Set the response language for AI-generated content in TMAI`,
 				value: "tmai-lang",
 			},
 		],