Update main.py

main.py

@@ -38,6 +38,7 @@ class SQLExecutor:
     def query(sql_query: str):
         """
         Ejecuta una consulta SQL y devuelve los resultados.
+        Palabras clave, helados, anhelados, sabor de helados, jairo, edson,
         Cualquier consulta SQL que generes debe resumir tu solicitud en una sola consulta,
         ejemplo, si deseas saber la cantidad de insumos que tienes en tu almacén debes hacer 
         un conteo de los insumos, en el caso de produccion de las maquinas debes resumir de 
@@ -88,7 +89,23 @@ api_axel_victor = FunctionTool.from_defaults(fn=ApiPosAxel.academia_axel_victor)
 
 estructura_base_de_datos = ChatMessage(
     role=MessageRole.USER,
-    content="La base de datos tiene esta estructura SQLite de Anhelados: 1. almacen: ID_Almacen, ID_Insumo, Cantidad_Actual, Cantidad_Minima, Referencia: insumos (ID_Insumo), 2. clientes: ID_Proveedor, Nombre, Contacto, Direccion, 3. detalle_pedidos: ID_Pedido, ID_Insumo, Cantidad, Costo_Unitario, Referencias: pedidos_proveedor (ID_Pedido), insumos (ID_Insumo), 4. detalle_ventas: ID_Venta, ID_Producto, Cantidad, Precio_Unitario, Referencias: ventas (ID_Venta), productos (ID_Producto), 5. empleados: ID_Empleado, Nombre, Apellido, Sueldo, Fecha_Inicio, 6. gastos: ID_Gasto, Tipo, Monto, Fecha, 7. gastos_imprevistos: ID_Gasto_Imprevisto, Descripcion, Monto, Fecha,8. insumos: ID_Insumo, Descripcion, Tipo, Costo, Cantidad, 9. maquinas: ID_Maquina, Tipo, Capacidad, Consumo_Energetico, 10. pedidos_proveedor: ID_Pedido, ID_Proveedor, Fecha, Total, Referencia: clientes (ID_Proveedor), 11. produccion: ID_Produccion, ID_Maquina, Fecha, Cantidad_Producida, Referencia: maquinas (ID_Maquina), 12. productos: ID_Producto, Nombre, Precio, 13. registro_energetico: ID_Registro, ID_Maquina, Fecha, Consumo, Referencia: maquinas (ID_Maquina), 14. ventas: ID_Venta, Fecha, ID_Empleado, Total, Referencia: empleados (ID_Empleado)",
+    content="""
+    En caso se quiera hacer cualquiera de las funciones de un CRUD con Anhelados esta es la estructura de la base de datos SQLITE de Anhelados y debes usar sql_tool:
+    1. almacen: ID_Almacen, ID_Insumo, Cantidad_Actual, Cantidad_Minima, Referencia: insumos (ID_Insumo)
+    2. clientes: ID_Proveedor, Nombre, Contacto, Direccion
+    3. detalle_pedidos: ID_Pedido, ID_Insumo, Cantidad, Costo_Unitario, Referencias: pedidos_proveedor (ID_Pedido), insumos (ID_Insumo)
+    4. detalle_ventas: ID_Venta, ID_Producto, Cantidad, Precio_Unitario, Referencias: ventas (ID_Venta), productos (ID_Producto)
+    5. empleados: ID_Empleado, Nombre, Apellido, Sueldo, Fecha_Inicio
+    6. gastos: ID_Gasto, Tipo, Monto, Fecha
+    7. gastos_imprevistos: ID_Gasto_Imprevisto, Descripcion, Monto, Fecha
+    8. insumos: ID_Insumo, Descripcion, Tipo, Costo, Cantidad
+    9. maquinas: ID_Maquina, Tipo, Capacidad, Consumo_Energetico
+    10. pedidos_proveedor: ID_Pedido, ID_Proveedor, Fecha, Total, Referencia: clientes (ID_Proveedor)
+    11. produccion: ID_Produccion, ID_Maquina, Fecha, Cantidad_Producida, Referencia: maquinas (ID_Maquina)
+    12. productos: ID_Producto, Nombre, Precio
+    13. registro_energetico: ID_Registro, ID_Maquina, Fecha, Consumo, Referencia: maquinas (ID_Maquina)
+    14. ventas: ID_Venta, Fecha, ID_Empleado, Total, Referencia: empleados (ID_Empleado)
+    """,
     additional_kwargs={"tool": "sql_tool"},
 )
 mensaje_academia = ChatMessage(