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

iatoolkit/static/js/chat_model_selector.js

@@ -0,0 +1,227 @@
+// src/iatoolkit/static/js/chat_model_selector.js
+// Gestión del selector de modelo LLM en la barra superior.
+
+// Estado global del modelo actual (visible también para otros scripts)
+window.currentLlmModel = window.currentLlmModel || null;
+
+(function () {
+    /**
+     * Lee el modelo guardado en localStorage (si existe y es válido).
+     */
+    function loadStoredModelId() {
+        try {
+            const stored = localStorage.getItem('iatoolkit.selected_llm_model');
+            return stored || null;
+        } catch (e) {
+            return null;
+        }
+    }
+
+    /**
+     * Guarda el modelo seleccionado en localStorage para esta instancia de navegador.
+     * No es crítico: si falla, simplemente no persistimos.
+     */
+    function storeModelId(modelId) {
+        try {
+            if (!modelId) {
+                localStorage.removeItem('iatoolkit.selected_llm_model');
+            } else {
+                localStorage.setItem('iatoolkit.selected_llm_model', modelId);
+            }
+        } catch (e) {
+            // No hacemos nada: fallo silencioso
+        }
+    }
+
+    /**
+     * Devuelve la lista de modelos disponibles desde la variable global.
+     */
+    function getAvailableModels() {
+        const raw = window.availableLlmModels;
+        if (!Array.isArray(raw)) {
+            return [];
+        }
+        return raw.map(m => ({
+            id: m.id,
+            label: m.label || m.id,
+            description: m.description || ''
+        })).filter(m => !!m.id);
+    }
+
+    /**
+     * Inicializa el estado de currentLlmModel usando SIEMPRE la config de company.yaml:
+     * 1) defaultLlmModel (company.yaml)
+     * 2) si no existe o no está en la lista, usa el primer modelo disponible.
+     *
+     * No se lee nada de localStorage en este punto: cada apertura de chat
+     * arranca desde la configuración de la compañía.
+     */
+    function initCurrentModel() {
+        const models = getAvailableModels();
+        const defaultId = (window.defaultLlmModel || '').trim() || null;
+
+        let resolved = null;
+
+        if (defaultId && models.some(m => m.id === defaultId)) {
+            resolved = defaultId;
+        } else if (models.length > 0) {
+            resolved = models[0].id;
+        }
+
+        window.currentLlmModel = resolved;
+        return resolved;
+    }
+
+    /**
+     * Pinta la lista de modelos en el popup y marca el seleccionado.
+     */
+    function renderModelList() {
+        const listEl = document.getElementById('llm-model-list');
+        if (!listEl) return;
+
+        const models = getAvailableModels();
+        const activeId = window.currentLlmModel;
+        listEl.innerHTML = '';
+
+        if (!models.length) {
+            const emptyItem = document.createElement('div');
+            emptyItem.className = 'list-group-item small text-muted';
+            emptyItem.textContent = 'No hay modelos configurados.';
+            listEl.appendChild(emptyItem);
+            return;
+        }
+
+        models.forEach(model => {
+            const item = document.createElement('button');
+            item.type = 'button';
+            item.className = 'list-group-item list-group-item-action small';
+
+            const isActive = model.id === activeId;
+            if (isActive) {
+                item.classList.add('active');
+            }
+
+            item.innerHTML = `
+                <div class="d-flex justify-content-between align-items-center">
+                    <div>
+                        <div class="fw-semibold">${model.label}</div>
+                        ${model.description
+                ? `<div class="text-muted" style="font-size: 0.8rem;">${model.description}</div>`
+                : ''
+            }
+                    </div>
+                    ${isActive ? '<i class="bi bi-check-lg ms-2"></i>' : ''}
+                </div>
+            `;
+
+            item.addEventListener('click', () => {
+                selectModel(model.id);
+            });
+
+            listEl.appendChild(item);
+        });
+    }
+
+    /**
+     * Actualiza el label del botón principal con el modelo actual.
+     */
+    function updateButtonLabel() {
+        const labelEl = document.getElementById('llm-model-button-label');
+        if (!labelEl) return;
+
+        const models = getAvailableModels();
+        const activeId = window.currentLlmModel;
+        const activeModel = models.find(m => m.id === activeId);
+
+        if (activeModel) {
+            labelEl.textContent = activeModel.label;
+        } else if (window.defaultLlmModel) {
+            labelEl.textContent = window.defaultLlmModel;
+        } else {
+            labelEl.textContent = 'Modelo IA';
+        }
+    }
+
+    /**
+     * Selecciona un modelo: actualiza estado global, UI y almacenamiento local.
+     */
+    function selectModel(modelId) {
+        if (!modelId) return;
+
+        const models = getAvailableModels();
+        const exists = models.some(m => m.id === modelId);
+        if (!exists) return;
+
+        window.currentLlmModel = modelId;
+        storeModelId(modelId);
+        updateButtonLabel();
+        renderModelList();
+        hidePopup();
+
+        if (typeof toastr !== 'undefined') {
+            toastr.info(`Modelo actualizado a "${models.find(m => m.id === modelId).label}".`);
+        }
+    }
+
+    /**
+     * Muestra/oculta el popup anclado al botón.
+     */
+    function togglePopup() {
+        const popup = document.getElementById('llm-model-popup');
+        const btn = document.getElementById('llm-model-button');
+        if (!popup || !btn) return;
+
+        const isVisible = popup.style.display === 'block';
+
+        if (isVisible) {
+            hidePopup();
+        } else {
+            const rect = btn.getBoundingClientRect();
+            popup.style.display = 'block';
+
+            // Posicionamos justo debajo del botón, alineado a la izquierda
+            popup.style.top = `${rect.bottom + window.scrollY + 4}px`;
+            popup.style.left = `${rect.left + window.scrollX}px`;
+        }
+    }
+
+    function hidePopup() {
+        const popup = document.getElementById('llm-model-popup');
+        if (popup) {
+            popup.style.display = 'none';
+        }
+    }
+
+    /**
+     * Cierra el popup si el usuario hace click fuera.
+     */
+    function setupOutsideClickHandler() {
+        document.addEventListener('click', (event) => {
+            const popup = document.getElementById('llm-model-popup');
+            const btn = document.getElementById('llm-model-button');
+            if (!popup || !btn) return;
+
+            if (popup.style.display !== 'block') return;
+
+            if (!popup.contains(event.target) && !btn.contains(event.target)) {
+                hidePopup();
+            }
+        });
+    }
+
+    document.addEventListener('DOMContentLoaded', () => {
+        // Inicializar estado inicial del modelo
+        initCurrentModel();
+        updateButtonLabel();
+        renderModelList();
+        setupOutsideClickHandler();
+
+        const btn = document.getElementById('llm-model-button');
+        if (btn) {
+            btn.addEventListener('click', (event) => {
+                event.stopPropagation();
+                togglePopup();
+            });
+        }
+    });
+})();

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 = (t_js('example') + ': ' + c.example) || '';
+          elExample.innerHTML = ('Example ' + ': ' + c.example) || '';
       }
       else
           elExample.innerHTML = '';

iatoolkit/static/js/chat_reload_button.js

@@ -18,7 +18,10 @@ $(document).ready(function () {
 
         // 2. prepare the api parameters
         const apiPath = '/api/init-context';
-        const payload = {'user_identifier': window.user_identifier};
+        const payload = {
+            'user_identifier': window.user_identifier,
+            'model': (window.currentLlmModel || window.defaultLlmModel || '')
+            };
 
         // 3. make the call to callToolkit
         const data = await callToolkit(apiPath, payload, 'POST');

iatoolkit/static/styles/chat_iatoolkit.css

@@ -499,5 +499,61 @@ li {
     cursor: pointer;
 }
 
-
-
+  /* Popup de selección de modelo LLM - versión minimalista, sin branding de fondo */
+    .llm-model-popup {
+        background-color: #f9fafb;           /* Fondo claro neutro */
+        border-radius: 0.5rem;
+        border: 1px solid #e5e7eb;          /* Borde gris muy suave */
+        color: #111827;                     /* Texto principal en gris casi negro */
+        box-shadow: 0 10px 30px rgba(15, 23, 42, 0.12); /* Sombra suave y limpia */
+    }
+
+    .llm-model-popup .card-body {
+        padding-left: 0.9rem;
+        padding-right: 0.9rem;
+    }
+
+    .llm-model-popup .llm-model-subtitle {
+        font-size: 0.8rem;
+        color: #6b7280;                     /* Gris medio para el subtítulo */
+    }
+
+    .llm-model-popup .list-group-item {
+        border: none;
+        padding-top: 0.4rem;
+        padding-bottom: 0.4rem;
+        background-color: transparent;
+        color: inherit;
+        border-radius: 0.35rem;
+        transition:
+            background-color 0.15s ease-in-out,
+            transform 0.1s ease-in-out,
+            box-shadow 0.15s ease-in-out;
+    }
+
+    /* Descripción del modelo: mismo color que el nombre, pero un poco más suave */
+    .llm-model-popup .list-group-item .text-muted {
+        color: currentColor !important;
+        opacity: 0.75;
+        font-size: 0.8rem;
+    }
+
+    /* Hover más visible, pero manteniendo el estilo minimalista */
+    .llm-model-popup .list-group-item:hover {
+        background-color: #e5e7eb;          /* Gris claro para resaltar bien */
+        transform: translateX(2px);
+        box-shadow: 0 2px 6px rgba(15, 23, 42, 0.12);
+    }
+
+    /* Item activo: resalta con el primario, pero fondo muy claro */
+    .llm-model-popup .list-group-item.active {
+        background-color: #eef2ff;          /* Azul/gris muy claro tipo focus */
+        color: var(--brand-primary-color, #4C6A8D);
+        border: 1px solid var(--brand-primary-color, #4C6A8D);
+        box-shadow: 0 0 0 1px rgba(37, 99, 235, 0.2);
+    }
+
+    .llm-model-popup .list-group-item.active .text-muted {
+        color: currentColor !important;
+        opacity: 0.85;
+    }

iatoolkit/static/styles/llm_output.css

@@ -1,7 +1,7 @@
 /* BLOQUE GENERAL */
 .llm-output {
   margin: 0;
-  padding: 20px;
+  padding: 14px;
   font-family: Arial, sans-serif;
   font-size: 16px;
   line-height: 1.5;
@@ -113,3 +113,36 @@
 .nowrap {
     white-space: nowrap;
   }
+
+/* =========================================================
+   REASONING BLOCK (LLM)
+   ========================================================= */
+
+.reasoning-block {
+}
+
+/* Botón de mostrar razonamiento */
+.reasoning-toggle {
+  font-size: 13px;
+  color: #6c757d; /* text-secondary */
+}
+
+.reasoning-toggle:hover {
+  color: #495057;
+  text-decoration: underline;
+}
+
+/* Contenedor del reasoning */
+.reasoning-card {
+  background-color: #f8f9fa;            /* bg-light */
+  border-left: 3px solid #6c757d;        /* border-secondary */
+  padding: 10px 14px;
+  font-size: 14px;                      /* consistente con llm-output */
+  line-height: 1.5;
+  color: #555;
+  white-space: pre-wrap;
+  max-height: 300px;
+  overflow-y: auto;
+  font-family: inherit;                 /* 👈 clave: usa la misma fuente base */
+}
+

iatoolkit/system_prompts/query_main.prompt

@@ -12,11 +12,35 @@ Eres un asistente que responde preguntas o ejecuta tareas según el contexto de
 - Rol de usuario: {{ user_rol }}
 
 
-## Servicios de datos (function calls) disponibles en {{company.name}}:
+## 🔧 Servicios de datos disponibles en {{ company.name }}
+
+A continuación se muestran los *function calls* (tools) que puedes usar para resolver tareas relacionadas con datos.
+Cada servicio incluye su **nombre**, **propósito** y **parámetros esperados**.
+
+### 📌 LISTA DE TOOLS DISPONIBLES
 {% for service in service_list %}
-# servicio {{ loop.index }}: {{ service.name }}
+#### Tool {{ loop.index }}: `{{ service.name }}`
+- **Descripción:** {{ service.description }}
+- **Parámetros:** {{ service.parameters | tojson }}
 {% endfor %}
 
+---
+
+### ⚠️ REGLAS IMPORTANTES PARA EL USO DE TOOLS
+
+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:**

iatoolkit/templates/base.html

@@ -1,6 +1,19 @@
 <!DOCTYPE html>
 <html lang="en">
 <head>
+    {% if google_analytics_id %}
+    <!-- Google tag (gtag.js) -->
+    <script async src="https://www.googletagmanager.com/gtag/js?id=G-96G1XCM2CC"></script>
+    <script>
+        window.dataLayer = window.dataLayer || [];
+        function gtag(){dataLayer.push(arguments);}
+        gtag('js', new Date());
+
+        gtag('config', '{{ google_analytics_id }}');
+    </script>
+    {% endif %}
+
+
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>{% block title %}Chatbot{% endblock %}</title>

iatoolkit/templates/chat.html

@@ -28,7 +28,8 @@
                   title="Powered by IAToolkit {{ license }} ver. {{ iatoolkit_version }}">
                 <i class="bi bi-info-circle" style="color: {{ branding.header_text_color }}; opacity: 0.7; font-size: 0.9rem;"></i>
             </span>
-        </div>
+
+            </div>
 
         <!-- Contenedor para la derecha que agrupa ID de usuario y botones -->
         <div class="d-flex flex-column flex-md-row align-items-center">
@@ -38,14 +39,39 @@
                 {{ user_identifier }}
             </span>
 
+
             <!-- Separador Vertical (Solo visible en Desktop) -->
             <div class="vr mx-3 d-none d-md-block"></div>
 
             <!-- Fila 3 (Móvil) / Parte 2 de la Columna Derecha (Desktop): Iconos de Acción -->
             <div class="d-flex align-items-center">
+                <!-- Selector de modelo LLM -->
+                <button type="button"
+                        id="llm-model-button"
+                        class="btn btn-sm py-1 px-3"
+                        style="border-radius: 0.4rem; border: 1px solid rgba(255,255,255,0.7); background: rgba(0,0,0,0.08); color: {{ branding.header_text_color }};">
+                    <i class="bi bi-cpu me-1"></i>
+                    <span id="llm-model-button-label">
+                        {{ (llm_available_models[0].label if llm_available_models and llm_available_models[0].label else llm_default_model) or 'Modelo IA' }}
+                    </span>
+                </button>
+                <!-- Popup simple para selección de modelo -->
+                <div id="llm-model-popup"
+                     class="card shadow-sm llm-model-popup"
+                     style="position: absolute; z-index: 9999; display: none; min-width: 260px;">
+                    <div class="card-body py-2">
+                        <div class="small mb-1 fw-bold">Modelo de IA</div>
+                        <div class="small mb-2 llm-model-subtitle">
+                            Este modelo se usará en las próximas consultas.
+                        </div>
+                        <div id="llm-model-list" class="list-group list-group-flush">
+                            {# El contenido se inyecta vía JavaScript con window.availableLlmModels #}
+                        </div>
+                    </div>
+                </div>
                 <a href="javascript:void(0);"
                    id="history-button"
-                   class="action-icon-style" title="{{ t('ui.tooltips.history') }}"
+                   class="ms-3 action-icon-style" title="{{ t('ui.tooltips.history') }}"
                    style="color: {{ branding.header_text_color }};">
                     <i class="bi bi-clock-history"></i>
                 </a>
@@ -77,7 +103,17 @@
                         title="{{ t('ui.tooltips.usage_guide') }}"
                        style="color: {{ branding.header_text_color }};">
                         <i class="bi bi-question-circle-fill"></i>
+                </a>
+                {% if user_role == "admin" and license == 'enterprise' %}
+                    <a href="/{{ company_short_name }}/admin/dashboard"
+                       target="_blank"
+                           id="preferences-button"
+                           class="ms-3 action-icon-style"
+                            title="{{ t('ui.tooltips.preferences') }}"
+                           style="color: {{ branding.header_text_color }};">
+                            <i class="bi bi-gear"></i>
                     </a>
+                {% endif %}
                 <a href="javascript:void(0);"
                    id="logout-button"
                    class="ms-3 action-icon-style"
@@ -220,6 +256,11 @@
     window.onboardingCards = {{ onboarding_cards | tojson }};
     window.sendButtonColor = "{{ branding.send_button_color }}";
 
+    // LLM configuration from backend
+    window.defaultLlmModel = {{ llm_default_model | tojson }};
+    window.availableLlmModels = {{ llm_available_models | tojson }};
+
+
     // JS translations helper
     window.i18n = {{ js_translations | tojson }};
     function t_js(key) {
@@ -238,6 +279,7 @@
     <script src="{{ url_for('static', filename='js/chat_logout_button.js', _external=True) }}"></script>
     <script src="{{ url_for('static', filename='js/chat_prompt_manager.js', _external=True) }}"></script>
     <script src="{{ url_for('static', filename='js/chat_filepond.js', _external=True) }}"></script>
+    <script src="{{ url_for('static', filename='js/chat_model_selector.js', _external=True) }}"></script>
     <script src="{{ url_for('static', filename='js/chat_main.js', _external=True) }}"></script>
 
     <script src="https://cdnjs.cloudflare.com/ajax/libs/jquery/3.6.0/jquery.min.js"></script>

iatoolkit/templates/onboarding_shell.html

@@ -38,7 +38,6 @@
     <div class="ob-stack">
       <h1 id="ob-brand-header" class="ob-brand-header">
         <span class="brand-name text-brand-primary">{{ branding.name | default('IAToolkit') }}</span>
-        <span class="brand-rest"> IA</span>
       </h1>
 
       <div id="card-container" class="ob-card">

iatoolkit/views/base_login_view.py

@@ -74,6 +74,9 @@ class BaseLoginView(MethodView):
             )
         else:
             # --- FAST PATH: Render the chat page directly ---
+            # 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)
 
             # Get the entire 'js_messages' block in the correct language.
@@ -87,5 +90,7 @@ class BaseLoginView(MethodView):
                 branding=branding_data,
                 onboarding_cards=onboarding_cards,
                 js_translations=js_translations,
-                redeem_token=redeem_token
-            )
+                redeem_token=redeem_token,
+                llm_default_model=default_llm_model,
+                llm_available_models = available_llm_models,
+                )

iatoolkit/views/chat_view.py

@@ -0,0 +1,76 @@
+from flask import render_template, redirect, url_for, request
+from flask.views import MethodView
+from injector import inject
+from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.branding_service import BrandingService
+from iatoolkit.services.configuration_service import ConfigurationService
+from iatoolkit.services.prompt_service import PromptService
+from iatoolkit.services.i18n_service import I18nService
+
+class ChatView(MethodView):
+    """
+    Handles direct access to the chat interface.
+    Validates if the user has an active session for the company.
+    """
+
+    @inject
+    def __init__(self,
+                 profile_service: ProfileService,
+                 branding_service: BrandingService,
+                 config_service: ConfigurationService,
+                 prompt_service: PromptService,
+                 i18n_service: I18nService):
+        self.profile_service = profile_service
+        self.branding_service = branding_service
+        self.config_service = config_service
+        self.prompt_service = prompt_service
+        self.i18n_service = i18n_service
+
+    def get(self, company_short_name: str):
+        # 1. Validate Company
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            return render_template('error.html',
+                                   message=self.i18n_service.t('errors.templates.company_not_found')), 404
+
+        # 2. Check Session
+        session_info = self.profile_service.get_current_session_info()
+        user_identifier = session_info.get('user_identifier')
+        session_company = session_info.get('company_short_name')
+
+        # If no user or session belongs to another company -> Redirect to Home
+        if not user_identifier or session_company != company_short_name:
+            return redirect(url_for('home', company_short_name=company_short_name))
+
+        # 3. Prepare Context for Chat
+        # (This logic mirrors the FAST PATH in BaseLoginView)
+        try:
+            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)
+            js_translations = self.i18n_service.get_translation_block('js_messages')
+
+            return render_template(
+                "chat.html",
+                company_short_name=company_short_name,
+                user_identifier=user_identifier,
+                prompts=prompts,
+                branding=branding_data,
+                onboarding_cards=onboarding_cards,
+                js_translations=js_translations,
+                llm_default_model=default_llm_model,
+                llm_available_models=available_llm_models,
+                # redeem_token is None for direct access
+                redeem_token=None
+            )
+        except Exception as e:
+            # Fallback error handling
+            message = self.i18n_service.t('errors.templates.processing_error', error=str(e))
+            branding_data = self.branding_service.get_company_branding(company_short_name)
+            return render_template(
+                "error.html",
+                company_short_name=company_short_name,
+                branding=branding_data,
+                message=message
+            ), 500

iatoolkit/views/load_company_configuration_api_view.py

@@ -0,0 +1,49 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import redirect, url_for, jsonify, request, g
+from injector import inject
+from iatoolkit.services.auth_service import AuthService
+from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.configuration_service import ConfigurationService
+import logging
+
+class LoadCompanyConfigurationApiView(MethodView):
+    @inject
+    def __init__(self,
+                 configuration_service: ConfigurationService,
+                 profile_service: ProfileService,
+                 auth_service: AuthService):
+        self.configuration_service = configuration_service
+        self.profile_service = profile_service
+        self.auth_service = auth_service
+
+    def get(self, company_short_name: str = None):
+        try:
+            # 1. Get the authenticated user's
+            auth_result = self.auth_service.verify(anonymous=True)
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code", 401)
+
+            company = self.profile_service.get_company_by_short_name(company_short_name)
+            if not company:
+                return jsonify({"error": "company not found."}), 404
+
+            config, errors = self.configuration_service.load_configuration(company_short_name)
+            if config:
+                self.configuration_service.register_data_sources(company_short_name)
+
+            # this is fo avoid serialization issues
+            if 'company' in config:
+                config.pop('company')
+
+            status_code = 200 if not errors else 400
+            return {'config': config, 'errors': [errors]}, status_code
+        except Exception as e:
+            logging.exception(f"Unexpected error: {e}")
+            return {'status': 'error'}, 500
+
+