draftify-cli 1.0.69 → 1.0.71

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 vertexai_1 = require("@google-cloud/vertexai");
+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) {
@@ -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 vertexai_1.VertexAI({ project: 'draftify-prod', location: 'us-central1' });
+        const ai = new genai_1.GoogleGenAI({ 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') {
@@ -300,21 +300,23 @@ Throughout, balance deep technical proficiency with an accessible, friendly, wel
                 abortSignal.addEventListener("abort", onAbort);
             }
             try {
-                const generativeModel = ai.getGenerativeModel({
+                const responseStream = await ai.models.generateContentStream({
                     model: geminiModel,
-                    systemInstruction: { role: 'system', parts: [{ text: systemInstruction }] }
-                });
-                const responseStream = await generativeModel.generateContentStream({
                     contents: contents,
-                    generationConfig: generationConfig
+                    config: {
+                        systemInstruction: systemInstruction,
+                        maxOutputTokens: generationConfig.maxOutputTokens,
+                        temperature: generationConfig.temperature,
+                        thinkingConfig: generationConfig.thinkingConfig
+                    }
                 });
                 let tempResult = "";
-                for await (const chunk of responseStream.stream) {
+                for await (const chunk of responseStream) {
                     if (abortSignal?.aborted) {
                         reject(new Error("Aborted"));
                         return;
                     }
-                    const text = chunk.candidates?.[0]?.content?.parts?.[0]?.text || "";
+                    const text = chunk.text || "";
                     if (text) {
                         tempResult += text;
                         if (onChunk)

package/package.json

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