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

iatoolkit/services/prompt_manager_service.py

@@ -5,21 +5,25 @@
 
 from injector import inject
 from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
-
-import logging
+from iatoolkit.services.i18n_service import I18nService
 from iatoolkit.repositories.profile_repo import ProfileRepo
 from collections import defaultdict
 from iatoolkit.repositories.models import Prompt, PromptCategory, Company
 import os
 from iatoolkit.common.exceptions import IAToolkitException
 import importlib.resources
+import logging
 
 
 class PromptService:
     @inject
-    def __init__(self, llm_query_repo: LLMQueryRepo, profile_repo: ProfileRepo):
+    def __init__(self,
+                 llm_query_repo: LLMQueryRepo,
+                 profile_repo: ProfileRepo,
+                 i18n_service: I18nService):
         self.llm_query_repo = llm_query_repo
         self.profile_repo = profile_repo
+        self.i18n_service = i18n_service
 
     def create_prompt(self,
                       prompt_name: str,
@@ -36,20 +40,20 @@ class PromptService:
         if is_system_prompt:
             if not importlib.resources.files('iatoolkit.system_prompts').joinpath(prompt_filename).is_file():
                 raise IAToolkitException(IAToolkitException.ErrorType.INVALID_NAME,
-                                f'No existe el archivo de prompt de sistemas: {prompt_filename}')
+                                f'missing system prompt file: {prompt_filename}')
         else:
             template_dir = f'companies/{company.short_name}/prompts'
 
             relative_prompt_path = os.path.join(template_dir, prompt_filename)
             if not os.path.exists(relative_prompt_path):
                 raise IAToolkitException(IAToolkitException.ErrorType.INVALID_NAME,
-                               f'No existe el archivo de prompt: {relative_prompt_path}')
+                               f'missing prompt file: {relative_prompt_path}')
 
         if custom_fields:
             for f in custom_fields:
                 if ('data_key' not in f) or ('label' not in f):
                     raise IAToolkitException(IAToolkitException.ErrorType.INVALID_PARAMETER,
-                               f'El campo custom_fields debe contener los campos: data_key y label')
+                               f'The field "custom_fields" must contain the following keys: data_key y label')
 
                 # add default value for data_type
                 if 'type' not in f:
@@ -82,20 +86,20 @@ class PromptService:
             user_prompt = self.llm_query_repo.get_prompt_by_name(company, prompt_name)
             if not user_prompt:
                 raise IAToolkitException(IAToolkitException.ErrorType.DOCUMENT_NOT_FOUND,
-                                   f"No se encontró el prompt '{prompt_name}' para la empresa '{company.short_name}'")
+                                   f"prompt not found '{prompt_name}' for company '{company.short_name}'")
 
             prompt_file = f'companies/{company.short_name}/prompts/{user_prompt.filename}'
             absolute_filepath = os.path.join(execution_dir, prompt_file)
             if not os.path.exists(absolute_filepath):
                 raise IAToolkitException(IAToolkitException.ErrorType.FILE_IO_ERROR,
-                                   f"El archivo para el prompt '{prompt_name}' no existe: {absolute_filepath}")
+                                   f"prompt file '{prompt_name}' does not exist: {absolute_filepath}")
 
             try:
                 with open(absolute_filepath, 'r', encoding='utf-8') as f:
                     user_prompt_content = f.read()
             except Exception as e:
                 raise IAToolkitException(IAToolkitException.ErrorType.FILE_IO_ERROR,
-                                   f"Error leyendo el archivo de prompt '{prompt_name}' en {absolute_filepath}: {e}")
+                                   f"error while reading prompt: '{prompt_name}' in this pathname {absolute_filepath}: {e}")
 
             return user_prompt_content
 
@@ -105,9 +109,9 @@ class PromptService:
             raise
         except Exception as e:
             logging.exception(
-                f"Error al obtener el contenido del prompt para la empresa '{company.short_name}' y prompt '{prompt_name}': {e}")
+                f"error loading prompt '{prompt_name}' content for '{company.short_name}': {e}")
             raise IAToolkitException(IAToolkitException.ErrorType.PROMPT_ERROR,
-                               f'Error al obtener el contenido del prompt "{prompt_name}" para la empresa {company.short_name}: {str(e)}')
+                               f'error loading prompt "{prompt_name}" content for company {company.short_name}: {str(e)}')
 
     def get_system_prompt(self):
         try:
@@ -121,10 +125,10 @@ class PromptService:
                     content = importlib.resources.read_text('iatoolkit.system_prompts', prompt.filename)
                     system_prompt_content.append(content)
                 except FileNotFoundError:
-                    logging.warning(f"El archivo para el prompt de sistema no existe en el paquete: {prompt.filename}")
+                    logging.warning(f"Prompt file does not exist in the package: {prompt.filename}")
                 except Exception as e:
                     raise IAToolkitException(IAToolkitException.ErrorType.FILE_IO_ERROR,
-                                             f"Error leyendo el archivo de prompt del sistema '{prompt.filename}': {e}")
+                                             f"error reading system prompt '{prompt.filename}': {e}")
 
             # join the system prompts into a single string
             return "\n".join(system_prompt_content)
@@ -135,14 +139,14 @@ class PromptService:
             logging.exception(
                 f"Error al obtener el contenido del prompt de sistema: {e}")
             raise IAToolkitException(IAToolkitException.ErrorType.PROMPT_ERROR,
-                               f'Error al obtener el contenido de los prompts de sistema": {str(e)}')
+                               f'error reading the system prompts": {str(e)}')
 
     def get_user_prompts(self, company_short_name: str) -> dict:
         try:
             # validate company
             company = self.profile_repo.get_company_by_short_name(company_short_name)
             if not company:
-                return {'error': f'No existe la empresa: {company_short_name}'}
+                return {"error": self.i18n_service.t('errors.company_not_found', company_short_name=company_short_name)}
 
             # get all the prompts
             all_prompts = self.llm_query_repo.get_prompts(company)
@@ -183,6 +187,6 @@ class PromptService:
             return {'message': categorized_prompts}
 
         except Exception as e:
-            logging.error(f"Error en get_prompts: {e}")
+            logging.error(f"error in get_prompts: {e}")
             return {'error': str(e)}
 

iatoolkit/services/query_service.py

@@ -8,6 +8,7 @@ from iatoolkit.services.profile_service import ProfileService
 from iatoolkit.repositories.document_repo import DocumentRepo
 from iatoolkit.repositories.profile_repo import ProfileRepo
 from iatoolkit.services.document_service import DocumentService
+from iatoolkit.services.i18n_service import I18nService
 from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
 from iatoolkit.repositories.models import Task
 from iatoolkit.services.dispatcher_service import Dispatcher
@@ -37,6 +38,7 @@ class QueryService:
                  llmquery_repo: LLMQueryRepo,
                  profile_repo: ProfileRepo,
                  prompt_service: PromptService,
+                 i18n_service: I18nService,
                  util: Utility,
                  dispatcher: Dispatcher,
                  session_context: UserSessionContextService
@@ -47,6 +49,7 @@ class QueryService:
         self.llmquery_repo = llmquery_repo
         self.profile_repo = profile_repo
         self.prompt_service = prompt_service
+        self.i18n_service = i18n_service
         self.util = util
         self.dispatcher = dispatcher
         self.session_context = session_context
@@ -56,7 +59,7 @@ class QueryService:
         self.model = os.getenv("LLM_MODEL", "")
         if not self.model:
             raise IAToolkitException(IAToolkitException.ErrorType.API_KEY,
-                               "La variable de entorno 'LLM_MODEL' no está configurada.")
+                               "missing ENV variable 'LLM_MODEL' configuration.")
 
     def _build_context_and_profile(self, company_short_name: str, user_identifier: str) -> tuple:
         # this method read the user/company context from the database and renders the system prompt
@@ -127,7 +130,7 @@ class QueryService:
         lock_key = f"lock:context:{company_short_name}/{user_identifier}"
         if not self.session_context.acquire_lock(lock_key, expire_seconds=60):
             logging.warning(
-                f"Intento de reconstruir contexto para {user_identifier} mientras ya estaba en progreso. Se omite.")
+                f"try to rebuild context for user {user_identifier} while is still in process, ignored.")
             return
 
         try:
@@ -138,11 +141,9 @@ class QueryService:
             prepared_context, version_to_save = self.session_context.get_and_clear_prepared_context(company_short_name,
                                                                                                     user_identifier)
             if not prepared_context:
-                logging.info(
-                    f"No se requiere reconstrucción de contexto para {company_short_name}/{user_identifier}. Finalización rápida.")
                 return
 
-            logging.info(f"Enviando contexto al LLM para {company_short_name}/{user_identifier}...")
+            logging.info(f"sending context to LLM for: {company_short_name}/{user_identifier}...")
 
             # Limpiar solo el historial de chat y el ID de respuesta anterior
             self.session_context.clear_llm_history(company_short_name, user_identifier)
@@ -161,9 +162,9 @@ class QueryService:
                 self.session_context.save_context_version(company_short_name, user_identifier, version_to_save)
 
             logging.info(
-                f"Contexto de {company_short_name}/{user_identifier} establecido en {int(time.time() - start_time)} seg.")
+                f"Context for: {company_short_name}/{user_identifier} settled in {int(time.time() - start_time)} sec.")
         except Exception as e:
-            logging.exception(f"Error en finalize_context_rebuild para {company_short_name}: {e}")
+            logging.exception(f"Error in finalize_context_rebuild for {company_short_name}: {e}")
             raise e
         finally:
             # --- Liberar el Bloqueo ---
@@ -181,11 +182,11 @@ class QueryService:
             company = self.profile_repo.get_company_by_short_name(short_name=company_short_name)
             if not company:
                 return {"error": True,
-                        "error_message": f'No existe Company ID: {company_short_name}'}
+                        "error_message": self.i18n_service.t('errors.company_not_found', company_short_name=company_short_name)}
 
             if not prompt_name and not question:
                 return {"error": True,
-                        "error_message": f'Hola, cual es tu pregunta?'}
+                        "error_message": self.i18n_service.t('services.start_query')}
 
             # get the previous response_id and context history
             previous_response_id = None
@@ -196,7 +197,7 @@ class QueryService:
                 previous_response_id = self.session_context.get_last_response_id(company.short_name, user_identifier)
                 if not previous_response_id:
                     return {'error': True,
-                            "error_message": f"No se encontró 'previous_response_id' para '{company.short_name}/{user_identifier}'. Reinicia el contexto para continuar."
+                            "error_message": self.i18n_service.t('errors.services.missing_response_id', company_short_name=company.short_name, user_identifier=user_identifier)
                             }
             elif self.util.is_gemini_model(self.model):
                 # check the length of the context_history and remove old messages
@@ -294,7 +295,7 @@ class QueryService:
                 return len(history) >= 1
             return False
         except Exception as e:
-            logging.warning(f"Error verificando caché de contexto: {e}")
+            logging.warning(f"error verifying context cache: {e}")
             return False
 
     def load_files_for_context(self, files: list) -> str:
@@ -353,7 +354,7 @@ class QueryService:
         try:
             total_tokens = sum(self.llm_client.count_tokens(json.dumps(message)) for message in context_history)
         except Exception as e:
-            logging.error(f"Error al calcular tokens del historial: {e}. No se pudo recortar el contexto.")
+            logging.error(f"error counting tokens for history: {e}.")
             return
 
         # Si se excede el límite, eliminar mensajes antiguos (empezando por el segundo)
@@ -364,8 +365,8 @@ class QueryService:
                 removed_tokens = self.llm_client.count_tokens(json.dumps(removed_message))
                 total_tokens -= removed_tokens
                 logging.warning(
-                    f"Historial de contexto ({total_tokens + removed_tokens} tokens) excedía el límite de {GEMINI_MAX_TOKENS_CONTEXT_HISTORY}. "
-                    f"Nuevo total: {total_tokens} tokens."
+                    f"history tokens ({total_tokens + removed_tokens} tokens) exceed the limit of: {GEMINI_MAX_TOKENS_CONTEXT_HISTORY}. "
+                    f"new context: {total_tokens} tokens."
                 )
             except IndexError:
                 # Se produce si solo queda el mensaje del sistema, el bucle debería detenerse.

iatoolkit/services/sql_service.py

@@ -6,6 +6,7 @@
 from iatoolkit.repositories.database_manager import DatabaseManager
 
 from iatoolkit.common.util import Utility
+from iatoolkit.services.i18n_service import I18nService
 from sqlalchemy import text
 from injector import inject
 import json
@@ -14,8 +15,11 @@ from iatoolkit.common.exceptions import IAToolkitException
 
 class SqlService:
     @inject
-    def __init__(self,util: Utility):
+    def __init__(self,
+                 util: Utility,
+                 i18n_service: I18nService):
         self.util = util
+        self.i18n_service = i18n_service
 
     def exec_sql(self, db_manager: DatabaseManager, sql_statement: str) -> str:
         """
@@ -54,7 +58,7 @@ class SqlService:
 
             error_message = str(e)
             if 'timed out' in str(e):
-                error_message = 'Intentalo de nuevo, se agoto el tiempo de espera'
+                error_message = self.i18n_service.t('errors.timeout')
 
             raise IAToolkitException(IAToolkitException.ErrorType.DATABASE_ERROR,
                                      error_message) from e

iatoolkit/services/user_feedback_service.py

@@ -6,8 +6,9 @@
 from iatoolkit.repositories.models import UserFeedback, Company
 from injector import inject
 from iatoolkit.repositories.profile_repo import ProfileRepo
+from iatoolkit.services.i18n_service import I18nService
 from iatoolkit.infra.google_chat_app import GoogleChatApp
-from iatoolkit.infra.mail_app import MailApp  # <-- 1. Importar MailApp
+from iatoolkit.infra.mail_app import MailApp
 import logging
 
 
@@ -15,9 +16,11 @@ class UserFeedbackService:
     @inject
     def __init__(self,
                  profile_repo: ProfileRepo,
+                 i18n_service: I18nService,
                  google_chat_app: GoogleChatApp,
                  mail_app: MailApp):
         self.profile_repo = profile_repo
+        self.i18n_service = i18n_service
         self.google_chat_app = google_chat_app
         self.mail_app = mail_app
 
@@ -31,9 +34,9 @@ class UserFeedbackService:
             }
             chat_result = self.google_chat_app.send_message(message_data=chat_data)
             if not chat_result.get('success'):
-                logging.warning(f"Error al enviar notificación a Google Chat: {chat_result.get('message')}")
+                logging.warning(f"error sending notification to Google Chat: {chat_result.get('message')}")
         except Exception as e:
-            logging.exception(f"Fallo inesperado al enviar notificación a Google Chat: {e}")
+            logging.exception(f"error sending notification to Google Chat: {e}")
 
     def _send_email_notification(self, destination_email: str, company_name: str, message_text: str):
         """Envía una notificación de feedback por correo electrónico."""
@@ -43,20 +46,20 @@ class UserFeedbackService:
             html_body = message_text.replace('\n', '<br>')
             self.mail_app.send_email(to=destination_email, subject=subject, body=html_body)
         except Exception as e:
-            logging.exception(f"Fallo inesperado al enviar email de feedback: {e}")
+            logging.exception(f"error sending email de feedback: {e}")
 
     def _handle_notification(self, company: Company, message_text: str):
         """Lee la configuración de la empresa y envía la notificación al canal correspondiente."""
         feedback_params = company.parameters.get('user_feedback')
         if not isinstance(feedback_params, dict):
-            logging.warning(f"No se encontró configuración de 'user_feedback' para la empresa {company.short_name}.")
+            logging.warning(f"missing 'user_feedback' configuration for company: {company.short_name}.")
             return
 
         # get channel and destination
         channel = feedback_params.get('channel')
         destination = feedback_params.get('destination')
         if not channel or not destination:
-            logging.warning(f"Configuración 'user_feedback' incompleta para {company.short_name}. Faltan 'channel' o 'destination'.")
+            logging.warning(f"invalid 'user_feedback' configuration for: {company.short_name}. Faltan 'channel' o 'destination'.")
             return
 
         if channel == 'google_chat':
@@ -64,7 +67,7 @@ class UserFeedbackService:
         elif channel == 'email':
             self._send_email_notification(destination_email=destination, company_name=company.short_name, message_text=message_text)
         else:
-            logging.warning(f"Canal de feedback '{channel}' no reconocido para la empresa {company.short_name}.")
+            logging.warning(f"unknown feedback channel: '{channel}' for company {company.short_name}.")
 
     def new_feedback(self,
                      company_short_name: str,
@@ -72,19 +75,18 @@ class UserFeedbackService:
                      user_identifier: str,
                      rating: int = None) -> dict:
         try:
-            # 1. Validar empresa
             company = self.profile_repo.get_company_by_short_name(company_short_name)
             if not company:
-                return {'error': f'No existe la empresa: {company_short_name}'}
+                return {'error': self.i18n_service.t('errors.company_not_found', company_short_name=company_short_name)}
 
-            # 2. Enviar notificación según la configuración de la empresa
+            # 2. send notification using company configuration
             notification_text = (f"*Nuevo feedback de {company_short_name}*:\n"
                                  f"*Usuario:* {user_identifier}\n"
                                  f"*Mensaje:* {message}\n"
                                  f"*Calificación:* {rating if rating is not None else 'N/A'}")
             self._handle_notification(company, notification_text)
 
-            # 3. Guardar el feedback en la base de datos (independientemente del éxito de la notificación)
+            # 3. always save the feedback in the database
             new_feedback_obj = UserFeedback(
                 company_id=company.id,
                 message=message,
@@ -93,10 +95,10 @@ class UserFeedbackService:
             )
             saved_feedback = self.profile_repo.save_feedback(new_feedback_obj)
             if not saved_feedback:
-                logging.error(f"No se pudo guardar el feedback para el usuario {user_identifier} en la empresa {company_short_name}")
-                return {'error': 'No se pudo guardar el feedback'}
+                logging.error(f"can't save feedback for user {user_identifier}/{company_short_name}")
+                return {'error': 'can not save the feedback'}
 
-            return {'message': 'Feedback guardado correctamente'}
+            return {'success': True, 'message': 'Feedback guardado correctamente'}
 
         except Exception as e:
             logging.exception(f"Error crítico en el servicio de feedback: {e}")

iatoolkit/static/js/chat_feedback_button.js

@@ -1,110 +1,80 @@
 $(document).ready(function () {
+    const feedbackModal = $('#feedbackModal');
+    $('#submit-feedback').on('click', function () {
+        sendFeedback(this);
+    });
 
     // Evento para enviar el feedback
-    $('#submit-feedback').on('click', async function() {
+    async function sendFeedback(submitButton) {
+        toastr.options = {"positionClass": "toast-bottom-right", "preventDuplicates": true};
         const feedbackText = $('#feedback-text').val().trim();
-        const submitButton = $(this);
-
-        // --- LÓGICA DE COMPATIBILIDAD BS3 / BS5 ---
-        // Detecta si Bootstrap 5 está presente.
-        const isBootstrap5 = (typeof bootstrap !== 'undefined');
-
-        // Define el HTML del botón de cierre según la versión.
-        const closeButtonHtml = isBootstrap5 ?
-            '<button type="button" class="btn-close" data-bs-dismiss="alert" aria-label="Close"></button>' : // Versión BS5
-            '<button type="button" class="close" data-dismiss="alert"><span>&times;</span></button>';     // Versión BS3/BS4
-        // --- FIN DE LA LÓGICA DE COMPATIBILIDAD ---
+        const activeStars = $('.star.active').length;
 
         if (!feedbackText) {
-            const alertHtml = `
-            <div class="alert alert-warning alert-dismissible fade show" role="alert">
-                <strong>¡Atención!</strong> Por favor, escribe tu comentario antes de enviar.
-                ${closeButtonHtml}
-            </div>`;
-            $('.modal-body .alert').remove();
-            $('.modal-body').prepend(alertHtml);
+            toastr.error(t_js('feedback_comment_error'));
             return;
         }
 
-        const activeStars = $('.star.active').length;
         if (activeStars === 0) {
-            const alertHtml = `
-            <div class="alert alert-warning alert-dismissible fade show" role="alert">
-                <strong>¡Atención!</strong> Por favor, califica al asistente con las estrellas.
-                ${closeButtonHtml}
-            </div>`;
-            $('.modal-body .alert').remove();
-            $('.modal-body').prepend(alertHtml);
+            toastr.error(t_js('feedback_rating_error'));
             return;
         }
 
-        submitButton.prop('disabled', true);
-        submitButton.html('<i class="bi bi-send me-1 icon-spaced"></i>Enviando...');
+        submitButton.disabled = true;
 
-        const response = await sendFeedback(feedbackText);
-        $('#feedbackModal').modal('hide');
-        if (response)
-            toastr.success('¡Gracias por tu comentario!', 'Feedback Enviado');
+        // call the IAToolkit API to send feedback
+        const data = {
+            "user_identifier": window.user_identifier,
+            "message": feedbackText,
+            "rating": activeStars,
+        };
+
+        const responseData = await callToolkit('/api/feedback', data, "POST");
+        if (responseData)
+            toastr.success(t_js('feedback_sent_success_body'), t_js('feedback_sent_success_title'));
         else
-            toastr.error('No se pudo enviar el feedback, por favor intente nuevamente.', 'Error');
-    });
+            toastr.error(t_js('feedback_sent_error'));
 
-    // Evento para abrir el modal de feedback
-    $('#send-feedback-button').on('click', function() {
-        $('#submit-feedback').prop('disabled', false);
-        $('#submit-feedback').html('<i class="bi bi-send me-1 icon-spaced"></i>Enviar');
-        $('.star').removeClass('active hover-active'); // Resetea estrellas
-        $('#feedback-text').val(''); // Limpia texto
-        $('.modal-body .alert').remove(); // Quita alertas previas
-        $('#feedbackModal').modal('show');
-    });
+        submitButton.disabled = false;
+        feedbackModal.modal('hide');
+    }
 
-    // Evento que se dispara DESPUÉS de que el modal se ha ocultado
-    $('#feedbackModal').on('hidden.bs.modal', function () {
-        $('#feedback-text').val('');
-        $('.modal-body .alert').remove();
-        $('.star').removeClass('active');
+// Evento para abrir el modal de feedback
+$('#send-feedback-button').on('click', function () {
+    $('#submit-feedback').prop('disabled', false);
+    $('.star').removeClass('active hover-active'); // Resetea estrellas
+    $('#feedback-text').val('');
+    feedbackModal.modal('show');
+});
+
+// Evento que se dispara DESPUÉS de que el modal se ha ocultado
+$('#feedbackModal').on('hidden.bs.modal', function () {
+    $('#feedback-text').val('');
+    $('.star').removeClass('active');
+});
+
+// Function for the star rating system
+window.gfg = function (rating) {
+    $('.star').removeClass('active');
+    $('.star').each(function (index) {
+        if (index < rating) {
+            $(this).addClass('active');
+        }
     });
+};
 
-    // Función para el sistema de estrellas
-    window.gfg = function(rating) {
-        $('.star').removeClass('active');
-        $('.star').each(function(index) {
-            if (index < rating) {
-                $(this).addClass('active');
+$('.star').hover(
+    function () {
+        const rating = $(this).data('rating');
+        $('.star').removeClass('hover-active');
+        $('.star').each(function (index) {
+            if ($(this).data('rating') <= rating) {
+                $(this).addClass('hover-active');
             }
         });
-    };
+    },
+    function () {
+        $('.star').removeClass('hover-active');
+    });
 
-    $('.star').hover(
-        function() {
-            const rating = $(this).data('rating');
-            $('.star').removeClass('hover-active');
-            $('.star').each(function(index) {
-                if ($(this).data('rating') <= rating) {
-                    $(this).addClass('hover-active');
-                }
-            });
-        },
-        function() {
-            $('.star').removeClass('hover-active');
-        }
-    );
 });
-
-const sendFeedback = async function(message) {
-    const activeStars = $('.star.active').length;
-    const data = {
-        "user_identifier": window.user_identifier,
-        "message": message,
-        "rating": activeStars,
-    };
-    try {
-        // Asumiendo que callLLMAPI está definido globalmente en otro archivo (ej. chat_main.js)
-        const responseData = await callToolkit('/api/feedback', data, "POST");
-        return responseData;
-    } catch (error) {
-        console.error("Error al enviar feedback:", error);
-        return null;
-    }
-}