@riflo/ryte 1.1.2 → 1.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@riflo/ryte",
-  "version": "1.1.2",
+  "version": "1.2.0",
   "description": "AI Git Workflow Assistant - Generate semantic commits and PRs",
   "main": "src/index.js",
   "bin": {

package/src/ai.js

@@ -1,28 +1,19 @@
-import dotenv from "dotenv";
-dotenv.config();
+import { getConfig } from "./config.js";
+import { getProviderConfig } from "./provider.js";
 
-export async function generateAIResponse(messages) {
-    const groqKey = process.env.GROQ_API_KEY;
-    const openAiKey = process.env.OPENAI_API_KEY;
+export async function generateAIResponse(messages, overrideConfig = null) {
+    const config = overrideConfig || getConfig();
 
-    if (!groqKey && !openAiKey) {
-        console.error("Error: Please set either GROQ_API_KEY (for free AI) or OPENAI_API_KEY in your environment variables.");
-        process.exit(1);
+    if (!config || !config.apiKey) {
+        throw new Error("API Key not found. Please run 'ryte config' or follow the setup flow.");
     }
 
-    const isGroq = !!groqKey;
-    const apiKey = isGroq ? groqKey : openAiKey;
+    const providerName = config.provider || "openai";
+    const pConfig = getProviderConfig(providerName, config.baseUrl);
 
-    // Groq API is 100% compatible with OpenAI's format! Just changing URL & Model.
-    const apiUrl = isGroq
-        ? "https://api.groq.com/openai/v1/chat/completions"
-        : "https://api.openai.com/v1/chat/completions";
-
-    // llama-3.1-8b-instant: 14,400 TPM (6x higher than llama-3.3-70b-versatile)
-    // Still very capable for commit messages and PR summaries
-    const model = isGroq
-        ? "llama-3.1-8b-instant"
-        : "gpt-4o-mini";
+    const apiUrl = pConfig.url;
+    const apiKey = config.apiKey;
+    const model = config.model || pConfig.model;
 
     const MAX_RETRIES = 3;
 
@@ -42,7 +33,6 @@ export async function generateAIResponse(messages) {
             });
 
             if (response.status === 429) {
-                // Rate limited — parse retry-after header or use exponential backoff
                 const retryAfter = parseInt(response.headers.get("retry-after") || "15", 10);
                 const waitSeconds = retryAfter + 1;
 
@@ -50,7 +40,7 @@ export async function generateAIResponse(messages) {
                     process.stdout.write(`\r\x1b[33m⚠ Rate limit hit. Waiting ${i}s before retry (${attempt}/${MAX_RETRIES})...\x1b[0m`);
                     await new Promise(r => setTimeout(r, 1000));
                 }
-                process.stdout.write("\r" + " ".repeat(80) + "\r"); // Clear the line
+                process.stdout.write("\r" + " ".repeat(80) + "\r");
                 continue;
             }
 
@@ -63,8 +53,7 @@ export async function generateAIResponse(messages) {
             return data.choices[0].message.content.trim();
         } catch (e) {
             if (attempt === MAX_RETRIES) {
-                console.error("AI Generation failed:", e.message);
-                process.exit(1);
+                throw e; // Let the caller handle the final failure
             }
         }
     }

package/src/config.js

@@ -0,0 +1,40 @@
+import fs from "fs";
+import os from "os";
+import path from "path";
+
+const CONFIG_DIR = path.join(os.homedir(), ".ryte");
+const CONFIG_FILE = path.join(CONFIG_DIR, "config.json");
+
+const DEFAULT_CONFIG = {
+    provider: "openai",
+    apiKey: "",
+    model: "gpt-4o-mini",
+    baseUrl: "" // Optional for local providers like OpenClaw/Ollama
+};
+
+export function getConfig() {
+    if (!fs.existsSync(CONFIG_FILE)) {
+        return null;
+    }
+    try {
+        const data = fs.readFileSync(CONFIG_FILE, "utf-8");
+        return { ...DEFAULT_CONFIG, ...JSON.parse(data) };
+    } catch (e) {
+        return null;
+    }
+}
+
+export function setConfig(updates) {
+    if (!fs.existsSync(CONFIG_DIR)) {
+        fs.mkdirSync(CONFIG_DIR, { recursive: true });
+    }
+    const current = getConfig() || DEFAULT_CONFIG;
+    const updated = { ...current, ...updates };
+    fs.writeFileSync(CONFIG_FILE, JSON.stringify(updated, null, 2), { mode: 0o600 });
+    return updated;
+}
+
+export function hasValidConfig() {
+    const config = getConfig();
+    return !!(config && config.apiKey);
+}

package/src/index.js

@@ -6,12 +6,16 @@ import path from "path";
 import { getStagedDiff, getCurrentBranch, getBranchCommits, applyCommit } from "./git.js";
 import { generateAIResponse } from "./ai.js";
 import { COMMIT_SYSTEM_PROMPT, PR_SYSTEM_PROMPT } from "./prompt.js";
+import { getConfig, setConfig, hasValidConfig } from "./config.js";
+import { PROVIDERS } from "./provider.js";
 
 const rl = readline.createInterface({
     input: process.stdin,
     output: process.stdout
 });
 
+const VERSION = "1.2.0";
+
 async function question(query) {
     return new Promise(resolve => rl.question(query, resolve));
 }
@@ -40,6 +44,43 @@ function editInteractively(initialText) {
     });
 }
 
+async function setupFlow() {
+    console.log("\n\x1b[36mWelcome to RYTE. Let's set up your Git Intelligence Layer.\x1b[0m");
+    console.log("------------------------------------------------------------");
+
+    console.log("\nSelect your LLM Provider:");
+    const providerList = Object.keys(PROVIDERS);
+    providerList.forEach((p, i) => console.log(`${i + 1}) ${p.charAt(0).toUpperCase() + p.slice(1)}`));
+
+    const choice = await question(`\nChoose [1-${providerList.length}]: `);
+    const providerKey = providerList[parseInt(choice) - 1] || "openai";
+
+    const apiKey = await question(`Paste your ${providerKey.toUpperCase()} API Key: `);
+    if (!apiKey) {
+        console.error("Error: API Key is required.");
+        process.exit(1);
+    }
+
+    setConfig({
+        provider: providerKey,
+        apiKey: apiKey,
+        model: PROVIDERS[providerKey].defaultModel
+    });
+
+    console.log("\n\x1b[32m✔ Configuration saved to ~/.ryte/config.json\x1b[0m");
+}
+
+async function handleConfig() {
+    const config = getConfig() || {};
+    console.log("\n\x1b[36mCurrent Configuration:\x1b[0m");
+    console.log(JSON.stringify(config, null, 2));
+
+    const choice = await question("\nWould you like to reset configuration? [y/N]: ");
+    if (choice.toLowerCase() === "y") {
+        await setupFlow();
+    }
+}
+
 async function interactiveLoop(initialResult, type) {
     let currentResult = initialResult;
 
@@ -125,30 +166,36 @@ async function main() {
     const args = process.argv.slice(2);
     const cmd = args[0]?.toLowerCase();
 
+    if (!hasValidConfig()) {
+        await setupFlow();
+    }
+
     if (cmd === "c" || cmd === "commit") {
         await handleCommit();
     } else if (cmd === "pr") {
         await handlePR();
+    } else if (cmd === "config") {
+        await handleConfig();
     } else {
         console.log(`
- \x1b[1;38;5;39m██████╗ \x1b[1;38;5;63m██╗   ██╗\x1b[1;38;5;129m████████╗\x1b[1;38;5;161m███████╗\x1b[0m
- \x1b[1;38;5;39m██╔══██╗\x1b[1;38;5;63m╚██╗ ██╔╝\x1b[1;38;5;129m╚══██╔══╝\x1b[1;38;5;161m██╔════╝\x1b[0m
- \x1b[1;38;5;39m██████╔╝\x1b[1;38;5;63m ╚████╔╝ \x1b[1;38;5;129m   ██║   \x1b[1;38;5;161m█████╗  \x1b[0m
- \x1b[1;38;5;39m██╔══██╗\x1b[1;38;5;63m  ╚██╔╝  \x1b[1;38;5;129m   ██║   \x1b[1;38;5;161m██╔══╝  \x1b[0m
- \x1b[1;38;5;39m██║  ██║\x1b[1;38;5;63m   ██║   \x1b[1;38;5;129m   ██║   \x1b[1;38;5;161m███████╗\x1b[0m
- \x1b[1;38;5;39m╚═╝  ╚═╝\x1b[1;38;5;63m   ╚═╝   \x1b[1;38;5;129m   ╚═╝   \x1b[1;38;5;161m╚══════╝\x1b[0m
-
-  \x1b[90mAI-Powered Git Workflow Assistant\x1b[0m
-  \x1b[90mv1.1.2 | by Riflo\x1b[0m
+  \x1b[1;38;5;39m██████╗ \x1b[1;38;5;63m██╗   ██╗\x1b[1;38;5;129m████████╗\x1b[1;38;5;161m███████╗\x1b[0m
+  \x1b[1;38;5;39m██╔══██╗\x1b[1;38;5;63m╚██╗ ██╔╝\x1b[1;38;5;129m╚══██╔══╝\x1b[1;38;5;161m██╔════╝\x1b[0m
+  \x1b[1;38;5;39m██████╔╝\x1b[1;38;5;63m ╚████╔╝ \x1b[1;38;5;129m   ██║   \x1b[1;38;5;161m█████╗  \x1b[0m
+  \x1b[1;38;5;39m██╔══██╗\x1b[1;38;5;63m  ╚██╔╝  \x1b[1;38;5;129m   ██║   \x1b[1;38;5;161m██╔══╝  \x1b[0m
+  \x1b[1;38;5;39m██║  ██║\x1b[1;38;5;63m   ██║   \x1b[1;38;5;129m   ██║   \x1b[1;38;5;161m███████╗\x1b[0m
+  \x1b[1;38;5;39m╚═╝  ╚═╝\x1b[1;38;5;63m   ╚═╝   \x1b[1;38;5;129m   ╚═╝   \x1b[1;38;5;161m╚══════╝\x1b[0m
+  
+  \x1b[1;38;5;46m[ THE AI-POWERED GIT INFRASTRUCTURE ]\x1b[0m
+  \x1b[90mv${VERSION} | by Riflo\x1b[0m
 
   \x1b[33mCOMMANDS:\x1b[0m
-  \x1b[32mryte c\x1b[0m     Generate semantic commit from diff
-  \x1b[32mryte pr\x1b[0m    Generate PR markdown from branch commits
+  \x1b[32mryte c\x1b[0m      Generate semantic commit from diff
+  \x1b[32mryte pr\x1b[0m     Generate PR markdown from branch commits
+  \x1b[32mryte config\x1b[0m Generate or edit your local configuration
 
-  \x1b[33mGETTING STARTED:\x1b[0m
-  Set either environment variable to unleash the AI:
-  • \x1b[36mGROQ_API_KEY\x1b[0m   (Recommended / Free tier)
-  • \x1b[36mOPENAI_API_KEY\x1b[0m (OpenAI API key)
+  \x1b[33mONBOARDING:\x1b[0m
+  No .env required. Run \x1b[32mryte config\x1b[0m or just run \x1b[32mryte c\x1b[0m to 
+  start the interactive setup.
         `);
     }
 

package/src/prompt.js

@@ -15,7 +15,9 @@ RULES:
 6. Scope: If the diff is localized, infer a scope (e.g., "auth", "ui", "config").
 7. Body (Optional): If the change is complex, add a brief body after 1 blank line to explain technical nuances.
 8. Context: If a branch name or ticket is provided, incorporate it into the scope or footer if applicable.
-9. Anti-Plagiarism: If the diff shows documentation changes containing example commit messages, DO NOT copy those examples. Always synthesize a new message describing the actual change.
+9. Anti-Hallucination: Documentation files (like README.md) often contain example commit messages (e.g., "feat(auth): ..."). DO NOT assume these examples are the topic of the current change.
+10. Logical Validation: Your suggested <scope> must be derived from actual modified logic in the diff, not from text inside code blocks, comments, or examples within a documentation file.
+11. If only README.md or docs are changed, the type MUST be "docs" and the scope should relate to the documentation structure (e.g., "readme", "config", "intro"), NOT the example code inside it.
 
 Example:
 feat(ui): add loading state to checkout button

package/src/provider.js

@@ -0,0 +1,31 @@
+export const PROVIDERS = {
+    openai: {
+        baseUrl: "https://api.openai.com/v1/chat/completions",
+        defaultModel: "gpt-4o-mini",
+        authHeader: (key) => `Bearer ${key}`
+    },
+    groq: {
+        baseUrl: "https://api.groq.com/openai/v1/chat/completions",
+        defaultModel: "llama-3.1-8b-instant",
+        authHeader: (key) => `Bearer ${key}`
+    },
+    openrouter: {
+        baseUrl: "https://openrouter.ai/api/v1/chat/completions",
+        defaultModel: "google/gemini-pro-1.5-exp",
+        authHeader: (key) => `Bearer ${key}`
+    },
+    local: {
+        baseUrl: "http://localhost:11434/v1/chat/completions", // Default Ollama/OpenClaw local port
+        defaultModel: "llama3",
+        authHeader: (key) => `Bearer ${key}`
+    }
+};
+
+export function getProviderConfig(name, customBaseUrl = "") {
+    const p = PROVIDERS[name] || PROVIDERS.openai;
+    return {
+        url: customBaseUrl || p.baseUrl,
+        model: p.defaultModel,
+        auth: p.authHeader
+    };
+}