iatoolkit 2.45.2__tar.gz → 2.45.3__tar.gz

{iatoolkit-2.45.2/src/iatoolkit.egg-info → iatoolkit-2.45.3}/PKG-INFO

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

{iatoolkit-2.45.2 → iatoolkit-2.45.3}/src/iatoolkit/__init__.py

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

{iatoolkit-2.45.2 → iatoolkit-2.45.3}/src/iatoolkit/services/llm_client_service.py

@@ -404,7 +404,7 @@ class llmClient:
                     if key == "request_source":
                         continue
                     if value is not None:
-                        combined_stats[key] = value
+                        combined_stats[key] = self._compact_execution_metadata_for_stats(key, value)
 
             # decode the LLM response
             decoded_response = self.decode_response(response)
@@ -1309,6 +1309,57 @@ class llmClient:
             stats_dict["provider_calls"] = provider_calls
         return stats_dict
 
+    @classmethod
+    def _compact_execution_metadata_for_stats(cls, key: str, value):
+        if key == "tool_router":
+            return cls._compact_tool_router_stats(value)
+        return value
+
+    @staticmethod
+    def _compact_tool_router_stats(metrics):
+        if not isinstance(metrics, dict):
+            return metrics
+
+        compact = {}
+        scalar_keys = (
+            "candidate_count",
+            "selected_count",
+            "selection_mode",
+            "fallback_reason",
+            "selector_latency_ms",
+            "router_skipped",
+            "hard_disable_tools",
+        )
+        for key in scalar_keys:
+            if key in metrics:
+                compact[key] = metrics.get(key)
+
+        list_count_keys = {
+            "configured_tool_names": "configured_count",
+            "matched_tool_names": "matched_count",
+            "missing_tool_names": "missing_count",
+            "forced_tool_names": "forced_count",
+            "selected_system_prompt_keys": "selected_system_prompt_count",
+        }
+        for source_key, target_key in list_count_keys.items():
+            value = metrics.get(source_key)
+            if isinstance(value, list):
+                compact[target_key] = len(value)
+
+        hook_metadata = metrics.get("hook_metadata")
+        if isinstance(hook_metadata, dict):
+            for key in ("top_k", "top_n", "min_confidence", "shadow_mode"):
+                if key in hook_metadata:
+                    compact[key] = hook_metadata.get(key)
+            selected_tool_names = hook_metadata.get("selected_tool_names")
+            if isinstance(selected_tool_names, list):
+                compact["hook_selected_count"] = len(selected_tool_names)
+            ranked_tools_preview = hook_metadata.get("ranked_tools_preview")
+            if isinstance(ranked_tools_preview, list):
+                compact["ranked_preview_count"] = len(ranked_tools_preview)
+
+        return compact
+
 
     def _create_sql_retry_prompt(self, function_name: str, sql_query: str, db_error: str) -> str:
         return f"""

{iatoolkit-2.45.2 → iatoolkit-2.45.3}/src/iatoolkit/services/operator_identity_service.py

@@ -39,6 +39,17 @@ class OperatorIdentityService:
     SETTING_NAMES = ("IAT_OPERATOR_COMPANY", "IAT_BILLING_COMPANY_SHORT_NAME")
     SETTING_NAME = SETTING_NAMES[0]
 
+    #: Held on the class, not the instance. Callers that have a repository at hand
+    #: build their own instance rather than asking the injector, so an instance
+    #: cache resolved — and logged, and queried the database — once per request
+    #: instead of once per process.
+    _resolved: str | None = None
+
+    @classmethod
+    def reset_cache(cls) -> None:
+        """Forget the resolved name. For tests: it cannot change at runtime."""
+        OperatorIdentityService._resolved = None
+
     @classmethod
     def stated_name(cls) -> str:
         """What this deployment calls its operator, without checking it exists.
@@ -57,7 +68,6 @@ class OperatorIdentityService:
     @inject
     def __init__(self, profile_repo: ProfileRepo):
         self.profile_repo = profile_repo
-        self._resolved: str | None = None
 
     def company_short_name(self) -> str:
         """The operator company, resolved once per process.
@@ -65,24 +75,24 @@ class OperatorIdentityService:
         Cached because the answer cannot change without a deployment, and this is
         read on paths that run per request and per invoice.
         """
-        if self._resolved:
-            return self._resolved
+        if OperatorIdentityService._resolved:
+            return OperatorIdentityService._resolved
 
         for setting in self.SETTING_NAMES:
             configured = (os.getenv(setting) or "").strip().lower()
             if configured:
-                self._resolved = configured
+                OperatorIdentityService._resolved = configured
                 logging.info("Operator company: '%s' (from %s).", configured, setting)
-                return self._resolved
+                return configured
 
         if self._company_exists(self.LEGACY_COMPANY_SHORT_NAME):
-            self._resolved = self.LEGACY_COMPANY_SHORT_NAME
+            OperatorIdentityService._resolved = self.LEGACY_COMPANY_SHORT_NAME
             logging.info(
                 "Operator company: '%s' (default). Set %s to name it explicitly.",
-                self._resolved,
+                OperatorIdentityService._resolved,
                 self.SETTING_NAME,
             )
-            return self._resolved
+            return OperatorIdentityService._resolved
 
         # Deliberately loud. The alternative is what used to happen: a name from
         # another deployment travels down to the payment configuration and the

{iatoolkit-2.45.2 → iatoolkit-2.45.3/src/iatoolkit.egg-info}/PKG-INFO

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