iatoolkit 0.3.7__py3-none-any.whl → 0.3.9__py3-none-any.whl

iatoolkit/__init__.py

@@ -21,6 +21,7 @@ from services.excel_service import ExcelService
 from services.dispatcher_service import Dispatcher
 from services.document_service import DocumentService
 from services.search_service import SearchService
+from services.query_service import QueryService
 from repositories.profile_repo import ProfileRepo
 from repositories.llm_query_repo import LLMQueryRepo
 from repositories.database_manager import DatabaseManager
@@ -39,6 +40,7 @@ __all__ = [
     'ExcelService',
     'Dispatcher',
     'DocumentService',
+    'QueryService',
     'SearchService',
     'ProfileRepo',
     'LLMQueryRepo',

iatoolkit/base_company.py

@@ -15,14 +15,19 @@ class BaseCompany(ABC):
 
     @abstractmethod
     # initialize all the database tables  needed
-    def init_db(self):
-        raise NotImplementedError("La subclase debe implementar el método init_db()")
+    def register_company(self):
+        raise NotImplementedError("La subclase debe implementar el método create_company()")
 
     @abstractmethod
     # get context specific for this company
     def get_company_context(self, **kwargs) -> str:
         raise NotImplementedError("La subclase debe implementar el método get_company_context()")
 
+    @abstractmethod
+    # get context specific for this company
+    def get_user_info(self, **kwargs) -> str:
+        raise NotImplementedError("La subclase debe implementar el método get_user_info()")
+
     @abstractmethod
     # execute the specific action configured in the intent table
     def handle_request(self, tag: str, params: dict) -> dict:
@@ -38,5 +43,11 @@ class BaseCompany(ABC):
     def get_metadata_from_filename(self, filename: str) -> dict:
         raise NotImplementedError("La subclase debe implementar el método get_query_context()")
 
+    def register_cli_commands(self, app):
+        """
+        optional method for a company definition of it's cli commands
+        """
+        pass
+
     def unsupported_operation(self, tag):
         raise NotImplementedError(f"La operación '{tag}' no está soportada por esta empresa.")

iatoolkit/cli_commands.py

@@ -0,0 +1,77 @@
+import click
+import logging
+from iatoolkit import IAToolkit
+from services.dispatcher_service import Dispatcher
+from services.profile_service import ProfileService
+
+def register_core_commands(app):
+    """Registra los comandos CLI del núcleo de IAToolkit."""
+    
+    @app.cli.command("setup-all-companies")
+    def setup_all_companies():
+        """🗄️ Inicializa todas las compañías registradas en la base de datos."""
+        try:
+            dispatcher = IAToolkit.get_instance().get_injector().get(Dispatcher)
+            click.echo("🚀 Inicializando base de datos y compañías...")
+            dispatcher.setup_all_companies()
+            click.echo("✅ Base de datos y compañías inicializadas correctamente.")
+        except Exception as e:
+            logging.exception(e)
+            click.echo(f"❌ Error: {e}")
+
+    @app.cli.command("setup-company")
+    @click.argument("company_short_name")
+    def setup_company(company_short_name: str):
+        """⚙️ Genera una nueva API key para una compañía ya registrada."""
+        try:
+            profile_service = IAToolkit.get_instance().get_injector().get(ProfileService)
+            click.echo(f"🔑 Generando API key para '{company_short_name}'...")
+            result = profile_service.new_api_key(company_short_name)
+
+            if 'error' in result:
+                click.echo(f"❌ Error: {result['error']}")
+                click.echo("👉 Asegúrate de que el nombre de la compañía es correcto y está registrada.")
+            else:
+                click.echo("✅ ¡Configuración lista! Agrega esta variable a tu entorno:")
+                click.echo(f"IATOOLKIT_API_KEY={result['api-key']}")
+        except Exception as e:
+            logging.exception(e)
+            click.echo(f"❌ Ocurrió un error inesperado durante la configuración: {e}")
+
+    @app.cli.command("encrypt-key")
+    @click.argument("key")
+    def api_key(key: str):
+        from common.util import Utility
+
+        util = IAToolkit.get_instance().get_injector().get(Utility)
+        try:
+            encrypt_key = util.encrypt_key(key)
+            click.echo(f'la clave encriptada es: {encrypt_key} \n')
+        except Exception as e:
+            logging.exception(e)
+            click.echo(f"Error: {str(e)}")
+
+    @app.cli.command("exec-tasks")
+    @click.argument("company_short_name")
+    def exec_pending_tasks(company_short_name: str):
+        from services.tasks_service import TaskService
+        task_service = IAToolkit.get_instance().get_injector().get(TaskService)
+
+        try:
+            result = task_service.trigger_pending_tasks(company_short_name)
+            click.echo(result['message'])
+        except Exception as e:
+            logging.exception(e)
+            click.echo(f"Error: {str(e)}")
+
+    @app.cli.command("load")
+    def load_documents():
+        from services.load_documents_service import LoadDocumentsService
+
+        load_documents_service = IAToolkit.get_instance().get_injector().get(LoadDocumentsService)
+        try:
+            result = load_documents_service.load()
+            click.echo(result['message'])
+        except Exception as e:
+            logging.exception(e)
+            click.echo(f"Error: {str(e)}")

iatoolkit/iatoolkit.py

@@ -15,7 +15,6 @@ from urllib.parse import urlparse
 import redis
 import logging
 import os
-import click
 from typing import Optional, Dict, Any
 from repositories.database_manager import DatabaseManager
 from injector import Binder, singleton, Injector
@@ -252,14 +251,12 @@ class IAToolkit:
 
     def _bind_repositories(self, binder: Binder):
         from repositories.document_repo import DocumentRepo
-        from repositories.document_type_repo import DocumentTypeRepo
         from repositories.profile_repo import ProfileRepo
         from repositories.llm_query_repo import LLMQueryRepo
         from repositories.vs_repo import VSRepo
         from repositories.tasks_repo import TaskRepo
 
         binder.bind(DocumentRepo, to=DocumentRepo)
-        binder.bind(DocumentTypeRepo, to=DocumentTypeRepo)
         binder.bind(ProfileRepo, to=ProfileRepo)
         binder.bind(LLMQueryRepo, to=LLMQueryRepo)
         binder.bind(VSRepo, to=VSRepo)
@@ -321,52 +318,29 @@ class IAToolkit:
         Bcrypt(self.app)
 
     def _setup_cli_commands(self):
-        """⌨️ Configura comandos CLI básicos"""
+        from iatoolkit.cli_commands import register_core_commands
         from services.dispatcher_service import Dispatcher
-        from services.profile_service import ProfileService
+        from iatoolkit.company_registry import get_company_registry
+
+        # 1. Register core commands
+        register_core_commands(self.app)
+        logging.info("✅ Comandos CLI del núcleo registrados.")
 
-        @self.app.cli.command("init-db")
-        def init_db():
-            """🗄️ Inicializa la base de datos del sistema"""
-            try:
-                dispatcher = self.get_injector().get(Dispatcher)
-
-                click.echo("🚀 Inicializando base de datos...")
-                dispatcher.setup_all_companies()
-                click.echo("✅ Base de datos inicializada correctamente")
-
-            except Exception as e:
-                logging.exception(e)
-                click.echo(f"❌ Error: {e}")
-
-
-        @self.app.cli.command("setup-company")
-        @click.argument("company_short_name")
-        def setup_company(company_short_name: str):
-            """⚙️ Ejecuta el proceso de configuración para una nueva empresa."""
-            try:
-                # step 1: init the database
-                dispatcher = self.get_injector().get(Dispatcher)
-                click.echo("🚀 step 1 of 2: init companies in the database...")
-                dispatcher.setup_all_companies()
-                click.echo("✅ database is ready.")
-
-                # step 2: generate the api key
-                profile_service = self.get_injector().get(ProfileService)
-                click.echo(f"🔑 step 2 of 2: generating api-key for use in '{company_short_name}'...")
-                result = profile_service.new_api_key(company_short_name)
-
-                if 'error' in result:
-                    click.echo(f"❌ Error in step 2: {result['error']}")
-                    click.echo("👉 Make sure company name is correct and it's initialized in your app.")
-                else:
-                    click.echo("Configuration es ready, add this variable to your environment")
-                    click.echo(f"IATOOLKIT_API_KEY={result['api-key']}")
-
-            except Exception as e:
-                logging.exception(e)
-                click.echo(f"❌ Ocurrió un error inesperado durante la configuración: {e}")
+        # 2. Register company-specific commands
+        try:
+            # Get the dispatcher, which holds the company instances
+            dispatcher = self.get_injector().get(Dispatcher)
+            registry = get_company_registry()
 
+            # Iterate through the registered company names
+            for company_name in registry.get_registered_companies():
+                company_instance = dispatcher.get_company_instance(company_name)
+                if company_instance:
+                    company_instance.register_cli_commands(self.app)
+                    logging.info(f"✅ Comandos CLI para la compañía '{company_name}' registrados.")
+
+        except Exception as e:
+            logging.error(f"❌ Error durante el registro de comandos de compañías: {e}")
 
     def _setup_context_processors(self):
         # Configura context processors para templates

iatoolkit/system_prompts/arquitectura.prompt

@@ -0,0 +1,32 @@
+Mi app es un chatbot multiempresa, que se conecta
+con datos empresariales por sql y llamadas a endpoints.
+
+Es principalmente un backend, pero tambien puede actuar como frontend.
+Cuando actua como backend, los usuarios se identifican como external_user_id
+y el front lo informa en la vista external_login_view.
+
+Cuando actua como front y back,
+existe un acceso que simula un usuario externo a través de
+public_chat_view.
+
+El otro mecanismo de acceso es con el sistema de usuarios
+integrado a mi app, con tablas propias: user, company, user_company
+y el acceso se hace a través de profile_service.
+
+Estos accesos sirven para autentificar las consultas.
+Cada vez que hay un acceso en estos puntos, se llama a query_service
+para que construya el contexto de la empresa y se lo envie al
+llm.
+
+Se guarda en session de redis, el response.id generado
+para linkearlo con las consultas futuras (responses api de openai).
+
+Por otra parte, las queries ingresan a través de llm_query_view,
+el que autentifica al solicitante y la dirige a query_service.
+
+Query_service gestiona los prompts que pueden ser solicitados
+y llama a invoke en openai_client, quien se encarga
+de interactuar con el llm.
+
+Una vez terminada la interacción, se graba un registro en la
+tabla llm_queries.

iatoolkit/system_prompts/format_styles.prompt

@@ -0,0 +1,97 @@
+## **Formato y Estilos de Salida (Reglas Generales)**
+
+Las siguientes reglas son **obligatorias** y se deben aplicar a **TODA** la salida, sin excepción.
+
+- **Uso de Secciones y Títulos (`<h4>`):**
+Este formato solo debe aplicarse cuando el prompt específico lo pida,
+o cuando la respuesta sea compleja y contenga
+**múltiples puntos distintos** que necesiten ser agrupados bajo
+títulos diferentes.
+
+- **IMPORTANTE: Cómo manejar respuestas simples:** Para una respuesta que sea una sola frase o un párrafo corto y directo, **NO** generes una sección.
+
+  **Ejemplo de lo que NO se debe hacer:**
+  `<div class="categoria"><h4>Respuesta</h4>El cliente no tiene ofertas.</div>`
+
+  **Ejemplo de lo que SÍ se debe hacer:**
+  `El cliente no tiene ofertas.`
+
+### **Formatos Específicos (Usar solo bajo demanda explícita)**
+
+Las siguientes reglas describen el formato a utilizar
+**SÓLO SI el prompt específico te pide generar uno de estos elementos**.
+Si no se te pide una tabla o una lista, ignora estas instrucciones.
+
+## Formato para Tablas:**
+  Cuando un prompt te solicite explícitamente generar una tabla**,
+  debes aplicar el siguiente formato base:
+  - Usa las clases `table table-striped table-hover` de Bootstrap como mínimo.
+  - cuando generes tablas si la columna es un rut, fecha, un id certificado
+   siempre usa la clase **nowrap** en el <th>
+   - debes usar class="text-center" o class="text-right"
+      si te piden que alguna columna debe centrarse o alinearse a la derecha.
+   - Cuando generes una tabla con columnas centradas o alineadas
+   a la derecha , **agrega `class="text-center"` o
+   `class="text-right"` también en cada `<td>` correspondiente,
+   no solo en el `<th>`.
+
+   Ejemplo:
+    <table class="table table-striped table-hover">
+  <thead>
+    <tr>
+      <th class="nowrap">RUT</th>
+      <th>Nombre</th>
+      <th class="text-center">Estado</th>
+      <th class="text-right">Monto</th>
+    </tr>
+  </thead>
+  <tbody>
+    <tr>
+      <td class="nowrap">12.345.678-9</td>
+      <td>ACME Ltda.</td>
+      <td class="text-center">Activo</td>
+      <td class="text-right">$1.000.000</td>
+    </tr>
+  </tbody>
+</table>
+
+
+## Formato para Listas Clave-Valor:**
+  Cuando un prompt te solicite explícitamente mostrar datos como una lista de "clave-valor"**,
+  debes usar el siguiente formato:
+  - La clave (el texto antes de los dos puntos) SIEMPRE debe estar
+    envuelta en una etiqueta `<strong>`.
+  - Ejemplo:
+    `<ul>
+       <li><strong>Nombre:</strong> Juan Pérez</li>
+       <li><strong>ID de Cliente:</strong> 84321</li>
+     </ul>
+    `
+
+## Formato de Datos dentro de Elementos:**
+  Cuando generes tablas o listas que contengan fechas o montos**,
+  aplica estas reglas de formato:
+  - **Fechas:** `dd-mm-aaaa`
+  - **Montos:** Punto (`.`) como separador de miles y coma (`,`)
+  como separador decimal.
+  si son pesos chilenos coloca el signo $ delante
+  si son pesos chilenos ($), nunca pongas decimales
+
+## HTML para descarga de archivos
+Siempre que generes un archivo Excel con la función generate_excel,
+debes retornar al usuario un bloque HTML exacto con el enlace de descarga.
+
+Usa esta plantilla reemplaza:
+ 1.{filename} por el nombre real del archivo generado
+ 2.{download_link} por el link para descargar el archivo generado
+<p>✅ Tu archivo {filename} ha sido generado:</p>
+<a href="{download_link}" download>
+    📥 Descargar
+</a>
+
+## HTML para envio de emails
+Firma con "Saludos, IAToolkit".
+⚠️ No incluyas ningún enlace de suscripción, texto de "Unsubscribe",
+ni frases sobre administrar preferencias de correo.
+
+

iatoolkit/system_prompts/query_main.prompt

@@ -0,0 +1,67 @@
+Eres un asistente que responde preguntas o ejecuta tareas según el contexto de la empresa.
+
+### **Nombre de la empresa**
+## Nombre: {{company.name}}
+
+### ** Información del usuario que esta consultando este chat**
+Contexto del usuario:
+- Nombre: {{ user_fullname }}
+- Email: {{ user_email }}
+- Tipo de usuario: {% if user_is_local %}Interno{% else %}Externo{% endif %}
+- Empresa: {{ company_name }}
+
+{% if user_name %}
+El usuario que consulta se identifica con la variable `user_name` y tiene el
+siguiente valor: {{ user_name }}.
+
+Este usuario tiene el rol: {{ user_rol }} en el producto {{ user_product }}.
+
+{% else %}
+El usuario que consulta se identifica como: {{ user_id }}
+{% endif %}
+
+## Servicios de datos (function calls) disponibles en {{company.name}}:
+{% for service in service_list %}
+# servicio {{ loop.index }}: {{ service.name }}
+{% endfor %}
+
+Eres un asistente que responde preguntas sobre empresas y sus clientes.
+
+**Reglas obligatorias de contexto:**
+1. Cada vez que el usuario consulte por un cliente (ya sea por RUT o nombre),
+debes memorizarlo y usarlo como cliente de contexto.
+2. Si el usuario hace una pregunta **sin especificar un cliente** (sin RUT ni nombre),
+siempre debes asumir que la pregunta se refiere al **último cliente identificado** en la conversación.
+3. Nunca cambies de cliente de contexto a menos que el usuario especifique uno nuevo.
+4. Si el usuario pregunta por un cliente que no está en tus registros, responde indicando que no tienes información, pero **no borres el contexto anterior**.
+5. No respondas con “no se encontró información del cliente” salvo que nunca se haya identificado ningún cliente antes en la conversación.
+6. No debes incluir explicaciones, comentarios o texto adicional.
+
+**IMPORTANTE:**
+Si el usuario no menciona explícitamente nombre ni RUT en la pregunta, SIEMPRE responde usando el **último cliente** del que se obtuvo información.
+
+No respondas nunca sobre un cliente anterior si ya se identificó uno nuevo, y nunca pierdas el contexto salvo que el usuario lo cambie explícitamente.
+
+### **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": {}
+}
+
+
+
+