iatoolkit 0.71.4__py3-none-any.whl → 0.91.1__py3-none-any.whl

iatoolkit/services/tool_service.py

@@ -0,0 +1,231 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from injector import inject
+from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
+from iatoolkit.repositories.models import Company, Tool
+from iatoolkit.common.exceptions import IAToolkitException
+from iatoolkit.services.sql_service import SqlService
+from iatoolkit.services.excel_service import ExcelService
+from iatoolkit.services.mail_service import MailService
+
+
+_SYSTEM_TOOLS = [
+    {
+        "function_name": "iat_generate_excel",
+        "description": "Generador de Excel."
+                       "Genera un archivo Excel (.xlsx) a partir de una lista de diccionarios. "
+                       "Cada diccionario representa una fila del archivo. "
+                       "el archivo se guarda en directorio de descargas."
+                       "retorna diccionario con filename, attachment_token (para enviar archivo por mail)"
+                       "content_type y download_link",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "filename": {
+                    "type": "string",
+                    "description": "Nombre del archivo de salida (ejemplo: 'reporte.xlsx')",
+                    "pattern": "^.+\\.xlsx?$"
+                },
+                "sheet_name": {
+                    "type": "string",
+                    "description": "Nombre de la hoja dentro del Excel",
+                    "minLength": 1
+                },
+                "data": {
+                    "type": "array",
+                    "description": "Lista de diccionarios. Cada diccionario representa una fila.",
+                    "minItems": 1,
+                    "items": {
+                        "type": "object",
+                        "properties": {},
+                        "additionalProperties": {
+                            "anyOf": [
+                                {"type": "string"},
+                                {"type": "number"},
+                                {"type": "boolean"},
+                                {"type": "null"},
+                                {
+                                    "type": "string",
+                                    "format": "date"
+                                }
+                            ]
+                        }
+                    }
+                }
+            },
+            "required": ["filename", "sheet_name", "data"]
+        }
+    },
+    {
+        'function_name': "iat_send_email",
+        'description': "iatoolkit mail system. "
+                       "envia mails cuando un usuario lo solicita.",
+        'parameters': {
+            "type": "object",
+            "properties": {
+                "recipient": {"type": "string", "description": "email del destinatario"},
+                "subject": {"type": "string", "description": "asunto del email"},
+                "body": {"type": "string", "description": "HTML del email"},
+                "attachments": {
+                    "type": "array",
+                    "description": "Lista de archivos adjuntos codificados en base64",
+                    "items": {
+                        "type": "object",
+                        "properties": {
+                            "filename": {
+                                "type": "string",
+                                "description": "Nombre del archivo con su extensión (ej. informe.pdf)"
+                            },
+                            "content": {
+                                "type": "string",
+                                "description": "Contenido del archivo en b64."
+                            },
+                            "attachment_token": {
+                                "type": "string",
+                                "description": "token para descargar el archivo."
+                            }
+                        },
+                        "required": ["filename", "content", "attachment_token"],
+                        "additionalProperties": False
+                    }
+                }
+            },
+            "required": ["recipient", "subject", "body", "attachments"]
+        }
+    },
+    {
+        "function_name": "iat_sql_query",
+        "description": "Servicio SQL de IAToolkit: debes utilizar este servicio para todas las consultas a base de datos.",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "database": {
+                    "type": "string",
+                    "description": "nombre de la base de datos a consultar: `database_name`"
+                },
+                "query": {
+                    "type": "string",
+                    "description": "string con la consulta en sql"
+                },
+            },
+            "required": ["database", "query"]
+        }
+    }
+]
+
+
+class ToolService:
+    @inject
+    def __init__(self,
+                 llm_query_repo: LLMQueryRepo,
+                 sql_service: SqlService,
+                 excel_service: ExcelService,
+                 mail_service: MailService):
+        self.llm_query_repo = llm_query_repo
+        self.sql_service = sql_service
+        self.excel_service = excel_service
+        self.mail_service = mail_service
+
+        # execution mapper for system tools
+        self.system_handlers = {
+            "iat_generate_excel": self.excel_service.excel_generator,
+            "iat_send_email": self.mail_service.send_mail,
+            "iat_sql_query": self.sql_service.exec_sql
+        }
+
+    def register_system_tools(self):
+        """Creates or updates system functions in the database."""
+        try:
+            # delete all system tools
+            self.llm_query_repo.delete_system_tools()
+
+            # create new system tools
+            for function in _SYSTEM_TOOLS:
+                new_tool = Tool(
+                    company_id=None,
+                    system_function=True,
+                    name=function['function_name'],
+                    description=function['description'],
+                    parameters=function['parameters']
+                )
+                self.llm_query_repo.create_or_update_tool(new_tool)
+
+            self.llm_query_repo.commit()
+        except Exception as e:
+            self.llm_query_repo.rollback()
+            raise IAToolkitException(IAToolkitException.ErrorType.DATABASE_ERROR, str(e))
+
+    def sync_company_tools(self, company_instance, tools_config: list):
+        """
+        Synchronizes tools from YAML config to Database (Create/Update/Delete strategy).
+        """
+        try:
+            # 1. Get existing tools map for later cleanup
+            existing_tools = {
+                f.name: f for f in self.llm_query_repo.get_company_tools(company_instance.company)
+            }
+            defined_tool_names = set()
+
+            # 2. Sync (Create or Update) from Config
+            for tool_data in tools_config:
+                name = tool_data['function_name']
+                defined_tool_names.add(name)
+
+                # Construct the tool object with current config values
+                # We create a new transient object and let the repo merge it
+                tool_obj = Tool(
+                    company_id=company_instance.company.id,
+                    name=name,
+                    description=tool_data['description'],
+                    parameters=tool_data['params'],
+                    system_function=False
+                )
+
+                # Always call create_or_update. The repo handles checking for existence by name.
+                self.llm_query_repo.create_or_update_tool(tool_obj)
+
+            # 3. Cleanup: Delete tools present in DB but not in Config
+            for name, tool in existing_tools.items():
+                # Ensure we don't delete system functions or active tools accidentally if logic changes,
+                # though get_company_tools filters by company_id so system functions shouldn't be here usually.
+                if not tool.system_function and (name not in defined_tool_names):
+                    self.llm_query_repo.delete_tool(tool)
+
+            self.llm_query_repo.commit()
+
+        except Exception as e:
+            self.llm_query_repo.rollback()
+            raise IAToolkitException(IAToolkitException.ErrorType.DATABASE_ERROR, str(e))
+
+
+    def get_tools_for_llm(self, company: Company) -> list[dict]:
+        """
+        Returns the list of tools (System + Company) formatted for the LLM (OpenAI Schema).
+        """
+        tools = []
+        # Obtiene tanto las de la empresa como las del sistema (la query del repo debería soportar esto con OR)
+        functions = self.llm_query_repo.get_company_tools(company)
+
+        for function in functions:
+            # Clonamos para no modificar el objeto de la sesión SQLAlchemy
+            params = function.parameters.copy() if function.parameters else {}
+            params["additionalProperties"] = False
+
+            ai_tool = {
+                "type": "function",
+                "name": function.name,
+                "description": function.description,
+                "parameters": params,
+                "strict": True
+            }
+            tools.append(ai_tool)
+        return tools
+
+    def get_system_handler(self, function_name: str):
+        return self.system_handlers.get(function_name)
+
+    def is_system_tool(self, function_name: str) -> bool:
+        return function_name in self.system_handlers

iatoolkit/services/user_feedback_service.py

@@ -8,7 +8,7 @@ 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
+from iatoolkit.services.mail_service import MailService
 import logging
 
 
@@ -18,11 +18,11 @@ class UserFeedbackService:
                  profile_repo: ProfileRepo,
                  i18n_service: I18nService,
                  google_chat_app: GoogleChatApp,
-                 mail_app: MailApp):
+                 mail_service: MailService):
         self.profile_repo = profile_repo
         self.i18n_service = i18n_service
         self.google_chat_app = google_chat_app
-        self.mail_app = mail_app
+        self.mail_service = mail_service
 
     def _send_google_chat_notification(self, space_name: str, message_text: str):
         """Envía una notificación de feedback a un espacio de Google Chat."""
@@ -38,13 +38,21 @@ class UserFeedbackService:
         except Exception as 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):
+    def _send_email_notification(self,
+                                 company_short_name: str,
+                                 destination_email: str,
+                                 company_name: str,
+                                 message_text: str):
         """Envía una notificación de feedback por correo electrónico."""
         try:
             subject = f"Nuevo Feedback de {company_name}"
             # Convertir el texto plano a un HTML simple para mantener los saltos de línea
             html_body = message_text.replace('\n', '<br>')
-            self.mail_app.send_email(to=destination_email, subject=subject, body=html_body)
+            self.mail_service.send_mail(
+                company_short_name=company_short_name,
+                to=destination_email,
+                subject=subject,
+                body=html_body)
         except Exception as e:
             logging.exception(f"error sending email de feedback: {e}")
 
@@ -65,7 +73,11 @@ class UserFeedbackService:
         if channel == 'google_chat':
             self._send_google_chat_notification(space_name=destination, message_text=message_text)
         elif channel == 'email':
-            self._send_email_notification(destination_email=destination, company_name=company.short_name, message_text=message_text)
+            self._send_email_notification(
+                company_short_name=company.short_name,
+                destination_email=destination,
+                company_name=company.short_name,
+                message_text=message_text)
         else:
             logging.warning(f"unknown feedback channel: '{channel}' for company {company.short_name}.")
 

iatoolkit/services/user_session_context_service.py

@@ -49,6 +49,24 @@ class UserSessionContextService:
         if session_key:
             RedisSessionManager.hset(session_key, 'last_response_id', response_id)
 
+    def get_initial_response_id(self, company_short_name: str, user_identifier: str) -> Optional[str]:
+        """
+        Obtiene el ID de respuesta inicial desde la sesión del usuario.
+        Este ID corresponde al estado del LLM justo después de haber configurado el contexto.
+        """
+        session_key = self._get_session_key(company_short_name, user_identifier)
+        if not session_key:
+            return None
+        return RedisSessionManager.hget(session_key, 'initial_response_id')
+
+    def save_initial_response_id(self, company_short_name: str, user_identifier: str, response_id: str):
+        """
+        Guarda el ID de respuesta inicial en la sesión del usuario.
+        """
+        session_key = self._get_session_key(company_short_name, user_identifier)
+        if session_key:
+            RedisSessionManager.hset(session_key, 'initial_response_id', response_id)
+
     def save_context_history(self, company_short_name: str, user_identifier: str, context_history: List[Dict]):
         session_key = self._get_session_key(company_short_name, user_identifier)
         if session_key:

iatoolkit/static/js/chat_feedback_button.js

@@ -53,7 +53,7 @@ $('#feedbackModal').on('hidden.bs.modal', function () {
     $('.star').removeClass('active');
 });
 
-// Function for the star rating system
+// Tool for the star rating system
 window.gfg = function (rating) {
     $('.star').removeClass('active');
     $('.star').each(function (index) {

iatoolkit/static/js/chat_help_content.js

@@ -59,7 +59,7 @@ $(document).ready(function () {
                 cat.questions.forEach(q => contentHtml += `<li>${q}</li>`);
                 contentHtml += `</ul>`;
             });
-            accordionHtml += createAccordionItem('examples', 'Preguntas de Ejemplo', contentHtml, true);
+            accordionHtml += createAccordionItem('examples', 'Sample questions', contentHtml, true);
         }
 
         if (data.data_sources) {
@@ -68,7 +68,7 @@ $(document).ready(function () {
                 contentHtml += `<dt>${p.source}</dt><dd>${p.description}</dd>`;
             });
             contentHtml += `</dl>`;
-            accordionHtml += createAccordionItem('sources', 'Datos disponibles', contentHtml );
+            accordionHtml += createAccordionItem('sources', 'Data available', contentHtml );
         }
 
         if (data.best_practices) {
@@ -81,7 +81,7 @@ $(document).ready(function () {
                 contentHtml += `</dd>`;
             });
             contentHtml += `</dl>`;
-            accordionHtml += createAccordionItem('practices', 'Mejores Prácticas', contentHtml);
+            accordionHtml += createAccordionItem('practices', 'Best practices', contentHtml);
         }
 
         if (data.capabilities) {
@@ -89,7 +89,7 @@ $(document).ready(function () {
             contentHtml += `<div class="col-md-6"><h6 class="fw-bold">Puede hacer:</h6><ul>${data.capabilities.can_do.map(item => `<li>${item}</li>`).join('')}</ul></div>`;
             contentHtml += `<div class="col-md-6"><h6 class="fw-bold">No puede hacer:</h6><ul>${data.capabilities.cannot_do.map(item => `<li>${item}</li>`).join('')}</ul></div>`;
             contentHtml += `</div>`;
-            accordionHtml += createAccordionItem('capabilities', 'Capacidades y Límites', contentHtml);
+            accordionHtml += createAccordionItem('capabilities', 'Capabilities and limits', contentHtml);
         }
 
         container.html(accordionHtml);

iatoolkit/static/js/chat_main.js

@@ -120,7 +120,6 @@ const handleChatMessage = async function () {
         }
     } catch (error) {
         if (error.name === 'AbortError') {
-            console.log('Petición abortada por el usuario.');
 
             // Usando jQuery estándar para construir el elemento ---
             const icon = $('<i>').addClass('bi bi-stop-circle me-2'); // Icono sin "fill" para un look más ligero
@@ -230,22 +229,35 @@ const callToolkit = async function(apiPath, data, method, timeoutMs = 500000) {
         }
         const response = await fetch(url, fetchOptions);
         clearTimeout(timeoutId);
+
+        // answer is NOT OK (status != 200)
         if (!response.ok) {
             try {
                 // Intentamos leer el error como JSON, que es el formato esperado de nuestra API.
                 const errorData = await response.json();
-                const errorMessage = errorData.error_message || t_js('unknown_server_error'); // <-- Translation
-                const errorIcon = '<i class="bi bi-exclamation-triangle"></i>';
-                const endpointError = $('<div>').addClass('error-section').html(errorIcon + `<p>${errorMessage}</p>`);
-                displayBotMessage(endpointError);
+
+                // if it's a iatoolkit error  (409 o 400 with a message), shot it on the chat
+                if (errorData && (errorData.error_message || errorData.error)) {
+                    const errorMessage = errorData.error_message || errorData.error || t_js('unknown_server_error');
+                    const errorIcon = '<i class="bi bi-exclamation-triangle"></i>';
+                    const endpointError = $('<div>').addClass('error-section').html(errorIcon + `<p>${errorMessage}</p>`);
+                    displayBotMessage(endpointError);
+                } else {
+                    // if there is not message, we show a generic error message
+                    throw new Error(`Server error: ${response.status}`);
+                }
             } catch (e) {
                 // Si response.json() falla, es porque el cuerpo no era JSON (ej. un 502 con HTML).
                 // Mostramos un error genérico y más claro para el usuario.
                 const errorMessage = `Error de comunicación con el servidor (${response.status}). Por favor, intente de nuevo más tarde.`;
                 toastr.error(errorMessage);
             }
+
+            // stop the flow on the calling function
             return null;
         }
+
+        // if the answer is OK
         return await response.json();
     } catch (error) {
         clearTimeout(timeoutId);

iatoolkit/static/js/chat_onboarding_button.js

@@ -45,7 +45,7 @@
       if (elTitle) elTitle.textContent = c.title || '';
       if (elText) elText.innerHTML = c.text || '';
       if (elExample && c.example) {
-          elExample.innerHTML = ('Ejemplo: ' + c.example) || '';
+          elExample.innerHTML = (t_js('example') + ': ' + c.example) || '';
       }
       else
           elExample.innerHTML = '';

iatoolkit/static/styles/chat_iatoolkit.css

@@ -172,7 +172,7 @@ li {
     max-width: 75%;
     min-width: 250px;
 
-    color: var(--brand-danger-text, #842029); /* Color de texto de error de la marca */
+    color: var(--brand-danger-text, #000000); /* Color de texto de error de la marca */
     background-color: var(--brand-danger-bg, #f8d7da); /* Fondo de error de la marca */
     border: 1px solid var(--brand-danger-border, #f5c2c7); /* Borde de error de la marca */
     font-style: italic;

iatoolkit/static/styles/chat_public.css

@@ -104,4 +104,32 @@
         margin-bottom: 1.5rem; /* Espacio consistente debajo del título */
     }
 
+    /* Links de idioma sobre fondo azul */
+.language-link {
+    text-decoration: none;
+    color: #ffffff;             /* texto blanco */
+    padding: 1px 4px;
+    font-size: 0.85rem;
+    font-weight: 500;
+    border-radius: 3px;
+    transition: background-color 0.2s ease;
+}
+
+/* Separador | puede heredar color blanco */
+.language-switcher .mx-1 {
+    color: #ffffff;
+}
+
+/* Hover: blanco translúcido encima del azul */
+.language-link:hover {
+    background-color: rgba(255, 255, 255, 0.25);
+}
+
+/* Idioma activo: un poco más marcado */
+.language-link.active {
+    background-color: rgba(255, 255, 255, 0.35);
+    font-weight: 700;
+    color: #ffffff;
+}
+