iatoolkit 0.55.1__tar.gz → 0.55.3__tar.gz

{iatoolkit-0.55.1 → iatoolkit-0.55.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.55.1
+Version: 0.55.3
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT

{iatoolkit-0.55.1 → iatoolkit-0.55.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "iatoolkit"
-version = "0.55.1"
+version = "0.55.3"
 requires-python = ">=3.12"
 description = "IAToolkit"
 readme = "readme.md"

{iatoolkit-0.55.1 → iatoolkit-0.55.3}/src/iatoolkit/common/routes.py

@@ -111,8 +111,8 @@ def register_views(injector, app):
         except FileNotFoundError:
             abort(404)
 
-    # login testing (old home page)
-    app.add_url_rule('/login_test/<company_short_name>/<external_user_id>',
+    # login testing
+    app.add_url_rule('/<company_short_name>/login_test',
                      view_func=LoginSimulationView.as_view('login_test'))
 
     app.add_url_rule(

{iatoolkit-0.55.1 → iatoolkit-0.55.3}/src/iatoolkit/iatoolkit.py

@@ -19,7 +19,7 @@ from werkzeug.middleware.proxy_fix import ProxyFix
 from injector import Binder, singleton, Injector
 from importlib.metadata import version as _pkg_version, PackageNotFoundError
 
-IATOOLKIT_VERSION = "0.55.1"
+IATOOLKIT_VERSION = "0.55.3"
 
 # global variable for the unique instance of IAToolkit
 _iatoolkit_instance: Optional['IAToolkit'] = None
@@ -188,6 +188,15 @@ class IAToolkit:
         self.db_manager.create_all()
         logging.info("✅ Base de datos configurada correctamente")
 
+        @self.app.teardown_appcontext
+        def remove_session(exception=None):
+            """
+            Flask calls this after each request.
+            It ensures the SQLAlchemy session is properly closed
+            and the DB connection is returned to the pool.
+            """
+            self.db_manager.scoped_session.remove()
+
     def _setup_redis_sessions(self):
         redis_url = self._get_config_value('REDIS_URL')
         if not redis_url:

{iatoolkit-0.55.1 → iatoolkit-0.55.3}/src/iatoolkit/repositories/database_manager.py

@@ -21,8 +21,23 @@ class DatabaseManager:
         :param echo: Si True, habilita logs de SQL.
         """
         self.url = make_url(database_url)
-        self._engine = create_engine(database_url, echo=False)
-        self.SessionFactory = sessionmaker(bind=self._engine)
+        if database_url.startswith('sqlite'):       # for tests
+            self._engine = create_engine(database_url, echo=False)
+        else:
+            self._engine = create_engine(
+                database_url,
+                echo=False,
+                pool_size=2,  # per worker
+                max_overflow=3,
+                pool_timeout=30,
+                pool_recycle=1800,
+                pool_pre_ping=True,
+                future=True,
+            )
+        self.SessionFactory = sessionmaker(bind=self._engine,
+                                           autoflush=False,
+                                           autocommit=False,
+                                           expire_on_commit=False)
         self.scoped_session = scoped_session(self.SessionFactory)
 
         # REGISTRAR pgvector para cada nueva conexión solo en postgres

{iatoolkit-0.55.1 → iatoolkit-0.55.3}/src/iatoolkit/services/profile_service.py

@@ -74,20 +74,20 @@ class ProfileService:
         except Exception as e:
             return {'success': False, "message": str(e)}
 
-    def create_external_user_session(self, company: Company, external_user_id: str):
+    def create_external_user_session(self, company: Company, user_identifier: str):
         """
         Public method for views to create a web session for an external user.
         """
         # 1. Fetch the profile from the external system via Dispatcher.
         user_profile = self.dispatcher.get_user_info(
             company_name=company.short_name,
-            user_identifier=external_user_id
+            user_identifier=user_identifier
         )
 
         # 2. Call the session creation helper with external_user_id as user_identifier
         self.create_web_session(
             company=company,
-            user_identifier=external_user_id,
+            user_identifier=user_identifier,
             user_profile=user_profile)
 
     def create_web_session(self, company: Company, user_identifier: str, user_profile: dict):

{iatoolkit-0.55.1 → iatoolkit-0.55.3}/src/iatoolkit/services/query_service.py

@@ -114,10 +114,6 @@ class QueryService:
                                  self._has_valid_cached_context(company_short_name, user_identifier))
 
         if rebuild_is_needed:
-            logging.info(
-                f"Se necesita reconstrucción de contexto para {company_short_name}/{user_identifier}. Preparando...")
-
-
             # Guardar el contexto preparado y su versión para que `finalize_context_rebuild` los use.
             self.session_context.save_prepared_context(company_short_name, user_identifier, final_system_context,
                                                        current_version)

{iatoolkit-0.55.1 → iatoolkit-0.55.3}/src/iatoolkit/templates/index.html

@@ -1,6 +1,6 @@
 {% extends "base.html" %}
 
-{% block title %}IAToolkit - Acelerador de IA {% endblock %}
+{% block title %}IAToolkit{% endblock %}
 
 {% block content %}
 
@@ -119,32 +119,31 @@
 
     <!-- 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 class="container">
+                <div class="author-card p-4">
+                    <div class="row align-items-center g-4">
+                        <!-- Columna para la Imagen -->
+                        <div class="col-md-2 text-center">
+                            <img src="{{ url_for('static', filename='images/fernando.jpeg') }}" alt="Foto de Fernando Libedinsky" class="img-fluid rounded-circle" style="max-width: 120px;">
+                        </div>
+                        <!-- Columna para el Texto -->
+                        <div class="col-md-10">
+                            <div class="d-flex flex-wrap align-items-center mb-2">
+                                <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 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 empresas con la inteligencia artificial.
+                            </p>
                         </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>
+        </section>
 
 
     <!-- 4. Footer (sin cambios) -->

iatoolkit-0.55.3/src/iatoolkit/templates/login_simulation.html

@@ -0,0 +1,91 @@
+{% extends "base.html" %}
+
+{% block title %}Prueba de Login para {{ company_short_name }}{% endblock %}
+
+{% block content %}
+
+<div class="container-fluid">
+  <div class="row flex-fill mt-5 justify-content-center">
+    <!-- login desde sistema externo -->
+    <div class="col-12 col-lg-6">
+      <div class="border rounded p-4 p-md-5 shadow-sm bg-light">
+        {# El título ahora muestra dinámicamente el nombre de la empresa #}
+        <h3 class="text-muted fw-semibold text-start mb-3">
+          Login Externo para <span style="color:#0d6efd;">{{ company_short_name }}</span>
+        </h3>
+        <div class="text-center mb-4">
+          <p class="text-muted widget-intro-text">
+            Este formulario permite testear el acceso a IAToolkit desde un portal externo utilizando una api-key. El external user ID es el nombre del usuario ya autentificado en algún portal interno de la empresa.
+          </p>
+        </div>
+
+        {# El 'action' y 'method' son manejados por JS, pero el id es crucial #}
+        <form id="external-login">
+
+          <div class="mb-3">
+            <label for="external_user_id" class="form-label d-block">External user ID</label>
+            <input type="text" id="external_user_id" name="external_user_id" class="form-control" required>
+          </div>
+
+          <button type="submit" id="loginButton" class="btn btn-primary">
+            Iniciar Sesión
+          </button>
+        </form>
+      </div>
+    </div>
+  </div>
+</div>
+{% endblock %}
+
+{% block scripts %}
+<script>
+  document.addEventListener('DOMContentLoaded', function() {
+    const companyShortName = "{{ company_short_name }}";
+    const apiKey = "{{ api_key }}";
+    const loginForm = document.getElementById('external-login');
+    const userIdInput = document.getElementById('external_user_id');
+
+    loginForm.addEventListener('submit', function(event) {
+      event.preventDefault();
+      const userIdentifier = userIdInput.value.trim();
+      if (!userIdentifier) return;
+
+      // 1. Reemplazamos la página actual con un mensaje de carga.
+      document.body.innerHTML = '<div style="display:flex; align-items:center; justify-content:center; height:100vh; font-family:sans-serif;">Iniciando sesión, por favor espera...</div>';
+
+      const apiUrl = `/${companyShortName}/external_login`;
+
+      fetch(apiUrl, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+            'Authorization': `Bearer ${apiKey}`
+        },
+        body: JSON.stringify({
+            // Usamos 'userIdentifier' como el valor para el campo 'external_user_id'
+            user_identifier: userIdentifier
+        })
+      })
+      .then(async response => {
+          if (response.ok) {
+              return response.text();
+          } else {
+              const errorText = await response.text();
+              throw new Error(errorText || `Error ${response.status}`);
+          }
+      })
+      .then(htmlContent => {
+          // 2. Reemplazamos todo el documento con el nuevo HTML recibido.
+          // Esto asegura que todos los scripts y estilos del nuevo HTML se carguen correctamente.
+          document.open();
+          document.write(htmlContent);
+          document.close();
+      })
+      .catch(error => {
+          // Si hay un error, lo mostramos en la página.
+          document.body.innerHTML = `<div style="padding:20px; font-family:monospace; color:red;"><h2>Error</h2><pre>${error.message}</pre></div>`;
+      });
+    });
+  });
+</script>
+{% endblock %}

{iatoolkit-0.55.1 → iatoolkit-0.55.3}/src/iatoolkit/views/external_login_view.py

@@ -43,16 +43,16 @@ class ExternalLoginView(BaseLoginView):
 
     def post(self, company_short_name: str):
         data = request.get_json()
-        if not data or 'external_user_id' not in data:
-            return jsonify({"error": "Falta external_user_id"}), 400
+        if not data or 'user_identifier' not in data:
+            return jsonify({"error": "Falta user_identifier"}), 400
 
         company = self.profile_service.get_company_by_short_name(company_short_name)
         if not company:
             return jsonify({"error": "Empresa no encontrada"}), 404
 
-        external_user_id = data['external_user_id']
-        if not external_user_id:
-            return jsonify({"error": "missing external_user_id"}), 404
+        user_identifier = data.get('user_identifier')
+        if not user_identifier:
+            return jsonify({"error": "missing user_identifier"}), 404
 
         # 1. Authenticate the API call.
         iaut = self.iauthentication.verify()
@@ -60,11 +60,11 @@ class ExternalLoginView(BaseLoginView):
             return jsonify(iaut), 401
 
         # 2. Create the external user session.
-        self.profile_service.create_external_user_session(company, external_user_id)
+        self.profile_service.create_external_user_session(company, user_identifier)
 
         # 3. Delegate the path decision to the centralized logic.
         try:
-            return self._handle_login_path(company_short_name, external_user_id, company)
+            return self._handle_login_path(company_short_name, user_identifier, company)
         except Exception as e:
-            logging.exception(f"Error processing external login path for {company_short_name}/{external_user_id}: {e}")
+            logging.exception(f"Error processing external login path for {company_short_name}/{user_identifier}: {e}")
             return jsonify({"error": f"Internal server error while starting chat. {str(e)}"}), 500

iatoolkit-0.55.3/src/iatoolkit/views/login_simulation_view.py

@@ -0,0 +1,27 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import render_template, request
+from injector import inject
+from iatoolkit.services.profile_service import ProfileService
+import os
+
+
+class LoginSimulationView(MethodView):
+    @inject
+    def __init__(self,
+                 profile_service: ProfileService):
+        self.profile_service = profile_service
+
+    def get(self, company_short_name: str = None):
+
+        # Esta API_KEY para el login
+        api_key_for_login = os.getenv("IATOOLKIT_API_KEY", "tu_api_key_por_defecto_o_error")
+
+        return render_template('login_simulation.html',
+                               company_short_name=company_short_name,
+                               api_key=api_key_for_login
+                               )

{iatoolkit-0.55.1 → iatoolkit-0.55.3}/src/iatoolkit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.55.1
+Version: 0.55.3
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT

{iatoolkit-0.55.1 → iatoolkit-0.55.3}/src/iatoolkit.egg-info/SOURCES.txt

@@ -62,15 +62,7 @@ src/iatoolkit/services/sql_service.py
 src/iatoolkit/services/tasks_service.py
 src/iatoolkit/services/user_feedback_service.py
 src/iatoolkit/services/user_session_context_service.py
-src/iatoolkit/static/images/arrow_up.png
-src/iatoolkit/static/images/diagrama_iatoolkit.jpg
-src/iatoolkit/static/images/logo_clinica.png
-src/iatoolkit/static/images/logo_iatoolkit.png
-src/iatoolkit/static/images/logo_maxxa.png
-src/iatoolkit/static/images/logo_notaria.png
-src/iatoolkit/static/images/logo_tarjeta.png
-src/iatoolkit/static/images/logo_umayor.png
-src/iatoolkit/static/images/upload.png
+src/iatoolkit/static/images/fernando.jpeg
 src/iatoolkit/static/js/chat_feedback.js
 src/iatoolkit/static/js/chat_filepond.js
 src/iatoolkit/static/js/chat_history.js
@@ -97,7 +89,7 @@ src/iatoolkit/templates/error.html
 src/iatoolkit/templates/forgot_password.html
 src/iatoolkit/templates/header.html
 src/iatoolkit/templates/index.html
-src/iatoolkit/templates/login_test.html
+src/iatoolkit/templates/login_simulation.html
 src/iatoolkit/templates/onboarding_shell.html
 src/iatoolkit/templates/signup.html
 src/iatoolkit/templates/test.html

iatoolkit-0.55.1/src/iatoolkit/templates/login_test.html

@@ -1,118 +0,0 @@
-{% extends "base.html" %}
-
-{% block title %}Prueba de Login para {{ branding.name }}{% endblock %}
-
-{% block styles %}
-{# Este bloque asegura que los colores de la marca estén disponibles como variables CSS #}
-{% if branding and branding.css_variables %}
-<style>
-  {{ branding.css_variables|safe }}
-  /* Usa la variable de CSS para el color primario del botón */
-  #initiateJwtChatButton {
-    background-color: var(--brand-primary-color, #0d6efd);
-    border-color: var(--brand-primary-color, #0d6efd);
-  }
-</style>
-{% endif %}
-{% endblock %}
-
-{% block content %}
-<div class="container-fluid">
-  <div class="row flex-fill mt-5 justify-content-center">
-    <!-- login desde sistema externo -->
-    <div class="col-12 col-lg-6">
-      <div class="border rounded p-4 p-md-5 shadow-sm bg-light">
-        {# El título ahora muestra dinámicamente el nombre de la empresa #}
-        <h3 class="text-muted fw-semibold text-start mb-3">
-          Login Externo para <span style="color: var(--brand-primary-color, #0d6efd);">{{ branding.name }}</span>
-        </h3>
-        <div class="text-center mb-4">
-          <p class="text-muted widget-intro-text">
-            Este formulario permite testear el acceso a IAToolkit desde un portal externo utilizando una api-key. El external user ID es el nombre del usuario ya autentificado en algún portal interno de la empresa.
-          </p>
-        </div>
-
-        {# El 'action' y 'method' son manejados por JS, pero el id es crucial #}
-        <form id="jwt-form">
-
-          <div class="mb-3">
-            <label for="external_user_id" class="form-label d-block">External user ID</label>
-            <input type="text" id="external_user_id" name="external_user_id" class="form-control" required>
-          </div>
-
-          <button type="submit" id="initiateJwtChatButton" class="btn btn-primary">
-            <span class="spinner-border spinner-border-sm d-none" role="status" aria-hidden="true"></span>
-            Iniciar Sesión
-          </button>
-        </form>
-      </div>
-    </div>
-  </div>
-</div>
-{% endblock %}
-
-{% block scripts %}
-<script>
-  document.addEventListener('DOMContentLoaded', function() {
-    const companyShortName = "{{ company_short_name }}";
-    const apiKey = "{{ api_key }}";
-    const loginForm = document.getElementById('jwt-form');
-    const userIdInput = document.getElementById('external_user_id');
-    const submitButton = document.getElementById('initiateJwtChatButton');
-    const spinner = submitButton.querySelector('.spinner-border');
-
-    loginForm.addEventListener('submit', function(event) {
-      event.preventDefault();
-      const userIdentifier = userIdInput.value.trim();
-      if (!userIdentifier) return;
-
-      // Deshabilitar botón y mostrar spinner
-      submitButton.disabled = true;
-      spinner.classList.remove('d-none');
-
-      const newWindow = window.open('', '_blank');
-      const apiUrl = `/${companyShortName}/external_login`;
-
-      fetch(apiUrl, {
-        method: 'POST',
-        headers: {
-                  'Content-Type': 'application/json',
-                  'Authorization': `Bearer ${apiKey}`
-              },
-        body: JSON.stringify({ external_user_id: userIdentifier })
-      })
-      .then(response => response.ok ? response.text() : response.text().then(text => { throw new Error(text) }))
-      .then(html => {
-          newWindow.document.documentElement.innerHTML = html;
-                  // Buscamos todos los scripts en el HTML inyectado y los re-ejecutamos.
-        const scripts = newWindow.document.querySelectorAll('script');
-        scripts.forEach(oldScript => {
-          const newScript = newWindow.document.createElement('script');
-
-          // Copiamos los atributos (como 'src' para archivos externos)
-          Array.from(oldScript.attributes).forEach(attr => {
-            newScript.setAttribute(attr.name, attr.value);
-          });
-
-          // Copiamos el contenido para scripts inline
-          newScript.appendChild(newWindow.document.createTextNode(oldScript.innerHTML));
-
-          // Reemplazamos el script viejo por el nuevo para que se ejecute.
-          oldScript.parentNode.replaceChild(newScript, oldScript);
-        });
-
-      })
-      .catch(error => {
-        console.error("Falló la prueba de login externo:", error);
-        const errorMessage = error.message || "Error desconocido.";
-        newWindow.document.body.innerHTML = `<div style="padding: 20px; font-family: monospace;"><h2>Error</h2><pre>${errorMessage}</pre></div>`;
-      })
-      .finally(() => {
-        // Volver a habilitar el botón y ocultar el spinner
-        submitButton.disabled = false;
-        spinner.classList.add('d-none');
-      });
-    });
-  });
-</script>
-{% endblock %}

iatoolkit-0.55.1/src/iatoolkit/views/login_simulation_view.py

@@ -1,60 +0,0 @@
-import requests
-import json
-import os
-from flask.views import MethodView
-from flask import Response, abort, request, make_response
-
-
-class LoginSimulationView(MethodView):
-    """
-    Simula un portal externo que llama a la API de IAToolkit de servidor a servidor,
-    replicando el flujo real de `dispatch_request`.
-    Para usarlo, visita /login_test/<company_short_name>/<external_user_id>
-    """
-
-    def get(self, company_short_name: str, external_user_id: str):
-        api_key = os.getenv("IATOOLKIT_API_KEY")
-        base_url = request.host_url.rstrip('/')
-
-        if not api_key:
-            abort(500, "Error: IATOOLKIT_API_KEY no está configurada en el servidor de prueba.")
-        if not external_user_id:
-            abort(400, "Error: Debes proporcionar un external_user_id en la URL.")
-
-        target_url = f"{base_url}/{company_short_name}/external_login"
-
-        # --- INICIO DE LA CORRECCIÓN ---
-        # Usamos el formato de header 'Authorization: Bearer' como solicitaste.
-        headers = {
-            'Content-Type': 'application/json',
-            'Authorization': f'Bearer {api_key}'
-        }
-        # --- FIN DE LA CORRECCIÓN ---
-
-        payload = {'external_user_id': external_user_id}
-
-        try:
-            # Llamada POST interna. 'stream=True' es importante para manejar la respuesta.
-            internal_response = requests.post(target_url, headers=headers, data=json.dumps(payload), timeout=120,
-                                              stream=True)
-            internal_response.raise_for_status()
-
-            # Creamos una nueva Response de Flask para el navegador del usuario.
-            user_response = Response(
-                internal_response.iter_content(chunk_size=1024),
-                status=internal_response.status_code
-            )
-
-            # Copiamos TODAS las cabeceras de la respuesta interna a la respuesta final,
-            # incluyendo 'Content-Type' y, crucialmente, 'Set-Cookie'.
-            for key, value in internal_response.headers.items():
-                if key.lower() not in ['content-encoding', 'content-length', 'transfer-encoding', 'connection']:
-                    user_response.headers[key] = value
-
-            return user_response
-
-        except requests.exceptions.HTTPError as e:
-            error_text = f"Error en la llamada interna a la API: {e.response.status_code}. Respuesta: {e.response.text}"
-            return Response(error_text, status=e.response.status_code, mimetype='text/plain')
-        except requests.exceptions.RequestException as e:
-            return Response(f'Error de conexión con el servicio de IA: {str(e)}', status=502, mimetype='text/plain')