iatoolkit 0.3.6__py3-none-any.whl → 0.3.8__py3-none-any.whl

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:

iatoolkit/iatoolkit.py

@@ -19,10 +19,10 @@ import click
 from typing import Optional, Dict, Any
 from repositories.database_manager import DatabaseManager
 from injector import Binder, singleton, Injector
+from importlib.metadata import version as _pkg_version, PackageNotFoundError
 
-VERSION = "2.0.0"
 
-# global variable for yhe unique instance of IAToolkit
+# global variable for the unique instance of IAToolkit
 _iatoolkit_instance: Optional['IAToolkit'] = None
 
 
@@ -53,6 +53,7 @@ class IAToolkit:
         self.app = None
         self.db_manager = None
         self._injector = None
+        self.version = "0.0.0+dev"
 
     @classmethod
     def get_instance(cls) -> 'IAToolkit':
@@ -69,6 +70,9 @@ class IAToolkit:
             Creates, configures, and returns the Flask application instance.
             this is the main entry point for the application factory.
         """
+        if self._initialized and self.app:
+            return self.app
+
         self._setup_logging()
 
         # Step 1: Create the Flask app instance
@@ -94,7 +98,13 @@ class IAToolkit:
         self._setup_cli_commands()
         self._setup_context_processors()
 
-        logging.info(f"🎉 IAToolkit v{VERSION} inicializado correctamente")
+        try:
+            self.version = _pkg_version("iatoolkit")
+        except PackageNotFoundError:
+            pass
+
+        logging.info(f"🎉 IAToolkit v{self.version} inicializado correctamente")
+        self._initialized = True
         return self.app
 
     def _get_config_value(self, key: str, default=None):
@@ -133,7 +143,7 @@ class IAToolkit:
         is_dev = self._get_config_value('FLASK_ENV') == 'development'
 
         self.app.config.update({
-            'VERSION': VERSION,
+            'VERSION': self.version,
             'SECRET_KEY': self._get_config_value('FLASK_SECRET_KEY', 'iatoolkit-default-secret'),
             'SESSION_COOKIE_SAMESITE': "None" if is_https else "Lax",
             'SESSION_COOKIE_SECURE': is_https,
@@ -192,9 +202,9 @@ class IAToolkit:
         """🌐 Configura CORS"""
         # Origins por defecto para desarrollo
         default_origins = [
-            "http://localhost:3000",
             "http://localhost:5001",
-            "http://127.0.0.1:5001"
+            "http://127.0.0.1:5001",
+            os.getenv('IATOOLKIT_BASE_URL')
         ]
 
         # Obtener origins adicionales desde configuración/env
@@ -242,14 +252,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)
@@ -315,8 +323,8 @@ class IAToolkit:
         from services.dispatcher_service import Dispatcher
         from services.profile_service import ProfileService
 
-        @self.app.cli.command("init-db")
-        def init_db():
+        @self.app.cli.command("setup_all_companies")
+        def setup_all_companies():
             """🗄️ Inicializa la base de datos del sistema"""
             try:
                 dispatcher = self.get_injector().get(Dispatcher)
@@ -357,6 +365,29 @@ class IAToolkit:
                 logging.exception(e)
                 click.echo(f"❌ Ocurrió un error inesperado durante la configuración: {e}")
 
+        @self.app.cli.command("populate-sample-db")
+        def populate_sample_db():
+            from companies.sample_company.sample_company import SampleCompany
+            """📦 Crea y puebla la base de datos de sample_company con datos de prueba."""
+            try:
+                company_instance = self.get_injector().get(SampleCompany)
+                click.echo("🚀 Obteniendo instancia de 'sample_company'...")
+
+                if not company_instance or not hasattr(company_instance, 'sample_database') or not company_instance.sample_database:
+                    click.echo("❌ Error: No se pudo obtener la instancia de 'sample_company' o su base de datos no está configurada.")
+                    click.echo("👉 Asegúrate de que 'sample_company' esté registrada y que la variable de entorno 'SAMPLE_DATABASE_URI' esté definida.")
+                    return
+
+                click.echo("⚙️  Creando y poblando la base de datos. Esto puede tardar unos momentos...")
+
+                company_instance.sample_database.create_database()
+                company_instance.sample_database.populate_database()
+
+                click.echo("✅ Base de datos de 'sample_company' poblada exitosamente.")
+
+            except Exception as e:
+                logging.exception(e)
+                click.echo(f"❌ Ocurrió un error inesperado: {e}")
 
     def _setup_context_processors(self):
         # Configura context processors para templates
@@ -364,7 +395,7 @@ class IAToolkit:
         def inject_globals():
             return {
                 'url_for': url_for,
-                'iatoolkit_version': VERSION,
+                'iatoolkit_version': self.version,
                 'app_name': 'IAToolkit',
                 'user': SessionManager.get('user'),
                 'user_company': SessionManager.get('company_short_name'),

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": {}
+}
+
+
+
+