feat(ai): add multi-provider support for AI models

Author: RajaRakoto

.taskmaster/config.json

@@ -1,20 +1,20 @@
 {
   "models": {
     "main": {
-      "provider": "gemini-cli",
-      "modelId": "gemini-2.5-pro",
+      "provider": "openrouter",
+      "modelId": "qwen/qwen3-coder:free",
       "maxTokens": 65536,
       "temperature": 0.2
     },
     "research": {
-      "provider": "gemini-cli",
-      "modelId": "gemini-2.5-pro",
+      "provider": "openrouter",
+      "modelId": "qwen/qwen3-coder:free",
       "maxTokens": 65536,
       "temperature": 0.1
     },
     "fallback": {
-      "provider": "gemini-cli",
-      "modelId": "gemini-2.5-pro",
+      "provider": "openrouter",
+      "modelId": "qwen/qwen3-coder:free",
       "maxTokens": 65536,
       "temperature": 0.2
     }

src/@types/index.d.ts

@@ -1 +1,17 @@
 export type T_PackageManager = "npm" | "pnpm" | "bun";
+
+export const VALID_PROVIDERS = [
+	"openrouter",
+	"ollama",
+	"bedrock",
+	"azure",
+	"vertex",
+] as const;
+
+export type T_Provider = (typeof VALID_PROVIDERS)[number];
+
+export interface I_AIModel {
+	name: string;
+	value: string;
+	provider: T_Provider | null;
+}

src/constants/index.ts

@@ -1,3 +1,11 @@
+/* libs */
+import chalk from "chalk";
+
+/* types */
+import type { I_AIModel } from "@/@types/index";
+
+// ===============================
+
 // dev
 export const DEV_MODE = true;
 
@@ -74,9 +82,22 @@ export const LANG = [
 ] 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" },
+export const AI_MODELS: I_AIModel[] = [
+	{
+		name: `Gemini 2.5 Pro ${chalk.gray("(google | free)")}`,
+		value: "gemini-2.5-pro",
+		provider: null,
+	},
+	{
+		name: `Gemini 2.5 Flash ${chalk.gray("(google | free)")}`,
+		value: "gemini-2.5-flash",
+		provider: null,
+	},
+	{
+		name: `Qwen 3 coder ${chalk.gray("(openrouter | free)")}`,
+		value: "qwen/qwen3-coder:free",
+		provider: "openrouter",
+	},
 ];
 
 // task conversion rules

src/core/TaskMaster.ts

@@ -627,11 +627,13 @@ export class TaskMaster {
 	 * @param mainModel The main AI model to use
 	 * @param researchModel The research AI model to use
 	 * @param fallbackModel The fallback AI model to use
+	 * @param provider Optional provider for the models
 	 */
 	public async configModelsAsync(
 		mainModel: string,
 		researchModel: string,
 		fallbackModel: string,
+		provider?: string,
 	): Promise<void> {
 		const oraOptions = {
 			text: "Configuring AI models...",
@@ -640,24 +642,21 @@ export class TaskMaster {
 		};
 
 		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,
-			);
+			const mainArgs = provider
+				? ["models", "--set-main", mainModel, `--${provider}`]
+				: ["models", "--set-main", mainModel];
+
+			const researchArgs = provider
+				? ["models", "--set-research", researchModel, `--${provider}`]
+				: ["models", "--set-research", researchModel];
+
+			const fallbackArgs = provider
+				? ["models", "--set-fallback", fallbackModel, `--${provider}`]
+				: ["models", "--set-fallback", fallbackModel];
+
+			await runCommandAsync(this._mainCommand, mainArgs, false, false);
+			await runCommandAsync(this._mainCommand, researchArgs, false, false);
+			await runCommandAsync(this._mainCommand, fallbackArgs, false, false);
 		}, oraOptions);
 	}
 

src/core/exec.ts

@@ -3,7 +3,12 @@ import inquirer from "inquirer";
 import path from "node:path";
 
 /* constants */
-import { MAIN_COMMAND, TASKS_PATH, TASKS_STATUSES } from "@/constants";
+import {
+	AI_MODELS,
+	MAIN_COMMAND,
+	TASKS_PATH,
+	TASKS_STATUSES,
+} from "@/constants";
 
 /* core */
 import { TaskMaster } from "@/core/TaskMaster";
@@ -70,7 +75,29 @@ export async function tmaiInitAsync() {
 		await tmai.interactiveConfigModelAsync();
 	} else if (choice.tmaiInitMenu === "tmai-config") {
 		const { mainModel, researchModel, fallbackModel } = await askModelsAsync();
-		await tmai.configModelsAsync(mainModel, researchModel, fallbackModel);
+
+		// Get provider from AI_MODELS configuration
+		const mainModelObj = AI_MODELS.find((model) => model.value === mainModel);
+		const researchModelObj = AI_MODELS.find(
+			(model) => model.value === researchModel,
+		);
+		const fallbackModelObj = AI_MODELS.find(
+			(model) => model.value === fallbackModel,
+		);
+
+		// Use the first provider that is defined and not null
+		const provider =
+			mainModelObj?.provider ||
+			researchModelObj?.provider ||
+			fallbackModelObj?.provider ||
+			undefined;
+
+		await tmai.configModelsAsync(
+			mainModel,
+			researchModel,
+			fallbackModel,
+			provider,
+		);
 	} else if (choice.tmaiInitMenu === "tmai-lang") {
 		const lang = await askLangAsync();
 		await tmai.setLangAsync(lang);