iatoolkit 0.22.1__py3-none-any.whl → 0.50.2__py3-none-any.whl

iatoolkit/services/user_session_context_service.py

@@ -6,80 +6,138 @@
 from iatoolkit.infra.redis_session_manager import RedisSessionManager
 from typing import List, Dict, Optional
 import json
+import logging
 
 
 class UserSessionContextService:
     """
-    Gestiona el contexto de la sesión del usuario, incluyendo el historial
-    de conversación con el LLM y datos de la sesión del usuario.
-
-    Usa RedisSessionManager para persistencia directa en Redis.
+    Gestiona el contexto de la sesión del usuario usando un único Hash de Redis por sesión.
+    Esto mejora la atomicidad y la eficiencia.
     """
 
-    def _get_llm_history_key(self, company_short_name: str, user_identifier: str) -> str:
-        user_identifier = (user_identifier or "").strip()
-        if not user_identifier:
-            return None
-        return f"llm_history:{company_short_name}/{user_identifier}"
-
-    def _get_user_data_key(self, company_short_name: str, user_identifier: str) -> str:
+    def _get_session_key(self, company_short_name: str, user_identifier: str) -> Optional[str]:
+        """Devuelve la clave única de Redis para el Hash de sesión del usuario."""
         user_identifier = (user_identifier or "").strip()
-        if not user_identifier:
+        if not company_short_name or not user_identifier:
             return None
-        return f"user_data:{company_short_name}/{user_identifier}"
+        return f"session:{company_short_name}/{user_identifier}"
 
     def clear_all_context(self, company_short_name: str, user_identifier: str):
-        """Limpia todo el contexto de sesión para un usuario."""
-        self.clear_llm_history(company_short_name, user_identifier)
-        self.clear_user_session_data(company_short_name, user_identifier)
+        """Limpia el contexto de sesión para un usuario de forma atómica."""
+        session_key = self._get_session_key(company_short_name, user_identifier)
+        if session_key:
+            # RedisSessionManager.remove(session_key)
+            # 'profile_data' should not be deleted
+            RedisSessionManager.hdel(session_key, 'context_version')
+            RedisSessionManager.hdel(session_key, 'context_history')
+            RedisSessionManager.hdel(session_key, 'last_response_id')
 
     def clear_llm_history(self, company_short_name: str, user_identifier: str):
-        history_key = self._get_llm_history_key(company_short_name, user_identifier)
-        if history_key:
-            RedisSessionManager.remove(history_key)
-
-    def get_last_response_id(self, company_short_name: str, user_identifier: str) -> str:
-        history_key = self._get_llm_history_key(company_short_name, user_identifier)
-        if not history_key:
+        """Limpia solo los campos relacionados con el historial del LLM (ID y chat)."""
+        session_key = self._get_session_key(company_short_name, user_identifier)
+        if session_key:
+            RedisSessionManager.hdel(session_key, 'last_response_id', 'context_history')
+
+    def get_last_response_id(self, company_short_name: str, user_identifier: str) -> Optional[str]:
+        session_key = self._get_session_key(company_short_name, user_identifier)
+        if not session_key:
             return None
-
-        return RedisSessionManager.get(history_key, '')
+        return RedisSessionManager.hget(session_key, 'last_response_id')
 
     def save_last_response_id(self, company_short_name: str, user_identifier: str, response_id: str):
-        user_identifier = (user_identifier or "").strip()
-        history_key = self._get_llm_history_key(company_short_name, user_identifier)
-        if not history_key or not user_identifier:
-            return
-
-        RedisSessionManager.set(history_key, response_id)
+        session_key = self._get_session_key(company_short_name, user_identifier)
+        if session_key:
+            RedisSessionManager.hset(session_key, 'last_response_id', response_id)
 
     def save_context_history(self, company_short_name: str, user_identifier: str, context_history: List[Dict]):
-        history_key = f"chat_history:{company_short_name}/{user_identifier}"
-        if not history_key:
-            return
-        RedisSessionManager.set(history_key, json.dumps(context_history))
+        session_key = self._get_session_key(company_short_name, user_identifier)
+        if session_key:
+            try:
+                history_json = json.dumps(context_history)
+                RedisSessionManager.hset(session_key, 'context_history', history_json)
+            except (TypeError, ValueError) as e:
+                logging.error(f"Error al serializar context_history para {session_key}: {e}")
 
     def get_context_history(self, company_short_name: str, user_identifier: str) -> Optional[List[Dict]]:
-        history_key = f"chat_history:{company_short_name}/{user_identifier}"
-        return RedisSessionManager.get_json(history_key, {})
+        session_key = self._get_session_key(company_short_name, user_identifier)
+        if not session_key:
+            return None
 
-    def save_user_session_data(self, company_short_name: str, user_identifier: str, data: dict):
-        """Guarda un diccionario de datos en la sesión del usuario."""
-        user_identifier = (user_identifier or "").strip()
-        data_key = self._get_user_data_key(company_short_name, user_identifier)
-        if data_key:
-            RedisSessionManager.set_json(data_key, data)
-
-    def get_user_session_data(self, company_short_name: str, user_identifier: str) -> dict:
-        """Recupera el diccionario de datos de la sesión del usuario."""
-        data_key = self._get_user_data_key(company_short_name, user_identifier)
-        if not data_key:
+        history_json = RedisSessionManager.hget(session_key, 'context_history')
+        if not history_json:
+            return []
+
+        try:
+            return json.loads(history_json)
+        except json.JSONDecodeError:
+            return []
+
+    def save_profile_data(self, company_short_name: str, user_identifier: str, data: dict):
+        session_key = self._get_session_key(company_short_name, user_identifier)
+        if session_key:
+            try:
+                data_json = json.dumps(data)
+                RedisSessionManager.hset(session_key, 'profile_data', data_json)
+            except (TypeError, ValueError) as e:
+                logging.error(f"Error al serializar profile_data para {session_key}: {e}")
+
+    def get_profile_data(self, company_short_name: str, user_identifier: str) -> dict:
+        session_key = self._get_session_key(company_short_name, user_identifier)
+        if not session_key:
             return {}
 
-        return RedisSessionManager.get_json(data_key, {})
+        data_json = RedisSessionManager.hget(session_key, 'profile_data')
+        if not data_json:
+            return {}
 
-    def clear_user_session_data(self, company_short_name: str, user_identifier: str):
-        """Limpia los datos de la sesión del usuario."""
-        data_key = self._get_user_data_key(company_short_name, user_identifier)
-        if data_key:
-            RedisSessionManager.remove(data_key)
+        try:
+            return json.loads(data_json)
+        except json.JSONDecodeError:
+            return {}
+
+    def save_context_version(self, company_short_name: str, user_identifier: str, version: str):
+        session_key = self._get_session_key(company_short_name, user_identifier)
+        if session_key:
+            RedisSessionManager.hset(session_key, 'context_version', version)
+
+    def get_context_version(self, company_short_name: str, user_identifier: str) -> Optional[str]:
+        session_key = self._get_session_key(company_short_name, user_identifier)
+        if not session_key:
+            return None
+        return RedisSessionManager.hget(session_key, 'context_version')
+
+    def save_prepared_context(self, company_short_name: str, user_identifier: str, context: str, version: str):
+        """Guarda un contexto de sistema pre-renderizado y su versión, listos para ser enviados al LLM."""
+        session_key = self._get_session_key(company_short_name, user_identifier)
+        if session_key:
+            RedisSessionManager.hset(session_key, 'prepared_context', context)
+            RedisSessionManager.hset(session_key, 'prepared_context_version', version)
+
+    def get_and_clear_prepared_context(self, company_short_name: str, user_identifier: str) -> tuple:
+        """Obtiene el contexto preparado y su versión, y los elimina para asegurar que se usan una sola vez."""
+        session_key = self._get_session_key(company_short_name, user_identifier)
+        if not session_key:
+            return None, None
+
+        pipe = RedisSessionManager.pipeline()
+        pipe.hget(session_key, 'prepared_context')
+        pipe.hget(session_key, 'prepared_context_version')
+        pipe.hdel(session_key, 'prepared_context', 'prepared_context_version')
+        results = pipe.execute()
+
+        # results[0] es el contexto, results[1] es la versión
+        return (results[0], results[1]) if results else (None, None)
+
+    # --- Métodos de Bloqueo ---
+    def acquire_lock(self, lock_key: str, expire_seconds: int) -> bool:
+        """Intenta adquirir un lock. Devuelve True si se adquiere, False si no."""
+        # SET con NX (solo si no existe) y EX (expiración) es una operación atómica.
+        return RedisSessionManager.set(lock_key, "1", ex=expire_seconds, nx=True)
+
+    def release_lock(self, lock_key: str):
+        """Libera un lock."""
+        RedisSessionManager.remove(lock_key)
+
+    def is_locked(self, lock_key: str) -> bool:
+        """Verifica si un lock existe."""
+        return RedisSessionManager.exists(lock_key)

iatoolkit/static/js/chat_feedback.js

@@ -98,7 +98,7 @@ $(document).ready(function () {
 const sendFeedback = async function(message) {
     const activeStars = $('.star.active').length;
     const data = {
-        "external_user_id": window.externalUserId,
+        "user_identifier": window.user_identifier,
         "message": message,
         "rating": activeStars,
         "space": "spaces/AAQAupQldd4", // Este valor podría necesitar ser dinámico

iatoolkit/static/js/chat_history.js

@@ -40,11 +40,7 @@ $(document).ready(function () {
         historyContent.hide();
 
         try {
-            const data = {
-                external_user_id: window.externalUserId
-            };
-
-            const responseData = await callLLMAPI("/history", data, "POST");
+            const responseData = await callLLMAPI("/api/history", {}, "POST");
 
             if (responseData && responseData.history) {
                 // Guardar datos globalmente

iatoolkit/static/js/chat_main.js

@@ -173,7 +173,7 @@ const handleChatMessage = async function () {
             prompt_name: promptName,
             client_data: clientData,
             files: filesBase64.map(f => ({ filename: f.name, content: f.base64 })),
-            external_user_id: window.externalUserId
+            user_identifier: window.user_identifier
         };
 
         const responseData = await callLLMAPI("/llm_query", data, "POST");

iatoolkit/static/styles/landing_page.css

@@ -49,8 +49,35 @@ body {
 }
 
 .hero-title {
-    font-weight: 700;
-    color: #111827;
+  font-size: 3rem;
+  font-weight: 700;
+  background: linear-gradient(90deg, #4F46E5, #818CF8);
+  -webkit-background-clip: text;
+  -webkit-text-fill-color: transparent;
+  line-height: 1.1;
+  letter-spacing: -0.5px;
+}
+
+.hero-bullets {
+  list-style: none;
+  padding-left: 2rem;
+  margin-top: 1.5rem;
+}
+
+.hero-bullets li {
+  font-size: 1.15rem;
+  color: #4b5563;
+  margin-bottom: 0.8rem;
+  display: flex;
+  align-items: flex-start;
+  gap: 0.6rem;
+  line-height: 1.5;
+}
+
+.hero-bullets i {
+  color: #4F46E5;
+  font-size: 1.3rem;
+  margin-top: 0.2rem;
 }
 
 .value-proposition .hero-subtitle {
@@ -69,6 +96,39 @@ body {
     color: var(--brand-primary-color, #0d6efd);
 }
 
+.author-section {
+  background: #ffffff;
+}
+
+.author-card {
+  background: #f8f9fa;
+  border: 1px solid #e9ecef;
+  border-radius: 10px;
+}
+
+.author-avatar {
+  width: 64px;
+  height: 64px;
+  border-radius: 50%;
+  background: var(--brand-primary-color, #0d6efd);
+  color: var(--brand-text-on-primary, #fff);
+  display: flex; align-items: center; justify-content: center;
+  font-size: 2rem;
+}
+
+.author-bio {
+  color: #4b5563;
+  line-height: 1.6;
+}
+
+.author-linkedin {
+  color: var(--brand-primary-color, #0d6efd);
+  text-decoration: none;
+}
+.author-linkedin:hover {
+  text-decoration: underline;
+}
+
 /* --- Estilo para el texto introductorio del widget --- */
 .widget-intro-text {
     font-style: italic;

iatoolkit/system_prompts/query_main.prompt

@@ -1,24 +1,15 @@
 Eres un asistente que responde preguntas o ejecuta tareas según el contexto de la empresa.
 
 ### **Nombre de la empresa**
-## Nombre: {{company.name}}
+## Nombre: {{company}}, tambien se conoce como {{ company_short_name }}
 
 ### ** Información del usuario que esta consultando este chat**
-Contexto del usuario:
+- Identificador unico de usuario: {{ user_identifier }}
 - Nombre: {{ user_fullname }}
 - Email: {{ user_email }}
 - Tipo de usuario: {% if user_is_local %}Interno{% else %}Externo{% endif %}
-- Empresa: {{ company_name }}
+- Rol de usuario: {{ user_rol }}
 
-{% if user_name %}
-El usuario que consulta se identifica con la variable `user_name` y tiene el
-siguiente valor: {{ user_name }}.
-
-Este usuario tiene el rol: {{ user_rol }} en el producto {{ user_product }}.
-
-{% else %}
-El usuario que consulta se identifica como: {{ user_id }}
-{% endif %}
 
 ## Servicios de datos (function calls) disponibles en {{company.name}}:
 {% for service in service_list %}

iatoolkit/templates/_login_widget.html

@@ -1,12 +1,10 @@
 <!-- templates/_login_widget.html -->
 <div class="border rounded p-4 shadow-sm bg-light">
-    <!-- 1. Nuevo Encabezado de Marketing -->
+    <!-- 1. Encabezado de Marketing -->
     <div class="text-center mb-4">
-        <h4 class="form-title fw-bold">Acceso a la Plataforma</h4>
-        <!-- Párrafo modificado con nuevo texto y nueva clase CSS -->
         <p class="text-muted widget-intro-text">
-            Aca puedes mirar como funciona  IAToolkit con una empresa ficticia (Sample Company).
-            Pregunta por clientes, productos, ventas, etc.
+            Explora IAToolkit con una empresa de ejemplo.
+            Regístrate para vivir la experiencia completa de un usuario real que consulta a la plataforma.
         </p>
     </div>
 
@@ -15,16 +13,16 @@
           action="{{ url_for('chat', company_short_name=company_short_name) }}"
           method="post">
       <div class="mb-3">
-        <label for="email" class="form-label d-block text-muted">Correo Electrónico</label>
+        <label for="email" class="form-label d-block">Correo Electrónico</label>
         <input type="email" id="email" name="email" class="form-control" required>
       </div>
       <div class="mb-3">
-        <label for="password" class="form-label d-block text-muted">Contraseña</label>
+        <label for="password" class="form-label d-block">Contraseña</label>
         <input type="password" id="password" name="password"
                class="form-control" required>
       </div>
       <button type="submit" class="btn btn-branded-primary w-100 fw-bold py-2">
-          Ingresar
+          Probar demo ahora
       </button>
     </form>
 

iatoolkit/templates/chat.html

@@ -27,7 +27,7 @@
     <div class="d-flex align-items-center">
         <!-- 1. ID de Usuario -->
         <span style="{{ branding.secondary_text_style }}">
-            {{ external_user_id or user_email }}
+            {{ user_identifier }}
         </span>
 
         <!-- 2. Separador Vertical -->
@@ -38,13 +38,20 @@
            class="action-icon-style" title="Historial con mis consultas" style="color: {{ branding.header_text_color }};">
             <i class="bi bi-clock-history"></i>
         </a>
+        <a href="javascript:void(0);"
+           id="force-reload-button"
+           class="ms-3 action-icon-style"
+           title="Forzar Recarga de Contexto"
+           style="color: {{ branding.header_text_color }};">
+            <i class="bi bi-arrow-clockwise"></i>
+        </a>
         <a href="javascript:void(0);" id="send-feedback-button"
            class="ms-3 action-icon-style" title="Tu feedback es muy importante" style="color: {{ branding.header_text_color }};">
             <i class="bi bi-emoji-smile"></i>
         </a>
 
         <!-- Icono de cerrar sesión (al final) -->
-        {% if user_email %}
+        {% if user_is_local %}
             <a href="{{ url_for('logout', company_short_name=company_short_name, _external=True) }}"
                  class="ms-3 action-icon-style" title="Cerrar sesión" style="color: {{ branding.header_text_color }} !important;">
                 <i class="bi bi-box-arrow-right"></i>
@@ -163,8 +170,8 @@
 <script>
     // --- Global Configuration from Backend ---
     window.companyShortName = "{{ company_short_name }}";
+    window.user_identifier = "{{ user_identifier }}";
     window.iatoolkit_base_url = "{{ iatoolkit_base_url }}";
-    window.externalUserId = "{{ external_user_id }}";
     window.availablePrompts = {{ prompts.message | tojson }};
     window.sendButtonColor = "{{ branding.send_button_color }}";
 
@@ -180,8 +187,13 @@
 <script src="{{ url_for('static', filename='js/chat_feedback.js', _external=True) }}"></script>
 <script src="{{ url_for('static', filename='js/chat_main.js', _external=True) }}"></script>
 
+
+<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/toastr.js/latest/toastr.min.css">
+<script src="https://cdnjs.cloudflare.com/ajax/libs/jquery/3.6.0/jquery.min.js"></script>
+<script src="https://cdnjs.cloudflare.com/ajax/libs/toastr.js/latest/toastr.min.js"></script>
+
 <script>
-    document.addEventListener('DOMContentLoaded', function() {
+    document.addEventListener('DOMContentLoaded', function () {
         const promptCollapse = document.getElementById('prompt-assistant-collapse');
         if (promptCollapse) {
             promptCollapse.addEventListener('shown.bs.collapse', function () {
@@ -193,6 +205,7 @@
             });
         }
     });
+
     document.addEventListener('DOMContentLoaded', function () {
         // Inicializar todos los tooltips de la página
         var tooltipTriggerList = [].slice.call(document.querySelectorAll('[data-bs-toggle="tooltip"]'))
@@ -200,5 +213,66 @@
             return new bootstrap.Tooltip(tooltipTriggerEl)
         })
     });
+
+document.addEventListener('DOMContentLoaded', function() {
+    const reloadButton = document.getElementById('force-reload-button');
+    if (!reloadButton) return;
+
+    const originalIconClass = 'bi bi-arrow-clockwise';
+    const spinnerIconClass = 'spinner-border spinner-border-sm';
+
+    // Configuración de Toastr para que aparezca abajo a la derecha
+    toastr.options = { "positionClass": "toast-bottom-right", "preventDuplicates": true };
+
+    reloadButton.addEventListener('click', function(event) {
+        event.preventDefault();
+
+        if (reloadButton.disabled) return; // Prevenir doble clic
+
+        // 1. Deshabilitar y mostrar spinner
+        reloadButton.disabled = true;
+        const icon = reloadButton.querySelector('i');
+        icon.className = spinnerIconClass;
+        toastr.info('Iniciando recarga de contexto en segundo plano...');
+
+        // 2. Construir la URL dinámicamente
+        const company = window.companyShortName;
+        const reloadUrl = `/${company}/api/init_context_api`;
+
+        // 3. Hacer la llamada AJAX con POST
+        fetch(reloadUrl, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json'
+            },
+            // Envía un cuerpo vacío o los datos necesarios
+            body: JSON.stringify({})
+        })
+        .then(response => {
+            if (!response.ok) {
+                return response.json().then(err => {
+                    throw new Error(err.error_message || `Error del servidor: ${response.status}`);
+                });
+            }
+            return response.json();
+        })
+        .then(data => {
+            if (data.status === 'OK') {
+                toastr.success(data.message || 'Contexto recargado exitosamente.');
+            } else {
+                toastr.error(data.error_message || 'Ocurrió un error desconocido.');
+            }
+        })
+        .catch(error => {
+            console.error('Error durante la recarga del contexto:', error);
+            toastr.error(error.message || 'Error de red al intentar recargar.');
+        })
+        .finally(() => {
+            // 4. Restaurar el botón
+            reloadButton.disabled = false;
+            icon.className = originalIconClass;
+        });
+    });
+});
 </script>
 {% endblock %}

iatoolkit/templates/error.html

@@ -9,7 +9,7 @@
         <p class="mt-3 text-muted">
             {{ message }}
         </p>
-        <a href="{{ url_for('home') }}" class="btn btn-primary mt-4">Volver al Inicio</a>
+        <a href="{{ url_for('index', company_short_name=company_short_name) }}" class="btn btn-primary mt-4">Volver al Inicio</a>
     </div>
 </div>
 {% endblock %}

iatoolkit/templates/index.html

@@ -1,6 +1,6 @@
 {% extends "base.html" %}
 
-{% block title %}IAToolkit - Acelerador de IA para tu Empresa{% endblock %}
+{% block title %}IAToolkit - Acelerador de IA {% endblock %}
 
 {% block content %}
 
@@ -23,17 +23,14 @@
             <div class="row align-items-center">
                 <div class="col-lg-7">
                     <div class="value-proposition">
-                        <h1 class="hero-title gradient-text">Construye IA que entiende tu Negocio</h1>
-                        <p class="hero-subtitle">IAToolkit es la plataforma de código abierto que conecta de forma segura tus
-                            datos internos —bases de datos SQL, API's y documentos— con un un asistente de IA
-                            que entiende tu negocio desde el primer día.</p>
-                        <p class="hero-subtitle">
-                            Es un producto para ingenieros de software que quieren construir aplicaciones de IA
-                            a gran velocidad. Provee la infraestructura fundamental
-                            para que tú te puedas enfocar en conectar tus datos y crear tus prompts.</p>
-                        <p class="hero-subtitle">
-                            Te invito a descargarlo y probar el "hello, world" de IAToolkit.</p>
+                        <h1 class="hero-title gradient-text">IA que entiende tu <span class="text-dark">negocio</span> desde el primer día.</h1>
+
                     </div>
+                    <ul class="hero-bullets">
+                      <li><i class="bi bi-database"></i> Conecta tus bases de datos, APIs y documentos.</li>
+                      <li><i class="bi bi-chat-dots"></i> Construye un asistente que entiende el lenguaje de tu empresa.</li>
+                      <li><i class="bi bi-rocket-takeoff"></i> Despliega donde quieras, con tu marca y tus reglas.</li>
+                    </ul>
                 </div>
                 <div class="col-lg-5">
                     {% include '_login_widget.html' %}
@@ -120,6 +117,36 @@
         </div>
     </section>
 
+    <!-- 3.5 Sobre el Autor -->
+    <section class="author-section py-5">
+        <div class="container">
+            <div class="row align-items-center g-4">
+                <div class="col-md-1 d-none d-md-block text-center">
+                    <!-- Opcional: avatar (si no tienes imagen, deja el ícono) -->
+                    <div class="author-avatar mx-auto">
+                        <i class="bi bi-person-circle"></i>
+                    </div>
+                </div>
+                <div class="col-md-11">
+                    <div class="author-card p-4">
+                        <div class="d-flex flex-wrap align-items-center">
+                            <h5 class="mb-0 me-3" style="color: var(--brand-primary-color);">Sobre el autor</h5>
+                            <a href="https://www.linkedin.com/in/fernandolibedinsky" target="_blank" rel="noopener" class="author-linkedin ms-auto">
+                                <i class="bi bi-linkedin me-1"></i> LinkedIn
+                            </a>
+                        </div>
+                        <p class="author-bio mt-2 mb-0">
+Soy <strong>Fernando Libedinsky</strong>, ingeniero de software y creador de <strong>IAToolkit</strong>.
+<br>Tras una extensa trayectoria en el desarrollo de tecnología, sigo movido por la misma curiosidad que me llevó a programar por primera vez: aprender, explorar y construir cosas nuevas.
+<br>IAToolkit es la continuación de ese impulso, una plataforma creada para conectar rapidamente una empresas  con la IA.
+</p>
+                    </div>
+                </div>
+            </div>
+        </div>
+    </section>
+
+
     <!-- 4. Footer (sin cambios) -->
     <footer class="landing-footer">
         <div class="container">