Merge remote-tracking branch 'origin/main'

Author: Dannkol

examples/.env

@@ -5,4 +5,4 @@ PASSWORD = "123456"
 DATABASE = "inpos_ayudakarla2"
 APIKEYGEMINI = 
 HOST_QDRANT = 
-APIKEY_QDRANT = 
+APIKEY_QDRANT = 

examples/api/app.py

@@ -1,6 +1,6 @@
 import os
-import ssl
-from fastapi import FastAPI
+# import ssl
+from fastapi import FastAPI, Body
 from pydantic import BaseModel
 from vertexai.generative_models import GenerativeModel, SafetySetting, Part
 from pydantic import BaseModel, Field
@@ -13,7 +13,7 @@
 import yipao as yp
 from yipao.databases import MySql
 from yipao.vectorstores import QdrantDB
-from yipao.LLM import VertexAiLLM
+from yipao.LLM import GoogleGenAi
 
 KEY_PATH='./credentials/gen-lang-client-0867205962-4a2d259770b4.json'
 PROJECT_ID = "gen-lang-client-0867205962"
@@ -47,10 +47,18 @@
 app = FastAPI(
         title="Simple Inference API",
         version="1.0.0",
-        description="This is the OpenAPI specification of a service to query a sql database using natural language. Its purpose is to illustrate how to declare your REST API as an OpenAPI tool."
-)
+        description="Esta es una demo en swagger ui de una api para realizar consultas sql a una base de datos por medio de lenguaje natural")
 app.openapi_version = "3.0.0"
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+
 #----------------
 # SSL
 #----------------
@@ -62,17 +70,27 @@
 # APP
 # --------------
 
-@app.get("/ping")
+@app.get("/ping", summary="Ping de validacion")
 def root():
     return {"Hello": "World"}
 
 
 class QueryModel(BaseModel):
-    Item: str = Field(..., description="Name Item")
-    q: str
+    Item: str = Field(...,  examples=["root:123456@127.0.0.1:3306/messagehistoryrag"], description="Url database")
+    q: str = Field(..., description='Pregunta del usuario')
 
 class InitializeModel(BaseModel):
-    Item: str = Field(..., description="Name Item")
+    Item: str = Field(
+        ..., 
+        examples=["root:123456@127.0.0.1:3306/messagehistoryrag"], 
+        description="Url database")
+
+    class Config:
+            schema_extra = {
+                "example": {
+                    "Item": "root:123456@127.0.0.1:3306/messagehistoryrag"
+                }
+            }
 
 
 def connect_sql():
@@ -85,7 +103,7 @@ def connect_sql():
     }
 
     try:
-        mysql = MySql(**connection)
+        mysql = MySql(**db)
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Failed to connect to MySQL: {str(e)}")
 
@@ -142,10 +160,12 @@ def initialize_qdrant():
         qdrant.add_ddls(formatted_output, connection["database"])
         added_excel = True
 
-    return {"message": "Qdrant initialized", "payload": data.Item}
-
+        return {"message": "Qdrant initialized", "payload": connection["database"]}
+    except Exception as e:
+        print('error',e)
+        raise HTTPException(status_code=401, detail="No fue posible conectar a la db")
 
-@app.post("/inference", summary="Executes a natural language query. Needs Item and 'q' is the query to be performed.")
+@app.post("/inference", summary="Ejecuta una consulta en lenguaje natural, devuelve el resultado de la consulta junto con la consulta creada")
 def query_inference(data: QueryModel):
 
     connection = parse_sql_url(data.Item)
@@ -159,7 +179,15 @@ def query_inference(data: QueryModel):
 
     try:
         res, _, _, sql_query = agent.invoke(data.q, debug=True, iterations=6)
+        
+        try:
+            res = res.to_dict(orient="records")
+        except Exception as e:
+            print(f"Error converting result to dict: {e}")
+            res = str(res)
+
+        return {"q": data.q, "res": res, "sql_query_generated": sql_query}
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Query execution failed: {str(e)}")
+        print('Error en el query',e)
+        raise HTTPException(status_code=401, detail=f"Query execution failed")
 
-    return {"q": data.q, "res": res, "sql_query_generated": sql_query}

examples/api/cert.der

@@ -0,0 +1,117 @@
+// Selecciona los elementos del DOM
+const messagesDiv = document.getElementById('messages');
+const messageInput = document.getElementById('message-input');
+const sendButton = document.getElementById('send-button');
+const usernameDialog = document.getElementById('db-dialog');
+const usernameInput = document.getElementById('db-input');
+const confirmUsernameButton = document.getElementById('db-username');
+const chatContainer = document.getElementById('chat-container');
+
+let username = '';
+
+// Función para agregar mensajes al chat
+function addMessageToChat(message, className) {
+    const p = document.createElement('p');
+    p.className = className;
+    p.textContent = message;
+    messagesDiv.appendChild(p);
+    messagesDiv.scrollTop = messagesDiv.scrollHeight; // Scroll al final del chat
+}
+
+// Función para mostrar el cuadro de diálogo del nombre de usuario
+function showUsernameDialog() {
+    usernameDialog.style.display = 'flex';
+}
+
+// Función para ocultar el cuadro de diálogo y mostrar el chat
+function hideUsernameDialog() {
+    usernameDialog.style.display = 'none';
+    chatContainer.style.display = 'block';
+}
+
+// Evento de clic en el botón "Confirmar" para el nombre de usuario
+confirmUsernameButton.addEventListener('click', () => {
+    const enteredUsername = usernameInput.value.trim();
+
+    if (enteredUsername) {
+        const initdb = enteredUsername; // Guardamos el nombre de usuario
+        const requestBody = {
+            Item: initdb
+        }
+        fetch('http://127.0.0.1:8080/init_item', {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify(requestBody)
+        }).then((response) => {
+            if (!response.ok) {  // Verifica si la respuesta no fue exitosa (status >= 400)
+                return response.json().then(err => {
+                    throw new Error(`Error ${response.status}: ${err.detail || 'Error desconocido'}`);
+                });
+            }
+            return response.json();
+        })  // convertir a json
+            .then((json) => {
+                hideUsernameDialog()
+                localStorage.setItem('db', initdb)
+                addMessageToChat(`Conexion establesida a la base de datos: ${json.payload}`, 'api-response');
+            })    //imprimir los datos en la consola
+            .catch((err) => {
+                console.log('Solicitud fallida', err)
+                alert('Error en la conexios revisa que el url este bien')
+            }); // Capturar errores
+
+    }
+});
+
+// Evento de clic en el botón "Enviar"
+sendButton.addEventListener('click', async () => {
+    const message = messageInput.value;
+
+    if (message.trim() === '') {
+        return; // Si el mensaje está vacío, no hacemos nada
+    }
+
+    // Agregar el mensaje del usuario al chat
+    addMessageToChat(`${username}: ${message}`, 'user-message');
+
+    // Limpiar el input después de enviar el mensaje
+    messageInput.value = '';
+
+    // Crear el cuerpo de la solicitud
+    const requestBody = {
+        Item: localStorage.getItem('db'),
+        q: message
+    };
+
+    try {
+        // Enviar el mensaje al endpoint '/inference' usando fetch
+        const response = await fetch('http://127.0.0.1:8080/inference', {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify(requestBody)
+        });
+
+        // Verificar si la respuesta es exitosa
+        if (response.ok) {
+            const jsonResponse = await response.json();
+            const apiResponse = jsonResponse.res; // Suponiendo que el backend devuelve una respuesta en el campo "response"
+            const sql_query_generated = jsonResponse.sql_query_generated
+            // Agregar la respuesta de la API al chat
+            addMessageToChat(`Respuesta: ${JSON.stringify(apiResponse)}`, 'api-response');
+            addMessageToChat(`Consulta SQL generada: ${sql_query_generated}`, 'api-response');
+
+        } else {
+            addMessageToChat('Error al conectarse con la API', 'api-response');
+        }
+    } catch (error) {
+        console.error('Error:', error);
+        addMessageToChat('Error al enviar el mensaje', 'api-response');
+    }
+});
+
+// Mostrar el cuadro de diálogo del nombre de usuario al cargar la página
+showUsernameDialog();