icoa-cli 1.0.0 → 1.1.1

package/dist/index.js

@@ -11,7 +11,7 @@ import { registerNoteCommand } from './commands/note.js';
 import { registerLogCommand } from './commands/log.js';
 import { registerLangCommand } from './commands/lang.js';
 import { registerSetupCommand } from './commands/setup.js';
-import { isConnected, getConfig } from './lib/config.js';
+import { isConnected, getConfig, saveConfig } from './lib/config.js';
 const BANNER = `
 ${chalk.cyan('╔══════════════════════════════════════════════════════════╗')}
 ${chalk.cyan('║')}                                                          ${chalk.cyan('║')}
@@ -22,8 +22,13 @@ ${chalk.cyan('║')}   ${chalk.bold.white('██║██║     ██║
 ${chalk.cyan('║')}   ${chalk.bold.white('██║╚██████╗╚██████╔╝██║  ██║')}                          ${chalk.cyan('║')}
 ${chalk.cyan('║')}   ${chalk.bold.white('╚═╝ ╚═════╝ ╚═════╝ ╚═╝  ╚═╝')}                         ${chalk.cyan('║')}
 ${chalk.cyan('║')}                                                          ${chalk.cyan('║')}
-${chalk.cyan('║')}   ${chalk.yellow('International Cybersecurity Olympiad 2026')}               ${chalk.cyan('║')}
-${chalk.cyan('║')}   ${chalk.gray('CLI-Native CTF Competition Terminal  v1.0.0')}              ${chalk.cyan('║')}
+${chalk.cyan('║')}   ${chalk.yellow('International Cyber Olympiad in AI 2026')}                ${chalk.cyan('║')}
+${chalk.cyan('║')}   ${chalk.bold.magenta("The World's First AI Security Olympiad")}                 ${chalk.cyan('║')}
+${chalk.cyan('║')}                                                          ${chalk.cyan('║')}
+${chalk.cyan('║')}   ${chalk.green.bold('AI4CTF')} ${chalk.gray('Use AI to solve challenges')}                      ${chalk.cyan('║')}
+${chalk.cyan('║')}   ${chalk.red.bold('CTF4AI')} ${chalk.gray('Hack, attack & evaluate AI systems')}              ${chalk.cyan('║')}
+${chalk.cyan('║')}                                                          ${chalk.cyan('║')}
+${chalk.cyan('║')}   ${chalk.gray('CLI-Native Competition Terminal  v1.1.1')}                ${chalk.cyan('║')}
 ${chalk.cyan('║')}                                                          ${chalk.cyan('║')}
 ${chalk.cyan('╚══════════════════════════════════════════════════════════╝')}
 `;
@@ -75,4 +80,30 @@ registerNoteCommand(program);
 registerLogCommand(program);
 registerLangCommand(program);
 registerSetupCommand(program);
+// Hidden command: switch AI model
+program
+    .command('model', { hidden: true })
+    .argument('[name]', 'model name to switch to')
+    .action((name) => {
+    const config = getConfig();
+    const current = config.geminiModel || 'gemini-2.5-flash';
+    if (!name) {
+        console.log();
+        console.log(chalk.gray('  Current model: ') + chalk.white(current));
+        console.log();
+        console.log(chalk.gray('  Available models:'));
+        console.log(chalk.white('    icoa model gemini-2.5-flash    ') + chalk.gray('Fast, free tier'));
+        console.log(chalk.white('    icoa model gemini-2.5-pro      ') + chalk.gray('Strongest reasoning'));
+        console.log(chalk.white('    icoa model gemma-4-31b-it      ') + chalk.gray('Open-source, free'));
+        console.log(chalk.white('    icoa model gemma-4-26b-a4b-it  ') + chalk.gray('Open-source, lightweight'));
+        console.log(chalk.white('    icoa model <any-model-id>      ') + chalk.gray('Custom model'));
+        console.log();
+    }
+    else {
+        saveConfig({ geminiModel: name });
+        console.log();
+        console.log(chalk.green('  Model switched: ') + chalk.gray(current) + chalk.white(' -> ') + chalk.bold.white(name));
+        console.log();
+    }
+});
 program.parse();

package/dist/lib/gemini.js

@@ -1,4 +1,4 @@
-import { GoogleGenerativeAI } from '@google/generative-ai';
+import { GoogleGenAI } from '@google/genai';
 import { getConfig, saveConfig } from './config.js';
 const SYSTEM_PROMPTS = {
     A: `You are an AI assistant in a cybersecurity CTF competition called ICOA.
@@ -41,7 +41,6 @@ function filterFlagPatterns(text) {
     return text.replace(/icoa\{[^}]*\}/gi, '[FLAG REDACTED]');
 }
 function getApiKey() {
-    // Priority: env var > config
     const envKey = process.env.GEMINI_API_KEY;
     if (envKey)
         return envKey;
@@ -50,10 +49,12 @@ function getApiKey() {
         return config.geminiApiKey;
     return '';
 }
+function getClient(apiKey) {
+    return new GoogleGenAI({ apiKey });
+}
 export async function generateHint(level, question, context) {
     let apiKey = getApiKey();
     if (!apiKey) {
-        // Try to prompt for key interactively
         try {
             const { input } = await import('@inquirer/prompts');
             console.log();
@@ -76,14 +77,15 @@ export async function generateHint(level, question, context) {
                 'Set GEMINI_API_KEY environment variable, or run: icoa setup');
         }
     }
-    const genAI = new GoogleGenerativeAI(apiKey);
-    const model = genAI.getGenerativeModel({
-        model: 'gemini-2.5-pro-preview-05-06',
-        systemInstruction: buildSystemPrompt(level, context),
+    const config = getConfig();
+    const modelName = config.geminiModel || 'gemini-2.5-flash';
+    const ai = getClient(apiKey);
+    const response = await ai.models.generateContent({
+        model: modelName,
+        config: { systemInstruction: buildSystemPrompt(level, context) },
+        contents: question,
     });
-    const result = await model.generateContent(question);
-    const response = result.response;
-    const text = filterFlagPatterns(response.text());
+    const text = filterFlagPatterns(response.text ?? '');
     const usage = response.usageMetadata;
     const tokensUsed = (usage?.promptTokenCount || 0) + (usage?.candidatesTokenCount || 0);
     return { text, tokensUsed };
@@ -93,15 +95,19 @@ export async function translateText(text, targetLang) {
     if (!apiKey) {
         throw new Error('Gemini API key not configured for translation.');
     }
-    const genAI = new GoogleGenerativeAI(apiKey);
-    const model = genAI.getGenerativeModel({ model: 'gemini-2.5-pro-preview-05-06' });
+    const config = getConfig();
+    const modelName = config.geminiModel || 'gemini-2.5-flash';
+    const ai = getClient(apiKey);
     const prompt = `Translate the following CTF challenge description to ${targetLang}.
 Keep all technical terms, code, commands, URLs, and flag formats in English.
 Only translate the narrative/descriptive text.
 
 ${text}`;
-    const result = await model.generateContent(prompt);
-    return result.response.text();
+    const response = await ai.models.generateContent({
+        model: modelName,
+        contents: prompt,
+    });
+    return response.text ?? '';
 }
 export function setApiKey(key) {
     saveConfig({ geminiApiKey: key });

package/dist/types/index.d.ts

@@ -97,6 +97,7 @@ export interface IcoaConfig {
     competitionStartsAt: string;
     competitionEndsAt: string;
     geminiApiKey: string;
+    geminiModel: string;
 }
 export type CompetitionState = 'pre_competition' | 'demo' | 'live' | 'finished' | 'unknown';
 export type HintLevel = 'A' | 'B' | 'C';

package/dist/types/index.js

@@ -25,5 +25,6 @@ export const DEFAULT_CONFIG = {
     competitionStartsAt: '',
     competitionEndsAt: '',
     geminiApiKey: '',
+    geminiModel: 'gemini-2.5-flash',
 };
 export const SUPPORTED_LANGUAGES = ['en', 'zh', 'ja', 'ko', 'es'];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "icoa-cli",
-  "version": "1.0.0",
+  "version": "1.1.1",
   "description": "ICOA CLI — The world's first CLI-native CTF competition terminal",
   "type": "module",
   "bin": {
@@ -24,11 +24,11 @@
   ],
   "license": "SEE LICENSE IN licenses/apache-2.0.txt",
   "dependencies": {
-    "@google/generative-ai": "^0.24.0",
+    "@google/genai": "^1.48.0",
+    "@inquirer/prompts": "^7.5.0",
     "chalk": "^5.4.1",
     "cli-table3": "^0.6.5",
     "commander": "^13.1.0",
-    "@inquirer/prompts": "^7.5.0",
     "marked": "^15.0.7",
     "marked-terminal": "^7.3.0",
     "ora": "^8.2.0"