glotto 1.0.2 → 2.9.0

package/script/src/models/gemini.js

@@ -1,85 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const generative_ai_1 = require("@google/generative-ai");
-const mod_js_1 = require("../../deps/jsr.io/@std/encoding/1.0.7/mod.js");
-const utilites_js_1 = require("../utilites.js");
-const logger_js_1 = require("../logger.js");
-const file_js_1 = require("../file.js");
-/**
- * Gemini AI Module
- */
-class Gemini {
-    constructor(key, fileContents, from, to) {
-        Object.defineProperty(this, "fileContents", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "from", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "to", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "genAI", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.fileContents = fileContents;
-        this.from = from;
-        this.to = to;
-        this.genAI = new generative_ai_1.GoogleGenerativeAI(key);
-    }
-    async translate() {
-        const { systemPrompt, userPrompt } = (0, utilites_js_1.generatePrompts)(this.from, this.to);
-        const model = this.genAI.getGenerativeModel({
-            model: 'gemini-2.0-flash',
-            systemInstruction: systemPrompt,
-            generationConfig: {
-                candidateCount: 1,
-                responseMimeType: 'application/json',
-                responseLogprobs: false,
-                temperature: 0.3,
-                topP: 0.7,
-                topK: 20,
-                presencePenalty: 0.2,
-                frequencyPenalty: 0.2,
-                maxOutputTokens: Number.POSITIVE_INFINITY,
-            },
-        });
-        const results = [];
-        let index = 0;
-        for (const file of this.fileContents) {
-            const fileBase64 = (0, mod_js_1.encodeBase64)(file);
-            const part = {
-                inlineData: {
-                    data: fileBase64,
-                    mimeType: 'text/plain',
-                },
-            };
-            const { response } = await model.generateContent([systemPrompt, userPrompt, part]);
-            const text = response.text();
-            const tempJsonFileName = `${this.to}_${index + 1}.json`;
-            await (0, file_js_1.writeTemp)(tempJsonFileName, text);
-            if (!(0, utilites_js_1.isValidJson)(text)) {
-                logger_js_1.logger.error('Json is not valid!', tempJsonFileName);
-                continue;
-            }
-            results.push(text);
-            index++;
-        }
-        const jsonInputs = results.map((r) => JSON.parse(r));
-        const mergedContent = (0, file_js_1.mergeInputs)(jsonInputs);
-        return JSON.stringify(mergedContent, null, 2);
-    }
-}
-exports.default = Gemini;