iatoolkit 1.9.0__py3-none-any.whl → 1.15.3__py3-none-any.whl

iatoolkit/static/js/chat_main.js

@@ -58,7 +58,6 @@ $(document).ready(function () {
 
 });
 
-
 /**
  * Main function to handle sending a chat message.
  */
@@ -93,16 +92,16 @@ const handleChatMessage = async function () {
             if (paramsString) { displayMessage += `: ${paramsString}`; }
         }
 
-        // Simplificado: Si no hay mensaje, el 'finally' se encargará de limpiar.
         if (!displayMessage) {
             return;
         }
 
-        displayUserMessage(displayMessage, isEditable, question);
+        const files = window.filePond.getFiles();
+
+        displayUserMessage(displayMessage, isEditable, question, files);
         showSpinner();
         resetAllInputs();
 
-        const files = window.filePond.getFiles();
         const filesBase64 = await Promise.all(files.map(fileItem => toBase64(fileItem.file)));
 
         const data = {
@@ -112,59 +111,18 @@ const handleChatMessage = async function () {
             files: filesBase64.map(f => ({ filename: f.name, content: f.base64 })),
             user_identifier: window.user_identifier,
             model: (window.currentLlmModel || window.defaultLlmModel || '')
-
         };
 
         const responseData = await callToolkit("/api/llm_query", data, "POST");
-        if (responseData && responseData.answer) {
-                // CAMBIO:  contenedor principal para la respuesta del bot
-                const botMessageContainer = $('<div>').addClass('bot-message-container');
-
-                // 1. Si hay reasoning_content, agregar el acordeón colapsable
-                if (responseData.reasoning_content) {
-                    const uniqueId = 'reasoning-' + Date.now(); // ID único para el collapse
-
-                    const reasoningBlock = $(`
-                            <div class="reasoning-block">
-                                <button class="reasoning-toggle btn btn-sm btn-link text-decoration-none p-0"
-                                    type="button"
-                                    data-bs-toggle="collapse"
-                                    data-bs-target="#${uniqueId}"
-                                    aria-expanded="false"
-                                    aria-controls="${uniqueId}">
-                                    <i class="bi bi-lightbulb me-1"></i> ${t_js('show_reasoning')}
-                                </button>
-                        
-                                <div class="collapse mt-2" id="${uniqueId}">
-                                    <div class="reasoning-card">
-                                        ${responseData.reasoning_content}
-                                    </div>
-                                </div>
-                            </div>
-                        `);
-                    botMessageContainer.append(reasoningBlock);
-                }
-
-                // 2. Agregar la respuesta final
-                const answerSection = $('<div>').addClass('answer-section llm-output').append(responseData.answer);
-                botMessageContainer.append(answerSection);
 
-                // 3. Mostrar el contenedor completo
-                displayBotMessage(botMessageContainer);
+        // Delegamos el procesamiento de la respuesta a la nueva función
+        processBotResponse(responseData);
 
-        }
     } catch (error) {
         if (error.name === 'AbortError') {
-
-            // 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
+            const icon = $('<i>').addClass('bi bi-stop-circle me-2');
             const textSpan = $('<span>').text('La generación de la respuesta ha sido detenida.');
-
-            const abortMessage = $('<div>')
-                .addClass('system-message')
-                .append(icon)
-                .append(textSpan);
-
+            const abortMessage = $('<div>').addClass('system-message').append(icon).append(textSpan);
             displayBotMessage(abortMessage);
         } else {
             console.error("Error in handleChatMessage:", error);
@@ -172,17 +130,78 @@ const handleChatMessage = async function () {
             displayBotMessage(errorSection);
         }
     } finally {
-        // Este bloque se ejecuta siempre, garantizando que el estado se limpie.
         isRequestInProgress = false;
         hideSpinner();
         toggleSendStopButtons(false);
         updateSendButtonState();
         if (window.filePond) {
-             window.filePond.removeFiles();
+            window.filePond.removeFiles();
         }
     }
 };
 
+/**
+ * Processes the response data from the LLM and displays it in the chat.
+ * Handles multimodal content: uses 'answer' for text (HTML) and 'content_parts' for images.
+ * @param {object} responseData - The JSON response from the server.
+ */
+function processBotResponse(responseData) {
+    if (!responseData || (!responseData.answer && !responseData.content_parts)) {
+        return;
+    }
+
+    const botMessageContainer = $('<div>').addClass('bot-message-container');
+
+    // 1. Si hay reasoning_content, agregar el acordeón colapsable
+    if (responseData.reasoning_content) {
+        const uniqueId = 'reasoning-' + Date.now();
+        const reasoningBlock = $(`
+                <div class="reasoning-block">
+                    <button class="reasoning-toggle btn btn-sm btn-link text-decoration-none p-0"
+                        type="button" data-bs-toggle="collapse" data-bs-target="#${uniqueId}"
+                        aria-expanded="false" aria-controls="${uniqueId}">
+                        <i class="bi bi-lightbulb me-1"></i> ${t_js('show_reasoning')}
+                    </button>
+                    <div class="collapse mt-2" id="${uniqueId}">
+                        <div class="reasoning-card">${responseData.reasoning_content}</div>
+                    </div>
+                </div>
+            `);
+        botMessageContainer.append(reasoningBlock);
+    }
+
+    // 2. Agregar la respuesta final
+    const answerSection = $('<div>').addClass('answer-section llm-output');
+
+    // A. Texto: Usamos 'answer' porque contiene el HTML procesado y limpio del backend.
+    // Evitamos usar content_parts[type=text] porque contiene el JSON crudo del LLM.
+    if (responseData.answer) {
+        answerSection.append(responseData.answer);
+    }
+
+    // B. Imágenes: Iteramos content_parts buscando SOLO imágenes para adjuntarlas.
+    if (responseData.content_parts && responseData.content_parts.length > 0) {
+        responseData.content_parts.forEach(part => {
+            if (part.type === 'image' && part.source && part.source.url) {
+                const imgContainer = $('<div>').addClass('image-part my-3 text-center');
+                const img = $('<img>')
+                    .attr('src', part.source.url)
+                    .addClass('img-fluid rounded shadow-sm border')
+                    .css({'max-height': '400px', 'cursor': 'pointer'})
+                    .on('click', () => window.open(part.source.url, '_blank'));
+
+                imgContainer.append(img);
+                answerSection.append(imgContainer);
+            }
+        });
+    }
+
+    botMessageContainer.append(answerSection);
+
+    // 3. Mostrar el contenedor completo
+    displayBotMessage(botMessageContainer);
+}
+
 
 /**
  * Resets all inputs to their initial state.
@@ -316,14 +335,48 @@ const callToolkit = async function(apiPath, data, method, timeoutMs = 500000) {
  * @param {string} message - The full message string to display.
  * @param {boolean} isEditable - Determines if the edit icon should be shown.
  * @param {string} originalQuestion - The original text to put back in the textarea for editing.
+ * @param {Array} [files] - Optional array of FilePond file items.
  */
-const displayUserMessage = function(message, isEditable, originalQuestion) {
+const displayUserMessage = function(message, isEditable, originalQuestion, files = []) {
     const chatContainer = $('#chat-container');
     const userMessage = $('<div>').addClass('message shadow-sm');
     const messageText = $('<span>').text(message);
 
     userMessage.append(messageText);
 
+    // Renderizar previsualizaciones de archivos si existen
+    if (files && files.length > 0) {
+        const attachmentsContainer = $('<div>').addClass('mt-2 d-flex flex-wrap gap-2 ms-3');
+
+        files.forEach(fileItem => {
+            const file = fileItem.file;
+
+            if (file.type && file.type.startsWith('image/')) {
+                // Previsualización de imagen usando URL temporal
+                const imgUrl = URL.createObjectURL(file);
+                const img = $('<img>')
+                    .attr('src', imgUrl)
+                    .addClass('rounded border')
+                    .css({
+                        'max-height': '80px',
+                        'max-width': '120px',
+                        'object-fit': 'cover',
+                        'cursor': 'pointer'
+                    })
+                    .on('click', () => window.open(imgUrl, '_blank')); // Click para ver en grande
+                attachmentsContainer.append(img);
+            } else {
+                // Icono genérico para documentos
+                const badge = $('<span>')
+                    .addClass('badge bg-light text-dark border p-2')
+                    .html(`<i class="bi bi-file-earmark-text me-1"></i> ${file.name}`);
+                attachmentsContainer.append(badge);
+            }
+        });
+
+        userMessage.append(attachmentsContainer);
+    }
+
     if (isEditable) {
         const editIcon = $('<i>').addClass('p-2 bi bi-pencil-fill edit-icon edit-pencil').attr('title', 'Edit query').on('click', function () {
             $('#question').val(originalQuestion)

iatoolkit/static/styles/chat_iatoolkit.css

@@ -303,6 +303,102 @@ li {
     color: #343a40;
 }
 
+/* --- NUEVO: Dropzone y Lista de Archivos Inline --- */
+
+/* Contenedor de la lista de archivos */
+.file-list-inline {
+    margin-bottom: 0.5rem;
+    display: none; /* Se oculta si no hay archivos */
+}
+
+.file-list-item {
+    background-color: #fff;
+    border: 1px solid #dee2e6;
+    border-radius: 6px;
+    padding: 6px 10px;
+    margin-bottom: 5px;
+    display: flex;
+    align-items: center;
+    font-size: 0.9rem;
+    transition: background-color 0.2s;
+}
+
+.file-list-item:hover {
+    background-color: #f8f9fa;
+}
+
+.file-list-item i.file-icon {
+    font-size: 1.1rem;
+    margin-right: 8px;
+    color: var(--brand-primary-color);
+}
+
+.file-list-item .file-name {
+    flex-grow: 1;
+    white-space: nowrap;
+    overflow: hidden;
+    text-overflow: ellipsis;
+    margin-right: 10px;
+    color: #495057;
+}
+
+.file-list-item .file-remove {
+    cursor: pointer;
+    color: #dc3545; /* Rojo Bootstrap estándar */
+    opacity: 0.6;
+    transition: opacity 0.2s, transform 0.2s;
+}
+
+.file-list-item .file-remove:hover {
+    opacity: 1;
+    transform: scale(1.1);
+}
+
+/* Dropzone estática */
+.chat-dropzone {
+    border: 2px dashed #ced4da;
+    border-radius: 0.5rem;
+    padding: 10px;
+    text-align: center;
+    background-color: rgba(255, 255, 255, 0.5);
+    color: #6c757d;
+    cursor: pointer;
+    transition: all 0.2s ease-in-out;
+    margin-top: 0.5rem;
+    font-size: 0.9rem;
+}
+
+.chat-dropzone:hover {
+    border-color: var(--brand-primary-color);
+    background-color: rgba(255, 255, 255, 0.8);
+    color: var(--brand-primary-color);
+}
+
+.chat-dropzone.drag-over {
+    border-color: var(--brand-primary-color);
+    background-color: #e9ecef;
+    transform: scale(1.01);
+}
+
+.chat-dropzone i {
+    font-size: 1.2rem;
+    vertical-align: middle;
+    margin-right: 5px;
+}
+
+/* Ajuste para el contador de archivos */
+.dropzone-counter {
+    font-size: 0.8rem;
+    margin-left: 5px;
+    background: #e9ecef;
+    padding: 2px 6px;
+    border-radius: 10px;
+    color: #495057;
+}
+
+/* --- FIN NUEVO --- */
+
+
 /* Anulación específica para el botón de ENVIAR usando su ID (Máxima Prioridad) */
 #send-button i {
     font-size: 1.7rem; /* Ligeramente más grande */

iatoolkit/system_prompts/query_main.prompt

@@ -1,4 +1,5 @@
-Eres un asistente que responde preguntas o ejecuta tareas según el contexto de la empresa.
+Eres un asistente que responde preguntas o ejecuta tareas según el contexto
+de la empresa.
 
 ### **Nombre de la empresa**
 ## Nombre: {{company}}, tambien se conoce como **{{ company_short_name }}**
@@ -12,9 +13,10 @@ Eres un asistente que responde preguntas o ejecuta tareas según el contexto de
 - Rol de usuario: {{ user_rol }}
 
 
-## 🔧 Servicios de datos disponibles en {{ company.name }}
+## 🔧 Servicios de datos disponibles en {{ company_short_name }}
 
-A continuación se muestran los *function calls* (tools) que puedes usar para resolver tareas relacionadas con datos.
+A continuación se muestran los *function calls* (tools) que tienes
+ disponibles para resolver consultas relacionadas con datos de la empresa.
 Cada servicio incluye su **nombre**, **propósito** y **parámetros esperados**.
 
 ### 📌 LISTA DE TOOLS DISPONIBLES
@@ -26,49 +28,30 @@ Cada servicio incluye su **nombre**, **propósito** y **parámetros esperados**.
 
 ---
 
-### ⚠️ REGLAS IMPORTANTES PARA EL USO DE TOOLS
+1. **NO inventes información** si un tool existe para obtenerla.
+2. Si un usuario solicita datos específicos, **elige el tool cuyo propósito coincida exactamente** con la solicitud.
+3. **Si ningún tool aplica**, responde siguiendo el estilo de un asistente normal.
 
-1. **Debes usar un tool cuando la tarea lo requiera.**
-   Ejemplos:
-   - Consultar bases de datos
-   - Obtener información corporativa
-   - Ejecutar SQL
-   - Cargar documentos
-
-2. **NO inventes información** si un tool existe para obtenerla.
-3. Si un usuario solicita datos específicos, **elige el tool cuyo propósito coincida exactamente** con la solicitud.
-4. **Si ningún tool aplica**, responde siguiendo el estilo de un asistente normal.
-
----
-
-Eres un asistente que responde preguntas sobre empresas y sus clientes.
 
 **Reglas obligatorias de contexto:**
 En caso que te hagan preguntas especificas sobre un cliente,  debes asumir que la pregunta se
 refiere al **último cliente identificado** en la conversación.
 
-**IMPORTANTE:**
-
-### **Instrucciones**
-1. Devuelve siempre la respuesta en formato JSON.
-2. Usa la información de contexto para responder la consulta del usuario de forma clara y concisa.
-    Si el contexto o los adjuntos no te proveen información para la respuesta utiliza tu conocimiento propio.
-3. Genera una única `"answer"` que integre la información de todos los servicios relevantes.
-4. En `"aditional_data"`, a menos que se te indique lo contrario retorna un diccionario vacio: {}
-5. El JSON de salida solo puede tener dos llaves: `"answer"` y `"aditional_data"`.
-6. `"answer"` siempre debe contener un unico string con tu respuesta
-7. Devuelve **únicamente** la respuesta en JSON válido, sin texto adicional.
-7. **NO devuelvas el JSON como un string ni como texto escapado.**
-9. **NO incluyas delimitadores como \`\`\`, \`\`\`json, ni comillas alrededor del objeto JSON.**
-10. **NO escribas ninguna explicación ni texto fuera del JSON. Devuelve solo el objeto JSON.**
-
-
-### **Ejemplo de salida esperada**
-{
-  "answer": "El iPhone 15 Pro está disponible por 1099 USD y tu ticket de soporte está en proceso.",
-  "aditional_data": {}
-}
-
-
+---
+### **Instrucciones de Formato de Salida**
+
+1. **Estructura Preferida (JSON):**
+   Para la parte **textual** de tu respuesta, utiliza preferentemente el siguiente formato JSON.
+   {
+      "answer": "Tu respuesta final al usuario. Usa Markdown para formato (negritas, listas, tablas).",
+      "aditional_data": {}
+   }
+
+2. **Generación de Imágenes y Contenido Multimodal:**
+   - Si el usuario te pide generar una imagen y tienes la capacidad nativa **HAZLO**.
+   - Genera la imagen como un bloque de contenido separado.
+   - **NO** intentes meter la imagen dentro del JSON.
+   - En el campo "answer" del JSON, describe o presenta la imagen que has generado.
+   - incluye la imagen como parte multimodal (output_image / file / base64) en la respuesta.
 
 

iatoolkit/templates/chat.html

@@ -208,12 +208,7 @@
                title="{{ t('ui.tooltips.attach_files') }}">
                 <i class="bi bi-plus-circle"></i>
             </a>
-            <div id="view-files-button-container" style="display: none;">
-                <a class="p-2" href="javascript:void(0);" id="view-files-button"
-                   title="{{ t('ui.tooltips.view_attached_files') }}">
-                    <i class="bi bi-file-earmark-text"></i>
-                </a>
-            </div>
+
         </div>
 
         <!-- Textarea Central -->
@@ -238,6 +233,20 @@
             </div>
         </div>
     </div>
+
+    <!-- NUEVO: Área de Adjuntos (Lista + Dropzone) -->
+    <div id="attachments-wrapper" class="px-1">
+        <!-- Lista de archivos cargados -->
+        <div id="inline-file-list" class="file-list-inline"></div>
+
+        <!-- Dropzone siempre visible -->
+        <div id="chat-dropzone" class="chat-dropzone">
+            <i class="bi bi-cloud-upload"></i>
+            <span>Arrastra archivos aquí o haz clic para adjuntar</span>
+            <span id="file-counter" class="dropzone-counter" style="display:none;">0/5</span>
+        </div>
+    </div>
+
 </div>
 
 <!-- Incluir los modales desde un archivo externo -->

iatoolkit/views/base_login_view.py

@@ -77,7 +77,7 @@ class BaseLoginView(MethodView):
             # LLM configuration: default model and availables
             default_llm_model, available_llm_models = self.config_service.get_llm_configuration(company_short_name)
 
-            prompts = self.prompt_service.get_user_prompts(company_short_name)
+            prompts = self.prompt_service.get_prompts(company_short_name)
 
             # Get the entire 'js_messages' block in the correct language.
             js_translations = self.i18n_service.get_translation_block('js_messages')

iatoolkit/views/categories_api_view.py

@@ -3,10 +3,11 @@
 #
 # IAToolkit is open source software.
 
-from flask import jsonify
+from flask import jsonify, request
 from flask.views import MethodView
 from injector import inject
 from iatoolkit.services.auth_service import AuthService
+from iatoolkit.services.prompt_service import PromptService
 from iatoolkit.services.profile_service import ProfileService
 from iatoolkit.services.configuration_service import ConfigurationService
 from iatoolkit.services.knowledge_base_service import KnowledgeBaseService
@@ -25,12 +26,15 @@ class CategoriesApiView(MethodView):
                  profile_service: ProfileService,
                  configuration_service: ConfigurationService,
                  knowledge_base_service: KnowledgeBaseService,
-                 llm_query_repo: LLMQueryRepo):
+                 llm_query_repo: LLMQueryRepo,
+                 prompt_service: PromptService):
         self.auth_service = auth_service
         self.profile_service = profile_service
         self.knowledge_base_service = knowledge_base_service
         self.llm_query_repo = llm_query_repo
         self.configuration_service = configuration_service
+        self.prompt_service = prompt_service
+
 
     def get(self, company_short_name):
         try:
@@ -68,4 +72,40 @@ class CategoriesApiView(MethodView):
 
         except Exception as e:
             logging.exception(f"Error fetching categories for {company_short_name}: {e}")
-            return jsonify({"status": "error", "message": str(e)}), 500
+            return jsonify({"status": "error", "message": str(e)}), 500
+
+    def post(self, company_short_name):
+        try:
+            # 1. Verify Authentication
+            auth_result = self.auth_service.verify()
+            if not auth_result.get("success"):
+                return jsonify(auth_result), 401
+
+            # 2. Get Company
+            company = self.profile_service.get_company_by_short_name(company_short_name)
+            if not company:
+                return jsonify({"error": "Company not found"}), 404
+
+            # 3. Parse Request
+            data = request.get_json() or {}
+
+            # 4. Sync Collection Types
+            # The service expects a list of names strings
+            if 'collection_types' in data:
+                self.knowledge_base_service.sync_collection_types(
+                    company_short_name,
+                    data.get('collection_types', [])
+                )
+
+            # 5. Sync Prompt Categories
+            if 'prompt_categories' in data:
+                self.prompt_service.sync_prompt_categories(
+                    company_short_name,
+                    data.get('prompt_categories', [])
+                )
+
+            return jsonify({"status": "success", "message": "Categories synchronized successfully"}), 200
+
+        except Exception as e:
+            logging.exception(f"Error syncing categories for {company_short_name}: {e}")
+            return jsonify({"status": "error", "message": str(e)}), 500

iatoolkit/views/chat_view.py

@@ -48,7 +48,7 @@ class ChatView(MethodView):
             branding_data = self.branding_service.get_company_branding(company_short_name)
             onboarding_cards = self.config_service.get_configuration(company_short_name, 'onboarding_cards')
             default_llm_model, available_llm_models = self.config_service.get_llm_configuration(company_short_name)
-            prompts = self.prompt_service.get_user_prompts(company_short_name)
+            prompts = self.prompt_service.get_prompts(company_short_name)
             js_translations = self.i18n_service.get_translation_block('js_messages')
 
             return render_template(

iatoolkit/views/login_view.py

@@ -142,7 +142,7 @@ class FinalizeContextView(MethodView):
             )
 
             # 3. render the chat page.
-            prompts = self.prompt_service.get_user_prompts(company_short_name)
+            prompts = self.prompt_service.get_prompts(company_short_name)
             onboarding_cards = self.config_service.get_configuration(company_short_name, 'onboarding_cards')
 
             # Get the entire 'js_messages' block in the correct language.

iatoolkit/views/prompt_api_view.py

@@ -58,7 +58,7 @@ class PromptApiView(MethodView):
                 include_all = request.args.get('all', 'false').lower() == 'true'
 
                 # return prompts based on filter
-                return jsonify(self.prompt_service.get_user_prompts(company_short_name, include_all=include_all))
+                return jsonify(self.prompt_service.get_prompts(company_short_name, include_all=include_all))
 
         except Exception as e:
             logging.exception(

{iatoolkit-1.9.0.dist-info → iatoolkit-1.15.3.dist-info}/METADATA

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