iatoolkit 2.57.0__tar.gz → 2.58.1__tar.gz

{iatoolkit-2.57.0/src/iatoolkit.egg-info → iatoolkit-2.58.1}/PKG-INFO

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

{iatoolkit-2.57.0 → iatoolkit-2.58.1}/src/iatoolkit/__init__.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-__version__ = "2.57.0"
+__version__ = "2.58.1"
 
 # Expose main classes and functions at the top level of the package
 

iatoolkit-2.58.1/src/iatoolkit/base_company.py

@@ -0,0 +1,48 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+# companies/base_company.py
+from abc import ABC, abstractmethod
+from pathlib import Path
+from typing import Optional
+
+from flask import Blueprint
+
+class BaseCompany(ABC):
+
+    @abstractmethod
+    # execute the specific action configured in the intent table
+    def handle_request(self, tag: str, params: dict) -> dict:
+        raise NotImplementedError("La subclase debe implementar el método handle_request()")
+
+    def register_cli_commands(self, app):
+        # Optional extension point. Community/Enterprise can override as needed.
+        return None
+
+    def get_blueprint(self) -> Optional[Blueprint]:
+        """
+        Optional extension point: the web routes, templates and static files this
+        company owns. The core registers the blueprint at boot for every company
+        that is loaded, so the host application never imports a company's views.
+
+        The blueprint must be named after the company short name: that name is
+        the namespace of its endpoints (``url_for('<company>.<endpoint>')``).
+        URLs are not prefixed by the core; the company chooses them.
+        """
+        return None
+
+    @classmethod
+    def get_resource_root(cls) -> Optional[Path]:
+        """
+        Directory that holds this company's files (config, prompts, schemas,
+        context, templates). ``None`` keeps the classic layout,
+        ``companies/<company_short_name>`` under the working directory. A company
+        installed as a package returns its own directory, usually
+        ``Path(__file__).resolve().parent``.
+        """
+        return None
+
+    def unsupported_operation(self, tag):
+        raise NotImplementedError(f"La operación '{tag}' no está soportada por esta empresa.")

iatoolkit-2.58.1/src/iatoolkit/common/company_paths.py

@@ -0,0 +1,25 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from pathlib import Path
+
+
+def get_company_root(company_short_name: str) -> Path:
+    """
+    Directory that holds a company's files: config, prompts, schemas, context,
+    templates.
+
+    A company module installed as a package declares it through
+    ``BaseCompany.get_resource_root()``. Every other company keeps the classic
+    layout, ``companies/<company_short_name>`` relative to the working directory,
+    which is why that is the fallback and not an error.
+    """
+    from iatoolkit.company_registry import get_registered_companies
+
+    company_class = get_registered_companies().get((company_short_name or "").lower())
+    root = company_class.get_resource_root() if company_class else None
+    if root:
+        return Path(root)
+    return Path("companies") / company_short_name

{iatoolkit-2.57.0 → iatoolkit-2.58.1}/src/iatoolkit/common/util.py

@@ -98,7 +98,8 @@ class Utility:
 
     def get_company_template(self, company_short_name: str, template_name: str) -> str:
         # 1. get the path to the company specific template
-        template_path = os.path.join(os.getcwd(), f'companies/{company_short_name}/templates/{template_name}')
+        from iatoolkit.common.company_paths import get_company_root
+        template_path = os.path.join(os.getcwd(), str(get_company_root(company_short_name) / 'templates' / template_name))
         if not os.path.exists(template_path):
             return None
 

{iatoolkit-2.57.0 → iatoolkit-2.58.1}/src/iatoolkit/core.py

@@ -114,6 +114,9 @@ class IAToolkit:
         # Step 7: load company configuration file
         self._hydrate_company_configuration()
 
+        # Step 7b: web routes, templates and static files owned by each company
+        self._register_company_blueprints()
+
         # Step 8: Finalize setup within the application context
         self._setup_redis_sessions()
 
@@ -181,7 +184,12 @@ class IAToolkit:
                 # connection "idle in transaction" until the thread's own session
                 # is removed - and nothing else ever removes it.
                 if self.db_manager is not None:
-                    self.db_manager.remove_session()
+                    try:
+                        self.db_manager.remove_session()
+                    except Exception as exc:
+                        # The server may have closed the connection while the
+                        # endpoint was waking; the pool discards it either way.
+                        logging.info("Startup warm-up: database session already closed by the server (%s).", exc.__class__.__name__)
 
         thread = threading.Thread(target=_run_warmup, name="iatoolkit-startup-warmup", daemon=True)
         thread.start()
@@ -560,6 +568,31 @@ class IAToolkit:
                 logging.error(f"❌ Failed to register configuration for '{company_short_name}': {e}")
                 raise e
 
+    def _register_company_blueprints(self):
+        from iatoolkit.company_registry import get_company_registry
+
+        registered = []
+        for company_short_name, company_instance in get_company_registry().get_all_company_instances().items():
+            blueprint = company_instance.get_blueprint()
+            if blueprint is None:
+                continue
+            # The blueprint name is the namespace of the company's endpoints, so
+            # it must be the company itself: anything else could collide with
+            # another tenant or with the platform.
+            if blueprint.name != company_short_name:
+                raise RuntimeError(
+                    f"Company '{company_short_name}' returned a blueprint named "
+                    f"'{blueprint.name}'; it must be named '{company_short_name}'."
+                )
+            self.app.register_blueprint(blueprint)
+            registered.append(company_short_name)
+
+        if registered:
+            # FlaskInjector wraps the views that exist when it runs, and the first
+            # pass ran before any company was loaded.
+            FlaskInjector(app=self.app, injector=self._injector)
+            logging.info(f"✅ Company blueprints registered: {', '.join(registered)}")
+
     def _setup_cli_commands(self):
         from iatoolkit.cli_commands import register_core_commands
         from iatoolkit.company_registry import get_company_registry

{iatoolkit-2.57.0 → iatoolkit-2.58.1}/src/iatoolkit/locales/en.yaml

@@ -332,6 +332,9 @@ ui:
     init_context: "Initializing the context ..."
 
   account:
+    mcp_connection_copy_json_button: "Copy JSON"
+    mcp_connection_json_copy_success: "JSON config copied."
+    mcp_connection_copy_error: "Could not copy the MCP connection snippet."
     title: "My account"
     back_to_chat: "Back to chat"
     profile: "Profile"
@@ -2626,6 +2629,7 @@ errors:
     password_no_special_char: "Password must contain at least one special character."
 
   services:
+    docling_disabled: "Docling parsing is disabled in this deployment."
     no_text_file: "The file is not text or the encoding is not UTF-8"
     no_output_file: "Missing output file name"
     no_data_for_excel: "Missing data or it is not a list of dictionaries"

{iatoolkit-2.57.0 → iatoolkit-2.58.1}/src/iatoolkit/locales/es.yaml

@@ -346,6 +346,9 @@ ui:
     init_context: "Inicializando el contexto de la IA ..."
 
   account:
+    mcp_connection_copy_json_button: "Copiar JSON"
+    mcp_connection_json_copy_success: "Configuración JSON copiada."
+    mcp_connection_copy_error: "No se pudo copiar el snippet de conexión MCP."
     title: "Mi cuenta"
     back_to_chat: "Volver al chat"
     profile: "Perfil"
@@ -2645,6 +2648,7 @@ errors:
     password_no_special_char: "La contraseña debe tener al menos un carácter especial."
 
   services:
+    docling_disabled: "El parseo con Docling está deshabilitado en esta instalación."
     no_text_file: "El archivo no es texto o la codificación no es UTF-8"
     no_output_file: "falta el nombre del archivo de salida"
     no_data_for_excel: "faltan los datos  o no es una lista de diccionarios"

{iatoolkit-2.57.0 → iatoolkit-2.58.1}/src/iatoolkit/repositories/filesystem_asset_repository.py

@@ -1,10 +1,11 @@
 from iatoolkit.common.interfaces.asset_storage import AssetRepository, AssetType
+from iatoolkit.common.company_paths import get_company_root
 from pathlib import Path
 
 
 class FileSystemAssetRepository(AssetRepository):
     def _get_path(self, company_short_name: str, asset_type: AssetType, filename: str = "") -> Path:
-        return Path("companies") / company_short_name / asset_type.value / filename
+        return get_company_root(company_short_name) / asset_type.value / filename
 
     def exists(self, company_short_name: str, asset_type: AssetType, filename: str) -> bool:
         return self._get_path(company_short_name, asset_type, filename).is_file()

{iatoolkit-2.57.0 → iatoolkit-2.58.1}/src/iatoolkit/services/warmup_service.py

@@ -86,6 +86,14 @@ class WarmupService:
             label = self._format_profile_labels([profile])[0]
             host = self._endpoint_host(endpoint_url)
             try:
+                # A cold endpoint can take minutes to answer. Whatever this
+                # thread read from the database (config, the endpoint's token)
+                # would hold its connection "idle in transaction" all that time,
+                # until the server cut it and the thread's session cleanup failed
+                # on a dead connection. Build the client first, so its secret is
+                # already read and cached, then hand the connection back.
+                self._prepare_client(company_short_name, model_type)
+                self._release_idle_db_transaction()
                 self.embedding_service.embed_text(
                     company_short_name,
                     "hello",
@@ -134,6 +142,31 @@ class WarmupService:
 
         return targets
 
+    def _prepare_client(self, company_short_name: str, model_type: str) -> None:
+        """Creates (and caches) the embedding client, which reads its secret."""
+        factory = getattr(self.embedding_service, "client_factory", None)
+        if factory is None:
+            return
+        try:
+            factory.get_client(company_short_name, model_type)
+        except Exception:
+            # embed_text raises the same error with full context; nothing to add.
+            pass
+
+    @staticmethod
+    def _release_idle_db_transaction() -> None:
+        """Hands the thread's database connection back to the pool before a network
+        wait. Resolved lazily, like LLMProxy: the service is also built outside a
+        running toolkit (tests, scripts), where there is nothing to release."""
+        try:
+            from iatoolkit import current_iatoolkit
+            from iatoolkit.repositories.database_manager import DatabaseManager
+
+            db_manager = current_iatoolkit().get_injector().get(DatabaseManager)
+        except Exception:
+            return
+        db_manager.release_idle_transaction()
+
     @staticmethod
     def _endpoint_host(endpoint_url: str) -> str:
         """Host only: the URL can come from a secret, and the host is enough to log."""

{iatoolkit-2.57.0 → iatoolkit-2.58.1/src/iatoolkit.egg-info}/PKG-INFO

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

{iatoolkit-2.57.0 → iatoolkit-2.58.1}/src/iatoolkit.egg-info/SOURCES.txt

@@ -15,6 +15,7 @@ src/iatoolkit.egg-info/requires.txt
 src/iatoolkit.egg-info/top_level.txt
 src/iatoolkit/common/__init__.py
 src/iatoolkit/common/bridge_validation.py
+src/iatoolkit/common/company_paths.py
 src/iatoolkit/common/exceptions.py
 src/iatoolkit/common/model_registry.py
 src/iatoolkit/common/routes.py

iatoolkit-2.57.0/src/iatoolkit/base_company.py

@@ -1,21 +0,0 @@
-# Copyright (c) 2024 Fernando Libedinsky
-# Product: IAToolkit
-#
-# IAToolkit is open source software.
-
-# companies/base_company.py
-from abc import ABC, abstractmethod
-
-class BaseCompany(ABC):
-
-    @abstractmethod
-    # execute the specific action configured in the intent table
-    def handle_request(self, tag: str, params: dict) -> dict:
-        raise NotImplementedError("La subclase debe implementar el método handle_request()")
-
-    def register_cli_commands(self, app):
-        # Optional extension point. Community/Enterprise can override as needed.
-        return None
-
-    def unsupported_operation(self, tag):
-        raise NotImplementedError(f"La operación '{tag}' no está soportada por esta empresa.")