ownkey 0.2.1 → 0.5.1

package/README.md

@@ -1,9 +1,24 @@
-# OwnKey CLI
+<div align="center">
 
-**Local-first AI coding agent for intelligent codebase analysis**
+<img src="assets/logo.png" alt="OwnKey Logo" width="600"/>
+
+# OwnKey
+
+**Privacy-First AI Coding Agent**
 
 [![npm version](https://badge.fury.io/js/ownkey.svg)](https://www.npmjs.com/package/ownkey)
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![Downloads](https://img.shields.io/npm/dm/ownkey.svg)](https://www.npmjs.com/package/ownkey)
+
+*Intelligent code analysis powered by Google Gemini 2.5 • Your code never leaves your machine*
+
+[Quick Start](#-quick-start) • [Features](#-features) • [Installation](#-installation) • [Documentation](#-commands)
+
+</div>
+
+---
+
+> ⚠️ **v0.5.0** - Currently supports Google Gemini. OpenAI, Anthropic, and Ollama coming soon!
 
 ## 🚀 Quick Start
 
@@ -11,24 +26,58 @@
 # Install globally
 npm install -g ownkey
 
-# Configure OwnKey
-ownkey config
-
-# Scan your project
-ownkey scan .
+# Configure with your Gemini API key
+ownkey config --provider gemini --api-key YOUR_KEY
 
-# Get AI suggestions (coming in v0.5.0)
+# Analyze your codebase
 ownkey suggest .
 ```
 
-## 📋 Features
+**Get your free Gemini API key:** [Google AI Studio](https://aistudio.google.com/app/apikey)
+
+## ✨ Features
+
+<table>
+<tr>
+<td width="50%">
+
+### 🤖 AI-Powered Analysis
+Intelligent code review using Google Gemini 2.5 Pro
+
+### 🔒 Privacy-First
+Your code never leaves your machine - all analysis is local
+
+### 💾 Optional Cloud Sync
+Store history in your own Supabase database
+
+</td>
+<td width="50%">
 
-- **Local-First**: Your code never leaves your machine
-- **AI-Powered**: Intelligent analysis using OpenAI, Anthropic, or local LLMs
-- **Privacy-First**: API keys stored securely in OS keychain
-- **Database Integration**: Optional Supabase for history tracking
-- **Smart Scanning**: Respects .gitignore and custom ignore patterns
-- **Safe Changes**: Backup files before applying suggestions
+### 🔐 Secure Storage
+API keys stored in OS keychain or encrypted
+
+### 📂 Smart Scanning
+Respects .gitignore and custom patterns
+
+### 🎯 Multi-Model Support
+Use any Gemini model (pro, flash, flash-lite)
+
+</td>
+</tr>
+</table>
+
+## ⚡ AI Provider Support
+
+<div align="center">
+
+| Provider | Status | Models | ETA |
+|:--------:|:------:|:------:|:---:|
+| **🟢 Google Gemini** | ✅ **Available Now** | gemini-2.5-pro, gemini-2.5-flash | - |
+| **🔵 OpenAI** | 🔜 Coming Soon | GPT-4, GPT-4 Turbo | v0.6.0 |
+| **🟣 Anthropic** | 🔜 Coming Soon | Claude 3.5 Sonnet | v0.7.0 |
+| **🟠 Ollama** | 🔜 Coming Soon | Any local model | v0.8.0 |
+
+</div>
 
 ## 🛠 Installation
 
@@ -41,7 +90,7 @@ npm install -g ownkey
 ### From Source
 
 ```bash
-git clone https://github.com/yourusername/ownkey-cli.git
+git clone https://github.com/ArpitK24/OwnKey.git
 cd ownkey-cli
 npm install
 npm run build
@@ -50,27 +99,30 @@ npm link
 
 ## ⚙️ Configuration
 
-Run the interactive configuration wizard:
+### Interactive Setup
 
 ```bash
 ownkey config
 ```
 
-Or use flags:
+### Using Flags
 
 ```bash
-# Configure Supabase
-ownkey config --supabase-url postgresql://... --supabase-key your-key
+# Configure Gemini
+ownkey config --provider gemini --api-key YOUR_GEMINI_API_KEY
+
+# Optional: Configure Supabase for history tracking
+ownkey config --supabase-url https://xxx.supabase.co --supabase-key YOUR_KEY
 
-# Configure AI provider
-ownkey config --provider openai --api-key sk-...
+# Use different Gemini model
+ownkey config --provider gemini --model gemini-2.5-flash
 ```
 
-### Supported AI Providers
+### Getting a Gemini API Key
 
-- **OpenAI** (GPT-4)
-- **Anthropic** (Claude)
-- **Ollama** (Local LLMs)
+1. Visit [Google AI Studio](https://aistudio.google.com/app/apikey)
+2. Create a new API key
+3. Run `ownkey config --provider gemini --api-key YOUR_KEY`
 
 ## 📖 Commands
 
@@ -81,6 +133,7 @@ Configure Supabase database and AI providers.
 ```bash
 ownkey config                    # Interactive wizard
 ownkey config --reset            # Reset to defaults
+ownkey config --provider gemini --api-key YOUR_KEY
 ```
 
 ### `ownkey scan [path]`
@@ -91,16 +144,32 @@ Scan a project directory.
 ownkey scan .                    # Scan current directory
 ownkey scan ./src                # Scan specific folder
 ownkey scan --max-file-size 1mb  # Custom file size limit
+ownkey scan --ignore "*.test.js" # Additional ignore patterns
 ```
 
-### `ownkey suggest [path]` *(Coming in v0.5.0)*
+### `ownkey suggest [path]`
 
-Generate AI-powered suggestions.
+Generate AI-powered code suggestions.
 
 ```bash
-ownkey suggest .
-ownkey suggest --type security
-ownkey suggest --severity high
+ownkey suggest .                 # Analyze current directory
+ownkey suggest ./src             # Analyze specific folder
+ownkey suggest --max-files 20    # Limit files analyzed
+ownkey suggest --local-only      # Skip database storage
+```
+
+**Output:**
+```
+🔍 Analyzing code...
+
+[1] bug (high) in src/auth.ts
+    Potential null pointer dereference
+
+[2] security (critical) in db/query.ts
+    SQL injection vulnerability detected
+
+✨ Analysis Complete!
+  Total suggestions: 12
 ```
 
 ### `ownkey apply <id>` *(Coming in v0.7.0)*
@@ -149,16 +218,16 @@ npm test
 - [x] v0.2.0 - CLI skeleton & configuration
 - [x] v0.3.0 - Config system with keychain
 - [x] v0.4.0 - File scanner
-- [ ] v0.5.0 - AI provider integration
-- [ ] v0.6.0 - Suggestion engine
+- [x] **v0.5.0 - Gemini AI integration** ← Current
+- [ ] v0.6.0 - OpenAI & Anthropic providers
 - [ ] v0.7.0 - Apply/diff system
-- [ ] v0.8.0 - Additional commands
+- [ ] v0.8.0 - Ollama (local LLM) support
 - [ ] v0.9.0 - Testing & polish
 - [ ] v1.0.0 - Stable release
 
 ## 🤝 Contributing
 
-Contributions are welcome! Please read our [Contributing Guide](CONTRIBUTING.md) first.
+Contributions are welcome! Please feel free to submit a Pull Request.
 
 ## 📄 License
 
@@ -166,10 +235,11 @@ MIT © 2026 OwnKey
 
 ## 🔗 Links
 
-- [Documentation](https://github.com/yourusername/ownkey-cli/wiki)
-- [Issue Tracker](https://github.com/yourusername/ownkey-cli/issues)
+- [GitHub Repository](https://github.com/ArpitK24/OwnKey)
+- [Issue Tracker](https://github.com/ArpitK24/OwnKey/issues)
 - [npm Package](https://www.npmjs.com/package/ownkey)
+- [Get Gemini API Key](https://aistudio.google.com/app/apikey)
 
 ---
 
-**Note**: OwnKey is currently in active development. Features marked as "coming soon" are planned for upcoming releases.
+**Built with ❤️ for developers who value privacy and AI-powered productivity**

package/dist/ai/factory.d.ts

@@ -0,0 +1,23 @@
+import { AIProvider } from './provider.js';
+/**
+ * Factory for creating AI provider instances
+ */
+export declare class AIProviderFactory {
+    /**
+     * Create an AI provider instance based on configuration
+     */
+    static createProvider(): Promise<AIProvider>;
+    /**
+     * Test provider connection
+     */
+    static testProvider(provider: AIProvider): Promise<boolean>;
+    /**
+     * Get recommended model for a provider
+     */
+    static getRecommendedModel(provider: string): string;
+    /**
+     * Get available models for a provider
+     */
+    static getAvailableModels(provider: string): string[];
+}
+//# sourceMappingURL=factory.d.ts.map

package/dist/ai/factory.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"factory.d.ts","sourceRoot":"","sources":["../../src/ai/factory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAM3C;;GAEG;AACH,qBAAa,iBAAiB;IAC1B;;OAEG;WACU,cAAc,IAAI,OAAO,CAAC,UAAU,CAAC;IAkClD;;OAEG;WACU,YAAY,CAAC,QAAQ,EAAE,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC;IAgBjE;;OAEG;IACH,MAAM,CAAC,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM;IAepD;;OAEG;IACH,MAAM,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE;CAkBxD"}

package/dist/ai/factory.js

@@ -0,0 +1,96 @@
+import { GeminiProvider } from './gemini.js';
+import { ConfigError } from '../types/index.js';
+import { configManager } from '../config/manager.js';
+import { logger } from '../utils/logger.js';
+/**
+ * Factory for creating AI provider instances
+ */
+export class AIProviderFactory {
+    /**
+     * Create an AI provider instance based on configuration
+     */
+    static async createProvider() {
+        const config = await configManager.load();
+        if (!config.activeProvider) {
+            throw new ConfigError('No active AI provider configured. Run: ownkey config');
+        }
+        const provider = config.activeProvider;
+        const apiKey = await configManager.getApiKey(provider);
+        logger.verbose(`Creating ${provider} provider...`);
+        switch (provider) {
+            case 'gemini':
+                if (!apiKey) {
+                    throw new ConfigError('Gemini API key not found. Run: ownkey config');
+                }
+                const geminiModel = config.providers.gemini?.model || GeminiProvider.getRecommendedModel();
+                return new GeminiProvider(apiKey, geminiModel);
+            case 'openai':
+                throw new ConfigError('OpenAI provider not yet implemented (coming soon)');
+            case 'anthropic':
+                throw new ConfigError('Anthropic provider not yet implemented (coming soon)');
+            case 'ollama':
+                throw new ConfigError('Ollama provider not yet implemented (coming soon)');
+            default:
+                throw new ConfigError(`Unknown provider: ${provider}`);
+        }
+    }
+    /**
+     * Test provider connection
+     */
+    static async testProvider(provider) {
+        try {
+            logger.verbose('Testing provider connection...');
+            const result = await provider.testConnection();
+            if (result) {
+                logger.success('Provider connection successful');
+            }
+            else {
+                logger.error('Provider connection failed');
+            }
+            return result;
+        }
+        catch (error) {
+            logger.error('Provider test error', error);
+            return false;
+        }
+    }
+    /**
+     * Get recommended model for a provider
+     */
+    static getRecommendedModel(provider) {
+        switch (provider) {
+            case 'gemini':
+                return GeminiProvider.getRecommendedModel();
+            case 'openai':
+                return 'gpt-4-turbo';
+            case 'anthropic':
+                return 'claude-3-5-sonnet-20241022';
+            case 'ollama':
+                return 'deepseek-coder:latest';
+            default:
+                return '';
+        }
+    }
+    /**
+     * Get available models for a provider
+     */
+    static getAvailableModels(provider) {
+        switch (provider) {
+            case 'gemini':
+                return GeminiProvider.getAvailableModels();
+            case 'openai':
+                return ['gpt-4-turbo', 'gpt-4', 'gpt-3.5-turbo'];
+            case 'anthropic':
+                return [
+                    'claude-3-5-sonnet-20241022',
+                    'claude-3-opus-20240229',
+                    'claude-3-haiku-20240307',
+                ];
+            case 'ollama':
+                return ['deepseek-coder:latest', 'codellama:latest', 'llama2:latest'];
+            default:
+                return [];
+        }
+    }
+}
+//# sourceMappingURL=factory.js.map

package/dist/ai/factory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"factory.js","sourceRoot":"","sources":["../../src/ai/factory.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C;;GAEG;AACH,MAAM,OAAO,iBAAiB;IAC1B;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,cAAc;QACvB,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,IAAI,EAAE,CAAC;QAE1C,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YACzB,MAAM,IAAI,WAAW,CAAC,sDAAsD,CAAC,CAAC;QAClF,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,CAAC,cAAc,CAAC;QACvC,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAEvD,MAAM,CAAC,OAAO,CAAC,YAAY,QAAQ,cAAc,CAAC,CAAC;QAEnD,QAAQ,QAAQ,EAAE,CAAC;YACf,KAAK,QAAQ;gBACT,IAAI,CAAC,MAAM,EAAE,CAAC;oBACV,MAAM,IAAI,WAAW,CAAC,8CAA8C,CAAC,CAAC;gBAC1E,CAAC;gBACD,MAAM,WAAW,GAAG,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,KAAK,IAAI,cAAc,CAAC,mBAAmB,EAAE,CAAC;gBAC3F,OAAO,IAAI,cAAc,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YAEnD,KAAK,QAAQ;gBACT,MAAM,IAAI,WAAW,CAAC,mDAAmD,CAAC,CAAC;YAE/E,KAAK,WAAW;gBACZ,MAAM,IAAI,WAAW,CAAC,sDAAsD,CAAC,CAAC;YAElF,KAAK,QAAQ;gBACT,MAAM,IAAI,WAAW,CAAC,mDAAmD,CAAC,CAAC;YAE/E;gBACI,MAAM,IAAI,WAAW,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;QAC/D,CAAC;IACL,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,QAAoB;QAC1C,IAAI,CAAC;YACD,MAAM,CAAC,OAAO,CAAC,gCAAgC,CAAC,CAAC;YACjD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,cAAc,EAAE,CAAC;YAC/C,IAAI,MAAM,EAAE,CAAC;gBACT,MAAM,CAAC,OAAO,CAAC,gCAAgC,CAAC,CAAC;YACrD,CAAC;iBAAM,CAAC;gBACJ,MAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAC/C,CAAC;YACD,OAAO,MAAM,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,qBAAqB,EAAE,KAAc,CAAC,CAAC;YACpD,OAAO,KAAK,CAAC;QACjB,CAAC;IACL,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,mBAAmB,CAAC,QAAgB;QACvC,QAAQ,QAAQ,EAAE,CAAC;YACf,KAAK,QAAQ;gBACT,OAAO,cAAc,CAAC,mBAAmB,EAAE,CAAC;YAChD,KAAK,QAAQ;gBACT,OAAO,aAAa,CAAC;YACzB,KAAK,WAAW;gBACZ,OAAO,4BAA4B,CAAC;YACxC,KAAK,QAAQ;gBACT,OAAO,uBAAuB,CAAC;YACnC;gBACI,OAAO,EAAE,CAAC;QAClB,CAAC;IACL,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,kBAAkB,CAAC,QAAgB;QACtC,QAAQ,QAAQ,EAAE,CAAC;YACf,KAAK,QAAQ;gBACT,OAAO,cAAc,CAAC,kBAAkB,EAAE,CAAC;YAC/C,KAAK,QAAQ;gBACT,OAAO,CAAC,aAAa,EAAE,OAAO,EAAE,eAAe,CAAC,CAAC;YACrD,KAAK,WAAW;gBACZ,OAAO;oBACH,4BAA4B;oBAC5B,wBAAwB;oBACxB,yBAAyB;iBAC5B,CAAC;YACN,KAAK,QAAQ;gBACT,OAAO,CAAC,uBAAuB,EAAE,kBAAkB,EAAE,eAAe,CAAC,CAAC;YAC1E;gBACI,OAAO,EAAE,CAAC;QAClB,CAAC;IACL,CAAC;CACJ"}

package/dist/ai/gemini.d.ts

@@ -0,0 +1,21 @@
+import { AIProvider } from './provider.js';
+import { CodeContext, Suggestion, AnalyzeOptions } from '../types/index.js';
+/**
+ * Google Gemini AI Provider
+ * Supports: gemini-1.5-pro, gemini-1.5-flash, gemini-pro
+ */
+export declare class GeminiProvider extends AIProvider {
+    private baseUrl;
+    getDefaultModel(): string;
+    testConnection(): Promise<boolean>;
+    analyze(context: CodeContext, options?: AnalyzeOptions): AsyncGenerator<Suggestion>;
+    /**
+     * Get available Gemini models
+     */
+    static getAvailableModels(): string[];
+    /**
+     * Get recommended model for code analysis
+     */
+    static getRecommendedModel(): string;
+}
+//# sourceMappingURL=gemini.d.ts.map

package/dist/ai/gemini.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../src/ai/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAC3C,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,cAAc,EAAmB,MAAM,mBAAmB,CAAC;AAyB7F;;;GAGG;AACH,qBAAa,cAAe,SAAQ,UAAU;IAC1C,OAAO,CAAC,OAAO,CAAkD;IAEjE,eAAe,IAAI,MAAM;IAInB,cAAc,IAAI,OAAO,CAAC,OAAO,CAAC;IAMjC,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,OAAO,CAAC,EAAE,cAAc,GAAG,cAAc,CAAC,UAAU,CAAC;IAsE1F;;OAEG;IACH,MAAM,CAAC,kBAAkB,IAAI,MAAM,EAAE;IAIrC;;OAEG;IACH,MAAM,CAAC,mBAAmB,IAAI,MAAM;CAGvC"}

package/dist/ai/gemini.js

@@ -0,0 +1,87 @@
+import { AIProvider } from './provider.js';
+import { AIProviderError } from '../types/index.js';
+import { logger } from '../utils/logger.js';
+/**
+ * Google Gemini AI Provider
+ * Supports: gemini-1.5-pro, gemini-1.5-flash, gemini-pro
+ */
+export class GeminiProvider extends AIProvider {
+    baseUrl = 'https://generativelanguage.googleapis.com/v1';
+    getDefaultModel() {
+        return 'gemini-2.5-pro';
+    }
+    async testConnection() {
+        // Temporarily bypass for testing - will fix endpoint later
+        logger.verbose('Skipping Gemini connection test (will test during analysis)');
+        return true;
+    }
+    async *analyze(context, options) {
+        try {
+            logger.verbose(`Analyzing with Gemini (${this.model})...`);
+            const systemPrompt = this.getSystemPrompt();
+            const userPrompt = this.buildUserPrompt(context);
+            const request = {
+                contents: [
+                    {
+                        role: 'user',
+                        parts: [
+                            { text: systemPrompt },
+                            { text: userPrompt },
+                        ],
+                    },
+                ],
+                generationConfig: {
+                    temperature: options?.temperature || 0.3,
+                    maxOutputTokens: options?.maxTokens || 8192,
+                },
+            };
+            const response = await fetch(`${this.baseUrl}/models/${this.model}:generateContent`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    'x-goog-api-key': this.apiKey || '',
+                },
+                body: JSON.stringify(request),
+            });
+            if (!response.ok) {
+                const errorText = await response.text();
+                logger.error(`Gemini API error: ${errorText}`);
+                throw new AIProviderError(`Gemini API error: ${response.status} ${errorText}`);
+            }
+            const data = (await response.json());
+            if (!data.candidates || data.candidates.length === 0) {
+                throw new AIProviderError('No response from Gemini');
+            }
+            const candidate = data.candidates[0];
+            const responseText = candidate.content.parts.map((p) => p.text).join('');
+            logger.verbose('Received response from Gemini');
+            logger.debug('Response:', responseText.substring(0, 200) + '...');
+            // Parse suggestions from response
+            const suggestions = this.parseSuggestions(responseText);
+            logger.success(`Generated ${suggestions.length} suggestions`);
+            // Yield suggestions one by one (simulating streaming)
+            for (const suggestion of suggestions) {
+                yield suggestion;
+            }
+        }
+        catch (error) {
+            if (error instanceof AIProviderError) {
+                throw error;
+            }
+            throw new AIProviderError('Failed to analyze with Gemini', error);
+        }
+    }
+    /**
+     * Get available Gemini models
+     */
+    static getAvailableModels() {
+        return ['gemini-2.5-pro', 'gemini-2.5-flash', 'gemini-2.5-flash-lite'];
+    }
+    /**
+     * Get recommended model for code analysis
+     */
+    static getRecommendedModel() {
+        return 'gemini-2.5-pro';
+    }
+}
+//# sourceMappingURL=gemini.js.map

package/dist/ai/gemini.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../src/ai/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAC3C,OAAO,EAA2C,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAC7F,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAwB5C;;;GAGG;AACH,MAAM,OAAO,cAAe,SAAQ,UAAU;IAClC,OAAO,GAAG,8CAA8C,CAAC;IAEjE,eAAe;QACX,OAAO,gBAAgB,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,cAAc;QAChB,2DAA2D;QAC3D,MAAM,CAAC,OAAO,CAAC,6DAA6D,CAAC,CAAC;QAC9E,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,CAAC,OAAO,CAAC,OAAoB,EAAE,OAAwB;QACzD,IAAI,CAAC;YACD,MAAM,CAAC,OAAO,CAAC,0BAA0B,IAAI,CAAC,KAAK,MAAM,CAAC,CAAC;YAE3D,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;YAC5C,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;YAEjD,MAAM,OAAO,GAAkB;gBAC3B,QAAQ,EAAE;oBACN;wBACI,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE;4BACH,EAAE,IAAI,EAAE,YAAY,EAAE;4BACtB,EAAE,IAAI,EAAE,UAAU,EAAE;yBACvB;qBACJ;iBACJ;gBACD,gBAAgB,EAAE;oBACd,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,GAAG;oBACxC,eAAe,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI;iBAC9C;aACJ,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,KAAK,CACxB,GAAG,IAAI,CAAC,OAAO,WAAW,IAAI,CAAC,KAAK,kBAAkB,EACtD;gBACI,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACL,cAAc,EAAE,kBAAkB;oBAClC,gBAAgB,EAAE,IAAI,CAAC,MAAM,IAAI,EAAE;iBACtC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;aAChC,CACJ,CAAC;YAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,MAAM,CAAC,KAAK,CAAC,qBAAqB,SAAS,EAAE,CAAC,CAAC;gBAC/C,MAAM,IAAI,eAAe,CAAC,qBAAqB,QAAQ,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC,CAAC;YACnF,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAmB,CAAC;YAEvD,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACnD,MAAM,IAAI,eAAe,CAAC,yBAAyB,CAAC,CAAC;YACzD,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,YAAY,GAAG,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAEzE,MAAM,CAAC,OAAO,CAAC,+BAA+B,CAAC,CAAC;YAChD,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC;YAElE,kCAAkC;YAClC,MAAM,WAAW,GAAG,IAAI,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;YAExD,MAAM,CAAC,OAAO,CAAC,aAAa,WAAW,CAAC,MAAM,cAAc,CAAC,CAAC;YAE9D,sDAAsD;YACtD,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;gBACnC,MAAM,UAAU,CAAC;YACrB,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,KAAK,YAAY,eAAe,EAAE,CAAC;gBACnC,MAAM,KAAK,CAAC;YAChB,CAAC;YACD,MAAM,IAAI,eAAe,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;QACtE,CAAC;IACL,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,kBAAkB;QACrB,OAAO,CAAC,gBAAgB,EAAE,kBAAkB,EAAE,uBAAuB,CAAC,CAAC;IAC3E,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,mBAAmB;QACtB,OAAO,gBAAgB,CAAC;IAC5B,CAAC;CACJ"}

package/dist/ai/provider.d.ts

@@ -0,0 +1,39 @@
+import { CodeContext, Suggestion, AnalyzeOptions, AIProviderInterface } from '../types/index.js';
+/**
+ * Base abstract class for AI providers
+ * All providers (OpenAI, Gemini, Anthropic, Ollama) must implement this interface
+ */
+export declare abstract class AIProvider implements AIProviderInterface {
+    protected apiKey?: string;
+    protected model: string;
+    protected endpoint?: string;
+    constructor(apiKey?: string, model?: string, endpoint?: string);
+    /**
+     * Get the default model for this provider
+     */
+    abstract getDefaultModel(): string;
+    /**
+     * Test if the provider connection is working
+     */
+    abstract testConnection(): Promise<boolean>;
+    /**
+     * Analyze code and generate suggestions
+     * @param context - Code context with files and project info
+     * @param options - Analysis options (streaming, max tokens, etc.)
+     * @returns AsyncGenerator yielding suggestions as they're generated
+     */
+    abstract analyze(context: CodeContext, options?: AnalyzeOptions): AsyncGenerator<Suggestion>;
+    /**
+     * Format the system prompt for code analysis
+     */
+    protected getSystemPrompt(): string;
+    /**
+     * Build the user prompt with code context
+     */
+    protected buildUserPrompt(context: CodeContext): string;
+    /**
+     * Parse AI response and extract suggestions
+     */
+    protected parseSuggestions(response: string): Suggestion[];
+}
+//# sourceMappingURL=provider.d.ts.map

package/dist/ai/provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.d.ts","sourceRoot":"","sources":["../../src/ai/provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,cAAc,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAEjG;;;GAGG;AACH,8BAAsB,UAAW,YAAW,mBAAmB;IAC3D,SAAS,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC;IACxB,SAAS,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;gBAEhB,MAAM,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM;IAM9D;;OAEG;IACH,QAAQ,CAAC,eAAe,IAAI,MAAM;IAElC;;OAEG;IACH,QAAQ,CAAC,cAAc,IAAI,OAAO,CAAC,OAAO,CAAC;IAE3C;;;;;OAKG;IACH,QAAQ,CAAC,OAAO,CACZ,OAAO,EAAE,WAAW,EACpB,OAAO,CAAC,EAAE,cAAc,GACzB,cAAc,CAAC,UAAU,CAAC;IAE7B;;OAEG;IACH,SAAS,CAAC,eAAe,IAAI,MAAM;IAgCnC;;OAEG;IACH,SAAS,CAAC,eAAe,CAAC,OAAO,EAAE,WAAW,GAAG,MAAM;IA2BvD;;OAEG;IACH,SAAS,CAAC,gBAAgB,CAAC,QAAQ,EAAE,MAAM,GAAG,UAAU,EAAE;CAwB7D"}

package/dist/ai/provider.js

@@ -0,0 +1,99 @@
+/**
+ * Base abstract class for AI providers
+ * All providers (OpenAI, Gemini, Anthropic, Ollama) must implement this interface
+ */
+export class AIProvider {
+    apiKey;
+    model;
+    endpoint;
+    constructor(apiKey, model, endpoint) {
+        this.apiKey = apiKey;
+        this.model = model || this.getDefaultModel();
+        this.endpoint = endpoint;
+    }
+    /**
+     * Format the system prompt for code analysis
+     */
+    getSystemPrompt() {
+        return `You are a senior software engineer performing code review and analysis.
+
+Your task is to analyze the provided codebase and identify:
+- Bugs and potential errors
+- Security vulnerabilities
+- Performance issues
+- Code quality improvements
+- Refactoring opportunities
+- Missing tests or documentation
+
+Guidelines:
+- Be conservative and precise
+- Only suggest changes you're confident about
+- Provide clear rationale for each suggestion
+- Use unified diff format for code changes
+- Categorize suggestions by type and severity
+
+Output Format:
+Return a JSON array of suggestions. Each suggestion must have:
+{
+  "type": "bug" | "security" | "performance" | "refactor" | "style" | "architecture" | "test" | "documentation",
+  "severity": "critical" | "high" | "medium" | "low" | "info",
+  "content": "Brief description of the issue",
+  "filePath": "path/to/file.ts",
+  "rationale": "Detailed explanation of why this matters",
+  "diff": "Unified diff format showing the change"
+}
+
+Be thorough but practical. Focus on high-impact suggestions.`;
+    }
+    /**
+     * Build the user prompt with code context
+     */
+    buildUserPrompt(context) {
+        let prompt = `# Project Analysis Request\n\n`;
+        prompt += `**Project:** ${context.projectInfo.name}\n`;
+        prompt += `**Path:** ${context.projectInfo.rootPath}\n`;
+        prompt += `**Files:** ${context.projectInfo.fileCount}\n\n`;
+        if (context.projectInfo.language) {
+            prompt += `**Primary Language:** ${context.projectInfo.language}\n`;
+        }
+        if (context.projectInfo.framework) {
+            prompt += `**Framework:** ${context.projectInfo.framework}\n`;
+        }
+        prompt += `\n## Code Files\n\n`;
+        for (const file of context.files) {
+            prompt += `### ${file.relativePath}\n\n`;
+            prompt += `\`\`\`${file.extension.slice(1)}\n`;
+            prompt += file.content;
+            prompt += `\n\`\`\`\n\n`;
+        }
+        prompt += `\nAnalyze the above code and provide suggestions in JSON format.`;
+        return prompt;
+    }
+    /**
+     * Parse AI response and extract suggestions
+     */
+    parseSuggestions(response) {
+        try {
+            // Try to extract JSON from response
+            const jsonMatch = response.match(/\[[\s\S]*\]/);
+            if (!jsonMatch) {
+                throw new Error('No JSON array found in response');
+            }
+            const suggestions = JSON.parse(jsonMatch[0]);
+            // Validate and normalize suggestions
+            return suggestions.map((s) => ({
+                type: s.type || 'refactor',
+                severity: s.severity || 'medium',
+                content: s.content || 'No description provided',
+                filePath: s.filePath || '',
+                rationale: s.rationale || '',
+                diff: s.diff || '',
+                tags: s.tags,
+            }));
+        }
+        catch (error) {
+            throw new Error(`Failed to parse suggestions: ${error}`);
+        }
+    }
+}
+//# sourceMappingURL=provider.js.map

package/dist/ai/provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../src/ai/provider.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,MAAM,OAAgB,UAAU;IAClB,MAAM,CAAU;IAChB,KAAK,CAAS;IACd,QAAQ,CAAU;IAE5B,YAAY,MAAe,EAAE,KAAc,EAAE,QAAiB;QAC1D,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;QAC7C,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC7B,CAAC;IAuBD;;OAEG;IACO,eAAe;QACrB,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;6DA4B8C,CAAC;IAC1D,CAAC;IAED;;OAEG;IACO,eAAe,CAAC,OAAoB;QAC1C,IAAI,MAAM,GAAG,gCAAgC,CAAC;QAC9C,MAAM,IAAI,gBAAgB,OAAO,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC;QACvD,MAAM,IAAI,aAAa,OAAO,CAAC,WAAW,CAAC,QAAQ,IAAI,CAAC;QACxD,MAAM,IAAI,cAAc,OAAO,CAAC,WAAW,CAAC,SAAS,MAAM,CAAC;QAE5D,IAAI,OAAO,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;YAC/B,MAAM,IAAI,yBAAyB,OAAO,CAAC,WAAW,CAAC,QAAQ,IAAI,CAAC;QACxE,CAAC;QACD,IAAI,OAAO,CAAC,WAAW,CAAC,SAAS,EAAE,CAAC;YAChC,MAAM,IAAI,kBAAkB,OAAO,CAAC,WAAW,CAAC,SAAS,IAAI,CAAC;QAClE,CAAC;QAED,MAAM,IAAI,qBAAqB,CAAC;QAEhC,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAC/B,MAAM,IAAI,OAAO,IAAI,CAAC,YAAY,MAAM,CAAC;YACzC,MAAM,IAAI,SAAS,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;YAC/C,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC;YACvB,MAAM,IAAI,cAAc,CAAC;QAC7B,CAAC;QAED,MAAM,IAAI,kEAAkE,CAAC;QAE7E,OAAO,MAAM,CAAC;IAClB,CAAC;IAED;;OAEG;IACO,gBAAgB,CAAC,QAAgB;QACvC,IAAI,CAAC;YACD,oCAAoC;YACpC,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAChD,IAAI,CAAC,SAAS,EAAE,CAAC;gBACb,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;YACvD,CAAC;YAED,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;YAE7C,qCAAqC;YACrC,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;gBAChC,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,UAAU;gBAC1B,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,QAAQ;gBAChC,OAAO,EAAE,CAAC,CAAC,OAAO,IAAI,yBAAyB;gBAC/C,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,EAAE;gBAC1B,SAAS,EAAE,CAAC,CAAC,SAAS,IAAI,EAAE;gBAC5B,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,EAAE;gBAClB,IAAI,EAAE,CAAC,CAAC,IAAI;aACf,CAAC,CAAC,CAAC;QACR,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,gCAAgC,KAAK,EAAE,CAAC,CAAC;QAC7D,CAAC;IACL,CAAC;CACJ"}

package/dist/commands/config.js

@@ -108,17 +108,19 @@ async function runInteractiveWizard() {
             choices: [
                 { name: 'OpenAI (GPT-4)', value: 'openai' },
                 { name: 'Anthropic (Claude)', value: 'anthropic' },
+                { name: 'Google Gemini', value: 'gemini' },
                 { name: 'Ollama (Local)', value: 'ollama' },
             ],
         },
     ]);
     // Get API key (not needed for Ollama)
     if (provider !== 'ollama') {
+        const providerName = provider === 'openai' ? 'OpenAI' : provider === 'anthropic' ? 'Anthropic' : 'Google Gemini';
         const { apiKey } = await inquirer.prompt([
             {
                 type: 'password',
                 name: 'apiKey',
-                message: `${provider === 'openai' ? 'OpenAI' : 'Anthropic'} API Key:`,
+                message: `${providerName} API Key:`,
                 validate: (input) => (input ? true : 'API key is required'),
             },
         ]);