iatoolkit 2.64.2__tar.gz → 2.64.4__tar.gz

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

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

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

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

{iatoolkit-2.64.2 → iatoolkit-2.64.4}/src/iatoolkit/services/query_service.py

@@ -17,6 +17,7 @@ from iatoolkit.services.memory_lookup_policy_service import MemoryLookupPolicySe
 from iatoolkit.services.telemetry_service import NoopTelemetryService
 from iatoolkit.services.prompt_service import PromptService
 from iatoolkit.common.model_registry import ModelRegistry
+from iatoolkit.common.exceptions import IAToolkitException
 from injector import inject
 import logging
 from typing import Optional, Callable
@@ -203,19 +204,66 @@ class QueryService:
                        model: Optional[str],
                        prompt_output_contract: dict | None = None) -> str:
         # Priority: 1. Explicit model override -> 2. Prompt model -> 3. Company config
-        effective_model = str(model or "").strip() or None
-        if not effective_model and isinstance(prompt_output_contract, dict):
+        # The default is asked of the same resolution the model picker uses, not of
+        # the raw `llm.model` key: on a company backed by the model catalogue the
+        # default travels with its entitlements, so a YAML default the company can
+        # no longer use does not keep being sent.
+        default_model, available = self.configuration_service.get_llm_configuration(company_short_name)
+        enabled = self._enabled_model_ids(available)
+
+        explicit_model = str(model or "").strip() or None
+        if explicit_model:
+            self._raise_if_not_enabled(company_short_name, explicit_model, enabled)
+            return explicit_model
+
+        if isinstance(prompt_output_contract, dict):
             prompt_model = str(prompt_output_contract.get("llm_model") or "").strip()
             if prompt_model:
-                effective_model = prompt_model
-        if not effective_model:
-            # Asked of the same resolution the model picker uses, not of the raw
-            # `llm.model` key: on a company backed by the model catalogue the
-            # default travels with its entitlements, so a YAML default the
-            # company can no longer use does not keep being sent.
-            default_model, _ = self.configuration_service.get_llm_configuration(company_short_name)
-            effective_model = self._coerce_model_name(default_model, company_short_name)
-        return effective_model
+                if not enabled or prompt_model in enabled:
+                    return prompt_model
+                # An agent configured with a model the company has since lost
+                # keeps running on the company default rather than going down.
+                logging.warning(
+                    "Company '%s': agent model '%s' is no longer enabled; using the company default.",
+                    company_short_name,
+                    prompt_model,
+                )
+
+        return self._coerce_model_name(default_model, company_short_name)
+
+    def ensure_model_enabled(self, company_short_name: str, model: Optional[str]) -> None:
+        """
+        Rejects a model the caller names that the company cannot use.
+
+        A company with a list of models (declared, from the model catalogue, or
+        the single default) may only run those: otherwise a request could run
+        any model on the platform's key. An empty list means no restriction.
+        """
+        explicit_model = str(model or "").strip()
+        if not explicit_model:
+            return
+        _, available = self.configuration_service.get_llm_configuration(company_short_name)
+        self._raise_if_not_enabled(company_short_name, explicit_model, self._enabled_model_ids(available))
+
+    @staticmethod
+    def _raise_if_not_enabled(company_short_name: str, model: str, enabled: set[str]) -> None:
+        if enabled and model not in enabled:
+            raise IAToolkitException(
+                IAToolkitException.ErrorType.INVALID_PARAMETER,
+                f"Model '{model}' is not enabled for company '{company_short_name}'.",
+            )
+
+    @staticmethod
+    def _enabled_model_ids(available) -> set[str]:
+        ids = set()
+        for entry in available or []:
+            if isinstance(entry, dict):
+                model_id = str(entry.get("id") or "").strip()
+            else:
+                model_id = str(entry or "").strip()
+            if model_id:
+                ids.add(model_id)
+        return ids
 
     @staticmethod
     def _coerce_model_name(raw_model, company_short_name: str) -> Optional[str]:

{iatoolkit-2.64.2 → iatoolkit-2.64.4}/src/iatoolkit/static/styles/wiki_page.css

@@ -11,19 +11,22 @@
     font: 16px/1.6 system-ui, sans-serif;
 }
 .wiki-page * { box-sizing: border-box; min-width: 0; }
-.wiki-page-header { display: flex; align-items: center; justify-content: space-between; gap: 16px; padding: 20px 0 16px; border-bottom: 1px solid var(--wiki-line); }
-.wiki-page-brand { font-size: 18px; font-weight: 650; color: var(--brand-primary-color, #25292d); text-decoration: none; }
-.wiki-page-trail { display: flex; flex-wrap: wrap; align-items: baseline; gap: 4px 12px; padding: 16px 0 8px; font-size: 14px; }
+/* One bar: the brand, then where the page sits in the wiki. */
+.wiki-page-header { display: flex; flex-wrap: wrap; align-items: center; gap: 6px 16px; padding: 18px 0 14px; border-bottom: 1px solid var(--wiki-line); }
+.wiki-page-brand { font-size: 18px; font-weight: 650; color: var(--brand-primary-color, #25292d); text-decoration: none; padding-right: 16px; border-right: 1px solid var(--wiki-line); }
+.wiki-page-trail { display: flex; flex-wrap: wrap; align-items: center; gap: 4px 8px; font-size: 14px; }
 .wiki-page-trail a { display: inline-flex; align-items: center; gap: 6px; font-weight: 600; color: var(--wiki-link); text-decoration: none; }
+.wiki-page-sep { color: var(--wiki-muted); font-size: 11px; }
 .wiki-page-path { color: var(--wiki-muted); font-family: ui-monospace, SFMono-Regular, Menlo, monospace; font-size: 13px; overflow-wrap: anywhere; }
-.wiki-page-main { padding-top: 16px; }
+.wiki-page-main { padding-top: 24px; }
 .wiki-page-notice { padding: 12px 16px; border-left: 3px solid #c9ced3; background: #f6f7f8; overflow-wrap: anywhere; }
 .wiki-page-body { overflow-wrap: anywhere; }
 .wiki-page-body h1 { font-size: 28px; line-height: 1.25; font-weight: 650; margin: 8px 0 20px; }
 .wiki-page-body h2 { font-size: 21px; line-height: 1.3; font-weight: 650; margin: 32px 0 12px; padding-bottom: 6px; border-bottom: 1px solid var(--wiki-line); }
 .wiki-page-body h3 { font-size: 17px; font-weight: 650; margin: 24px 0 8px; }
 .wiki-page-body a { color: var(--wiki-link); }
-.wiki-page-body code { font-size: 0.9em; background: #f3f4f6; padding: 1px 4px; border-radius: 4px; overflow-wrap: anywhere; }
+/* Code stays in the text's own color: Bootstrap's pink shouts, more so in a heading. */
+.wiki-page-body code { color: inherit; font-size: 0.875em; background: #eff1f3; padding: 1px 5px; border-radius: 4px; overflow-wrap: anywhere; }
 .wiki-page-body pre { background: #f6f8fa; padding: 12px 16px; border-radius: 6px; overflow-x: auto; }
 .wiki-page-body pre code { background: none; padding: 0; }
 .wiki-page-body blockquote { margin: 0 0 16px; padding: 4px 16px; border-left: 3px solid var(--wiki-line); color: var(--wiki-muted); }
@@ -32,4 +35,5 @@
 .wiki-page-body thead { background: #f6f8fa; }
 @media (max-width: 600px) {
     .wiki-page { padding: 0 16px 48px; }
+    .wiki-page-brand { padding-right: 0; border-right: 0; }
 }

{iatoolkit-2.64.2 → iatoolkit-2.64.4}/src/iatoolkit/templates/wiki_page.html

@@ -9,11 +9,11 @@
 <div class="wiki-page">
     <header class="wiki-page-header">
         <a class="wiki-page-brand" href="{{ url_for('chat', company_short_name=company_short_name) }}">{{ branding.name }}</a>
+        <nav class="wiki-page-trail" aria-label="{{ t('ui.wiki_page.trail_label') }}">
+            <a href="{{ root_url }}"><i class="bi bi-journal-text" aria-hidden="true"></i>{{ wiki_name }}</a>
+            {% if page_path %}<i class="bi bi-chevron-right wiki-page-sep" aria-hidden="true"></i><span class="wiki-page-path">{{ page_path }}</span>{% endif %}
+        </nav>
     </header>
-    <nav class="wiki-page-trail" aria-label="{{ t('ui.wiki_page.trail_label') }}">
-        <a href="{{ root_url }}"><i class="bi bi-journal-text" aria-hidden="true"></i>{{ wiki_name }}</a>
-        {% if page_path %}<span class="wiki-page-path">{{ page_path }}</span>{% endif %}
-    </nav>
     <main class="wiki-page-main">
         {% if message %}
         <div class="wiki-page-notice" role="status">{{ message }}</div>

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

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