draftify-cli 1.0.54 → 1.0.56

package/dist/utils/api.js

@@ -6,7 +6,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.refactorCodeApi = refactorCodeApi;
 exports.getUserProfile = getUserProfile;
 const config_1 = require("./config");
-const genai_1 = require("@google/genai");
+const vertexai_1 = require("@google-cloud/vertexai");
 const dotenv_1 = __importDefault(require("dotenv"));
 dotenv_1.default.config();
 async function refactorCodeApi(fileName, code, instruction, history, modelName, skillsData, thinkingLevel, onChunk, abortSignal) {
@@ -14,7 +14,7 @@ async function refactorCodeApi(fileName, code, instruction, history, modelName,
     // 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 (apiKey) {
-        const ai = new genai_1.GoogleGenAI({ apiKey: apiKey });
+        const ai = new vertexai_1.VertexAI({ project: 'draftify-prod', location: 'us-central1' });
         // 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') {
@@ -160,23 +160,24 @@ Throughout, balance deep technical proficiency with an accessible, friendly, wel
                 abortSignal.addEventListener("abort", onAbort);
             }
             try {
-                const responseStream = await ai.models.generateContentStream({
+                const generativeModel = ai.getGenerativeModel({
                     model: geminiModel,
-                    contents: contents,
-                    config: {
-                        systemInstruction: systemInstruction
-                    }
+                    systemInstruction: { role: 'system', parts: [{ text: systemInstruction }] }
+                });
+                const responseStream = await generativeModel.generateContentStream({
+                    contents: contents
                 });
                 let tempResult = "";
-                for await (const chunk of responseStream) {
+                for await (const chunk of responseStream.stream) {
                     if (abortSignal?.aborted) {
                         reject(new Error("Aborted"));
                         return;
                     }
-                    if (chunk.text) {
-                        tempResult += chunk.text;
+                    const text = chunk.candidates?.[0]?.content?.parts?.[0]?.text || "";
+                    if (text) {
+                        tempResult += text;
                         if (onChunk)
-                            onChunk(chunk.text);
+                            onChunk(text);
                     }
                 }
                 resolve(tempResult);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "draftify-cli",
-  "version": "1.0.54",
+  "version": "1.0.56",
   "description": "Draftify AI CLI tool",
   "main": "dist/index.js",
   "bin": {
@@ -24,6 +24,7 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
+    "@google-cloud/vertexai": "^1.12.0",
     "@google/genai": "^2.8.0",
     "ansi-escapes": "^7.3.0",
     "axios": "^1.7.2",