@yamzzdev/aiyamz 1.0.0

package/bin/aiyamz.js

@@ -0,0 +1,113 @@
+#!/usr/bin/env node
+
+const { Command } = require('commander');
+const chalk = require('chalk');
+const inquirer = require('inquirer');
+const ora = require('ora');
+const figlet = require('figlet');
+const AIYamz = require('../lib/index');
+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'));
+
+program
+  .version('1.0.0')
+  .description('AI Chat CLI by yamzzdev - Powered by Google Gemini (FREE)');
+
+program
+  .command('chat')
+  .description('Start interactive chat with AI')
+  .option('-k, --key <key>', 'Gemini API key')
+  .action(async (options) => {
+    const apiKey = options.key || process.env.GEMINI_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'));
+      return;
+    }
+    
+    const ai = new AIYamz(apiKey);
+    
+    console.log(chalk.green('✅ AI Ready! Type your message (type "exit" to quit)\n'));
+    
+    while (true) {
+      const { message } = await inquirer.prompt([
+        {
+          type: 'input',
+          name: 'message',
+          message: chalk.cyan('You:'),
+          prefix: ''
+        }
+      ]);
+      
+      if (message.toLowerCase() === 'exit') {
+        console.log(chalk.yellow('\n👋 Bye!'));
+        break;
+      }
+      
+      if (message.toLowerCase() === 'clear') {
+        ai.clearHistory();
+        console.log(chalk.green('✅ History cleared!\n'));
+        continue;
+      }
+      
+      const spinner = ora(chalk.gray('AI is thinking...')).start();
+      const response = await ai.chat(message);
+      spinner.stop();
+      
+      console.log(chalk.magenta('\n🤖 AI:'), response, '\n');
+    }
+  });
+
+program
+  .command('ask <question>')
+  .description('Ask a single question')
+  .option('-k, --key <key>', 'Gemini API key')
+  .action(async (question, options) => {
+    const apiKey = options.key || process.env.GEMINI_API_KEY;
+    
+    if (!apiKey) {
+      console.log(chalk.red('❌ No API key found!'));
+      return;
+    }
+    
+    const ai = new AIYamz(apiKey);
+    const spinner = ora(chalk.gray('Thinking...')).start();
+    const response = await ai.chat(question);
+    spinner.stop();
+    
+    console.log(chalk.magenta('\n🤖 Answer:'), response);
+  });
+
+program
+  .command('config')
+  .description('Setup API key')
+  .action(async () => {
+    const { apiKey } = await inquirer.prompt([
+      {
+        type: 'input',
+        name: 'apiKey',
+        message: 'Enter your Gemini API key:',
+        validate: (input) => input.length > 0 || 'API key is required'
+      }
+    ]);
+    
+    const envPath = process.env.HOME + '/.aiyamz.env';
+    const envContent = `GEMINI_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'));
+  });
+
+program.parse(process.argv);
+
+if (!process.argv.slice(2).length) {
+  program.outputHelp();
+}

package/lib/index.js

@@ -0,0 +1,74 @@
+const GeminiProvider = require('./providers/gemini');
+const chalk = require('chalk');
+const fs = require('fs');
+const path = require('path');
+
+class AIYamz {
+  constructor(apiKey) {
+    this.apiKey = apiKey || process.env.GEMINI_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'));
+    }
+    this.provider = new GeminiProvider(this.apiKey);
+    this.history = [];
+  }
+
+  async chat(message, options = {}) {
+    if (!this.apiKey) {
+      return '❌ No API key configured. Please set GEMINI_API_KEY environment variable.';
+    }
+    
+    try {
+      const response = await this.provider.chat(message, options);
+      
+      // Save to history
+      this.history.push({
+        role: 'user',
+        content: message,
+        timestamp: new Date().toISOString()
+      });
+      this.history.push({
+        role: 'ai',
+        content: response,
+        timestamp: new Date().toISOString()
+      });
+      
+      return response;
+    } catch (error) {
+      return `❌ Error: ${error.message}`;
+    }
+  }
+
+  async stream(message, callback) {
+    if (!this.apiKey) {
+      callback('❌ No API key configured.');
+      return;
+    }
+    
+    try {
+      await this.provider.streamChat(message, (chunk) => {
+        if (callback) callback(chunk);
+      });
+    } catch (error) {
+      callback(`❌ Error: ${error.message}`);
+    }
+  }
+
+  getHistory() {
+    return this.history;
+  }
+
+  clearHistory() {
+    this.history = [];
+    return 'History cleared!';
+  }
+
+  async askQuestion(question) {
+    const response = await this.chat(question);
+    return response;
+  }
+}
+
+module.exports = AIYamz;

package/lib/providers/gemini.js

@@ -0,0 +1,40 @@
+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;

package/package.json

@@ -0,0 +1,21 @@
+{
+  "name": "@yamzzdev/aiyamz",
+  "version": "1.0.0",
+  "description": "AI chat package by yamzzdev - powered by Google Gemini (FREE)",
+  "main": "lib/index.js",
+  "bin": {
+    "aiyamz": "./bin/aiyamz.js"
+  },
+  "keywords": ["ai", "chat", "gemini", "yamzzdev", "gabut"],
+  "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",
+    "figlet": "^1.7.0",
+    "dotenv": "^16.4.5"
+  }
+}