iatoolkit 2.24.1__tar.gz → 2.25.0__tar.gz

{iatoolkit-2.24.1/src/iatoolkit.egg-info → iatoolkit-2.25.0}/PKG-INFO

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

{iatoolkit-2.24.1 → iatoolkit-2.25.0}/src/iatoolkit/__init__.py

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

{iatoolkit-2.24.1 → iatoolkit-2.25.0}/src/iatoolkit/locales/en.yaml

@@ -1859,6 +1859,10 @@ ui:
     company_config_desc: "Core runtime, branding, connector, and company behavior configuration."
     company_card_title: "Company"
     company_card_desc: "Core identity and runtime information for this company."
+    company_editor_desc: "Review and update the full company configuration for identity, runtime, and branding."
+    policies_card_title: "Policies"
+    policies_card_desc: "Manage runtime access policies, quotas, and tool or model controls."
+    policies_editor_desc: "Review and update the runtime policy file used for access decisions."
     company_name: "Company name"
     short_name: "Short name"
     runtime_mode: "Runtime mode"

{iatoolkit-2.24.1 → iatoolkit-2.25.0}/src/iatoolkit/locales/es.yaml

@@ -1863,6 +1863,10 @@ ui:
     company_config_desc: "Configuración central de runtime, branding, conectores y comportamiento de la empresa."
     company_card_title: "Empresa"
     company_card_desc: "Identidad principal e información de runtime para esta empresa."
+    company_editor_desc: "Revisa y actualiza la configuración completa de empresa para identidad, runtime y branding."
+    policies_card_title: "Políticas"
+    policies_card_desc: "Administra políticas de acceso, cuotas y controles de tools o modelos en runtime."
+    policies_editor_desc: "Revisa y actualiza el archivo de políticas usado para decisiones de acceso en runtime."
     company_name: "Nombre de la empresa"
     short_name: "Nombre corto"
     runtime_mode: "Modo runtime"

{iatoolkit-2.24.1 → iatoolkit-2.25.0}/src/iatoolkit/services/dispatcher_service.py

@@ -10,10 +10,13 @@ from iatoolkit.common.util import Utility
 from injector import inject
 import inspect
 import logging
+from time import perf_counter
+from uuid import uuid4
 
 
 class Dispatcher:
     USER_SCOPED_SYSTEM_TOOLS = {"iat_memory_search", "iat_memory_get_page"}
+    _tool_execution_hook = None
 
     @inject
     def __init__(self,
@@ -55,6 +58,32 @@ class Dispatcher:
             for parameter in signature.parameters.values()
         )
 
+    @classmethod
+    def register_tool_execution_hook(cls, hook):
+        cls._tool_execution_hook = hook
+
+    @classmethod
+    def clear_tool_execution_hook(cls):
+        cls._tool_execution_hook = None
+
+    def _notify_tool_execution_hook(self, event: str, **payload) -> None:
+        hook = type(self)._tool_execution_hook
+        if not callable(hook):
+            return
+
+        try:
+            hook(event=event, **payload)
+        except IAToolkitException:
+            raise
+        except Exception as exc:
+            logging.warning(
+                "Tool execution hook failed for event=%s company=%s tool=%s: %s",
+                event,
+                payload.get("company_short_name"),
+                payload.get("function_name"),
+                exc,
+            )
+
     @property
     def tool_service(self):
         """Lazy-loads and returns the ToolService instance to avoid circular imports."""
@@ -91,7 +120,14 @@ class Dispatcher:
         return self._company_instances
 
 
-    def dispatch(self, company_short_name: str, function_name: str, user_identifier: str | None = None, **kwargs) -> dict:
+    def dispatch(
+        self,
+        company_short_name: str,
+        function_name: str,
+        user_identifier: str | None = None,
+        _iat_runtime_source: str | None = None,
+        **kwargs,
+    ) -> dict:
         # 1. Consult the Database (Source of Truth) for the tool definition
         tool_def = self.tool_service.get_tool_definition(company_short_name, function_name)
         if not tool_def:
@@ -100,6 +136,65 @@ class Dispatcher:
                 f"Tool '{function_name}' not registered for company '{company_short_name}'"
             )
 
+        execution_id = uuid4().hex
+        self._notify_tool_execution_hook(
+            "before_dispatch",
+            execution_id=execution_id,
+            company_short_name=company_short_name,
+            function_name=function_name,
+            user_identifier=user_identifier,
+            runtime_source=_iat_runtime_source,
+            tool_def=tool_def,
+            arguments=dict(kwargs),
+        )
+        started_at = perf_counter()
+        try:
+            result = self._dispatch_resolved_tool(
+                company_short_name=company_short_name,
+                function_name=function_name,
+                user_identifier=user_identifier,
+                tool_def=tool_def,
+                **kwargs,
+            )
+            self._notify_tool_execution_hook(
+                "after_dispatch",
+                execution_id=execution_id,
+                company_short_name=company_short_name,
+                function_name=function_name,
+                user_identifier=user_identifier,
+                runtime_source=_iat_runtime_source,
+                tool_def=tool_def,
+                arguments=dict(kwargs),
+                status="success",
+                duration_ms=int(round((perf_counter() - started_at) * 1000)),
+                result=result,
+            )
+            return result
+        except Exception as exc:
+            self._notify_tool_execution_hook(
+                "after_dispatch",
+                execution_id=execution_id,
+                company_short_name=company_short_name,
+                function_name=function_name,
+                user_identifier=user_identifier,
+                runtime_source=_iat_runtime_source,
+                tool_def=tool_def,
+                arguments=dict(kwargs),
+                status="error",
+                duration_ms=int(round((perf_counter() - started_at) * 1000)),
+                error=exc,
+            )
+            raise
+
+    def _dispatch_resolved_tool(
+        self,
+        *,
+        company_short_name: str,
+        function_name: str,
+        user_identifier: str | None,
+        tool_def,
+        **kwargs,
+    ) -> dict:
         # 2. Dispatch based on Tool Type
         if tool_def.tool_type == 'SYSTEM':
             # Map to internal handler

{iatoolkit-2.24.1 → iatoolkit-2.25.0}/src/iatoolkit/services/inference_service.py

@@ -101,7 +101,7 @@ class InferenceService:
             payload["parameters"] = parameters
 
         # 4. Execute Call
-        logging.info("Calling inference tool %s with model %s.", tool_name, model_id)
+        logging.debug("Calling inference tool %s with model %s.", tool_name, model_id)
         response_data = self._call_endpoint(
             endpoint_url,
             api_key,

{iatoolkit-2.24.1 → iatoolkit-2.25.0}/src/iatoolkit/services/llm_client_service.py

@@ -250,6 +250,7 @@ class llmClient:
                                 company_short_name=company.short_name,
                                 function_name=function_name,
                                 user_identifier=user_identifier,
+                                _iat_runtime_source=request_source,
                                 **call_kwargs
                             )
                             force_tool_name = None

{iatoolkit-2.24.1 → iatoolkit-2.25.0}/src/iatoolkit/views/tool_api_view.py

@@ -188,6 +188,8 @@ class ToolApiView(MethodView):
             result = self.dispatcher.dispatch(
                 company_short_name=company_short_name,
                 function_name=tool_name,
+                user_identifier=auth_result.get("user_identifier"),
+                _iat_runtime_source="api",
                 **call_kwargs
             )
 
@@ -211,6 +213,8 @@ class ToolApiView(MethodView):
             status_code = 404
         elif e.error_type in [IAToolkitException.ErrorType.MISSING_PARAMETER, IAToolkitException.ErrorType.INVALID_PARAMETER]:
             status_code = 400
+        elif e.error_type == IAToolkitException.ErrorType.PERMISSION:
+            status_code = 403
         elif e.error_type in [IAToolkitException.ErrorType.DUPLICATE_ENTRY, IAToolkitException.ErrorType.INVALID_OPERATION]:
             status_code = 409
 

{iatoolkit-2.24.1 → iatoolkit-2.25.0/src/iatoolkit.egg-info}/PKG-INFO

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