Agrego prompt eval

add_prompt_eval.py

@@ -0,0 +1,70 @@
+#!/usr/bin/env python3
+"""Script para agregar el campo prompt_eval a la tabla registro"""
+
+from connection import get_connection
+import sys
+
+def agregar_prompt_eval():
+    """Agrega la columna prompt_eval a la tabla registro"""
+    
+    conn = get_connection()
+    
+    if not conn:
+        print("❌ Error: No se pudo conectar a la base de datos")
+        sys.exit(1)
+    
+    try:
+        cursor = conn.cursor()
+        
+        print("🔧 Agregando campo prompt_eval a la tabla...")
+        print("-" * 60)
+        
+        # Verificar si la columna ya existe
+        cursor.execute("""
+            SELECT COLUMN_NAME 
+            FROM INFORMATION_SCHEMA.COLUMNS 
+            WHERE TABLE_NAME = 'registro' 
+            AND COLUMN_NAME = 'prompt_eval'
+        """)
+        
+        if cursor.fetchone():
+            print("⚠️  El campo 'prompt_eval' ya existe en la tabla")
+            cursor.close()
+            conn.close()
+            return
+        
+        # Agregar la columna después de 'prompt'
+        sql = """
+        ALTER TABLE `registro` 
+        ADD COLUMN `prompt_eval` VARCHAR(255) COMMENT 'Evaluación del prompt'
+        AFTER `prompt`
+        """
+        
+        cursor.execute(sql)
+        conn.commit()
+        
+        print("✅ Campo 'prompt_eval' agregado correctamente")
+        
+        # Verificar estructura
+        cursor.execute("DESCRIBE `registro`")
+        columnas = cursor.fetchall()
+        
+        print("\n📋 Estructura actualizada de la tabla:")
+        print("-" * 60)
+        for col in columnas:
+            print(f"  {col[0]:<20} {col[1]:<30} {col[2]}")
+        
+        cursor.close()
+        conn.close()
+        
+        print("\n✅ Tabla actualizada correctamente")
+        
+    except Exception as e:
+        print(f"❌ Error: {e}")
+        sys.exit(1)
+
+if __name__ == "__main__":
+    print("\n🔄 Actualizando estructura de BD...")
+    print("=" * 60)
+    agregar_prompt_eval()
+    print("=" * 60)

funciones.py

@@ -38,23 +38,24 @@ def crear_registro(datos: RegistroRequest) -> dict:
         fecha_registro = datos.fecha_registro or datetime.now()
         usos = datos.usos or 0
         prompt = datos.prompt
+        prompt_eval = datos.prompt_eval
         proveedor = datos.proveedor
         seed = datos.seed
         
-        logger.info(f"📊 Datos a insertar - UID: {uid}, Usuario: {display_name}, País: {pais}, Email: {correo}, Prompt: {prompt}, Proveedor: {proveedor}")
+        logger.info(f"📊 Datos a insertar - UID: {uid}, Usuario: {display_name}, País: {pais}, Email: {correo}, Prompt: {prompt}, Prompt Eval: {prompt_eval}, Proveedor: {proveedor}")
         
         # Insertar en la BD
         cursor = conn.cursor()
         
         sql = """
         INSERT INTO `registro` 
-        (uid, display_name, pais, correo, fecha_registro, usos, prompt, proveedor, seed)
-        VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s)
+        (uid, display_name, pais, correo, fecha_registro, usos, prompt, prompt_eval, proveedor, seed)
+        VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
         """
         
         valores = (
             uid, display_name, pais, correo, fecha_registro,
-            usos, prompt, proveedor, seed
+            usos, prompt, prompt_eval, proveedor, seed
         )
         
         cursor.execute(sql, valores)
@@ -70,7 +71,7 @@ def crear_registro(datos: RegistroRequest) -> dict:
         cursor.execute(
             """
             SELECT id, uid, display_name, pais, correo, fecha_registro, 
-                   usos, prompt, proveedor, seed, created_at 
+                   usos, prompt, prompt_eval, proveedor, seed, created_at 
             FROM `registro` 
             WHERE id = %s
             """,
@@ -94,9 +95,10 @@ def crear_registro(datos: RegistroRequest) -> dict:
                     "fecha_registro": registro[5],
                     "usos": registro[6],
                     "prompt": registro[7],
-                    "proveedor": registro[8],
-                    "seed": registro[9],
-                    "created_at": registro[10]
+                    "prompt_eval": registro[8],
+                    "proveedor": registro[9],
+                    "seed": registro[10],
+                    "created_at": registro[11]
                 }
             }
         else:
@@ -148,7 +150,7 @@ def obtener_registros(limit: int = 10, offset: int = 0) -> dict:
         cursor.execute(
             """
             SELECT id, uid, display_name, pais, correo, fecha_registro, 
-                   usos, prompt, proveedor, seed, created_at 
+                   usos, prompt, prompt_eval, proveedor, seed, created_at 
             FROM `registro` 
             ORDER BY created_at DESC 
             LIMIT %s OFFSET %s
@@ -173,9 +175,10 @@ def obtener_registros(limit: int = 10, offset: int = 0) -> dict:
                 "fecha_registro": reg[5],
                 "usos": reg[6],
                 "prompt": reg[7],
-                "proveedor": reg[8],
-                "seed": reg[9],
-                "created_at": reg[10]
+                "prompt_eval": reg[8],
+                "proveedor": reg[9],
+                "seed": reg[10],
+                "created_at": reg[11]
             })
         
         logger.info(f"📈 Retornando {len(datos)} registros")

imagen_log.sql

@@ -7,6 +7,7 @@ CREATE TABLE `registro` (
   `fecha_registro` datetime,
   `usos` integer,
   `prompt` varchar(255),
+  `prompt_eval` varchar(255),
   `proveedor` varchar(255),
   `seed` int,
   `created_at` timestamp

models.py

@@ -14,6 +14,7 @@ class RegistroRequest(BaseModel):
     fecha_registro: Optional[datetime] = Field(None, description="Fecha de registro")
     usos: Optional[int] = Field(0, ge=0, description="Cantidad de usos")
     prompt: Optional[str] = Field(None, max_length=255, description="Prompt utilizado")
+    prompt_eval: Optional[str] = Field(None, max_length=255, description="Evaluación del prompt")
     proveedor: Optional[str] = Field(None, max_length=255, description="Proveedor del servicio")
     seed: Optional[int] = Field(None, description="Seed para generación")
 
@@ -38,6 +39,7 @@ class RegistroResponse(BaseModel):
     fecha_registro: Optional[datetime]
     usos: int
     prompt: Optional[str]
+    prompt_eval: Optional[str]
     proveedor: Optional[str]
     seed: Optional[int]
     created_at: datetime