draftify-cli 1.0.2 → 1.0.3

package/dist/utils/api.js

@@ -1,9 +1,70 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.refactorCodeApi = refactorCodeApi;
 exports.getUserProfile = getUserProfile;
 const config_1 = require("./config");
+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) {
+    // 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 });
+        // 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') {
+            geminiModel = 'gemini-3.1-pro-preview';
+        }
+        // Üzenetek összeállítása a Gemini számára
+        const contents = [];
+        // Előzmények hozzáadása (a 'assistant' szerepkört 'model'-re kell cserélni)
+        if (history && history.length > 0) {
+            for (const msg of history) {
+                contents.push({
+                    role: msg.role === 'assistant' ? 'model' : 'user',
+                    parts: [{ text: msg.content }]
+                });
+            }
+        }
+        // Aktuális prompt összeállítása
+        let currentPrompt = `Task/Instruction: ${instruction}\n\n`;
+        if (fileName && code) {
+            currentPrompt += `Context from workspace:\n${code}\n\n`;
+        }
+        if (skillsData && skillsData.length > 0) {
+            currentPrompt += `Available skills context:\n`;
+            skillsData.forEach(skill => {
+                currentPrompt += `--- Skill: ${skill.name} ---\n${skill.content}\n\n`;
+            });
+        }
+        contents.push({
+            role: 'user',
+            parts: [{ text: currentPrompt }]
+        });
+        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."
+            }
+        });
+        let fullResult = "";
+        for await (const chunk of responseStream) {
+            if (abortSignal?.aborted)
+                throw new Error("Aborted");
+            if (chunk.text) {
+                fullResult += chunk.text;
+                if (onChunk)
+                    onChunk(chunk.text);
+            }
+        }
+        return fullResult;
+    }
+    // 2. TÁVOLI VÉGREHAJTÁS (Eredeti Vercel backend kód, ha nincs helyi API kulcs)
     const token = (0, config_1.getToken)();
     if (!token) {
         throw new Error("No token found. Please run 'draftify login' first.");
@@ -69,7 +130,6 @@ async function refactorCodeApi(fileName, code, instruction, history, modelName,
                 }
             }
             else if (line.startsWith('{"result":')) {
-                // Fallback for non-streaming response
                 try {
                     const dataObj = JSON.parse(line);
                     if (dataObj.result) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "draftify-cli",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "Draftify AI CLI tool",
   "main": "dist/index.js",
   "bin": {
@@ -24,9 +24,12 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
+    "@google-cloud/vertexai": "^1.12.0",
+    "@google/genai": "^2.7.0",
     "ansi-escapes": "^7.3.0",
     "axios": "^1.7.2",
     "commander": "^12.1.0",
+    "dotenv": "^17.4.2",
     "enquirer": "^2.4.1",
     "kleur": "^4.1.5",
     "open": "^8.4.2",