iatoolkit 0.3.3__py3-none-any.whl → 0.3.4__py3-none-any.whl

iatoolkit/__init__.py

@@ -22,7 +22,11 @@ from services.dispatcher_service import Dispatcher
 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 repositories.database_manager import DatabaseManager
+from infra.call_service import CallServiceClient
+from common.util import Utility
+from repositories.models import Base, Company, Function, TaskType
 
 
 __all__ = [
@@ -37,5 +41,12 @@ __all__ = [
     'DocumentService',
     'SearchService',
     'ProfileRepo',
+    'LLMQueryRepo',
     'DatabaseManager',
+    'CallServiceClient',
+    'Utility',
+    'Company',
+    'Function',
+    'TaskType',
+    'Base',
 ]

iatoolkit/company_registry.py

@@ -8,10 +8,8 @@ import logging
 
 class CompanyRegistry:
     """
-    Registro centralizado de empresas para iatoolkit.
-
-    Permite a los clientes registrar sus clases de empresa de forma explícita
-    en lugar de usar autodiscovery.
+    Company registry with dependency injection support.
+    Allow the client to register companies and instantiate them with dependency injection.
     """
 
     def __init__(self):
@@ -19,22 +17,6 @@ class CompanyRegistry:
         self._company_instances: Dict[str, BaseCompany] = {}
         self._injector = None
 
-    def register_company(self, name: str, company_class: Type[BaseCompany]) -> None:
-        """
-        Registra una clase de empresa.
-
-        Args:
-            name: Nombre de la empresa (ej: 'maxxa')
-            company_class: Clase que hereda de BaseCompany
-        """
-        if not issubclass(company_class, BaseCompany):
-            raise ValueError(f"La clase {company_class.__name__} debe heredar de BaseCompany")
-
-        company_key = name.lower()
-        self._company_classes[company_key] = company_class
-
-        logging.info(f"Empresa registrada: {company_key} -> {company_class.__name__}")
-
     def set_injector(self, injector) -> None:
         """Establece el injector para crear instancias con dependencias"""
         self._injector = injector
@@ -65,34 +47,33 @@ class CompanyRegistry:
         return self._company_instances.copy()
 
     def get_registered_companies(self) -> Dict[str, Type[BaseCompany]]:
-        """Retorna las clases registradas"""
         return self._company_classes.copy()
 
-    def get_company_instances(self) -> Dict[str, BaseCompany]:
-        """Retorna las instancias de empresas"""
-        return self._company_instances.copy()
-
     def clear(self) -> None:
         """Limpia el registro (útil para tests)"""
         self._company_classes.clear()
         self._company_instances.clear()
 
 
-# Instancia global del registry
+# global instance of the company registry
 _company_registry = CompanyRegistry()
 
 
 def register_company(name: str, company_class: Type[BaseCompany]) -> None:
     """
-    Función pública para registrar empresas.
+    Public function to register a company.
 
     Args:
-        name: Nombre de la empresa
-        company_class: Clase que hereda de BaseCompany
+        name: Name of the company
+        company_class: Class that inherits from BaseCompany
     """
-    _company_registry.register_company(name, company_class)
+    if not issubclass(company_class, BaseCompany):
+        raise ValueError(f"La clase {company_class.__name__} debe heredar de BaseCompany")
+
+    company_key = name.lower()
+    _company_registry._company_classes[company_key] = company_class
 
 
 def get_company_registry() -> CompanyRegistry:
-    """Obtiene el registry global"""
+    """get the global company registry instance"""
     return _company_registry

iatoolkit/iatoolkit.py

@@ -19,7 +19,6 @@ import click
 from typing import Optional, Dict, Any
 from repositories.database_manager import DatabaseManager
 from injector import Binder, singleton, Injector
-from .toolkit_config import IAToolkitConfig
 
 VERSION = "2.0.0"
 
@@ -51,9 +50,9 @@ class IAToolkit:
             return
 
         self.config = config or {}
-        self.app: Optional[Flask] = None
-        self.db_manager: Optional[DatabaseManager] = None
-        self._injector: Optional[Injector] = None
+        self.app = None
+        self.db_manager = None
+        self._injector = None
 
     @classmethod
     def get_instance(cls) -> 'IAToolkit':
@@ -68,7 +67,7 @@ class IAToolkit:
     def create_iatoolkit(self):
         """
             Creates, configures, and returns the Flask application instance.
-            his is the main entry point for the application factory.
+            this is the main entry point for the application factory.
         """
         self._setup_logging()
 
@@ -78,12 +77,8 @@ class IAToolkit:
         # Step 2: Set up the core components that DI depends on
         self._setup_database()
 
-        # Step 3: Create the Injector with CORE dependencies (NO VIEWS)
-        toolkit_config_module = IAToolkitConfig(app=self.app, db_manager=self.db_manager)
-        self._injector = Injector([
-            toolkit_config_module,
-            self._configure_core_dependencies # This method binds services, repos, etc.
-        ])
+        # Step 3: Create the Injector and configure all dependencies in one place
+        self._injector = Injector(self._configure_core_dependencies)
 
         # Step 4: Register routes using the fully configured injector
         self._register_routes()
@@ -102,7 +97,6 @@ class IAToolkit:
         logging.info(f"🎉 IAToolkit v{VERSION} inicializado correctamente")
         return self.app
 
-
     def _get_config_value(self, key: str, default=None):
         """Obtiene un valor de configuración, primero del dict config, luego de env vars"""
         return self.config.get(key, os.getenv(key, default))
@@ -228,7 +222,8 @@ class IAToolkit:
         """⚙️ Configures all system dependencies."""
         try:
             # Core dependencies
-            binder.bind(Injector, to=self._injector, scope=singleton)
+            binder.bind(Flask, to=self.app, scope=singleton)
+            binder.bind(DatabaseManager, to=self.db_manager, scope=singleton)
 
             # Bind all application components by calling the specific methods
             self._bind_repositories(binder)
@@ -317,16 +312,17 @@ class IAToolkit:
 
     def _setup_cli_commands(self):
         """⌨️ Configura comandos CLI básicos"""
+        from services.dispatcher_service import Dispatcher
+        from services.profile_service import ProfileService
 
         @self.app.cli.command("init-db")
         def init_db():
             """🗄️ Inicializa la base de datos del sistema"""
             try:
-                from services.dispatcher_service import Dispatcher
-                dispatcher = self._get_injector().get(Dispatcher)
+                dispatcher = self.get_injector().get(Dispatcher)
 
                 click.echo("🚀 Inicializando base de datos...")
-                dispatcher.init_db()
+                dispatcher.setup_all_companies()
                 click.echo("✅ Base de datos inicializada correctamente")
 
             except Exception as e:
@@ -334,6 +330,34 @@ class IAToolkit:
                 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}")
+
+
     def _setup_context_processors(self):
         # Configura context processors para templates
         @self.app.context_processor
@@ -362,7 +386,7 @@ class IAToolkit:
         except:
             return 'templates'
 
-    def _get_injector(self) -> Injector:
+    def get_injector(self) -> Injector:
         """Obtiene el injector actual"""
         if not self._injector:
             raise IAToolkitException(

{iatoolkit-0.3.3.dist-info → iatoolkit-0.3.4.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.3.3
+Version: 0.3.4
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT
-Requires-Python: >=3.10
+Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 Requires-Dist: aiohappyeyeballs==2.4.4
 Requires-Dist: aiohttp==3.11.9

{iatoolkit-0.3.3.dist-info → iatoolkit-0.3.4.dist-info}/RECORD

@@ -1,11 +1,11 @@
-iatoolkit/__init__.py,sha256=Yu9XbQmEjs0gA1HBnqiaORL7HUfoKuEElUIs8scg4IQ,1125
+iatoolkit/__init__.py,sha256=JdcXIcvFFTMy2XI9ccEws_pCklilDkPPEb7RaQNf7oY,1444
 iatoolkit/base_company.py,sha256=VWfpNofFlmgHQQK8BCw2KSOPLd2DM9eA68lGQ3SDE7M,1639
-iatoolkit/company_registry.py,sha256=saOf5JxsEtvAs-JqkNp-UfOIQvtVCXyujvv6G5_zi5k,3332
-iatoolkit/iatoolkit.py,sha256=YX5Oh-WWBKX0DDrzeOeaNoBrPLm0O7t4ttntYC_rk-8,15186
+iatoolkit/company_registry.py,sha256=cRaez-VBo1icnUNKmkQqo_Xlr8UKWFoYEMZ70XP6Jgk,2702
+iatoolkit/iatoolkit.py,sha256=K7d8fKKTk-Q-sMW6Ip58oleiosRP4e-pdc3E5GSoSMw,16450
 services/__init__.py,sha256=fSvSfIcPW1dHwTBY1hQ5dBEhaoorzk_GzR4G46gD8tY,173
-services/api_service.py,sha256=VTM5OfKF7-eCT-dpl46ZnZ3ptbTxLUDH2IxnbS4T3jA,1113
+services/api_service.py,sha256=InIKTc64BWcp4U4tYKHz28x4ErPxIfvR9x3ZlxJZlXs,2911
 services/benchmark_service.py,sha256=g9JVrmAqIe_iI0D1DwdQ6DJ2_FJRCTndarESNSVfhbw,5907
-services/dispatcher_service.py,sha256=iNu_gUZS_AgbwmCxtLBk0A33-jxsnMGYAKI5ScaJ-yE,13207
+services/dispatcher_service.py,sha256=CBslIE5FsrST46JjpKW1k_XHnIRtN9WehEr6LKadZlU,13360
 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
@@ -13,15 +13,15 @@ services/history_service.py,sha256=dl-D7qgdnzpY9QhjuxJokBYZZ1AF0y59HbRzwpPet58,1
 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=z1dSX45HDv8VJE4w8-ucpb3kBpE2fkyOdtgnrHc9vuE,17918
-services/prompt_manager_service.py,sha256=nnr_iNIFeJzOY_1fbuu5-reyJS04RvGx5JW_d3YFUbk,7966
+services/profile_service.py,sha256=YtnlXofXtvud4AHOFMmPlX9VO7mhs_Fglpc1PTulExc,17861
+services/prompt_manager_service.py,sha256=kKsqZyt2ZUWIHYTA5C6sfBk8sbXhvYF4QAUN9sOYk_s,7915
 services/query_service.py,sha256=Fx_P1WcuoAp_TyocN5LGlv_hc_03ImzU5QdAIqQg0ek,15591
 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.3.dist-info/METADATA,sha256=mYfAO_d2KXSqC3Zcf61oURv1MDabe7ax7calsdKECG0,8801
-iatoolkit-0.3.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-iatoolkit-0.3.3.dist-info/top_level.txt,sha256=dqlBbmgo9okD9d_WMR9uYzdup7Rxgj26yFF85jRGeu4,19
-iatoolkit-0.3.3.dist-info/RECORD,,
+iatoolkit-0.3.4.dist-info/METADATA,sha256=fiepVsfr8K9meQEu6b5xlOFZX4KJ1npM2m9mMgC6Y9Q,8801
+iatoolkit-0.3.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+iatoolkit-0.3.4.dist-info/top_level.txt,sha256=dqlBbmgo9okD9d_WMR9uYzdup7Rxgj26yFF85jRGeu4,19
+iatoolkit-0.3.4.dist-info/RECORD,,

services/api_service.py

@@ -7,24 +7,69 @@ from infra.call_service import CallServiceClient
 from injector import inject
 from common.exceptions import IAToolkitException
 import json
-
+from typing import Optional, Dict, Any, Union
 
 class ApiService:
     @inject
     def __init__(self, call_service: CallServiceClient):
         self.call_service = call_service
 
-    def call_api(self, endpoint: str, method: str, **kwargs):
-        if method == 'get':
-            response, status_code = self.call_service.get(endpoint)
-        elif method == 'post':
-            response, status_code = self.call_service.post(endpoint=endpoint, json_dict=kwargs)
+    def call_api(
+        self,
+        endpoint: str,
+        method: str,
+        headers: Optional[Dict[str, str]] = None,
+        params: Optional[Dict[str, Union[str, int, float, bool]]] = None,
+        body: Optional[Dict[str, Any]] = None,
+        files: Optional[Dict[str, Any]] = None,
+        timeout: Union[int, float, tuple] = 10
+    ) -> str:
+        """
+        Ejecuta una llamada HTTP genérica.
+
+        - endpoint: URL completa
+        - method: GET | POST | PUT | DELETE (case-insensitive)
+        - headers: dict opcional de cabeceras
+        - params: dict opcional de query string
+        - body: dict opcional para JSON (POST/PUT/DELETE)
+        - files: dict opcional para multipart/form-data (prioriza POST files)
+        - timeout: segundos (int/float) o tuple (connect, read)
+        """
+        m = (method or "").strip().lower()
+
+        if m == "get":
+            response, status_code = self.call_service.get(
+                endpoint, params=params, headers=headers, timeout=timeout
+            )
+        elif m == "post":
+            # Si vienen files → multipart; si no → JSON
+            if files:
+                response, status_code = self.call_service.post_files(
+                    endpoint, data=files, params=params, headers=headers, timeout=timeout
+                )
+            else:
+                response, status_code = self.call_service.post(
+                    endpoint=endpoint, json_dict=body, params=params, headers=headers, timeout=timeout
+                )
+        elif m == "put":
+            response, status_code = self.call_service.put(
+                endpoint, json_dict=body, params=params, headers=headers, timeout=timeout
+            )
+        elif m == "delete":
+            response, status_code = self.call_service.delete(
+                endpoint, json_dict=body, params=params, headers=headers, timeout=timeout
+            )
         else:
-            raise IAToolkitException(IAToolkitException.ErrorType.INVALID_PARAMETER,
-                               f'API error, {method} not supported')
+            raise IAToolkitException(
+                IAToolkitException.ErrorType.INVALID_PARAMETER,
+                f"API error: método '{method}' no soportado"
+            )
 
-        if status_code != 200:
-            raise IAToolkitException(IAToolkitException.ErrorType.CALL_ERROR,
-                               f'API {endpoint} error: {status_code}')
+        if status_code < 200 or status_code >= 300:
+            raise IAToolkitException(
+                IAToolkitException.ErrorType.CALL_ERROR,
+                f"API {endpoint} error: {status_code}"
+            )
 
+        # Normalizamos a string JSON (para que el LLM lo consuma consistente)
         return json.dumps(response)

services/dispatcher_service.py

@@ -6,6 +6,7 @@
 from iatoolkit import current_iatoolkit
 from common.exceptions import IAToolkitException
 from services.prompt_manager_service import PromptService
+from services.api_service import ApiService
 from repositories.llm_query_repo import LLMQueryRepo
 from repositories.models import Company, Function
 from services.excel_service import ExcelService
@@ -23,10 +24,12 @@ class Dispatcher:
                  prompt_service: PromptService,
                  llmquery_repo: LLMQueryRepo,
                  util: Utility,
+                 api_service: ApiService,
                  excel_service: ExcelService,
                  mail_service: MailService):
         self.prompt_service = prompt_service
         self.llmquery_repo = llmquery_repo
+        self.api_service = api_service
         self.util = util
         self.excel_service = excel_service
         self.mail_service = mail_service
@@ -36,41 +39,42 @@ class Dispatcher:
         # Use the global registry
         self.company_registry = get_company_registry()
 
-        # The dispatcher starts "empty" and will be initialized later.
+        # load into the dispatcher the configured companies
         self.company_classes = {}
         self.initialize_companies()
 
+        # run the statrtup logic for all companies
+        self.start_execution()
+
         self.tool_handlers = {
             "iat_generate_excel": self.excel_service.excel_generator,
             "iat_send_email": self.mail_service.send_mail,
+            "iat_api_call": self.api_service.call_api
         }
 
     def initialize_companies(self):
         """
         Initializes and instantiates all registered company classes.
-        This method should be called *after* the main injector is fully configured.
+        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
             return
 
         # ✅ NOW it is safe to get the injector and instantiate companies.
-        injector = current_iatoolkit()._get_injector()
+        injector = current_iatoolkit().get_injector()
         self.company_registry.set_injector(injector)
         self.company_classes = self.company_registry.instantiate_companies()
 
     def start_execution(self):
         """Runs the startup logic for all registered companies."""
-        # Ensure companies are initialized before starting them
-        if not self.company_classes:
-            self.initialize_companies()
-
         for company_name, company_instance in self.company_classes.items():
             logging.info(f'Starting execution for company: {company_name}')
             company_instance.start_execution()
 
         return True
 
-    def init_db(self):
+    def setup_all_companies(self):
         # create system functions
         for function in self.system_functions:
             self.llmquery_repo.create_or_update_function(
@@ -96,10 +100,9 @@ class Dispatcher:
 
         # initialize the database for every company class
         for company in self.company_classes.values():
-            print(f'inicializando clase: {company.__class__.__name__}')
+            print(f'company: {company.__class__.__name__}')
             company.init_db()
 
-
     def dispatch(self, company_name: str, action: str, **kwargs) -> str:
         company_key = company_name.lower()
 

services/profile_service.py

@@ -266,7 +266,7 @@ class ProfileService:
 
         api_key = ApiKey(key=key, company_id=company.id)
         self.profile_repo.create_api_key(api_key)
-        return {"message": f"La nueva clave de API para {company_short_name} es: {key}"}
+        return {"api-key": key}
 
 
     def send_verification_email(self, new_user: User, company_short_name):

services/prompt_manager_service.py

@@ -11,6 +11,7 @@ from collections import defaultdict
 from repositories.models import Prompt, PromptCategory, Company
 import os
 from common.exceptions import IAToolkitException
+from pathlib import Path
 
 
 class PromptService:
@@ -32,7 +33,7 @@ class PromptService:
 
         prompt_filename = prompt_name.lower() + '.prompt'
         if is_system_prompt:
-            template_dir = 'prompts'
+            template_dir = 'src/system_prompts'
         else:
             template_dir = f'companies/{company.short_name}/prompts'
 
@@ -103,16 +104,15 @@ class PromptService:
             system_prompt_content = []
 
             # get the filepaths for all system prompts
-            current_dir = os.path.dirname(os.path.abspath(__file__))
-            src_dir = os.path.dirname(current_dir)  # ../src
-            system_prompt_dir = os.path.join(src_dir, "prompts")
+            current_dir = Path(__file__).resolve().parent
+            project_root = current_dir.parent.parent
 
-            # Obtener, ordenar y leer los system prompts
+            # read all the system prompts from the database
             system_prompts = self.llm_query_repo.get_system_prompts()
 
             for prompt in system_prompts:
-                # Construir la ruta absoluta para leer el archivo
-                absolute_filepath = os.path.join(system_prompt_dir, prompt.filepath)
+                # build the absolute filepath for reading it
+                absolute_filepath = os.path.join(project_root, prompt.filepath)
                 if not os.path.exists(absolute_filepath):
                     logging.warning(f"El archivo para el prompt de sistema no existe: {absolute_filepath}")
                     continue
@@ -123,7 +123,7 @@ class PromptService:
                     raise IAToolkitException(IAToolkitException.ErrorType.FILE_IO_ERROR,
                                        f"Error leyendo el archivo de prompt del sistema {absolute_filepath}: {e}")
 
-            # Unir todo el contenido en un solo string
+            # join the system prompts into a single string
             return "\n".join(system_prompt_content)
 
         except IAToolkitException: