iatoolkit 2.64.1__tar.gz → 2.64.2__tar.gz

{iatoolkit-2.64.1/src/iatoolkit.egg-info → iatoolkit-2.64.2}/PKG-INFO

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

{iatoolkit-2.64.1 → iatoolkit-2.64.2}/src/iatoolkit/__init__.py

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

{iatoolkit-2.64.1 → iatoolkit-2.64.2}/src/iatoolkit/services/system_agent_runner.py

@@ -132,8 +132,16 @@ class SystemAgentRunner:
     # Model
     # =====================================================================
 
-    def resolve_model(self, company_short_name: str, agent: SystemAgent) -> ResolvedModel:
-        default_model, available = self.configuration_service.get_llm_configuration(company_short_name)
+    def resolve_model(self,
+                      company_short_name: str,
+                      agent: SystemAgent,
+                      llm_configuration: tuple | None = None) -> ResolvedModel:
+        """
+        `llm_configuration` is get_llm_configuration's (default, available) when
+        the caller already read it: resolving many agents then costs one read,
+        which in enterprise is a query to the model catalogue.
+        """
+        default_model, available = llm_configuration or self.configuration_service.get_llm_configuration(company_short_name)
         enabled = self._model_ids(available)
 
         def usable(model: str | None) -> bool:
@@ -230,11 +238,12 @@ class SystemAgentRunner:
     # Describing (for the admin screens)
     # =====================================================================
 
-    def describe(self, company_short_name: str) -> list[dict]:
+    def describe(self, company_short_name: str, llm_configuration: tuple | None = None) -> list[dict]:
         """Every system agent with the model it runs with for this company."""
+        llm_configuration = llm_configuration or self.configuration_service.get_llm_configuration(company_short_name)
         result = []
         for agent in SystemAgentCatalog.list():
-            resolved = self.resolve_model(company_short_name, agent)
+            resolved = self.resolve_model(company_short_name, agent, llm_configuration)
             result.append({
                 "key": agent.key,
                 "name": agent.name,

{iatoolkit-2.64.1 → iatoolkit-2.64.2/src/iatoolkit.egg-info}/PKG-INFO

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