garriga 1.0.0

package/index.js

@@ -0,0 +1,298 @@
+const axios = require("axios");
+
+class LLMAdapter {
+    constructor(provider = "chatgpt") {
+        switch (provider) {
+            case "chatgpt":
+                this.adapter = new ChatGptAPI();
+                break;
+            case "localLLM":
+                this.adapter = new localLLM();
+                break;
+            case "gemini":
+                this.adapter = new GeminiAPI();
+                break;
+
+            default:
+                throw new Error("Proveedor de IA no disponible");
+        }
+    }
+
+    async generateResponse(prompt, userToken, source) {
+        return this.adapter.generateResponse(prompt, userToken, source);
+    }
+    async generateRateResponse(responseId, ratePoints, rateText, userToken) {
+        return this.adapter.generateRateResponse(responseId, ratePoints, rateText, userToken);
+    }
+    async getAIhistorial(fecha_inicial, fecha_final, id_usuario) {
+        return this.adapter.getAIhistorial(fecha_inicial, fecha_final, id_usuario);
+    }
+    async getAIQueryTypes() {
+        return this.adapter.getAIQueryTypes();
+    }
+}
+
+class ChatGptAPI {
+    constructor() {
+        this.apiKey = process.env.CHATGPTAPIKEY;
+        this.apiUrl = "https://api.openai.com/v1/chat/completions";
+        this.model = "gpt-3.5-turbo";
+    }
+    async generateResponse(prompt) {
+        try {
+            const response = await axios.post(
+                this.apiUrl,
+                {
+                    model: this.model,
+                    messages: [
+                        { role: "system", content: "You are a helpful assistant." },
+                        { role: "user", content: prompt },
+                    ],
+                },
+                {
+                    headers: {
+                        "Content-Type": "application/json",
+                        Authorization: "Bearer " + this.apiKey,
+                    },
+                },
+            );
+
+            return response.data.choices[0].message.content;
+        } catch (error) {
+            console.error(`Error al hacer la llamada de la API ChatGPT: ${error.message}`);
+            throw error;
+        }
+    }
+    async generateRateResponse(responseId, ratePoints, rateText, userToken) {
+        const LLMToken = await this.getToken({ username: "admin", password: "1234" });
+        try {
+            const response = await axios.post(
+                this.apiUrlRateResponse,
+                {
+                    responseId,
+                    ratePoints,
+                    rateText,
+                    userToken,
+                },
+                {
+                    headers: {
+                        "Content-Type": "application/json",
+                        Authorization: "Bearer " + LLMToken,
+                    },
+                },
+            );
+
+            return response;
+        } catch (error) {
+            console.error(`Error al hacer la llamada de la API Local: ${error}`);
+            throw error;
+        }
+    }
+    async getAIhistorial() {}
+    async getAIQueryTypes() {}
+}
+
+class GeminiAPI {
+    constructor() {
+        this.apiKey = process.env.GEMINIAPIKEY;
+        this.baseURL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent";
+    }
+
+    get URL() {
+        return this.baseURL + "?key=" + this.apiKey;
+    }
+
+    async generateResponse(prompt) {
+        try {
+            const response = await axios.post(
+                this.URL,
+                {
+                    contents: [
+                        {
+                            parts: [{ text: prompt }],
+                        },
+                    ],
+                },
+                {
+                    headers: {
+                        "Content-Type": "application/json",
+                    },
+                },
+            );
+
+            return response.data?.candidates?.[0].content?.parts?.[0]?.text || null;
+        } catch (error) {
+            console.error(`Error al hacer la llamada de la API Gemini: ${error.message}`);
+            throw error;
+        }
+    }
+    async generateRateResponse(responseId, ratePoints, rateText, userToken) {
+        const LLMToken = await this.getToken({ username: "admin", password: "1234" });
+        try {
+            const response = await axios.post(
+                this.apiUrlRateResponse,
+                {
+                    responseId,
+                    ratePoints,
+                    rateText,
+                    userToken,
+                },
+                {
+                    headers: {
+                        "Content-Type": "application/json",
+                        Authorization: "Bearer " + LLMToken,
+                    },
+                },
+            );
+
+            return response;
+        } catch (error) {
+            console.error(`Error al hacer la llamada de la API Local: ${error}`);
+            throw error;
+        }
+    }
+    async getAIhistorial() {}
+    async getAIQueryTypes() {}
+}
+
+class localLLM {
+    constructor() {
+        this.apiUrlGetToken = "http://iaService:29000/token";
+        this.apiUrlSendQuery = "http://iaService:29000/query";
+        this.apiUrlRateResponse = "http://iaService:29000/rate-response";
+        this.apiUrlHistorial = "http://iaService:29000/historial";
+        this.apiUrlQueryTypes = "http://iaService:29000/query-types";
+        this.apiUrlOllama = "http://10.20.150.102:11434/api/generate";
+        this.model = "GrupoImpuLLMtec";
+    }
+    async getToken({ username, password }) {
+        try {
+            const responseToken = await axios.post(
+                this.apiUrlGetToken,
+                {
+                    username,
+                    password,
+                },
+                {
+                    headers: {
+                        "Content-Type": "application/json",
+                    },
+                },
+            );
+            return responseToken.data.access_token;
+        } catch (error) {
+            console.error(`Error al hacer la llamada de gettoken de la API Local: ${error}`);
+            throw error;
+        }
+    }
+
+    async generateResponse(prompt, userToken, source = "mistral7b-genei:latest") {
+        if (userToken === "modeloLocalOllama") {
+            try {
+                const response = await axios.post(
+                    this.apiUrlOllama,
+                    {
+                        prompt,
+                        model: source,
+                        stream: false,
+                    },
+                    {
+                        headers: {
+                            "Content-Type": "application/json",
+                        },
+                    },
+                );
+
+                return response?.data.response;
+            } catch (error) {
+                console.error(`Error al hacer la llamada de la API Local Ollama: ${error}`);
+                throw error;
+            }
+        }
+        const LLMToken = await this.getToken({ username: "admin", password: "1234" });
+        try {
+            const response = await axios.post(
+                this.apiUrlSendQuery,
+                {
+                    question: prompt,
+                    userToken,
+                    source,
+                },
+                {
+                    headers: {
+                        "Content-Type": "application/json",
+                        Authorization: "Bearer " + LLMToken,
+                    },
+                },
+            );
+
+            return response;
+        } catch (error) {
+            console.error(`Error al hacer la llamada de la API Local: ${error}`);
+            throw error;
+        }
+    }
+    async generateRateResponse(responseId, ratePoints, rateText, userToken) {
+        const LLMToken = await this.getToken({ username: "admin", password: "1234" });
+        try {
+            const response = await axios.post(
+                this.apiUrlRateResponse,
+                {
+                    responseId,
+                    ratePoints,
+                    rateText,
+                    userToken,
+                },
+                {
+                    headers: {
+                        "Content-Type": "application/json",
+                        Authorization: "Bearer " + LLMToken,
+                    },
+                },
+            );
+
+            return response;
+        } catch (error) {
+            console.error(`Error al hacer la llamada de la API Local: ${error}`);
+            throw error;
+        }
+    }
+
+    async getAIhistorial(fecha_inicial, fecha_final, id_usuario) {
+        const LLMToken = await this.getToken({ username: "admin", password: "1234" });
+        const urlCompletaHistorial = `${this.apiUrlHistorial}/${fecha_inicial}/${fecha_final}/${id_usuario}`;
+        try {
+            const response = await axios.get(urlCompletaHistorial, {
+                headers: {
+                    "Content-Type": "application/json",
+                    Authorization: "Bearer " + LLMToken,
+                },
+            });
+
+            return response;
+        } catch (error) {
+            console.error(`Error al hacer la llamada de la API Local: ${error}`);
+            throw error;
+        }
+    }
+
+    async getAIQueryTypes() {
+        const LLMToken = await this.getToken({ username: "admin", password: "1234" });
+        const urlCompletaQueryTypes = `${this.apiUrlQueryTypes}`;
+        try {
+            const response = await axios.get(urlCompletaQueryTypes, {
+                headers: {
+                    "Content-Type": "application/json",
+                    Authorization: "Bearer " + LLMToken,
+                },
+            });
+
+            return response;
+        } catch (error) {
+            console.error(`Error al hacer la llamada de la API Local: ${error}`);
+            throw error;
+        }
+    }
+}
+
+module.exports = LLMAdapter;

package/package.json

@@ -0,0 +1,12 @@
+{
+  "name": "garriga",
+  "version": "1.0.0",
+  "description": "AI connector",
+  "main": "index.js",
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "keywords": [],
+  "author": "Alexandra",
+  "license": "ISC"
+}