aico-ai 1.0.1 → 1.0.5

package/README.md

@@ -1,4 +1,4 @@
-# Aico 🚀
+# Aico AI - Gatekeeper for your code
 
 **Aico** is an intelligent CLI tool that acts as a gatekeeper for your code. It uses AI (powered by Groq) to review your changes before you push them, ensuring high quality, security, and consistency across your projects.
 
@@ -7,6 +7,7 @@
 - 🔍 **AI Code Review**: Semantic analysis of your git diffs to catch bugs, security issues, and code smells.
 - ✨ **Apply Fixes**: Automatically apply suggested improvements with a single click.
 - 📝 **AI Commit Messages**: Generate high-quality, Conventional Commit messages based on your changes.
+- 🤖 **Multi-Provider Support**: Use Groq, OpenAI, Gemini, DeepSeek, or even local models via Ollama.
 - 🛡️ **Git Hook Integration**: Seamlessly integrates with Husky to run reviews as a `pre-push` hook.
 - 🤫 **Silent Mode**: Run reviews without blocking your workflow.
 - 🌍 **Global Config**: Configure your API key once and use it across all your projects.

package/index.js

@@ -13,17 +13,50 @@ import { execSync } from 'child_process';
 async function init() {
     console.log(pc.bold(pc.blue('Aico: Initializing...')));
 
-    const { apiKey } = await prompt({
-        type: 'input',
-        name: 'apiKey',
-        message: 'Enter your Groq API Key:'
+    const { provider } = await prompt({
+        type: 'select',
+        name: 'provider',
+        message: 'Which AI provider would you like to use?',
+        choices: [
+            { name: 'groq', message: 'Groq (Fast & Free tier)' },
+            { name: 'openai', message: 'OpenAI (GPT-4o, etc.)' },
+            { name: 'deepseek', message: 'DeepSeek (Powerful & Cheap)' },
+            { name: 'ollama', message: 'Ollama (Local & Private)' },
+            { name: 'gemini', message: 'Google Gemini' }
+        ]
     });
 
-    if (apiKey) {
-        saveGlobalConfig({ GROQ_API_KEY: apiKey });
-        console.log(pc.green('API Key saved globally in ~/.aicorc'));
+    let config = { provider, providers: {} };
+    config.providers[provider] = {};
+
+    if (provider === 'ollama') {
+        const { baseUrl } = await prompt({
+            type: 'input',
+            name: 'baseUrl',
+            message: 'Ollama Base URL:',
+            initial: 'http://localhost:11434'
+        });
+        config.providers[provider].baseUrl = baseUrl;
+    } else {
+        const { apiKey } = await prompt({
+            type: 'input',
+            name: 'apiKey',
+            message: `Enter your ${provider} API Key:`
+        });
+        config.providers[provider].apiKey = apiKey;
     }
 
+    const { model } = await prompt({
+        type: 'input',
+        name: 'model',
+        message: 'Model name (leave empty for default):',
+        initial: ''
+    });
+    if (model) config.providers[provider].model = model;
+
+    saveGlobalConfig(config);
+    console.log(pc.green(`\nConfiguration saved globally in ~/.aicorc for ${provider}!`));
+
     const { setupHusky } = await prompt({
         type: 'confirm',
         name: 'setupHusky',
@@ -46,16 +79,57 @@ async function init() {
     }
 }
 
+function displayHelp() {
+    console.log(`
+${pc.bold(pc.blue('Aico AI - Gatekeeper for your code'))}
+
+${pc.bold('Usage:')}
+  aico <command> [options]
+
+${pc.bold('Commands:')}
+  ${pc.cyan('review')}    Analyze staged changes and suggest improvements (default)
+  ${pc.cyan('commit')}    Generate and apply an AI-suggested commit message
+  ${pc.cyan('init')}      Setup AI providers and Git hooks
+  ${pc.cyan('help')}      Display this help message
+
+${pc.bold('Options:')}
+  ${pc.cyan('--silent, -s')}    Run review without blocking the push
+  ${pc.cyan('--version, -v')}   Display version number
+  ${pc.cyan('--help, -h')}      Display this help message
+
+${pc.bold('Examples:')}
+  aico review --silent
+  aico commit
+    `);
+}
+
 async function main() {
     const args = process.argv.slice(2);
     const command = args[0] || 'review';
     const isSilent = args.includes('--silent') || args.includes('-s');
 
+    if (args.includes('--help') || args.includes('-h') || command === 'help') {
+        displayHelp();
+        return;
+    }
+
+    if (args.includes('--version') || args.includes('-v')) {
+        const pkg = JSON.parse(fs.readFileSync(new URL('./package.json', import.meta.url), 'utf-8'));
+        console.log(`aico-ai v${pkg.version}`);
+        return;
+    }
+
     if (command === 'init') {
         await init();
         return;
     }
 
+    if (command !== 'review' && command !== 'commit') {
+        console.error(pc.red(`Unknown command: ${command}`));
+        console.log(`Run ${pc.cyan('aico help')} to see available commands.`);
+        process.exit(1);
+    }
+
     if (command === 'commit') {
         console.log(pc.bold(pc.blue('Aico: Generating commit message...')));
         try {

package/lib/ai-service.js

@@ -1,54 +1,118 @@
 import Groq from 'groq-sdk';
 import dotenv from 'dotenv';
-
-import { getApiKey } from './config-utils.js';
+import { getActiveProvider } from './config-utils.js';
 
 dotenv.config();
 
-const groq = new Groq({
-    apiKey: getApiKey(),
-});
-
 /**
- * Sends code diff to Groq for review.
- * @param {string} diff 
- * @returns {Promise<string>} AI review response.
+ * Generic AI Client Wrapper
  */
+class AIClient {
+    constructor() {
+        this.updateConfig();
+    }
+
+    updateConfig() {
+        const { provider, apiKey, model, baseUrl } = getActiveProvider();
+        this.provider = provider;
+        this.apiKey = apiKey;
+        this.model = model || this.getDefaultModel(provider);
+        this.baseUrl = baseUrl;
+
+        if (provider === 'groq') {
+            this.client = new Groq({ apiKey });
+        } else if (provider === 'openai' || provider === 'deepseek' || provider === 'ollama') {
+            // OpenAI compatible APIs
+            const url = provider === 'ollama' ? (baseUrl || 'http://localhost:11434/v1') :
+                provider === 'deepseek' ? 'https://api.deepseek.com' : undefined;
+
+            this.client = {
+                chat: {
+                    completions: {
+                        create: async (params) => {
+                            const res = await fetch(`${url || 'https://api.openai.com/v1'}/chat/completions`, {
+                                method: 'POST',
+                                headers: {
+                                    'Content-Type': 'application/json',
+                                    'Authorization': `Bearer ${apiKey}`
+                                },
+                                body: JSON.stringify(params)
+                            });
+                            if (!res.ok) {
+                                const err = await res.json();
+                                throw new Error(err.error?.message || 'AI API Error');
+                            }
+                            return res.json();
+                        }
+                    }
+                }
+            };
+        }
+    }
+
+    getDefaultModel(provider) {
+        const models = {
+            groq: 'llama-3.3-70b-versatile',
+            openai: 'gpt-4o-mini',
+            deepseek: 'deepseek-chat',
+            ollama: 'llama3'
+        };
+        return models[provider] || 'gpt-3.5-turbo';
+    }
+
+    async createChatCompletion(messages) {
+        if (!this.client) this.updateConfig();
+
+        if (this.provider === 'groq') {
+            return this.client.chat.completions.create({
+                messages,
+                model: this.model,
+            });
+        } else {
+            // OpenAI compatible
+            return this.client.chat.completions.create({
+                messages,
+                model: this.model,
+            });
+        }
+    }
+}
+
+const ai = new AIClient();
+
 export async function reviewCode(diff) {
-    const apiKey = getApiKey();
-    if (!apiKey) {
-        throw new Error('GROQ_API_KEY is not set. Run "aico init" to configure it.');
+    const { apiKey, provider } = getActiveProvider();
+    if (!apiKey && provider !== 'ollama') {
+        throw new Error(`API Key for ${provider} is not set. Run "aico init" to configure it.`);
     }
 
     if (!diff || diff.trim() === '') {
-        return 'No changes detected to review.';
+        throw new Error('No diff provided for review.');
     }
 
     const systemPrompt = `
-You are an expert code reviewer. Analyze the following git diff and provide constructive feedback.
-Focus on:
-1. Security vulnerabilities.
-2. Performance bottlenecks.
-3. Code readability and clean code principles.
-4. Potential bugs.
-
-Format your response as a list of issues. For each issue, provide:
-- File: [filename]
-- Issue: [description]
-- Suggestion: [how to fix it]
-- CorrectedCode: [the FULL corrected content of the file, wrapped in triple backticks]
-
-Keep it concise and professional. Do not use emojis.
+You are a senior code reviewer. Analyze the git diff and identify potential bugs, security issues, or performance bottlenecks.
+For each issue, provide:
+1. File: The filename.
+2. Issue: A concise description of the problem.
+3. Suggestion: How to fix it.
+4. CorrectedCode: The FULL corrected content of the file. This is CRITICAL.
+
+Format your response as a list of issues. Use the following markers:
+[FILE] filename
+[ISSUE] description
+[SUGGESTION] how to fix
+[CORRECTED_CODE]
+\`\`\`
+full file content here
+\`\`\`
 `;
 
     try {
-        const chatCompletion = await groq.chat.completions.create({
-            messages: [
-                { role: 'system', content: systemPrompt },
-                { role: 'user', content: `Review this diff:\n\n${diff}` },
-            ],
-            model: 'llama-3.3-70b-versatile',
-        });
+        const chatCompletion = await ai.createChatCompletion([
+            { role: 'system', content: systemPrompt },
+            { role: 'user', content: `Review this diff:\n\n${diff}` },
+        ]);
 
         return chatCompletion.choices[0]?.message?.content || 'No feedback provided.';
     } catch (error) {
@@ -57,19 +121,10 @@ Keep it concise and professional. Do not use emojis.
     }
 }
 
-/**
- * Generates a commit message based on the diff.
- * @param {string} diff 
- * @returns {Promise<string>} Suggested commit message.
- */
 export async function generateCommitMessage(diff) {
-    const apiKey = getApiKey();
-    if (!apiKey) {
-        throw new Error('GROQ_API_KEY is not set. Run "aico init" to configure it.');
-    }
-
-    if (!diff || diff.trim() === '') {
-        return 'chore: update files';
+    const { apiKey, provider } = getActiveProvider();
+    if (!apiKey && provider !== 'ollama') {
+        throw new Error(`API Key for ${provider} is not set. Run "aico init" to configure it.`);
     }
 
     const systemPrompt = `
@@ -78,18 +133,14 @@ Analyze the diff and provide a single, concise commit message.
 Format: <type>(<scope>): <description>
 Types: feat, fix, docs, style, refactor, test, chore.
 Description: Imperative, present tense, no period at the end.
-
 Do not include any other text, only the commit message itself. Do not use emojis.
 `;
 
     try {
-        const chatCompletion = await groq.chat.completions.create({
-            messages: [
-                { role: 'system', content: systemPrompt },
-                { role: 'user', content: `Generate a commit message for this diff:\n\n${diff}` },
-            ],
-            model: 'llama-3.3-70b-versatile',
-        });
+        const chatCompletion = await ai.createChatCompletion([
+            { role: 'system', content: systemPrompt },
+            { role: 'user', content: `Generate a commit message for this diff:\n\n${diff}` },
+        ]);
 
         return chatCompletion.choices[0]?.message?.content?.trim() || 'chore: update files';
     } catch (error) {

package/lib/config-utils.js

@@ -5,11 +5,13 @@ import path from 'path';
 const CONFIG_PATH = path.join(os.homedir(), '.aicorc');
 
 /**
- * Saves the API key to a global config file in the user's home directory.
- * @param {string} apiKey 
+ * Saves the global config.
+ * @param {object} config 
  */
 export function saveGlobalConfig(config) {
-    fs.writeFileSync(CONFIG_PATH, JSON.stringify(config, null, 2));
+    const currentConfig = loadGlobalConfig() || {};
+    const newConfig = { ...currentConfig, ...config };
+    fs.writeFileSync(CONFIG_PATH, JSON.stringify(newConfig, null, 2));
 }
 
 /**
@@ -28,13 +30,31 @@ export function loadGlobalConfig() {
 }
 
 /**
- * Gets the API key from environment variables or global config.
- * @returns {string|null}
+ * Gets the active provider configuration.
+ * @returns {object} { provider, apiKey, model, baseUrl }
  */
-export function getApiKey() {
-    // Priority: Process Env > Local .env (handled by dotenv) > Global Config
-    if (process.env.GROQ_API_KEY) return process.env.GROQ_API_KEY;
+export function getActiveProvider() {
+    const config = loadGlobalConfig();
+    const provider = process.env.AICO_PROVIDER || config?.provider || 'groq';
 
-    const globalConfig = loadGlobalConfig();
-    return globalConfig?.GROQ_API_KEY || null;
+    // Priority: Process Env > Global Config
+    const providers = config?.providers || {};
+    const providerConfig = providers[provider] || {};
+
+    const apiKeyEnvMap = {
+        groq: 'GROQ_API_KEY',
+        openai: 'OPENAI_API_KEY',
+        gemini: 'GEMINI_API_KEY',
+        deepseek: 'DEEPSEEK_API_KEY',
+        anthropic: 'ANTHROPIC_API_KEY'
+    };
+
+    const apiKey = process.env[apiKeyEnvMap[provider]] || providerConfig.apiKey;
+
+    return {
+        provider,
+        apiKey,
+        model: providerConfig.model,
+        baseUrl: providerConfig.baseUrl
+    };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aico-ai",
-  "version": "1.0.1",
+  "version": "1.0.5",
   "type": "module",
   "main": "index.js",
   "bin": {
@@ -17,10 +17,21 @@
     "commit",
     "groq",
     "cli",
-    "developer-tools"
+    "developer-tools",
+    "chatgpt",
+    "gemini",
+    "artificial-intelligence"
   ],
   "author": "Lucas Silva from CodeTech Software <projetos@codetechsoftware.com.br>",
   "license": "ISC",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/LukasdeSouza/aico-ai.git"
+  },
+  "bugs": {
+    "url": "https://github.com/LukasdeSouza/aico-ai/issues"
+  },
+  "homepage": "https://github.com/LukasdeSouza/aico-ai#readme",
   "dependencies": {
     "dotenv": "^17.2.3",
     "enquirer": "^2.4.1",