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

iatoolkit/__init__.py

@@ -23,6 +23,7 @@ from services.document_service import DocumentService
 from services.search_service import SearchService
 from repositories.profile_repo import ProfileRepo
 from repositories.llm_query_repo import LLMQueryRepo
+from services.query_service import QueryService
 from repositories.database_manager import DatabaseManager
 from infra.call_service import CallServiceClient
 from common.util import Utility
@@ -40,6 +41,7 @@ __all__ = [
     'Dispatcher',
     'DocumentService',
     'SearchService',
+    'QueryService',
     'ProfileRepo',
     'LLMQueryRepo',
     'DatabaseManager',

iatoolkit/base_company.py

@@ -43,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/company_registry.py

@@ -46,6 +46,10 @@ class CompanyRegistry:
 
         return self._company_instances.copy()
 
+    def get_all_company_instances(self) -> Dict[str, BaseCompany]:
+        """Devuelve un diccionario con todas las instancias de empresas creadas."""
+        return self._company_instances.copy()
+
     def get_registered_companies(self) -> Dict[str, Type[BaseCompany]]:
         return self._company_classes.copy()
 

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
@@ -319,75 +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.")
+
+        # 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.")
 
-        @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)
-
-                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}")
-
-        @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}")
+        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-0.3.8.dist-info → iatoolkit-0.4.0.dist-info}/METADATA

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

{iatoolkit-0.3.8.dist-info → iatoolkit-0.4.0.dist-info}/RECORD

@@ -1,7 +1,8 @@
-iatoolkit/__init__.py,sha256=JdcXIcvFFTMy2XI9ccEws_pCklilDkPPEb7RaQNf7oY,1444
-iatoolkit/base_company.py,sha256=gIFaMbQcZ5RAegfNxpjhFLacnOtCvB3sBiXpMtjfM1w,1859
-iatoolkit/company_registry.py,sha256=cRaez-VBo1icnUNKmkQqo_Xlr8UKWFoYEMZ70XP6Jgk,2702
-iatoolkit/iatoolkit.py,sha256=OKGh5bYQAlyBn8_iI5UKgVykD9V0lW9x4wZE5rCSp5A,18054
+iatoolkit/__init__.py,sha256=WImnnjDcaWy8_CoIS4UA2PwqYcMb0dDNUhg-OlkZ2X8,1512
+iatoolkit/base_company.py,sha256=FlB-HFYH8FoTl4nbtsYgfKjkdZtizJbKwXqaosxmRqc,2009
+iatoolkit/cli_commands.py,sha256=oWd5kwDYd0W1Lcpuk3N2cEnusPAVefaCrYveMQ1zDvY,3223
+iatoolkit/company_registry.py,sha256=HnDpVyCc41OAn-exVF53b_HMES7GelWZcvxR39S_nI4,2900
+iatoolkit/iatoolkit.py,sha256=OwlGujwtNLBYtfZuCpcX_yzrgB8BVo9Jfh72owM8FFc,15651
 iatoolkit/system_prompts/arquitectura.prompt,sha256=2W-7NWy6P6y1Gh5_-zD1iK-BWq1Siu8TuvGCouP67bQ,1267
 iatoolkit/system_prompts/format_styles.prompt,sha256=MSMe1qvR3cF_0IbFshn8R0z6Wx6VCHQq1p37rpu5wwk,3576
 iatoolkit/system_prompts/query_main.prompt,sha256=Eu5VOQzUygJ45Ct1WKYGbi0JMltgI6FQIZWlGmN1bdk,3214
@@ -9,7 +10,7 @@ iatoolkit/system_prompts/sql_rules.prompt,sha256=y4nURVnb9AyFwt-lrbMNBHHtZlhk6kC
 services/__init__.py,sha256=fSvSfIcPW1dHwTBY1hQ5dBEhaoorzk_GzR4G46gD8tY,173
 services/api_service.py,sha256=InIKTc64BWcp4U4tYKHz28x4ErPxIfvR9x3ZlxJZlXs,2911
 services/benchmark_service.py,sha256=g9JVrmAqIe_iI0D1DwdQ6DJ2_FJRCTndarESNSVfhbw,5907
-services/dispatcher_service.py,sha256=wWIw9QwyNZ6yaDbANkNdAkdp-f_EAHbJFegjWDcxTAM,14781
+services/dispatcher_service.py,sha256=jGixvvQ4DTQGZye8aa05q56B6U-s6NaDb6he6UTXmQc,15534
 services/document_service.py,sha256=sm5QtbrKs2dF9hpLuSLMB-IMWYNBD7yWHv3rd80aD0o,5960
 services/excel_service.py,sha256=wE9Udbyb96kGRSnZZ6KM2mbE484rKjTEhta9GKKpy-8,3630
 services/file_processor_service.py,sha256=82UArWtwpr94CAMkkoRP0_nPtoqItymdKSIABS0Xkxw,2943
@@ -18,14 +19,14 @@ services/jwt_service.py,sha256=dC45Sn6FyzdzRiQJnzgkjN3Hy21V1imRxB0hTyWRvlA,3979
 services/load_documents_service.py,sha256=_-OTUih8Zk0m4dHqAhkE7kAwU2mbz_QoMrOKnrq7ZWs,8821
 services/mail_service.py,sha256=ystFit1LuYUC4ekYYebyiy1rqYQmxeL6K8h58MxEkOY,2233
 services/profile_service.py,sha256=vZV0cregZQiPKYcNLaD7xjez2y6-3Mq97cDndC8NL8w,17922
-services/prompt_manager_service.py,sha256=g499zeWZODqoDvqQZX6eHWmsWj7oLWkEhge7UV_y8IE,7679
+services/prompt_manager_service.py,sha256=bWG4SIgt0u45PVUfm0xRLbLfKC7bk6uozVHRdkdgCmc,7761
 services/query_service.py,sha256=zpaDzjzh2HqAG1F2Ap8WHBpfAMVtzZ_6v1JGVEguwvs,15687
 services/search_service.py,sha256=oJD6WRXCJBD7WUVHWWKxexRkhR8nQSrFtcPV3pFO2KQ,1153
 services/sql_service.py,sha256=H7CIPpXTcxLXLojD2fBFr_mIAD0PW1vEJhKHLfJi4Hk,1418
 services/tasks_service.py,sha256=hHJDlcsSOPtEleD6_Vv3pocfxWNmthIhmZSdnoWFpEM,6861
 services/user_feedback_service.py,sha256=YtCndRBekDEWYEbac431Ksn2gMO5iBrI3WqKK0xtShE,2513
 services/user_session_context_service.py,sha256=5qn7fqpuiU8KgMpU4M5-iRUsETumz1raBw-EeZLuE1A,3868
-iatoolkit-0.3.8.dist-info/METADATA,sha256=ojUeydEjMKJIbnIpXnuYhgzdN3Q0PcWnDf_MbBcojtY,8801
-iatoolkit-0.3.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-iatoolkit-0.3.8.dist-info/top_level.txt,sha256=dqlBbmgo9okD9d_WMR9uYzdup7Rxgj26yFF85jRGeu4,19
-iatoolkit-0.3.8.dist-info/RECORD,,
+iatoolkit-0.4.0.dist-info/METADATA,sha256=1muE8emXWndqmPwC3xhtpVYjrFiktvWdWdD_UckbW10,8801
+iatoolkit-0.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+iatoolkit-0.4.0.dist-info/top_level.txt,sha256=dqlBbmgo9okD9d_WMR9uYzdup7Rxgj26yFF85jRGeu4,19
+iatoolkit-0.4.0.dist-info/RECORD,,

services/dispatcher_service.py

@@ -3,7 +3,6 @@
 # Todos los derechos reservados.
 # En trámite de registro en el Registro de Propiedad Intelectual de Chile.
 
-from iatoolkit import current_iatoolkit
 from common.exceptions import IAToolkitException
 from services.prompt_manager_service import PromptService
 from services.api_service import ApiService
@@ -11,10 +10,10 @@ from repositories.llm_query_repo import LLMQueryRepo
 from repositories.models import Company, Function
 from services.excel_service import ExcelService
 from services.mail_service import MailService
-from iatoolkit.company_registry import get_company_registry
 from common.session_manager import SessionManager
 from common.util import Utility
 from injector import inject
+from typing import Dict
 import logging
 import os
 
@@ -37,11 +36,10 @@ class Dispatcher:
         self.system_functions = _FUNCTION_LIST
         self.system_prompts = _SYSTEM_PROMPT
 
-        # Use the global registry
-        self.company_registry = get_company_registry()
+        self._company_registry = None
+        self._company_instances = None
 
         # load into the dispatcher the configured companies
-        self.company_classes = {}
         self.initialize_companies()
 
         # run the statrtup logic for all companies
@@ -53,23 +51,40 @@ class Dispatcher:
             "iat_api_call": self.api_service.call_api
         }
 
+    @property
+    def company_registry(self):
+        """Lazy-loads and returns the CompanyRegistry instance."""
+        if self._company_registry is None:
+            from iatoolkit.company_registry import get_company_registry
+            self._company_registry = get_company_registry()
+        return self._company_registry
+
+    @property
+    def company_instances(self):
+        """Lazy-loads and returns the instantiated company classes."""
+        if self._company_instances is None:
+            self._company_instances = self.company_registry.get_all_company_instances()
+        return self._company_instances
+
     def initialize_companies(self):
+        from iatoolkit import current_iatoolkit
         """
         Initializes and instantiates all registered company classes.
         This method should be called *after* the main injector is fully configured
         and the company registry is populated.
         """
-        if self.company_classes: # Prevent re-initialization
+        if self.company_registry.get_all_company_instances():  # Check if already instantiated
             return
 
         # ✅ NOW it is safe to get the injector and instantiate companies.
         injector = current_iatoolkit().get_injector()
         self.company_registry.set_injector(injector)
-        self.company_classes = self.company_registry.instantiate_companies()
+        self.company_registry.instantiate_companies()
+
 
     def start_execution(self):
         """Runs the startup logic for all registered companies."""
-        for company_name, company_instance in self.company_classes.items():
+        for company_name, company_instance in self.company_instances.items():
             logging.info(f'Starting execution for company: {company_name}')
             company_instance.start_execution()
 
@@ -100,14 +115,14 @@ class Dispatcher:
                 i += 1
 
         # register in the database  every company class
-        for company in self.company_classes.values():
+        for company in self.company_instances.values():
             company.register_company()
 
     def dispatch(self, company_name: str, action: str, **kwargs) -> str:
         company_key = company_name.lower()
 
-        if company_key not in self.company_classes:
-            available_companies = list(self.company_classes.keys())
+        if company_key not in self.company_instances:
+            available_companies = list(self.company_instances.keys())
             raise IAToolkitException(
                 IAToolkitException.ErrorType.EXTERNAL_SOURCE_ERROR,
                 f"Empresa '{company_name}' no configurada. Empresas disponibles: {available_companies}"
@@ -117,7 +132,7 @@ class Dispatcher:
         if action in self.tool_handlers:
             return  self.tool_handlers[action](**kwargs)
 
-        company_instance = self.company_classes[company_name]
+        company_instance = self.company_instances[company_name]
         try:
             return company_instance.handle_request(action, **kwargs)
         except IAToolkitException as e:
@@ -130,7 +145,7 @@ class Dispatcher:
                                f"Error en function call '{action}': {str(e)}") from e
 
     def get_company_context(self, company_name: str, **kwargs) -> str:
-        if company_name not in self.company_classes:
+        if company_name not in self.company_instances:
             raise IAToolkitException(IAToolkitException.ErrorType.EXTERNAL_SOURCE_ERROR,
                                f"Empresa no configurada: {company_name}")
 
@@ -152,7 +167,7 @@ class Dispatcher:
             filepath = os.path.join(schema_dir, file)
             company_context += self.util.generate_context_for_schema(schema_name, filepath)
 
-        company_instance = self.company_classes[company_name]
+        company_instance = self.company_instances[company_name]
         try:
             return company_context + company_instance.get_company_context(**kwargs)
         except Exception as e:
@@ -180,7 +195,7 @@ class Dispatcher:
         return tools
 
     def get_user_info(self, company_name: str, user_identifier: str, is_local_user: bool) -> dict:
-        if company_name not in self.company_classes:
+        if company_name not in self.company_instances:
             raise IAToolkitException(IAToolkitException.ErrorType.EXTERNAL_SOURCE_ERROR,
                                      f"Empresa no configurada: {company_name}")
 
@@ -190,7 +205,7 @@ class Dispatcher:
             raw_user_data = SessionManager.get('user', {})
         else:
             # source 2: external company user
-            company_instance = self.company_classes[company_name]
+            company_instance = self.company_instances[company_name]
             try:
                 raw_user_data = company_instance.get_user_info(user_identifier)
             except Exception as e:
@@ -226,11 +241,11 @@ class Dispatcher:
         return normalized_user
 
     def get_metadata_from_filename(self, company_name: str, filename: str) -> dict:
-        if company_name not in self.company_classes:
+        if company_name not in self.company_instances:
             raise IAToolkitException(IAToolkitException.ErrorType.EXTERNAL_SOURCE_ERROR,
                                f"Empresa no configurada: {company_name}")
 
-        company_instance = self.company_classes[company_name]
+        company_instance = self.company_instances[company_name]
         try:
             return company_instance.get_metadata_from_filename(filename)
         except Exception as e:
@@ -238,12 +253,16 @@ class Dispatcher:
             raise IAToolkitException(IAToolkitException.ErrorType.EXTERNAL_SOURCE_ERROR,
                                f"Error en get_metadata_from_filename de {company_name}: {str(e)}") from e
 
+    def get_company_instance(self, company_name: str):
+        """Returns the instance for a given company name."""
+        return self.company_instances.get(company_name)
+
     def get_registered_companies(self) -> dict:
-        """Obtiene todas las empresas registradas (para debugging/admin)"""
+        """Gets all registered companies (for debugging/admin purposes)"""
         return {
             "registered_classes": list(self.company_registry.get_registered_companies().keys()),
-            "instantiated": list(self.company_classes.keys()),
-            "count": len(self.company_classes)
+            "instantiated": list(self.company_instances.keys()),
+            "count": len(self.company_instances)
         }
 
 

services/prompt_manager_service.py

@@ -74,7 +74,8 @@ class PromptService:
                 raise IAToolkitException(IAToolkitException.ErrorType.DOCUMENT_NOT_FOUND,
                                    f"No se encontró el prompt '{prompt_name}' para la empresa '{company.short_name}'")
 
-            absolute_filepath = os.path.join(execution_dir, user_prompt.filename)
+            prompt_file = f'companies/{company.short_name}/prompts/{user_prompt.filename}'
+            absolute_filepath = os.path.join(execution_dir, prompt_file)
             if not os.path.exists(absolute_filepath):
                 raise IAToolkitException(IAToolkitException.ErrorType.FILE_IO_ERROR,
                                    f"El archivo para el prompt '{prompt_name}' no existe: {absolute_filepath}")