iatoolkit 0.67.0__py3-none-any.whl → 0.69.0__py3-none-any.whl

iatoolkit/services/sql_service.py

@@ -4,16 +4,22 @@
 # IAToolkit is open source software.
 
 from iatoolkit.repositories.database_manager import DatabaseManager
-
 from iatoolkit.common.util import Utility
 from iatoolkit.services.i18n_service import I18nService
+from iatoolkit.common.exceptions import IAToolkitException
 from sqlalchemy import text
-from injector import inject
+from injector import inject, singleton
 import json
-from iatoolkit.common.exceptions import IAToolkitException
+import logging
 
 
+@singleton
 class SqlService:
+    """
+    Manages database connections and executes SQL statements.
+    It maintains a cache of named DatabaseManager instances to avoid reconnecting.
+    """
+
     @inject
     def __init__(self,
                  util: Utility,
@@ -21,44 +27,64 @@ class SqlService:
         self.util = util
         self.i18n_service = i18n_service
 
-    def exec_sql(self, db_manager: DatabaseManager, sql_statement: str) -> str:
+        # Cache for database connections
+        self._db_connections: dict[str, DatabaseManager] = {}
+
+    def register_database(self, db_name: str, db_uri: str):
+        """
+        Creates and caches a DatabaseManager instance for a given database name and URI.
+        If a database with the same name is already registered, it does nothing.
         """
-        Executes a raw SQL statement and returns the result as a JSON string.
+        if db_name in self._db_connections:
+            return
 
-        This method takes a DatabaseManager instance and a SQL query, executes it
-        against the database, and fetches all results. The results are converted
-        into a list of dictionaries, where each dictionary represents a row.
-        This list is then serialized to a JSON string.
-        If an exception occurs during execution, the transaction is rolled back,
-        and a custom IAToolkitException is raised.
+        logging.debug(f"Registering and creating connection for database: '{db_name}'")
+        db_manager = DatabaseManager(db_uri, register_pgvector=False)
+        self._db_connections[db_name] = db_manager
 
-        Args:
-            db_manager: The DatabaseManager instance to get the database session from.
-            sql_statement: The raw SQL statement to be executed.
+    def get_database_manager(self, db_name: str) -> DatabaseManager:
+        """
+        Retrieves a registered DatabaseManager instance from the cache.
+        """
+        try:
+            return self._db_connections[db_name]
+        except KeyError:
+            logging.error(f"Attempted to access unregistered database: '{db_name}'")
+            raise IAToolkitException(
+                IAToolkitException.ErrorType.DATABASE_ERROR,
+                f"Database '{db_name}' is not registered with the SqlService."
+            )
 
-        Returns:
-            A JSON string representing the list of rows returned by the query.
+    def exec_sql(self, db_name: str, sql_statement: str) -> str:
+        """
+        Executes a raw SQL statement against a registered database and returns the result as a JSON string.
         """
         try:
-            # here the SQL is executed
-            result = db_manager.get_session().execute(text(sql_statement))
+            # 1. Get the database manager from the cache
+            db_manager = self.get_database_manager(db_name)
 
-            # get the column names
+            # 2. Execute the SQL statement
+            result = db_manager.get_session().execute(text(sql_statement))
             cols = result.keys()
-
-            # convert rows to dict
             rows_context = [dict(zip(cols, row)) for row in result.fetchall()]
 
-            # Serialize to JSON with type convertion
+            # seialize the result
             sql_result_json = json.dumps(rows_context, default=self.util.serialize)
 
             return sql_result_json
+        except IAToolkitException:
+            # Re-raise exceptions from get_database_manager to preserve the specific error
+            raise
         except Exception as e:
-            db_manager.get_session().rollback()
+            # Attempt to rollback if a session was active
+            db_manager = self._db_connections.get(db_name)
+            if db_manager:
+                db_manager.get_session().rollback()
 
             error_message = str(e)
             if 'timed out' in str(e):
                 error_message = self.i18n_service.t('errors.timeout')
 
+            logging.error(f"Error executing SQL statement: {error_message}")
             raise IAToolkitException(IAToolkitException.ErrorType.DATABASE_ERROR,
                                      error_message) from e

iatoolkit/static/js/chat_onboarding_button.js

@@ -28,6 +28,7 @@
     const elIcon = qs(root, ui.icon);
     const elTitle = qs(root, ui.title);
     const elText = qs(root, ui.text);
+    const elExample = qs(root, ui.example);
     const elDots = qs(root, ui.dots);
     const elPrev = qs(root, ui.prev);
     const elNext = qs(root, ui.next);
@@ -43,6 +44,11 @@
       if (elIcon) elIcon.innerHTML = `<i class="${c.icon || 'bi bi-lightbulb'}"></i>`;
       if (elTitle) elTitle.textContent = c.title || '';
       if (elText) elText.innerHTML = c.text || '';
+      if (elExample && c.example) {
+          elExample.innerHTML = ('Ejemplo: ' + c.example) || '';
+      }
+      else
+          elExample.innerHTML = '';
       if (elDots) createDots(elDots, cards.length, idx);
     }
 

iatoolkit/static/styles/onboarding.css

@@ -36,6 +36,13 @@
   min-height: 60px;
 }
 
+.ob-example {
+  font-size: 0.95rem;
+  color: #444;
+  line-height: 1.5;
+  min-height: 60px;
+}
+
 /* Navegación */
 .ob-nav {
   display: flex;

iatoolkit/templates/change_password.html

@@ -1,6 +1,6 @@
 {% extends "base.html" %}
 
-{% block title %}{{ t('ui.change_password.title') }} - {{ company.name }}{% endblock %}
+{% block title %}{{ t('ui.change_password.title') }} - {{ branding.name }} {% endblock %}
 
 {% block styles %}
     <style>

iatoolkit/templates/chat.html

@@ -272,6 +272,7 @@ document.addEventListener('DOMContentLoaded', function () {
       icon: '#ob-icon',
       title: '#ob-title',
       text: '#ob-text',
+        example: '#ob-example',
       dots: '#ob-dots',
       prev: '#ob-prev',
       next: '#ob-next'

iatoolkit/templates/chat_modals.html

@@ -114,6 +114,7 @@
           <div id="ob-icon" class="ob-icon"><i class="bi bi-lightbulb"></i></div>
           <h6 id="ob-title" class="ob-title"></h6>
           <div id="ob-text" class="ob-text"></div>
+            <div id="ob-example" class="ob-example"></div>
           <div class="ob-nav">
             <button id="ob-prev" class="ob-btn" aria-label="Anterior"><i class="bi bi-chevron-left"></i></button>
             <div id="ob-dots" class="ob-dots"></div>

iatoolkit/templates/login_simulation.html

@@ -1,6 +1,6 @@
 {% extends "base.html" %}
 
-{% block title %}Prueba de Login para {{ company_short_name }}{% endblock %}
+{% block title %}Login Test - {{ branding.name }}{% endblock %}
 {% block styles %}
     <style>
         {{ branding.css_variables | safe }}

iatoolkit/templates/onboarding_shell.html

@@ -1,6 +1,6 @@
 {% extends "base.html" %}
 
-{% block title %}Iniciando {{ branding.name | default('IAToolkit') }} IA...{% endblock %}
+{% block title %}Iniciando {{ branding.name }} IA...{% endblock %}
 
 {% block styles %}
 {% endblock %}
@@ -45,6 +45,8 @@
         <div id="card-icon" class="ob-icon"><i class="fas fa-lightbulb"></i></div>
         <h3 id="card-title" class="ob-title">Título de la Tarjeta</h3>
         <p id="card-text" class="ob-text">Descripción de la tarjeta de capacitación.</p>
+        <p id="card-example" class="ob-example"></p>
+
         <div class="ob-nav">
           <button id="prev-card" class="ob-btn btn-branded-primary" aria-label="Anterior">
             <i class="fas fa-chevron-left"></i>
@@ -88,6 +90,7 @@
         icon: '#card-icon',
         title: '#card-title',
         text: '#card-text',
+          example: '#card-example',
         dots: '#progress-dots',
         prev: '#prev-card',
         next: '#next-card',

iatoolkit/views/base_login_view.py

@@ -11,7 +11,7 @@ from iatoolkit.services.profile_service import ProfileService
 from iatoolkit.services.auth_service import AuthService
 from iatoolkit.services.query_service import QueryService
 from iatoolkit.services.branding_service import BrandingService
-from iatoolkit.services.onboarding_service import OnboardingService
+from iatoolkit.services.configuration_service import ConfigurationService
 from iatoolkit.services.prompt_manager_service import PromptService
 from iatoolkit.services.i18n_service import I18nService
 from iatoolkit.common.util import Utility
@@ -31,7 +31,7 @@ class BaseLoginView(MethodView):
                  jwt_service: JWTService,
                  branding_service: BrandingService,
                  prompt_service: PromptService,
-                 onboarding_service: OnboardingService,
+                 config_service: ConfigurationService,
                  query_service: QueryService,
                  i18n_service: I18nService,
                  utility: Utility
@@ -41,14 +41,14 @@ class BaseLoginView(MethodView):
         self.jwt_service = jwt_service
         self.branding_service = branding_service
         self.prompt_service = prompt_service
-        self.onboarding_service = onboarding_service
+        self.config_service = config_service
         self.query_service = query_service
         self.i18n_service = i18n_service
         self.utility = utility
 
 
     def _handle_login_path(self,
-                           company: Company,
+                           company_short_name: str,
                            user_identifier: str,
                            target_url: str,
                            redeem_token: str = None):
@@ -56,13 +56,12 @@ class BaseLoginView(MethodView):
         Centralized logic to decide between the fast path and the slow path.
         """
         # --- Get the company branding and onboarding_cards
-        branding_data = self.branding_service.get_company_branding(company)
-        onboarding_cards = self.onboarding_service.get_onboarding_cards(company)
-        company_short_name = company.short_name
+        branding_data = self.branding_service.get_company_branding(company_short_name)
+        onboarding_cards = self.config_service.get_configuration(company_short_name, 'onboarding_cards')
 
         # this service decides is the context needs to be rebuilt or not
         prep_result = self.query_service.prepare_context(
-            company_short_name=company.short_name, user_identifier=user_identifier
+            company_short_name=company_short_name, user_identifier=user_identifier
         )
 
         if prep_result.get('rebuild_needed'):

iatoolkit/views/change_password_view.py

@@ -34,7 +34,7 @@ class ChangePasswordView(MethodView):
                 return render_template('error.html',
                                        message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
-            branding_data = self.branding_service.get_company_branding(company)
+            branding_data = self.branding_service.get_company_branding(company_short_name)
 
             try:
                 # Decodificar el token
@@ -66,7 +66,7 @@ class ChangePasswordView(MethodView):
             return render_template('error.html',
                                    message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
-        branding_data = self.branding_service.get_company_branding(company)
+        branding_data = self.branding_service.get_company_branding(company_short_name)
         try:
             # Decodificar el token
             email = self.serializer.loads(token, salt='password-reset', max_age=3600)
@@ -98,7 +98,6 @@ class ChangePasswordView(MethodView):
                     'change_password.html',
                     token=token,
                     company_short_name=company_short_name,
-                    company=company,
                     branding=branding_data,
                     form_data={"temp_code": temp_code,
                                "new_password": new_password,

iatoolkit/views/external_login_view.py

@@ -47,7 +47,7 @@ class ExternalLoginView(BaseLoginView):
 
         # 5. Delegate the path decision to the centralized logic.
         try:
-            return self._handle_login_path(company, user_identifier, target_url, redeem_token)
+            return self._handle_login_path(company_short_name, user_identifier, target_url, redeem_token)
         except Exception as 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/views/forgot_password_view.py

@@ -30,9 +30,8 @@ class ForgotPasswordView(MethodView):
             return render_template('error.html',
                                    message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
-        branding_data = self.branding_service.get_company_branding(company)
+        branding_data = self.branding_service.get_company_branding(company_short_name)
         return render_template('forgot_password.html',
-                               company=company,
                                company_short_name=company_short_name,
                                branding=branding_data
                                )
@@ -45,7 +44,7 @@ class ForgotPasswordView(MethodView):
                 return render_template('error.html',
                                        message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
-            branding_data = self.branding_service.get_company_branding(company)
+            branding_data = self.branding_service.get_company_branding(company_short_name)
             email = request.form.get('email')
 
             # create a safe token and url for it
@@ -59,7 +58,6 @@ class ForgotPasswordView(MethodView):
                 flash(response["error"], 'error')
                 return render_template(
                     'forgot_password.html',
-                    company=company,
                     company_short_name=company_short_name,
                     branding=branding_data,
                     form_data={"email": email}), 400

iatoolkit/views/help_content_api_view.py

@@ -37,7 +37,7 @@ class HelpContentApiView(MethodView):
             user_identifier = auth_result.get('user_identifier')
 
             # 2. Call the config service with the unified identifier.
-            response = self.config_service.get_company_content(
+            response = self.config_service.get_configuration(
                 company_short_name=company_short_name,
                 content_key='help_content'      # specific key for this service
             )

iatoolkit/views/home_view.py

@@ -31,7 +31,7 @@ class HomeView(MethodView):
                 return render_template('error.html',
                                        message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
-            branding_data = self.branding_service.get_company_branding(company)
+            branding_data = self.branding_service.get_company_branding(company_short_name)
             home_template = self.util.get_company_template(company_short_name, "home.html")
 
             # 2. Verificamos si el archivo de plantilla personalizado no existe.
@@ -47,7 +47,6 @@ class HomeView(MethodView):
             # 3. Si el archivo existe, intentamos leerlo y renderizarlo.
             return render_template_string(
                 home_template,
-                company=company,
                 company_short_name=company_short_name,
                 branding=branding_data,
             )

iatoolkit/views/login_simulation_view.py

@@ -29,7 +29,7 @@ class LoginSimulationView(MethodView):
                                    company_short_name=company_short_name,
                                    message="Empresa no encontrada"), 404
 
-        branding_data = self.branding_service.get_company_branding(company)
+        branding_data = self.branding_service.get_company_branding(company_short_name)
 
         return render_template('login_simulation.html',
                                branding=branding_data,

iatoolkit/views/login_view.py

@@ -12,7 +12,7 @@ from iatoolkit.services.jwt_service import JWTService
 from iatoolkit.services.query_service import QueryService
 from iatoolkit.services.prompt_manager_service import PromptService
 from iatoolkit.services.branding_service import BrandingService
-from iatoolkit.services.onboarding_service import OnboardingService
+from iatoolkit.services.configuration_service import ConfigurationService
 from iatoolkit.services.i18n_service import I18nService
 from iatoolkit.views.base_login_view import BaseLoginView
 import logging
@@ -29,7 +29,7 @@ class LoginView(BaseLoginView):
             return render_template('error.html',
                                    message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
-        branding_data = self.branding_service.get_company_branding(company)
+        branding_data = self.branding_service.get_company_branding(company_short_name)
         email = request.form.get('email')
         password = request.form.get('password')
 
@@ -61,7 +61,7 @@ class LoginView(BaseLoginView):
 
         # 2. Delegate the path decision to the centralized logic.
         try:
-            return self._handle_login_path(company, user_identifier, target_url)
+            return self._handle_login_path(company_short_name, user_identifier, target_url)
         except Exception as e:
             message = self.i18n_service.t('errors.templates.processing_error', error=str(e))
             return render_template(
@@ -83,7 +83,7 @@ class FinalizeContextView(MethodView):
                  query_service: QueryService,
                  prompt_service: PromptService,
                  branding_service: BrandingService,
-                 onboarding_service: OnboardingService,
+                 config_service: ConfigurationService,
                  jwt_service: JWTService,
                  i18n_service: I18nService
                  ):
@@ -92,7 +92,7 @@ class FinalizeContextView(MethodView):
         self.query_service = query_service
         self.prompt_service = prompt_service
         self.branding_service = branding_service
-        self.onboarding_service = onboarding_service
+        self.config_service = config_service
         self.i18n_service = i18n_service
 
     def get(self, company_short_name: str, token: str = None):
@@ -111,7 +111,7 @@ class FinalizeContextView(MethodView):
 
                 user_identifier = payload.get('user_identifier')
             else:
-                logging.warning("Fallo crítico: missing session information or auth token")
+                logging.error("missing session information or auth token")
                 return redirect(url_for('home', company_short_name=company_short_name))
 
             company = self.profile_service.get_company_by_short_name(company_short_name)
@@ -119,7 +119,7 @@ class FinalizeContextView(MethodView):
                 return render_template('error.html',
                             company_short_name=company_short_name,
                             message="Empresa no encontrada"), 404
-            branding_data = self.branding_service.get_company_branding(company)
+            branding_data = self.branding_service.get_company_branding(company_short_name)
 
             # 2. Finalize the context rebuild (the heavy task).
             self.query_service.finalize_context_rebuild(
@@ -129,7 +129,7 @@ class FinalizeContextView(MethodView):
 
             # 3. render the chat page.
             prompts = self.prompt_service.get_user_prompts(company_short_name)
-            onboarding_cards = self.onboarding_service.get_onboarding_cards(company)
+            onboarding_cards = self.config_service.get_configuration(company_short_name, 'onboarding_cards')
 
             # Get the entire 'js_messages' block in the correct language.
             js_translations = self.i18n_service.get_translation_block('js_messages')

iatoolkit/views/signup_view.py

@@ -32,9 +32,8 @@ class SignupView(MethodView):
             return render_template('error.html',
                                    message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
-        branding_data = self.branding_service.get_company_branding(company)
+        branding_data = self.branding_service.get_company_branding(company_short_name)
         return render_template('signup.html',
-                               company=company,
                                company_short_name=company_short_name,
                                branding=branding_data)
 
@@ -45,7 +44,7 @@ class SignupView(MethodView):
                 return render_template('error.html',
                                      message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
-            branding_data = self.branding_service.get_company_branding(company)
+            branding_data = self.branding_service.get_company_branding(company_short_name)
 
             first_name = request.form.get('first_name')
             last_name = request.form.get('last_name')
@@ -70,7 +69,6 @@ class SignupView(MethodView):
                 flash(response["error"], 'error')
                 return render_template(
                     'signup.html',
-                    company=company,
                     company_short_name=company_short_name,
                     branding=branding_data,
                     form_data={

iatoolkit/views/verify_user_view.py

@@ -32,14 +32,13 @@ class VerifyAccountView(MethodView):
                 return render_template('error.html',
                                        message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
-            branding_data = self.branding_service.get_company_branding(company)
+            branding_data = self.branding_service.get_company_branding(company_short_name)
             try:
                 # decode the token from the URL
                 email = self.serializer.loads(token, salt='email-confirm', max_age=3600*5)
             except SignatureExpired:
                 flash(self.i18n_service.t('errors.verification.token_expired'), 'error')
                 return render_template('signup.html',
-                                       company=company,
                                        company_short_name=company_short_name,
                                        branding=branding_data,
                                        token=token), 400
@@ -49,7 +48,6 @@ class VerifyAccountView(MethodView):
                 flash(response["error"], 'error')
                 return render_template(
                     'signup.html',
-                    company=company,
                     company_short_name=company_short_name,
                     branding=branding_data,
                     token=token), 400

{iatoolkit-0.67.0.dist-info → iatoolkit-0.69.0.dist-info}/METADATA

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