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

iatoolkit/__init__.py

@@ -21,9 +21,9 @@ 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 services.query_service import QueryService
 from repositories.database_manager import DatabaseManager
 from infra.call_service import CallServiceClient
 from common.util import Utility
@@ -40,8 +40,8 @@ __all__ = [
     'ExcelService',
     'Dispatcher',
     'DocumentService',
-    'QueryService',
     'SearchService',
+    'QueryService',
     'ProfileRepo',
     'LLMQueryRepo',
     'DatabaseManager',

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-0.3.9.dist-info → iatoolkit-0.4.0.dist-info}/METADATA

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

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

@@ -1,7 +1,7 @@
-iatoolkit/__init__.py,sha256=dKjc0IsztMClwbalF5hbtz418TRtu0xE2pdb8J3iQQw,1512
+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=cRaez-VBo1icnUNKmkQqo_Xlr8UKWFoYEMZ70XP6Jgk,2702
+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
@@ -10,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=vZbz-XCP0Aa-oE4nh5ZKeUKWlJ78VPSRYkpzMHlVFM0,14952
+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
@@ -19,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.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,,
+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:
@@ -240,14 +255,14 @@ class Dispatcher:
 
     def get_company_instance(self, company_name: str):
         """Returns the instance for a given company name."""
-        return self.company_classes.get(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}")