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

iatoolkit/__init__.py

@@ -10,7 +10,6 @@ from .iatoolkit import IAToolkit, current_iatoolkit, create_app
 # for registering the client companies
 from .company_registry import register_company
 from .base_company import BaseCompany
-from iatoolkit.repositories.database_manager import DatabaseManager
 
 # --- Services ---
 from iatoolkit.services.query_service import QueryService
@@ -27,7 +26,6 @@ __all__ = [
     'current_iatoolkit',
     'register_company',
     'BaseCompany',
-    'DatabaseManager',
     'QueryService',
     'SqlService',
     'ExcelService',

iatoolkit/base_company.py

@@ -7,7 +7,6 @@
 from abc import ABC, abstractmethod
 from iatoolkit.repositories.profile_repo import ProfileRepo
 from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
-
 from iatoolkit.services.prompt_manager_service import PromptService
 from iatoolkit.repositories.models import Company, Function, PromptCategory
 from iatoolkit import IAToolkit
@@ -22,22 +21,14 @@ class BaseCompany(ABC):
         self.prompt_service: PromptService = injector.get(PromptService)
         self.company: Company | None = None
 
-    def _load_company_by_short_name(self, short_name: str) -> Company:
-        self.company = self.profile_repo.get_company_by_short_name(short_name)
-        return self.company
-
     def _create_company(self,
                         short_name: str,
                         name: str,
                         parameters: dict | None = None,
-                        branding: dict | None = None,
-                        onboarding_cards: dict | None = None,
                         ) -> Company:
         company_obj = Company(short_name=short_name,
                               name=name,
-                              parameters=parameters,
-                              branding=branding,
-                              onboarding_cards=onboarding_cards)
+                              parameters=parameters)
         self.company = self.profile_repo.create_company(company_obj)
         return self.company
 
@@ -77,17 +68,6 @@ class BaseCompany(ABC):
             **kwargs
         )
 
-
-    @abstractmethod
-    # initialize all the database tables  needed
-    def register_company(self):
-        raise NotImplementedError("La subclase debe implementar el método create_company()")
-
-    @abstractmethod
-    # get context specific for this company
-    def get_company_context(self, **kwargs) -> str:
-        raise NotImplementedError("La subclase debe implementar el método get_company_context()")
-
     @abstractmethod
     # get context specific for this company
     def get_user_info(self, user_identifier: str) -> dict:
@@ -98,11 +78,6 @@ class BaseCompany(ABC):
     def handle_request(self, tag: str, params: dict) -> dict:
         raise NotImplementedError("La subclase debe implementar el método handle_request()")
 
-    @abstractmethod
-    # get context specific for the query
-    def start_execution(self):
-        raise NotImplementedError("La subclase debe implementar el método start_execution()")
-
     @abstractmethod
     # get context specific for the query
     def get_metadata_from_filename(self, filename: str) -> dict:

iatoolkit/common/routes.py

@@ -5,7 +5,6 @@
 
 from flask import render_template, redirect, url_for,send_from_directory, current_app, abort
 from flask import jsonify
-from iatoolkit.views.history_api_view import HistoryApiView
 
 
 # this function register all the views
@@ -24,6 +23,10 @@ def register_views(injector, app):
     from iatoolkit.views.file_store_api_view import FileStoreApiView
     from iatoolkit.views.user_feedback_api_view import UserFeedbackApiView
     from iatoolkit.views.prompt_api_view import PromptApiView
+    from iatoolkit.views.history_api_view import HistoryApiView
+    from iatoolkit.views.help_content_api_view import HelpContentApiView
+    from iatoolkit.views.profile_api_view import UserLanguageApiView  # <-- Importa la nueva vista
+
     from iatoolkit.views.login_view import LoginView, FinalizeContextView
     from iatoolkit.views.external_login_view import ExternalLoginView, RedeemTokenApiView
     from iatoolkit.views.logout_api_view import LogoutApiView
@@ -53,6 +56,11 @@ def register_views(injector, app):
         view_func=FinalizeContextView.as_view('finalize_with_token')
     )
 
+    app.add_url_rule(
+        '/api/profile/language',
+        view_func=UserLanguageApiView.as_view('user_language_api')
+    )
+
     # logout
     app.add_url_rule('/<company_short_name>/api/logout',
                      view_func=LogoutApiView.as_view('logout'))
@@ -79,9 +87,10 @@ def register_views(injector, app):
     # open the promt directory
     app.add_url_rule('/<company_short_name>/api/prompts', view_func=PromptApiView.as_view('prompt'))
 
-    # feedback and history
+    # toolbar buttons
     app.add_url_rule('/<company_short_name>/api/feedback', view_func=UserFeedbackApiView.as_view('feedback'))
     app.add_url_rule('/<company_short_name>/api/history', view_func=HistoryApiView.as_view('history'))
+    app.add_url_rule('/<company_short_name>/api/help-content', view_func=HelpContentApiView.as_view('help-content'))
 
     # tasks management endpoints: create task, and review answer
     app.add_url_rule('/tasks', view_func=TaskApiView.as_view('tasks'))

iatoolkit/common/session_manager.py

@@ -3,6 +3,8 @@
 #
 # IAToolkit is open source software.
 
+# This is the Flask connected session manager for IAToolkit
+
 from flask import session
 
 class SessionManager:

iatoolkit/common/util.py

@@ -81,6 +81,23 @@ class Utility:
                                f'No se pudo renderizar el template desde el string, error: {str(e)}') from e
 
 
+    def get_company_template(self, company_short_name: str, template_name: str) -> str:
+        # 1. get the path to the company specific template
+        template_path = os.path.join(os.getcwd(), f'companies/{company_short_name}/templates/{template_name}')
+        if not os.path.exists(template_path):
+            return None
+
+        # 2. read the file
+        try:
+            with open(template_path, 'r') as f:
+                template_string = f.read()
+
+            return template_string
+        except Exception as e:
+            logging.exception(e)
+            return None
+
+
     def serialize(self, obj):
         if isinstance(obj, datetime) or isinstance(obj, date):
             return obj.isoformat()

iatoolkit/company_registry.py

@@ -31,10 +31,9 @@ class CompanyRegistry:
 
                     # save the created instance in the registry
                     self._company_instances[company_key] = company_instance
-                    logging.info(f"company '{company_key}' instantiated")
 
                 except Exception as e:
-                    logging.error(f"Error instanciando empresa {company_key}: {e}")
+                    logging.error(f"Error while creating company instance for {company_key}: {e}")
                     logging.exception(e)
                     raise
 

iatoolkit/iatoolkit.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-from flask import Flask, url_for
+from flask import Flask, url_for, get_flashed_messages
 from flask_session import Session
 from flask_injector import FlaskInjector
 from flask_bcrypt import Bcrypt
@@ -19,7 +19,7 @@ from werkzeug.middleware.proxy_fix import ProxyFix
 from injector import Binder, Injector, singleton
 from importlib.metadata import version as _pkg_version, PackageNotFoundError
 
-IATOOLKIT_VERSION = "0.63.1"
+IATOOLKIT_VERSION = "0.69.0"
 
 # global variable for the unique instance of IAToolkit
 _iatoolkit_instance: Optional['IAToolkit'] = None
@@ -96,6 +96,7 @@ class IAToolkit:
         self._setup_cors()
         self._setup_additional_services()
         self._setup_cli_commands()
+        self._setup_request_globals()
         self._setup_context_processors()
 
         # Step 8: define the download_dir for excel's
@@ -109,6 +110,22 @@ class IAToolkit:
         # get a value from the config dict or the environment variable
         return self.config.get(key, os.getenv(key, default))
 
+    def _setup_request_globals(self):
+        """
+        Configures functions to run before each request to set up
+        request-global variables, such as language.
+        """
+        injector = self._injector
+
+        @self.app.before_request
+        def set_request_language():
+            """
+            Determines and caches the language for the current request in g.lang.
+            """
+            from iatoolkit.services.language_service import LanguageService
+            language_service = injector.get(LanguageService)
+            language_service.get_current_language()
+
     def _setup_logging(self):
         # Lee el nivel de log desde una variable de entorno, con 'INFO' como valor por defecto.
         log_level_name = os.getenv('LOG_LEVEL', 'INFO').upper()
@@ -276,7 +293,6 @@ class IAToolkit:
         from iatoolkit.repositories.document_repo import DocumentRepo
         from iatoolkit.repositories.profile_repo import ProfileRepo
         from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
-
         from iatoolkit.repositories.vs_repo import VSRepo
         from iatoolkit.repositories.tasks_repo import TaskRepo
 
@@ -299,6 +315,9 @@ class IAToolkit:
         from iatoolkit.services.jwt_service import JWTService
         from iatoolkit.services.dispatcher_service import Dispatcher
         from iatoolkit.services.branding_service import BrandingService
+        from iatoolkit.services.i18n_service import I18nService
+        from iatoolkit.services.language_service import LanguageService
+        from iatoolkit.services.configuration_service import ConfigurationService
 
         binder.bind(QueryService, to=QueryService)
         binder.bind(TaskService, to=TaskService)
@@ -312,6 +331,9 @@ class IAToolkit:
         binder.bind(JWTService, to=JWTService)
         binder.bind(Dispatcher, to=Dispatcher)
         binder.bind(BrandingService, to=BrandingService)
+        binder.bind(I18nService, to=I18nService)
+        binder.bind(LanguageService, to=LanguageService)
+        binder.bind(ConfigurationService, to=ConfigurationService)
 
     def _bind_infrastructure(self, binder: Binder):
         from iatoolkit.infra.llm_client import llmClient
@@ -338,9 +360,9 @@ class IAToolkit:
         # instantiate all the registered companies
         get_company_registry().instantiate_companies(self._injector)
 
-        # use the dispatcher to start the execution of every company
+        # use the dispatcher to load the company config.yaml file and prepare the execution
         dispatcher = self._injector.get(Dispatcher)
-        dispatcher.start_execution()
+        dispatcher.load_company_configs()
 
     def _setup_cli_commands(self):
         from iatoolkit.cli_commands import register_core_commands
@@ -366,8 +388,18 @@ class IAToolkit:
         def inject_globals():
             from iatoolkit.common.session_manager import SessionManager
             from iatoolkit.services.profile_service import ProfileService
+            from iatoolkit.services.i18n_service import I18nService
 
+            # Get services from the injector
             profile_service = self._injector.get(ProfileService)
+            i18n_service = self._injector.get(I18nService)
+
+            # The 't' function wrapper no longer needs to determine the language itself.
+            # It will be automatically handled by the refactored I18nService.
+            def translate_for_template(key: str, **kwargs):
+                return i18n_service.t(key, **kwargs)
+
+            # Get user profile if a session exists
             user_profile = profile_service.get_current_session_info().get('profile', {})
 
             return {
@@ -379,9 +411,10 @@ class IAToolkit:
                 'user_is_local': user_profile.get('user_is_local'),
                 'user_email': user_profile.get('user_email'),
                 'iatoolkit_base_url': os.environ.get('IATOOLKIT_BASE_URL', ''),
+                'flashed_messages': get_flashed_messages(with_categories=True),
+                't': translate_for_template
             }
 
-
     def _get_default_static_folder(self) -> str:
         try:
             current_dir = os.path.dirname(os.path.abspath(__file__))  # .../src/iatoolkit

iatoolkit/locales/en.yaml

@@ -0,0 +1,167 @@
+# Language: English
+ui:
+  login_widget:
+    title: "Sign In"
+    welcome_message: "Enter your credentials or register to access Sample Company’s AI platform."
+    email_placeholder: "Email address"
+    password_placeholder: "Password"
+    login_button: "Login"
+    forgot_password_link: "Forgot your password?"
+    no_account_prompt: "Don't have an account?"
+    signup_link: "Sign Up"
+
+  signup:
+    title: "Create an Account"
+    subtitle: "Start your journey with us today."
+    first_name_label: "First Name"
+    last_name_label: "Last Name"
+    email_label: "Email Address"
+    password_label: "Password"
+    confirm_password_label: "Confirm Password"
+    signup_button: "Create Account"
+    already_have_account: "Already have an account?"
+    login_link: "Log In"
+    disclaimer: "🔒We value your privacy. Your data will be used exclusively for the operation of the platform."
+
+  forgot_password:
+    title: "Recover Password"
+    subtitle: "Enter your email address and we will send you a link to reset your password."
+    submit_button: "Send Recovery Link"
+    back_to_login: "Back to Login"
+
+  change_password:
+    title: "Create New Password"
+    subtitle: "You are changing the password for <strong>{email}</strong>."
+    temp_code_label: "Temporary Code"
+    temp_code_placeholder: "Check your email"
+    new_password_label: "New Password"
+    password_instructions: "Must contain at least 8 characters, an uppercase letter, a lowercase letter, a number, and a special character."
+    confirm_password_label: "Confirm New Password"
+    save_button: "Save Password"
+    back_to_home: "Back to home"
+
+  chat:
+    welcome_message: "Hello! How can I help you today?"
+    input_placeholder: "Type your query here..."
+    prompts_available: "Available prompts"
+
+  tooltips:
+    history: "History of my queries"
+    reload_context: "Force Context Reload"
+    feedback: "Your feedback is very important"
+    usage_guide: "Usage Guide"
+    onboarding: "How to ask better questions"
+    logout: "Log out"
+    use_prompt_assistant: "Use Prompt Assistant"
+    attach_files: "Attach files"
+    view_attached_files: "View attached files"
+    send: "Send"
+    stop: "Stop"
+
+  modals:
+    files_title: "Uploaded Files"
+    feedback_title: "Your Opinion is Important"
+    feedback_prompt: "How useful was the assistant's response?"
+    feedback_comment_label: "Your feedback helps us improve:"
+    feedback_comment_placeholder: "Write your opinion, suggestions, or comments here..."
+    history_title: "Query History"
+    history_table_date: "Date"
+    history_table_query: "Query"
+    loading_history: "Loading history..."
+    no_history_found: "No query history found."
+    help_title: "AI Assistant User Guide"
+
+  buttons:
+    cancel: "Close"
+    send: "Send"
+    stop: "Stop"
+
+errors:
+  company_not_found: "The company {company_short_name} does not exist."
+  timeout: "timeout expired."
+  auth:
+    invalid_password: "The provided password is incorrect."
+    user_not_found: "A user with that email address was not found."
+    invalid_or_expired_token: "Invalid or expired token."
+    session_creation_failed: "Could not create user session."
+    authentication_required: "Authentication required. No session cookie or API Key provided."
+    invalid_api_key: "Invalid or inactive API Key."
+    no_user_identifier_api: "No user_identifier provided for API call."
+  templates:
+    company_not_found: "Company not found."
+    home_template_not_found: "The home page template for the company '{company_name}' is not configured."
+    template_not_found: "Template not found: '{template_name}'."
+
+    processing_error: "An error occurred while processing the custom home page template: {error}"
+  general:
+    unexpected_error: "An unexpected error has occurred. Please contact support."
+    unsupported_language: "The selected language is not supported."
+  signup:
+    company_not_found: "The company {company_name} does not exist."
+    incorrect_password_for_existing_user: "The password for the user {email} is incorrect."
+    user_already_registered: "The user with email '{email}' already exists in this company."
+    password_mismatch: "The passwords do not match. Please try again."
+  change_password:
+    token_expired: "The password change link has expired. Please request a new one."
+    password_mismatch: "The passwords do not match. Please try again."
+    invalid_temp_code: "The temporary code is not valid. Please check it or request a new one."
+  forgot_password:
+    user_not_registered: "The user with email {email} is not registered."
+  verification:
+    token_expired: "The verification link has expired. Please contact support if you need a new one."
+    user_not_found: "The user you are trying to verify does not exist."
+  validation:
+    password_too_short: "Password must be at least 8 characters long."
+    password_no_uppercase: "Password must contain at least one uppercase letter."
+    password_no_lowercase: "Password must contain at least one lowercase letter."
+    password_no_digit: "Password must contain at least one number."
+    password_no_special_char: "Password must contain at least one special character."
+
+  services:
+    no_text_file: "The file is not text or the encoding is not UTF-8"
+    no_output_file: "Missing output file name"
+    no_data_for_excel: "Missing data or it is not a list of dictionaries"
+    no_download_directory: "Temporary directory for saving Excel files is not configured"
+    cannot_create_excel: "Could not create the Excel file"
+    invalid_filename: "Invalid filename"
+    file_not_exist: "File not found"
+    path_is_not_a_file: "The path does not correspond to a file"
+    file_validation_error: "Error validating file"
+    user_not_authorized: "user is not authorized for this company"
+    account_not_verified: "Your account has not been verified. Please check your email."
+    missing_response_id: "Can not found 'previous_response_id' for '{company_short_name}/{user_identifier}'. Reinit context"
+
+
+api_responses:
+  context_reloaded_success: "The context has been successfully reloaded."
+
+services:
+  mail_sent: "Email sent successfully."
+  start_query: "Hello, what can I help you with today?"
+
+
+flash_messages:
+  password_changed_success: "Your password has been successfully reset. You can now log in."
+  login_required: "Please log in to continue."
+  signup_success: "Registration successful. Please check your email to verify your account."
+  user_associated_success: "Existing user successfully associated with the new company."
+  account_verified_success: "Your account has been successfully verified. Welcome!"
+  forgot_password_success: "If your email is registered, you will receive a link to reset your password."
+
+# Keys specifically for JavaScript
+js_messages:
+  feedback_sent_success_title: "Feedback Sent"
+  feedback_sent_success_body: "Thank you for your feedback!"
+  feedback_sent_error: "Could not send feedback, please try again."
+  feedback_rating_error: "Please rate the assistant using the stars."
+  feedback_comment_error: "Please write your comment before sending."
+  context_reloaded: "Context has been reloaded."
+  error_loading_history: "An error occurred while loading the history."
+  request_aborted: "The response generation has been stopped."
+  processing_error: "An error occurred while processing the request."
+  server_comm_error: "Server communication error ({status}). Please try again later."
+  network_error: "A network error occurred. Please try again in a few moments."
+  unknown_server_error: "Unknown server error."
+  loading: "Loading..."
+  reload_init: "init reloading context in background..."
+  no_history_found: "No query history found."

iatoolkit/locales/es.yaml

@@ -0,0 +1,163 @@
+    # locales/es.yaml
+    ui:
+      login_widget:
+        title: "Iniciar Sesión"
+        welcome_message: "Ingresa tus credenciales o registrate para acceder a la plataforma IA de Sample Company."
+        email_placeholder: "Correo electrónico"
+        login_button: "Acceder"
+        forgot_password_link: "¿Olvidaste tu contraseña?"
+        no_account_prompt: "¿No tienes una cuenta?"
+        signup_link: "Regístrate"
+
+      signup:
+        title: "Crear una Cuenta"
+        subtitle: "Comienza tu viaje con nosotros hoy."
+        first_name_label: "Nombre"
+        last_name_label: "Apellido"
+        email_label: "Correo Electrónico"
+        password_label: "Contraseña"
+        confirm_password_label: "Confirmar Contraseña"
+        signup_button: "Crear Cuenta"
+        already_have_account: "¿Ya tienes una cuenta?"
+        login_link: "Inicia Sesión"
+        disclaimer: "🔒 Valoramos tu privacidad. Tus datos se usarán exclusivamente para el funcionamiento de la plataforma."
+
+      forgot_password:
+        title: "Recuperar Contraseña"
+        subtitle: "Ingresa tu correo electrónico y te enviaremos un enlace para restablecer tu contraseña."
+        submit_button: "Enviar Enlace de Recuperación"
+        back_to_login: "Volver a Iniciar Sesión"
+
+      change_password:
+        title: "Crear Nueva Contraseña"
+        subtitle: "Estás cambiando la contraseña para <strong>{email}</strong>."
+        temp_code_label: "Código Temporal"
+        temp_code_placeholder: "Revisa tu correo electrónico"
+        new_password_label: "Nueva Contraseña"
+        password_instructions: "Debe contener al menos 8 caracteres, mayúscula, minúscula, número y un carácter especial."
+        confirm_password_label: "Confirmar Nueva Contraseña"
+        save_button: "Guardar Contraseña"
+        back_to_home: "Volver al inicio"
+
+      chat:
+        welcome_message: "¡Hola! ¿En qué te puedo ayudar hoy?"
+        input_placeholder: "Escribe tu consulta aquí..."
+        prompts_available: "Prompts disponibles"
+
+      tooltips:
+        history: "Historial con mis consultas"
+        reload_context: "Forzar Recarga de Contexto"
+        feedback: "Tu feedback es muy importante"
+        usage_guide: "Guía de Uso"
+        onboarding: "Cómo preguntar mejor"
+        logout: "Cerrar sesión"
+        use_prompt_assistant: "Usar Asistente de Prompts"
+        attach_files: "Adjuntar archivos"
+        view_attached_files: "Ver archivos adjuntos"
+      modals:
+        files_title: "Archivos Cargados"
+        feedback_title: "Tu Opinión es Importante"
+        feedback_prompt: "¿Qué tan útil fue la respuesta del asistente?"
+        feedback_comment_label: "Tu comentario nos ayuda a mejorar:"
+        feedback_comment_placeholder: "Escribe aquí tu opinión, sugerencias o comentarios..."
+        history_title: "Historial de Consultas"
+        history_table_date: "Fecha"
+        history_table_query: "Consulta"
+        loading_history: "Cargando historial..."
+        no_history_found: "No se encontró historial de consultas."
+        help_title: "Guía de uso del Asistente IA"
+
+      buttons:
+        cancel: "Cerrar"
+        send: "Enviar"
+        stop: "Detener"
+
+    errors:
+      company_not_found: "La empresa {company_short_name} no existe."
+      timeout: "El tiempo de espera ha expirado."
+
+
+      auth:
+        invalid_password: "La contraseña proporcionada es incorrecta."
+        user_not_found: "No se encontró un usuario con ese correo."
+        invalid_or_expired_token: "Token inválido o expirado."
+        session_creation_failed: "No se pudo crear la sesión del usuario."
+        authentication_required: "Autenticación requerida. No se proporcionó cookie de sesión o clave de API."
+        invalid_api_key: "Clave de API inválida o inactiva."
+        no_user_identifier_api: "No se proporcionó user_identifier para la llamada a la API."
+      templates:
+        company_not_found: "Empresa no encontrada."
+        home_template_not_found: "La plantilla de la página de inicio para la empresa '{company_name}' no está configurada."
+        processing_error: "Error al procesar el template: {error}"
+        template_not_found: "No se encontro el template: '{template_name}'."
+
+      general:
+        unexpected_error: "Ha ocurrido un error inesperado: {error}."
+        unsupported_language: "El idioma seleccionado no es válido."
+      signup:
+        company_not_found: "La empresa {company_name} no existe."
+        incorrect_password_for_existing_user: "La contraseña para el usuario {email} es incorrecta."
+        user_already_registered: "El usuario con email '{email}' ya existe en esta empresa."
+        password_mismatch: "Las contraseñas no coinciden. Por favor, inténtalo de nuevo."
+      change_password:
+        token_expired: "El enlace de cambio de contraseña ha expirado. Por favor, solicita uno nuevo."
+        password_mismatch: "Las contraseñas no coinciden. Por favor, inténtalo nuevamente."
+        invalid_temp_code: "El código temporal no es válido. Por favor, verifica o solicita uno nuevo."
+      forgot_password:
+        user_not_registered: "El usuario con correo {email} no está registrado."
+      verification:
+        token_expired: "El enlace de verificación ha expirado. Por favor, contacta a soporte si necesitas uno nuevo."
+        user_not_found: "El usuario que intentas verificar no existe."
+      validation:
+        password_too_short: "La contraseña debe tener al menos 8 caracteres."
+        password_no_uppercase: "La contraseña debe tener al menos una letra mayúscula."
+        password_no_lowercase: "La contraseña debe tener al menos una letra minúscula."
+        password_no_digit: "La contraseña debe tener al menos un número."
+        password_no_special_char: "La contraseña debe tener al menos un carácter especial."
+
+      services:
+        no_text_file: "El archivo no es texto o la codificación no es UTF-8"
+        no_output_file: "falta el nombre del archivo de salida"
+        no_data_for_excel: "faltan los datos  o no es una lista de diccionarios"
+        no_download_directory: "no esta configurado el directorio temporal para guardar excels"
+        cannot_create_excel: "no se pudo crear el archivo excel"
+        invalid_filename: "Nombre de archivo inválido"
+        file_not_exist : "Archivo no encontrado"
+        path_is_not_a_file : "La ruta no corresponde a un archivo"
+        file_validation_error : "Error validando archivo"
+        user_not_authorized: "Usuario no esta autorizado para esta empresa"
+        account_not_verified: "Tu cuenta no ha sido verificada. Por favor, revisa tu correo."
+        missing_response_id: "No se encontró 'previous_response_id' para '{company_short_name}/{user_identifier}'. Reinicia el contexto."
+
+    api_responses:
+      context_reloaded_success: "El contexto se ha recargado con éxito."
+
+    services:
+      mail_sent: "mail enviado exitosamente."
+      start_query: "Hola, cual es tu pregunta?"
+
+    flash_messages:
+      password_changed_success: "Tu contraseña ha sido restablecida exitosamente. Ahora puedes iniciar sesión."
+      signup_success: "Registro exitoso. Por favor, revisa tu correo para verificar tu cuenta."
+      user_associated_success: "Usuario existente asociado exitosamente a la nueva empresa."
+      account_verified_success: "Tu cuenta ha sido verificada exitosamente. ¡Bienvenido!"
+      forgot_password_success: "Si tu correo está registrado, recibirás un enlace para restablecer tu contraseña."
+
+    # Claves específicas para JavaScript
+    js_messages:
+      feedback_sent_success_title: "Feedback Enviado"
+      feedback_sent_success_body: "¡Gracias por tu comentario!"
+      feedback_sent_error: "No se pudo enviar el feedback, por favor intente nuevamente."
+      feedback_rating_error: "Por favor, califica al asistente con las estrellas."
+      feedback_comment_error: "Por favor, escribe tu comentario antes de enviar."
+      context_reloaded: "El contexto ha sido recargado."
+      error_loading_history: "Ocurrió un error al cargar el historial."
+      request_aborted: "La generación de la respuesta ha sido detenida."
+      processing_error: "Ocurrió un error al procesar la solicitud."
+      server_comm_error: "Error de comunicación con el servidor ({status}). Por favor, intente de nuevo más tarde."
+      network_error: "Ocurrió un error de red. Por favor, inténtalo de nuevo en unos momentos."
+      unknown_server_error: "Error desconocido del servidor."
+      loading: "Cargando..."
+      reload_init: "Iniciando recarga de contexto en segundo plano..."
+      no_history_found: "No existe historial de consultas."
+

iatoolkit/repositories/database_manager.py

@@ -27,8 +27,8 @@ class DatabaseManager:
             self._engine = create_engine(
                 database_url,
                 echo=False,
-                pool_size=2,  # per worker
-                max_overflow=3,
+                pool_size=10,  # per worker
+                max_overflow=20,
                 pool_timeout=30,
                 pool_recycle=1800,
                 pool_pre_ping=True,
@@ -70,6 +70,11 @@ class DatabaseManager:
     def remove_session(self):
         self.scoped_session.remove()
 
+    def get_all_table_names(self) -> list[str]:
+        # Returns a list of all table names in the database
+        inspector = inspect(self._engine)
+        return inspector.get_table_names()
+
     def get_table_schema(self,
                          table_name: str,
                          schema_name: str | None = None,
@@ -77,7 +82,7 @@ class DatabaseManager:
         inspector = inspect(self._engine)
 
         if table_name not in inspector.get_table_names():
-            raise RuntimeError(f"La tabla '{table_name}' no existe en la BD.")
+            raise RuntimeError(f"Table '{table_name}' does not exist.")
 
         if exclude_columns is None:
             exclude_columns = []

iatoolkit/repositories/document_repo.py

@@ -22,7 +22,7 @@ class DocumentRepo:
     def get(self, company_id, filename: str ) -> Document:
         if not company_id or not filename:
             raise IAToolkitException(IAToolkitException.ErrorType.PARAM_NOT_FILLED,
-                               'Falta empresa o filename')
+                               'missing company_id or filename')
 
         return self.session.query(Document).filter_by(company_id=company_id, filename=filename).first()