iatoolkit 0.3.8__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

@@ -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/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.3.9.dist-info}/METADATA

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

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

@@ -1,7 +1,8 @@
-iatoolkit/__init__.py,sha256=JdcXIcvFFTMy2XI9ccEws_pCklilDkPPEb7RaQNf7oY,1444
-iatoolkit/base_company.py,sha256=gIFaMbQcZ5RAegfNxpjhFLacnOtCvB3sBiXpMtjfM1w,1859
+iatoolkit/__init__.py,sha256=dKjc0IsztMClwbalF5hbtz418TRtu0xE2pdb8J3iQQw,1512
+iatoolkit/base_company.py,sha256=FlB-HFYH8FoTl4nbtsYgfKjkdZtizJbKwXqaosxmRqc,2009
+iatoolkit/cli_commands.py,sha256=oWd5kwDYd0W1Lcpuk3N2cEnusPAVefaCrYveMQ1zDvY,3223
 iatoolkit/company_registry.py,sha256=cRaez-VBo1icnUNKmkQqo_Xlr8UKWFoYEMZ70XP6Jgk,2702
-iatoolkit/iatoolkit.py,sha256=OKGh5bYQAlyBn8_iI5UKgVykD9V0lW9x4wZE5rCSp5A,18054
+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=vZbz-XCP0Aa-oE4nh5ZKeUKWlJ78VPSRYkpzMHlVFM0,14952
 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
@@ -25,7 +26,7 @@ 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.3.9.dist-info/METADATA,sha256=wLPKZh6G6AXabwamba8ywnayHiCCwHYiNWVfqH7JK9I,8801
+iatoolkit-0.3.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+iatoolkit-0.3.9.dist-info/top_level.txt,sha256=dqlBbmgo9okD9d_WMR9uYzdup7Rxgj26yFF85jRGeu4,19
+iatoolkit-0.3.9.dist-info/RECORD,,

services/dispatcher_service.py

@@ -238,6 +238,10 @@ 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_classes.get(company_name)
+
     def get_registered_companies(self) -> dict:
         """Obtiene todas las empresas registradas (para debugging/admin)"""
         return {