@rabts/cli 1.0.0

package/bin/cli.js

@@ -0,0 +1,70 @@
+#!/usr/bin/env node
+
+import { Command } from 'commander';
+import chalk from 'chalk';
+import fs from 'fs';
+import path from 'path';
+import { fileURLToPath } from 'url';
+
+import { ensureConfig, configureApiKey } from '../src/config.js';
+import { startInteractiveLoop } from '../src/agent.js';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+const packageJson = JSON.parse(fs.readFileSync(path.join(__dirname, '../package.json'), 'utf8'));
+
+// Cor primária da marca
+const rabtsColor = chalk.cyan; // Azul claro para padronizar
+
+const logo = rabtsColor(`
+         %%%%                             
+         %%%%%%%%%%                       
+        %%%%%%%%%%%%%%%%%@                
+       %%%%%%%%%%%%%%%%%%%%%%%%%          
+       %%%   %%%%%  %%%%%%%%%%%%%%%%%%    
+      @%%%%     %%%   %%%%%%%%%%%%%%%%%%  
+      %%%%%%      %    %%%%%%%%%%%%%%%%%  
+     @%%%%%%%      %%  %%%%%%%%%%%%%%%%%% 
+     %%%%%%%%%%     %   %%%%%%%%%%%%%%%%  
+    %%%%%%%%%%%%%    %  %%%%%%%%%%%%%%%%  
+    %%%%%%%%%%%%%%%%%         %%%%%%%%%%  
+   %%%%%%%%%%%%%%%%%%     %%%   %%%%%%%   
+   %%%%%%%%%%%%%%%%%             %%%%%    
+  %%%%%%%%%%%%%%%%%%            %%%%%%    
+  %%%%%%%%%%%%%%%            %%%%%%%%     
+ @%%%%%%%%%%%%%             %%%%%%%%%     
+  %%%%%%%%%%%                %%%%%%%      
+  %%%%%%%%%%       %%%       %%%%%%%      
+    %%%%%%%@         %%%    @%%%%%%       
+          %           %%    %%%%%%%       
+                      %%%  @%%%%%%        
+                       %%%  %%%%%@        
+                              %%%         
+`);
+
+const program = new Command();
+
+program
+  .name('rabts')
+  .description('Rabts Studio AI CLI')
+  .version(packageJson.version)
+  .action(async () => {
+    console.clear();
+    console.log(logo);
+    console.log(chalk.gray(`v${packageJson.version} - Seu Agente Inteligente\n`));
+    
+    const config = await ensureConfig();
+    await startInteractiveLoop(config);
+  });
+
+program
+  .command('config')
+  .description('Força a reconfiguração de provedor e chave de API')
+  .action(async () => {
+    console.clear();
+    console.log(logo);
+    const config = await configureApiKey();
+    await startInteractiveLoop(config);
+  });
+
+program.parse(process.argv);

package/package.json

@@ -0,0 +1,23 @@
+{
+  "name": "@rabts/cli",
+  "version": "1.0.0",
+  "description": "Rabts Studio CLI",
+  "main": "src/index.js",
+  "type": "module",
+  "bin": {
+    "rabts": "bin/cli.js"
+  },
+  "scripts": {
+    "start": "node ./bin/cli.js"
+  },
+  "dependencies": {
+    "@anthropic-ai/sdk": "^0.105.0",
+    "@google/genai": "^2.9.0",
+    "axios": "^1.7.2",
+    "chalk": "^5.3.0",
+    "commander": "^12.1.0",
+    "dotenv": "^16.4.5",
+    "inquirer": "^10.0.0",
+    "openai": "^4.52.0"
+  }
+}

package/src/agent.js

@@ -0,0 +1,183 @@
+import fs from 'fs';
+import path from 'path';
+import { exec } from 'child_process';
+import util from 'util';
+import chalk from 'chalk';
+import inquirer from 'inquirer';
+import { sendMessage } from './llm.js';
+
+const execPromise = util.promisify(exec);
+
+// Tema Rabts
+const themeColor = chalk.cyan; // Azul claro para padronizar
+const infoColor = chalk.gray;
+
+const tools = [
+  {
+    type: "function",
+    function: {
+      name: "read_file",
+      description: "Lê o conteúdo de um arquivo do sistema local.",
+      parameters: {
+        type: "object",
+        properties: { filepath: { type: "string" } },
+        required: ["filepath"]
+      }
+    }
+  },
+  {
+    type: "function",
+    function: {
+      name: "write_file",
+      description: "Escreve ou sobrescreve um arquivo no sistema local.",
+      parameters: {
+        type: "object",
+        properties: {
+          filepath: { type: "string" },
+          content: { type: "string" }
+        },
+        required: ["filepath", "content"]
+      }
+    }
+  },
+  {
+    type: "function",
+    function: {
+      name: "run_command",
+      description: "Executa um comando no terminal local.",
+      parameters: {
+        type: "object",
+        properties: { command: { type: "string" } },
+        required: ["command"]
+      }
+    }
+  }
+];
+
+async function executeToolCall(toolCall) {
+  const args = JSON.parse(toolCall.function.arguments);
+  const cwd = process.cwd();
+
+  if (toolCall.function.name === 'read_file') {
+    try {
+      const fullPath = path.resolve(cwd, args.filepath);
+      return fs.readFileSync(fullPath, 'utf8');
+    } catch (e) {
+      return `Erro ao ler arquivo: ${e.message}`;
+    }
+  }
+
+  if (toolCall.function.name === 'write_file') {
+    try {
+      const fullPath = path.resolve(cwd, args.filepath);
+      fs.mkdirSync(path.dirname(fullPath), { recursive: true });
+      fs.writeFileSync(fullPath, args.content, 'utf8');
+      console.log(themeColor(`  ↳ Arquivo salvo: `) + chalk.white(args.filepath));
+      return `Arquivo salvo em ${fullPath}`;
+    } catch (e) {
+      return `Erro ao salvar arquivo: ${e.message}`;
+    }
+  }
+
+  if (toolCall.function.name === 'run_command') {
+    const { confirm } = await inquirer.prompt([{
+      type: 'confirm',
+      name: 'confirm',
+      message: chalk.yellow(`Permitir comando: '${args.command}'?`),
+      default: false
+    }]);
+
+    if (!confirm) return `Usuário negou o comando: ${args.command}`;
+
+    try {
+      console.log(themeColor(`  ↳ Executando: `) + chalk.white(args.command));
+      const { stdout, stderr } = await execPromise(args.command, { cwd });
+      return stdout || stderr || "Sucesso sem output.";
+    } catch (e) {
+      return `Erro: ${e.message}\nStderr: ${e.stderr || ''}`;
+    }
+  }
+
+  return "Função desconhecida.";
+}
+
+export async function startInteractiveLoop(config) {
+  console.log(infoColor(`\nConectado via ${config.provider} (Modelo: ${config.model})`));
+  console.log(infoColor('Você pode conversar ou pedir para ler/criar arquivos. Digite "sair" para encerrar.\n'));
+
+  let messages = [
+    { 
+      role: 'system', 
+      content: 'Você é o Rabts AI, um assistente de terminal. Você pode conversar com o usuário ou usar ferramentas para manipular arquivos e comandos quando requisitado.' 
+    }
+  ];
+
+  const supportsTools = (config.provider === 'rabts' || config.provider === 'openai');
+
+  while (true) {
+    const { userPrompt } = await inquirer.prompt([
+      {
+        type: 'input',
+        name: 'userPrompt',
+        message: chalk.cyan('❯')
+      }
+    ]);
+
+    if (userPrompt.toLowerCase() === 'sair' || userPrompt.toLowerCase() === 'exit') {
+      console.log(infoColor('Até logo!'));
+      break;
+    }
+
+    if (!userPrompt.trim()) continue;
+
+    messages.push({ role: 'user', content: userPrompt });
+
+    let isProcessingTools = true;
+    let totalPromptTokens = 0;
+    let totalCompletionTokens = 0;
+    let startTime = Date.now();
+
+    while (isProcessingTools) {
+      process.stdout.write(infoColor('  Pensando...'));
+      
+      const responseMessage = await sendMessage(config, messages, supportsTools ? tools : null);
+      
+      process.stdout.clearLine(0);
+      process.stdout.cursorTo(0);
+
+      if (responseMessage.usage) {
+        totalPromptTokens += responseMessage.usage.prompt_tokens || 0;
+        totalCompletionTokens += responseMessage.usage.completion_tokens || 0;
+      }
+
+      messages.push({
+        role: responseMessage.role,
+        content: responseMessage.content,
+        tool_calls: responseMessage.tool_calls
+      });
+
+      if (responseMessage.tool_calls && responseMessage.tool_calls.length > 0) {
+        for (const toolCall of responseMessage.tool_calls) {
+          console.log(themeColor(`  ✦ Usando ferramenta: `) + chalk.white(toolCall.function.name));
+          const result = await executeToolCall(toolCall);
+          messages.push({
+            tool_call_id: toolCall.id,
+            role: "tool",
+            name: toolCall.function.name,
+            content: result,
+          });
+        }
+      } else {
+        const duration = ((Date.now() - startTime) / 1000).toFixed(1);
+        
+        if (responseMessage.content) {
+          console.log(themeColor('\nRabts:'));
+          console.log(chalk.white(responseMessage.content));
+        }
+
+        console.log(infoColor(`\n  ↳ Finalizado em ${duration}s • ${totalPromptTokens + totalCompletionTokens} tokens (${totalPromptTokens} in, ${totalCompletionTokens} out)\n`));
+        isProcessingTools = false;
+      }
+    }
+  }
+}

package/src/config.js

@@ -0,0 +1,97 @@
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+import inquirer from 'inquirer';
+import chalk from 'chalk';
+
+const CONFIG_FILE_PATH = path.join(os.homedir(), '.rabts-config.json');
+
+const RABTS_DEFAULT_KEY = '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';
+
+export function getConfig() {
+  if (fs.existsSync(CONFIG_FILE_PATH)) {
+    try {
+      const data = fs.readFileSync(CONFIG_FILE_PATH, 'utf8');
+      return JSON.parse(data);
+    } catch (e) {
+      return null;
+    }
+  }
+  return null;
+}
+
+export function saveConfig(config) {
+  fs.writeFileSync(CONFIG_FILE_PATH, JSON.stringify(config, null, 2));
+}
+
+export async function ensureConfig() {
+  let config = getConfig();
+  if (config) {
+    return config;
+  }
+  return await configureApiKey();
+}
+
+export async function configureApiKey() {
+  console.log(chalk.blue('\n=== Bem-vindo ao Rabts Studio CLI ===\n'));
+  
+  const { provider } = await inquirer.prompt([
+    {
+      type: 'list',
+      name: 'provider',
+      message: 'Qual provedor de IA você quer usar?',
+      choices: ['rabts', 'openai', 'anthropic', 'google']
+    }
+  ]);
+
+  let config = { provider };
+
+  if (provider === 'rabts') {
+    config.apiKey = RABTS_DEFAULT_KEY;
+    const { model } = await inquirer.prompt([
+      {
+        type: 'list',
+        name: 'model',
+        message: 'Escolha um modelo gratuito do Rabts:',
+        choices: [
+          'qwen/qwen3-32b',
+          'openai/gpt-oss-120b',
+          'moonshotai/kimi-k2-instruct',
+          'llama-3.1-8b-instant',
+          'openai/gpt-oss-20b',
+          'meta-llama/llama-4-scout-17b-16e-instruct'
+        ]
+      }
+    ]);
+    config.model = model;
+    console.log(chalk.green('\nConfigurado para usar o Rabts Studio (Modelos Gratuitos).\n'));
+  } else {
+    const { apiKey } = await inquirer.prompt([
+      {
+        type: 'input',
+        name: 'apiKey',
+        message: `Digite a sua chave de API da ${provider}:`
+      }
+    ]);
+    config.apiKey = apiKey.trim();
+
+    let defaultModel = '';
+    if (provider === 'openai') defaultModel = 'gpt-4o';
+    if (provider === 'anthropic') defaultModel = 'claude-3-5-sonnet-20240620';
+    if (provider === 'google') defaultModel = 'gemini-1.5-pro';
+
+    const { model } = await inquirer.prompt([
+      {
+        type: 'input',
+        name: 'model',
+        message: `Digite o modelo desejado (Padrão: ${defaultModel}):`,
+        default: defaultModel
+      }
+    ]);
+    config.model = model.trim();
+    console.log(chalk.green('\nConfiguração salva com sucesso!\n'));
+  }
+
+  saveConfig(config);
+  return config;
+}

package/src/llm.js

@@ -0,0 +1,120 @@
+import OpenAI from 'openai';
+import { GoogleGenAI } from '@google/genai';
+import Anthropic from '@anthropic-ai/sdk';
+import chalk from 'chalk';
+
+const clients = {};
+
+export function getClient(config) {
+  if (clients[config.provider]) return clients[config.provider];
+
+  if (config.provider === 'rabts') {
+    clients[config.provider] = new OpenAI({
+      baseURL: 'https://api.groq.com/openai/v1',
+      apiKey: config.apiKey,
+    });
+  } else if (config.provider === 'openai') {
+    clients[config.provider] = new OpenAI({
+      apiKey: config.apiKey,
+    });
+  } else if (config.provider === 'anthropic') {
+    clients[config.provider] = new Anthropic({
+      apiKey: config.apiKey,
+    });
+  } else if (config.provider === 'google') {
+    clients[config.provider] = new GoogleGenAI({
+      apiKey: config.apiKey,
+    });
+  }
+  return clients[config.provider];
+}
+
+export async function sendMessage(config, messages, tools = null) {
+  const client = getClient(config);
+  let usage = null;
+  
+  try {
+    if (config.provider === 'rabts' || config.provider === 'openai') {
+      const params = {
+        model: config.model,
+        messages: messages,
+      };
+
+      if (tools) {
+        params.tools = tools;
+        params.tool_choice = "auto";
+      }
+
+      const response = await client.chat.completions.create(params);
+      const msg = response.choices[0].message;
+      usage = response.usage;
+      
+      return {
+        role: msg.role,
+        content: msg.content || '',
+        tool_calls: msg.tool_calls,
+        usage
+      };
+    } 
+    
+    else if (config.provider === 'anthropic') {
+      const systemMessage = messages.find(m => m.role === 'system')?.content || '';
+      const anthropicMessages = messages.filter(m => m.role !== 'system').map(m => ({
+        role: m.role === 'user' ? 'user' : 'assistant',
+        content: m.content
+      }));
+
+      const response = await client.messages.create({
+        model: config.model,
+        max_tokens: 4096,
+        system: systemMessage,
+        messages: anthropicMessages
+      });
+
+      usage = {
+        prompt_tokens: response.usage.input_tokens,
+        completion_tokens: response.usage.output_tokens,
+      };
+
+      return {
+        role: 'assistant',
+        content: response.content[0].text,
+        usage
+      };
+    }
+    
+    else if (config.provider === 'google') {
+      const systemMessage = messages.find(m => m.role === 'system')?.content || '';
+      const googleMessages = messages.filter(m => m.role !== 'system').map(m => ({
+        role: m.role === 'user' ? 'user' : 'model',
+        parts: [{ text: m.content }]
+      }));
+
+      const response = await client.models.generateContent({
+        model: config.model,
+        contents: googleMessages,
+        config: {
+          systemInstruction: systemMessage ? { parts: [{text: systemMessage}]} : undefined
+        }
+      });
+
+      // GoogleGenAI might have usageMetadata
+      if (response.usageMetadata) {
+        usage = {
+          prompt_tokens: response.usageMetadata.promptTokenCount,
+          completion_tokens: response.usageMetadata.candidatesTokenCount,
+        };
+      }
+
+      return {
+        role: 'assistant',
+        content: response.text,
+        usage
+      };
+    }
+
+  } catch (error) {
+    console.error(chalk.red(`\nErro na comunicação com a API (${config.provider}):`), error.message);
+    process.exit(1);
+  }
+}