@yamzzdev/aiyamz 1.0.1 → 1.0.3

package/bin/aiyamz.js

@@ -10,31 +10,39 @@ require('dotenv').config();
 
 const program = new Command();
 
-// Show banner
 console.log(chalk.cyan(figlet.textSync('AIYamz', { font: 'Small' })));
-console.log(chalk.gray('by yamzzdev - AI Chat Package (FREE)\n'));
+console.log(chalk.gray('by yamzzdev - AI Chat Package\n'));
 
 program
-  .version('1.0.0')
-  .description('AI Chat CLI by yamzzdev - Powered by Google Gemini (FREE)');
+  .version('1.0.2')
+  .description('AI Chat CLI by yamzzdev - Powered by Groq (FREE)');
 
 program
   .command('chat')
   .description('Start interactive chat with AI')
-  .option('-k, --key <key>', 'Gemini API key')
+  .option('-k, --key <key>', 'Groq API key')
+  .option('-m, --model <model>', 'Model to use (llama3-8b-8192, llama3-70b-8192, mixtral-8x7b-32768, gemma2-9b-it)')
   .action(async (options) => {
-    const apiKey = options.key || process.env.GEMINI_API_KEY;
+    const apiKey = options.key || process.env.GROQ_API_KEY;
     
     if (!apiKey) {
-      console.log(chalk.red('❌ No API key found!'));
-      console.log(chalk.yellow('Get your free API key: https://aistudio.google.com/apikey'));
-      console.log(chalk.gray('Then run: export GEMINI_API_KEY=your_key_here\n'));
+      console.log(chalk.red('Error: No API key found'));
+      console.log(chalk.yellow('Get your free API key: https://console.groq.com/keys'));
+      console.log(chalk.gray('Then run: export GROQ_API_KEY=your_key_here\n'));
+      console.log(chalk.gray('Or use: aiyamz chat --key your_key_here\n'));
       return;
     }
     
     const ai = new AIYamz(apiKey);
     
-    console.log(chalk.green('✅ AI Ready! Type your message (type "exit" to quit)\n'));
+    if (options.model) {
+      ai.provider.setModel(options.model);
+      console.log(chalk.gray(`Using model: ${options.model}\n`));
+    } else {
+      console.log(chalk.gray(`Using model: ${ai.provider.model}\n`));
+    }
+    
+    console.log(chalk.green('AI Ready! Type your message (type "exit" to quit, "clear" to clear history)\n'));
     
     while (true) {
       const { message } = await inquirer.prompt([
@@ -47,13 +55,28 @@ program
       ]);
       
       if (message.toLowerCase() === 'exit') {
-        console.log(chalk.yellow('\n👋 Bye!'));
+        console.log(chalk.yellow('\nGoodbye!\n'));
         break;
       }
       
       if (message.toLowerCase() === 'clear') {
         ai.clearHistory();
-        console.log(chalk.green('✅ History cleared!\n'));
+        console.log(chalk.green('History cleared\n'));
+        continue;
+      }
+      
+      if (message.toLowerCase() === 'history') {
+        const history = ai.getHistory();
+        if (history.length === 0) {
+          console.log(chalk.gray('No history yet\n'));
+        } else {
+          console.log(chalk.yellow('\n--- Chat History ---'));
+          for (const entry of history.slice(-10)) {
+            const role = entry.role === 'user' ? chalk.cyan('You:') : chalk.magenta('AI:');
+            console.log(`${role} ${entry.content.substring(0, 100)}...`);
+          }
+          console.log(chalk.yellow('--------------------\n'));
+        }
         continue;
       }
       
@@ -61,28 +84,36 @@ program
       const response = await ai.chat(message);
       spinner.stop();
       
-      console.log(chalk.magenta('\n🤖 AI:'), response, '\n');
+      console.log(chalk.magenta('\nAI:'), response, '\n');
     }
   });
 
 program
   .command('ask <question>')
   .description('Ask a single question')
-  .option('-k, --key <key>', 'Gemini API key')
+  .option('-k, --key <key>', 'Groq API key')
+  .option('-m, --model <model>', 'Model to use')
   .action(async (question, options) => {
-    const apiKey = options.key || process.env.GEMINI_API_KEY;
+    const apiKey = options.key || process.env.GROQ_API_KEY;
     
     if (!apiKey) {
-      console.log(chalk.red('❌ No API key found!'));
+      console.log(chalk.red('Error: No API key found'));
+      console.log(chalk.yellow('Get your free API key: https://console.groq.com/keys'));
       return;
     }
     
     const ai = new AIYamz(apiKey);
+    
+    if (options.model) {
+      ai.provider.setModel(options.model);
+    }
+    
     const spinner = ora(chalk.gray('Thinking...')).start();
     const response = await ai.chat(question);
     spinner.stop();
     
-    console.log(chalk.magenta('\n🤖 Answer:'), response);
+    console.log(chalk.magenta('\nAnswer:'), response);
+    console.log();
   });
 
 program
@@ -93,17 +124,58 @@ program
       {
         type: 'input',
         name: 'apiKey',
-        message: 'Enter your Gemini API key:',
+        message: 'Enter your Groq API key:',
         validate: (input) => input.length > 0 || 'API key is required'
       }
     ]);
     
     const envPath = process.env.HOME + '/.aiyamz.env';
-    const envContent = `GEMINI_API_KEY=${apiKey}`;
+    const envContent = `GROQ_API_KEY=${apiKey}`;
     
     require('fs').writeFileSync(envPath, envContent);
-    console.log(chalk.green(`✅ API key saved to ${envPath}`));
-    console.log(chalk.gray('Add this to your shell config: export GEMINI_API_KEY=your_key'));
+    console.log(chalk.green(`API key saved to ${envPath}`));
+    console.log(chalk.gray('Add this to your shell config: export GROQ_API_KEY=your_key'));
+  });
+
+program
+  .command('models')
+  .description('List available models')
+  .action(() => {
+    console.log(chalk.yellow('\nAvailable Models:'));
+    console.log(chalk.cyan('  - llama3-8b-8192 (fast, good for general chat)'));
+    console.log(chalk.cyan('  - llama3-70b-8192 (smarter, more accurate)'));
+    console.log(chalk.cyan('  - mixtral-8x7b-32768 (large context window)'));
+    console.log(chalk.cyan('  - gemma2-9b-it (lightweight, fast)\n'));
+  });
+
+program
+  .command('history')
+  .description('Show chat history')
+  .option('-k, --key <key>', 'Groq API key')
+  .action(async (options) => {
+    const apiKey = options.key || process.env.GROQ_API_KEY;
+    
+    if (!apiKey) {
+      console.log(chalk.red('Error: No API key found'));
+      return;
+    }
+    
+    const ai = new AIYamz(apiKey);
+    const history = ai.getHistory();
+    
+    if (history.length === 0) {
+      console.log(chalk.gray('No chat history found\n'));
+      return;
+    }
+    
+    console.log(chalk.yellow('\n=== Chat History ===\n'));
+    for (const entry of history) {
+      const role = entry.role === 'user' ? chalk.cyan('You:') : chalk.magenta('AI:');
+      const time = chalk.gray(`[${new Date(entry.timestamp).toLocaleString()}]`);
+      console.log(`${time}`);
+      console.log(`${role} ${entry.content}`);
+      console.log(chalk.gray('---\n'));
+    }
   });
 
 program.parse(process.argv);

package/lib/index.js

@@ -1,29 +1,34 @@
-const GeminiProvider = require('./providers/gemini');
+const GroqProvider = require('./providers/groq');
 const chalk = require('chalk');
 const fs = require('fs');
 const path = require('path');
 
 class AIYamz {
   constructor(apiKey) {
-    this.apiKey = apiKey || process.env.GEMINI_API_KEY;
+    this.apiKey = apiKey || process.env.GROQ_API_KEY;
     if (!this.apiKey) {
-      console.log(chalk.yellow('⚠️  No API key found!'));
-      console.log(chalk.cyan('Get your free API key at: https://aistudio.google.com/apikey'));
-      console.log(chalk.gray('Then set it: export GEMINI_API_KEY=your_key_here\n'));
+      console.log(chalk.yellow('Warning: No API key found'));
+      console.log(chalk.cyan('Get your free API key at: https://console.groq.com/keys'));
+      console.log(chalk.gray('Then set it: export GROQ_API_KEY=your_key_here\n'));
     }
-    this.provider = new GeminiProvider(this.apiKey);
+    this.provider = new GroqProvider(this.apiKey);
     this.history = [];
+    this.configPath = path.join(process.env.HOME || process.env.USERPROFILE, '.aiyamz.json');
+    this.loadHistory();
   }
 
   async chat(message, options = {}) {
     if (!this.apiKey) {
-      return '❌ No API key configured. Please set GEMINI_API_KEY environment variable.';
+      return 'Error: No API key configured. Please set GROQ_API_KEY environment variable.';
+    }
+    
+    if (!message || message.trim() === '') {
+      return 'Error: Message cannot be empty.';
     }
     
     try {
       const response = await this.provider.chat(message, options);
       
-      // Save to history
       this.history.push({
         role: 'user',
         content: message,
@@ -35,39 +40,113 @@ class AIYamz {
         timestamp: new Date().toISOString()
       });
       
+      this.saveHistory();
       return response;
+      
     } catch (error) {
-      return `❌ Error: ${error.message}`;
+      return `Error: ${error.message}`;
     }
   }
 
-  async stream(message, callback) {
+  async stream(message, callback, options = {}) {
     if (!this.apiKey) {
-      callback('❌ No API key configured.');
+      callback('Error: No API key configured.');
+      return;
+    }
+    
+    if (!message || message.trim() === '') {
+      callback('Error: Message cannot be empty.');
       return;
     }
     
+    let fullResponse = '';
+    
     try {
       await this.provider.streamChat(message, (chunk) => {
+        fullResponse += chunk;
         if (callback) callback(chunk);
+      }, options);
+      
+      this.history.push({
+        role: 'user',
+        content: message,
+        timestamp: new Date().toISOString()
       });
+      this.history.push({
+        role: 'ai',
+        content: fullResponse,
+        timestamp: new Date().toISOString()
+      });
+      
+      this.saveHistory();
+      
     } catch (error) {
-      callback(`❌ Error: ${error.message}`);
+      callback(`Error: ${error.message}`);
     }
   }
 
+  async ask(question, options = {}) {
+    return this.chat(question, options);
+  }
+
   getHistory() {
     return this.history;
   }
 
   clearHistory() {
     this.history = [];
-    return 'History cleared!';
+    this.saveHistory();
+    return 'History cleared successfully';
+  }
+
+  getHistoryByRole(role) {
+    return this.history.filter(entry => entry.role === role);
+  }
+
+  getLastMessage() {
+    return this.history.length > 0 ? this.history[this.history.length - 1] : null;
+  }
+
+  saveHistory() {
+    try {
+      fs.writeFileSync(this.configPath, JSON.stringify(this.history, null, 2));
+    } catch (error) {
+      console.log(chalk.red(`Failed to save history: ${error.message}`));
+    }
   }
 
-  async askQuestion(question) {
-    const response = await this.chat(question);
-    return response;
+  loadHistory() {
+    try {
+      if (fs.existsSync(this.configPath)) {
+        const data = fs.readFileSync(this.configPath, 'utf8');
+        this.history = JSON.parse(data);
+      }
+    } catch (error) {
+      console.log(chalk.red(`Failed to load history: ${error.message}`));
+      this.history = [];
+    }
+  }
+
+  async exportHistory(format = 'json') {
+    if (format === 'json') {
+      return JSON.stringify(this.history, null, 2);
+    } else if (format === 'txt') {
+      let text = '';
+      for (const entry of this.history) {
+        text += `[${entry.timestamp}] ${entry.role.toUpperCase()}: ${entry.content}\n\n`;
+      }
+      return text;
+    }
+    return 'Unsupported format. Use json or txt';
+  }
+
+  async setApiKey(apiKey) {
+    if (!apiKey || apiKey.trim() === '') {
+      return 'Error: API key cannot be empty';
+    }
+    this.apiKey = apiKey;
+    this.provider = new GroqProvider(this.apiKey);
+    return 'API key updated successfully';
   }
 }
 

package/lib/providers/groq.js

@@ -0,0 +1,120 @@
+const Groq = require('groq-sdk');
+
+class GroqProvider {
+  constructor(apiKey) {
+    if (!apiKey) {
+      throw new Error('API Key is required. Get your API key at: https://console.groq.com/keys');
+    }
+    this.client = new Groq({ apiKey });
+    this.model = "llama3-8b-8192";
+    this.availableModels = [
+      "llama3-8b-8192",
+      "llama3-70b-8192",
+      "mixtral-8x7b-32768",
+      "gemma2-9b-it"
+    ];
+  }
+
+  setModel(modelName) {
+    if (this.availableModels.includes(modelName)) {
+      this.model = modelName;
+      return true;
+    }
+    return false;
+  }
+
+  getModels() {
+    return this.availableModels;
+  }
+
+  async chat(message, options = {}) {
+    try {
+      const messages = [];
+      
+      if (options.systemPrompt) {
+        messages.push({
+          role: "system",
+          content: options.systemPrompt
+        });
+      }
+      
+      messages.push({
+        role: "user",
+        content: message
+      });
+
+      const completion = await this.client.chat.completions.create({
+        messages: messages,
+        model: options.model || this.model,
+        temperature: options.temperature || 0.7,
+        max_tokens: options.maxTokens || 2048,
+        top_p: options.topP || 1,
+        stop: options.stop || null
+      });
+
+      if (!completion.choices || completion.choices.length === 0) {
+        throw new Error('No response from API');
+      }
+
+      return completion.choices[0].message.content || "No response content";
+      
+    } catch (error) {
+      if (error.status === 401) {
+        throw new Error('Invalid API key. Please check your GROQ_API_KEY');
+      } else if (error.status === 429) {
+        throw new Error('Rate limit exceeded. Please wait a moment and try again');
+      } else if (error.status === 404) {
+        throw new Error('Model not found. Please check the model name');
+      } else {
+        throw new Error(`API Error: ${error.message}`);
+      }
+    }
+  }
+
+  async streamChat(message, onChunk, options = {}) {
+    try {
+      const stream = await this.client.chat.completions.create({
+        messages: [{ role: "user", content: message }],
+        model: options.model || this.model,
+        temperature: options.temperature || 0.7,
+        max_tokens: options.maxTokens || 2048,
+        top_p: options.topP || 1,
+        stream: true,
+      });
+
+      for await (const chunk of stream) {
+        const content = chunk.choices[0]?.delta?.content || "";
+        if (content && onChunk) {
+          onChunk(content);
+        }
+      }
+      
+    } catch (error) {
+      if (error.status === 401) {
+        throw new Error('Invalid API key. Please check your GROQ_API_KEY');
+      } else if (error.status === 429) {
+        throw new Error('Rate limit exceeded. Please wait a moment and try again');
+      } else {
+        throw new Error(`Stream error: ${error.message}`);
+      }
+    }
+  }
+
+  async chatWithHistory(messages, options = {}) {
+    try {
+      const completion = await this.client.chat.completions.create({
+        messages: messages,
+        model: options.model || this.model,
+        temperature: options.temperature || 0.7,
+        max_tokens: options.maxTokens || 2048,
+      });
+
+      return completion.choices[0].message.content || "No response content";
+      
+    } catch (error) {
+      throw new Error(`API Error: ${error.message}`);
+    }
+  }
+}
+
+module.exports = GroqProvider;

package/package.json

@@ -1,34 +1,29 @@
 {
   "name": "@yamzzdev/aiyamz",
-  "version": "1.0.1",
-  "description": "AI chat package by yamzzdev - powered by Google Gemini (FREE)",
+  "version": "1.0.3",
+  "description": "AI chat package by yamzzdev - powered by Groq (FREE)",
   "main": "lib/index.js",
   "bin": {
-    "aiyamz": "bin/aiyamz.js"
+    "aiyamz": "./bin/aiyamz.js"
   },
   "keywords": [
     "ai",
     "chat",
-    "gemini",
+    "groq",
+    "llama",
     "yamzzdev",
-    "gabut"
+    "gabut",
+    "cli"
   ],
   "author": "yamzzdev",
   "license": "MIT",
   "dependencies": {
-    "@google/generative-ai": "^0.21.0",
     "chalk": "^4.1.2",
     "commander": "^12.1.0",
-    "inquirer": "^8.2.6",
-    "ora": "^5.4.1",
+    "dotenv": "^16.4.5",
     "figlet": "^1.7.0",
-    "dotenv": "^16.4.5"
-  },
-  "directories": {
-    "lib": "lib"
-  },
-  "scripts": {
-    "test": "echo \"Error: no test specified\" && exit 1"
-  },
-  "type": "commonjs"
+    "groq-sdk": "^0.5.0",
+    "inquirer": "^8.2.6",
+    "ora": "^5.4.1"
+  }
 }

package/lib/providers/gemini.js

@@ -1,40 +0,0 @@
-const { GoogleGenerativeAI } = require('@google/generative-ai');
-const chalk = require('chalk');
-
-class GeminiProvider {
-  constructor(apiKey) {
-    if (!apiKey) {
-      throw new Error('API Key is required! Get your API key at: https://aistudio.google.com/apikey');
-    }
-    this.genAI = new GoogleGenerativeAI(apiKey);
-    this.model = this.genAI.getGenerativeModel({ model: "gemini-1.5-flash" });
-  }
-
-  async chat(message, options = {}) {
-    try {
-      const prompt = options.systemPrompt 
-        ? `${options.systemPrompt}\n\nUser: ${message}` 
-        : message;
-      
-      const result = await this.model.generateContent(prompt);
-      const response = await result.response;
-      return response.text();
-    } catch (error) {
-      throw new Error(`Gemini API Error: ${error.message}`);
-    }
-  }
-
-  async streamChat(message, onChunk) {
-    try {
-      const result = await this.model.generateContentStream(message);
-      for await (const chunk of result.stream) {
-        const chunkText = chunk.text();
-        if (onChunk) onChunk(chunkText);
-      }
-    } catch (error) {
-      throw new Error(`Stream error: ${error.message}`);
-    }
-  }
-}
-
-module.exports = GeminiProvider;