draftify-cli 1.0.3 → 1.0.6

package/dist/repl.js

@@ -432,6 +432,7 @@ async function startRepl(initialUsername) {
                     console.log(`  ${(0, kleur_1.cyan)("/thinking-level")} - Adjust the AI's thinking process depth`);
                     console.log(`  ${(0, kleur_1.cyan)("/login")}     - Log in to your Draftify account`);
                     console.log(`  ${(0, kleur_1.cyan)("/logout")}    - Log out of your account`);
+                    console.log(`  ${(0, kleur_1.cyan)("/usage")}     - Show your token usage and quota`);
                     console.log(`  ${(0, kleur_1.cyan)("/help")}      - List commands`);
                     console.log(`  ${(0, kleur_1.cyan)("/clear")}     - Clear the terminal screen`);
                     console.log(`  ${(0, kleur_1.cyan)("/exit")}      - Exit the CLI`);
@@ -440,6 +441,11 @@ async function startRepl(initialUsername) {
                 else if (cmd === "clear") {
                     renderWelcome();
                 }
+                else if (cmd === "usage") {
+                    ui_1.ui.header("Usage & Quota");
+                    console.log(`  Quota: ${(0, kleur_1.cyan)("Infinite")}`);
+                    console.log(`  Tokens used: ${(0, kleur_1.cyan)((0, config_1.getTokensUsed)().toLocaleString())}\n`);
+                }
                 else if (cmd === "chats") {
                     const chats = (0, chats_1.loadChats)();
                     if (chats.length === 0) {

package/dist/utils/api.js

@@ -10,10 +10,11 @@ const genai_1 = require("@google/genai");
 const dotenv_1 = __importDefault(require("dotenv"));
 dotenv_1.default.config();
 async function refactorCodeApi(fileName, code, instruction, history, modelName, skillsData, thinkingLevel, onChunk, abortSignal) {
+    const apiKey = process.env.GEMINI_API_KEY || (0, config_1.getGeminiApiKey)();
     // 1. HELYI VÉGREHAJTÁS (Ha van GEMINI_API_KEY a .env-ben, közvetlenül a Geminit hívjuk)
     // Ez tökéletes a helyi teszteléshez, mielőtt a backendre (Vercel) kerül a kód.
-    if (process.env.GEMINI_API_KEY) {
-        const ai = new genai_1.GoogleGenAI({ apiKey: process.env.GEMINI_API_KEY });
+    if (apiKey) {
+        const ai = new genai_1.GoogleGenAI({ apiKey: apiKey });
         // Modell kiválasztása a kívánt logika alapján
         let geminiModel = 'gemini-3.5-flash'; // Alapértelmezett (Medium, Low, Adaptive)
         if (thinkingLevel === 'High') {
@@ -45,11 +46,46 @@ async function refactorCodeApi(fileName, code, instruction, history, modelName,
             role: 'user',
             parts: [{ text: currentPrompt }]
         });
+        const systemInstruction = `You are Draftify AI, an expert software engineering CLI assistant.
+You have the ability to automatically create files, modify files, delete files, and run terminal commands on the user's machine.
+To do this, you MUST format your response using the following XML tags:
+
+1. To create a file:
+<FILE_CREATE path="relative/path/to/file.ext">
+file content here
+</FILE_CREATE>
+
+2. To modify an existing file:
+<FILE_MODIFY path="relative/path/to/file.ext">
+<<<<<<< SEARCH
+exact lines to replace
+=======
+new lines
+>>>>>>>
+</FILE_MODIFY>
+
+3. To delete a file:
+<FILE_DELETE path="relative/path/to/file.ext" />
+
+4. To run a terminal command:
+<RUN_COMMAND>npm install express</RUN_COMMAND>
+
+5. To ask clarifying questions when the user's intent is ambiguous:
+<ASK_QUESTIONS>
+[
+  {
+    "question": "What tech stack do you prefer?",
+    "options": ["React + Vite", "Next.js", "Vanilla HTML/JS"]
+  }
+]
+</ASK_QUESTIONS>
+
+Always use these tags when you write code or ask questions so the CLI can apply it automatically! Keep explanations short and outside the tags.`;
         const responseStream = await ai.models.generateContentStream({
             model: geminiModel,
             contents: contents,
             config: {
-                systemInstruction: "You are an expert AI software engineering CLI assistant. You provide solutions, refactorings, and terminal commands."
+                systemInstruction: systemInstruction
             }
         });
         let fullResult = "";
@@ -62,6 +98,9 @@ async function refactorCodeApi(fileName, code, instruction, history, modelName,
                     onChunk(chunk.text);
             }
         }
+        // Rough estimation of tokens
+        const estimatedTokens = Math.ceil((JSON.stringify(contents).length + fullResult.length) / 4);
+        (0, config_1.addTokensUsed)(estimatedTokens);
         return fullResult;
     }
     // 2. TÁVOLI VÉGREHAJTÁS (Eredeti Vercel backend kód, ha nincs helyi API kulcs)
@@ -142,6 +181,9 @@ async function refactorCodeApi(fileName, code, instruction, history, modelName,
             }
         }
     }
+    // Rough estimation of tokens
+    const estimatedTokens = Math.ceil((JSON.stringify({ fileName, code, instruction, history, modelName, skillsData }).length + fullResult.length) / 4);
+    (0, config_1.addTokensUsed)(estimatedTokens);
     return fullResult;
 }
 async function getUserProfile() {

package/dist/utils/config.js

@@ -13,6 +13,9 @@ exports.getSkills = getSkills;
 exports.setSkills = setSkills;
 exports.getThinkingLevel = getThinkingLevel;
 exports.setThinkingLevel = setThinkingLevel;
+exports.getGeminiApiKey = getGeminiApiKey;
+exports.getTokensUsed = getTokensUsed;
+exports.addTokensUsed = addTokensUsed;
 const os_1 = __importDefault(require("os"));
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
@@ -71,3 +74,15 @@ function getThinkingLevel() {
 function setThinkingLevel(level) {
     saveConfig({ thinkingLevel: level });
 }
+function getGeminiApiKey() {
+    const config = getConfig();
+    return config.geminiApiKey || null;
+}
+function getTokensUsed() {
+    const config = getConfig();
+    return config.tokensUsed || 0;
+}
+function addTokensUsed(amount) {
+    const current = getTokensUsed();
+    saveConfig({ tokensUsed: current + amount });
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "draftify-cli",
-  "version": "1.0.3",
+  "version": "1.0.6",
   "description": "Draftify AI CLI tool",
   "main": "dist/index.js",
   "bin": {