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

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

@@ -0,0 +1,111 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+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
+from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
+from iatoolkit.repositories.models import PromptType, PromptCategory
+import logging
+
+class CategoriesApiView(MethodView):
+    """
+    Endpoint to retrieve all available categories and types in the system.
+    Useful for populating dropdowns in the frontend.
+    """
+    @inject
+    def __init__(self,
+                 auth_service: AuthService,
+                 profile_service: ProfileService,
+                 configuration_service: ConfigurationService,
+                 knowledge_base_service: KnowledgeBaseService,
+                 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:
+            # 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. Gather Categories
+            response_data = {
+                "prompt_types": [t.value for t in PromptType],
+                "prompt_categories": [],
+                "collection_types": [],
+                # Future categories can be added here (e.g., tool_types, user_roles)
+            }
+
+            # A. Prompt Categories (from DB)
+            prompt_cats = self.llm_query_repo.get_all_categories(company_id=company.id)
+            response_data["prompt_categories"] = [c.name for c in prompt_cats]
+
+            # B. Collection Types (from KnowledgeBaseService)
+            response_data["collection_types"] = self.knowledge_base_service.get_collection_names(company_short_name)
+
+            # C. LLM Models (from ConfigurationService)
+            _, llm_models = self.configuration_service.get_llm_configuration(company_short_name)
+            # Extract only IDs
+            response_data["llm_models"] = [m['id'] for m in llm_models if 'id' in m]
+
+            return jsonify(response_data)
+
+        except Exception as e:
+            logging.exception(f"Error fetching categories for {company_short_name}: {e}")
+            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/configuration_api_view.py

@@ -62,7 +62,7 @@ class ConfigurationApiView(MethodView):
         Body: { "key": "llm.model", "value": "gpt-4" }
         """
         try:
-            auth_result = self.auth_service.verify(anonymous=False) # Require valid user for updates
+            auth_result = self.auth_service.verify()
             if not auth_result.get("success"):
                 return jsonify(auth_result), 401
 

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

@@ -3,9 +3,11 @@
 #
 # IAToolkit is open source software.
 
-from flask import jsonify
+from flask import jsonify, request
 from flask.views import MethodView
 from iatoolkit.services.prompt_service import PromptService
+from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.repositories.llm_query_repo import LLMQueryRepo
 from iatoolkit.services.auth_service import AuthService
 from injector import inject
 import logging
@@ -15,23 +17,102 @@ class PromptApiView(MethodView):
     @inject
     def __init__(self,
                  auth_service: AuthService,
-                 prompt_service: PromptService ):
+                 prompt_service: PromptService,
+                 profile_service: ProfileService,
+                 llm_query_repo: LLMQueryRepo):
         self.auth_service = auth_service
         self.prompt_service = prompt_service
+        self.profile_service = profile_service
+        self.llm_query_repo = llm_query_repo
 
-    def get(self, company_short_name):
+    def get(self, company_short_name, prompt_name=None):
+        """
+        GET /: Lista el árbol de prompts (Categorías > Prompts).
+        GET /<name>: Devuelve detalle completo: metadata + contenido texto.
+        """
         try:
             # get access credentials
             auth_result = self.auth_service.verify(anonymous=True)
             if not auth_result.get("success"):
                 return jsonify(auth_result), auth_result.get('status_code')
 
-            response = self.prompt_service.get_user_prompts(company_short_name)
-            if "error" in response:
-                return {'error_message': response["error"]}, 402
+            company = self.profile_service.get_company_by_short_name(company_short_name)
+            if not company:
+                 return jsonify({"error": "Company not found"}), 404
+
+            if prompt_name:
+                # get the prompt object from database
+                prompt_obj = self.llm_query_repo.get_prompt_by_name(company, prompt_name)
+                if not prompt_obj:
+                     return jsonify({"error": "Prompt not found"}), 404
+
+                # get the prompt content
+                content = self.prompt_service.get_prompt_content(company, prompt_name)
+
+                return jsonify({
+                    "meta": prompt_obj.to_dict(),
+                    "content": content
+                })
+            else:
+                # Check for query param to include all prompts (admin view)
+                include_all = request.args.get('all', 'false').lower() == 'true'
+
+                # return prompts based on filter
+                return jsonify(self.prompt_service.get_prompts(company_short_name, include_all=include_all))
 
-            return response, 200
         except Exception as e:
             logging.exception(
                 f"unexpected error getting company prompts: {e}")
             return jsonify({"error_message": str(e)}), 500
+
+    def put(self, company_short_name, prompt_name):
+        try:
+            auth_result = self.auth_service.verify()
+            if not auth_result.get("success"):
+                return jsonify(auth_result), 401
+
+            data = request.get_json()
+
+            # The service handles file magic and YAML sync
+            self.prompt_service.save_prompt(company_short_name, prompt_name, data)
+
+            return jsonify({"status": "success"})
+        except Exception as e:
+            logging.exception(f"Error saving prompt {prompt_name}: {e}")
+            return jsonify({"status": "error", "message": str(e)}), 500
+
+    def post(self, company_short_name, prompt_name=None):
+        """Creates a new prompt."""
+        try:
+            auth_result = self.auth_service.verify()
+            if not auth_result.get("success"):
+                return jsonify(auth_result), 401
+
+            data = request.get_json()
+            # If prompt_name is not in URL, check body
+            target_name = prompt_name if prompt_name else data.get('name')
+
+            if not target_name:
+                return jsonify({"status": "error", "message": "Prompt name is required"}), 400
+
+            # Reuse save_prompt logic which handles create/update
+            self.prompt_service.save_prompt(company_short_name, target_name, data)
+
+            return jsonify({"status": "success"})
+        except Exception as e:
+            logging.exception(f"Error creating prompt: {e}")
+            return jsonify({"status": "error", "message": str(e)}), 500
+
+    def delete(self, company_short_name, prompt_name):
+        """Deletes a prompt."""
+        try:
+            auth_result = self.auth_service.verify()
+            if not auth_result.get("success"):
+                return jsonify(auth_result), 401
+
+            self.prompt_service.delete_prompt(company_short_name, prompt_name)
+
+            return jsonify({"status": "success"})
+        except Exception as e:
+            logging.exception(f"Error deleting prompt {prompt_name}: {e}")
+            return jsonify({"status": "error", "message": str(e)}), 500

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

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