@runnerpro/backend 1.12.26 → 1.13.1

package/lib/cjs/chat/api/conversation.js

@@ -38,6 +38,7 @@ const saveResponseTime_1 = require("../saveResponseTime");
 const fluent_ffmpeg_1 = __importDefault(require("fluent-ffmpeg"));
 const ffmpeg_static_1 = __importDefault(require("ffmpeg-static"));
 const ffprobe_static_1 = __importDefault(require("ffprobe-static"));
+const prompt_1 = require("../../prompt");
 fluent_ffmpeg_1.default.setFfmpegPath(ffmpeg_static_1.default);
 fluent_ffmpeg_1.default.setFfprobePath(ffprobe_static_1.default.path);
 const conversationRoute = (_a) => {
@@ -214,13 +215,17 @@ const getChatFile = ({ bucket, id, isClient, userid }) => __awaiter(void 0, void
     }
 });
 const sendMessage = (req, res, { sendNotification, firebaseMessaging, isClient }) => __awaiter(void 0, void 0, void 0, function* () {
-    const { text, replyMessageId, new: newMessage } = req.body;
+    const { replyMessageId, new: newMessage } = req.body;
+    let { text } = req.body;
     const { userid } = req.session;
     const idCliente = isClient ? req.session.userid : req.body.idCliente;
     let idWorkout, type = 1;
-    // Si es entrenador, se marca leído cuando se envía un mensaje
-    if (!isClient)
+    if (!isClient) {
+        // Si es entrenador, se marca leído cuando se envía un mensaje
         yield markReadMessage({ isClient, idCliente });
+        // Cambiar el tono del texto a uno neutro y corregir la ortografía
+        text = cambiarTonoEntrenadorNeutro(text);
+    }
     // Devuelve el texto en el otro idioma si el cliente no habla español y el idioma del cliente
     const { textSpanish, textPreferredLanguage, preferredLanguage } = yield getPreferredLanguageForChat({
         text,
@@ -395,6 +400,14 @@ const markReadMessage = ({ isClient, idCliente }) => __awaiter(void 0, void 0, v
     if (!isClient)
         yield (0, saveResponseTime_1.saveResponseTime)(idCliente);
 });
+const cambiarTonoEntrenadorNeutro = (text) => __awaiter(void 0, void 0, void 0, function* () {
+    const textCorregido = yield (0, prompt_1.sendPrompt)(`
+    Eres un asistente de IA que cambia el tono de los mensajes de un entrenador a uno neutro y corrige la ortografía.
+    El mensaje es: ${text}
+    El mensaje corregido es:
+  `);
+    return textCorregido;
+});
 const resizeImage = (data, filePath, mimetype) => __awaiter(void 0, void 0, void 0, function* () {
     if (mimetype.includes('webp'))
         return data;

package/lib/cjs/prompt/index.js

@@ -0,0 +1,28 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.sendPrompt = exports.getVertexChat = void 0;
+const vertexai_1 = require("@google-cloud/vertexai");
+const getVertexChat = () => {
+    const vertexAI = new vertexai_1.VertexAI({ project: process.env.GCP_PROYECT_ID, location: 'us-central1' });
+    const generativeModel = vertexAI.preview.getGenerativeModel({
+        model: 'gemini-2.5-flash',
+    });
+    return generativeModel.startChat({});
+};
+exports.getVertexChat = getVertexChat;
+const sendPrompt = (message) => __awaiter(void 0, void 0, void 0, function* () {
+    var _a, _b;
+    const chat = getVertexChat();
+    const result = yield chat.sendMessage(message);
+    return ((_b = (_a = result.response.candidates[0].content) === null || _a === void 0 ? void 0 : _a.parts[0]) === null || _b === void 0 ? void 0 : _b.text) || '';
+});
+exports.sendPrompt = sendPrompt;

package/lib/cjs/types/chat/api/conversation.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"conversation.d.ts","sourceRoot":"","sources":["../../../../../src/chat/api/conversation.ts"],"names":[],"mappings":"AAkBA,QAAA,MAAM,iBAAiB,0BAA2B,GAAG,SAuBpD,CAAC;AAuZF,OAAO,EAAE,iBAAiB,EAAE,CAAC"}
+{"version":3,"file":"conversation.d.ts","sourceRoot":"","sources":["../../../../../src/chat/api/conversation.ts"],"names":[],"mappings":"AAmBA,QAAA,MAAM,iBAAiB,0BAA2B,GAAG,SAuBpD,CAAC;AAqaF,OAAO,EAAE,iBAAiB,EAAE,CAAC"}

package/lib/cjs/types/prompt/index.d.ts

@@ -0,0 +1,4 @@
+declare const getVertexChat: () => import("@google-cloud/vertexai").ChatSessionPreview;
+declare const sendPrompt: (message: any) => Promise<string>;
+export { getVertexChat, sendPrompt };
+//# sourceMappingURL=index.d.ts.map

package/lib/cjs/types/prompt/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/prompt/index.ts"],"names":[],"mappings":"AAEA,QAAA,MAAM,aAAa,2DAQlB,CAAC;AAEF,QAAA,MAAM,UAAU,mCAIf,CAAC;AAEF,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@runnerpro/backend",
-  "version": "1.12.26",
+  "version": "1.13.1",
   "description": "A collection of common backend functions",
   "exports": {
     ".": "./lib/cjs/index.js"
@@ -64,6 +64,7 @@
   "dependencies": {
     "@google-cloud/storage": "^7.11.3",
     "@google-cloud/translate": "^8.3.0",
+    "@google-cloud/vertexai": "^1.10.0",
     "@notionhq/client": "^2.2.15",
     "exifr": "^7.1.3",
     "ffmpeg-static": "^5.2.0",