iatoolkit 2.28.0__tar.gz → 2.28.2__tar.gz

{iatoolkit-2.28.0/src/iatoolkit.egg-info → iatoolkit-2.28.2}/PKG-INFO

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

{iatoolkit-2.28.0 → iatoolkit-2.28.2}/src/iatoolkit/__init__.py

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

{iatoolkit-2.28.0 → iatoolkit-2.28.2}/src/iatoolkit/common/model_registry.py

@@ -100,8 +100,9 @@ class ModelRegistry:
         }
         supports_text_verbosity = normalized_provider in {"openai", "xai", "openrouter"}
         supports_store = normalized_provider in {"openai", "xai"}
+        supports_native_images = self._resolve_native_image_support(normalized_provider)
 
-        return {
+        capabilities = {
             "provider": normalized_provider,
             "history_type": self.get_history_type(model),
             "supports_reasoning_effort": supports_reasoning_effort,
@@ -110,6 +111,9 @@ class ModelRegistry:
             "allowed_text_verbosity": list(self._text_verbosity_options) if supports_text_verbosity else [],
             "supports_store": supports_store,
         }
+        if supports_native_images is not None:
+            capabilities["supports_native_images"] = supports_native_images
+        return capabilities
 
     def get_request_defaults(self, model: str) -> dict:
         """
@@ -191,3 +195,12 @@ class ModelRegistry:
 
     def is_anthropic_model(self, model: str) -> bool:
         return self.get_provider(model) == "anthropic"
+
+    @staticmethod
+    def _resolve_native_image_support(provider: ProviderType) -> bool | None:
+        # Keep this conservative and only publish a hint when we know the family behavior.
+        if provider in {"openai", "gemini", "anthropic"}:
+            return True
+        if provider == "deepseek":
+            return False
+        return None

{iatoolkit-2.28.0 → iatoolkit-2.28.2}/src/iatoolkit/infra/llm_gateway_resolver.py

@@ -18,6 +18,7 @@ class LLMGatewayResolver:
         "deepseek": "deepseek",
         "anthropic": "anthropic",
         "gemini": "google-ai-studio",
+        "openrouter": "openrouter",
     }
     SUPPORTED_PROVIDERS = frozenset(PROVIDER_SLUGS)
     CREDENTIAL_MODE_PROVIDER_KEY = "provider_key_in_request"
@@ -117,7 +118,7 @@ class LLMGatewayResolver:
                     ),
                 )
             # Some SDKs insist on a non-null api_key even when upstream credentials are injected by Cloudflare.
-            resolved_api_key = "" if normalized_provider in {"openai", "deepseek", "gemini"} else None
+            resolved_api_key = "" if normalized_provider in {"openai", "deepseek", "gemini", "openrouter"} else None
         else:
             raise IAToolkitException(
                 IAToolkitException.ErrorType.CONFIG_ERROR,

{iatoolkit-2.28.0 → iatoolkit-2.28.2}/src/iatoolkit/infra/llm_proxy.py

@@ -178,6 +178,8 @@ class LLMProxy:
             return self.PROVIDER_XAI
         if provider_key == "anthropic":
             return self.PROVIDER_ANTHROPIC
+        if provider_key == "openrouter":
+            return self.PROVIDER_OPENROUTER
 
         raise IAToolkitException(
             IAToolkitException.ErrorType.MODEL,

{iatoolkit-2.28.0 → iatoolkit-2.28.2}/src/iatoolkit/services/attachment_policy_service.py

@@ -60,10 +60,12 @@ class AttachmentPolicyService:
         provider: str,
         files: list | None,
         policy: dict | None = None,
+        model_capabilities: dict | None = None,
     ) -> dict:
         mode = self.normalize_mode((policy or {}).get("attachment_mode"))
         fallback = self.normalize_fallback((policy or {}).get("attachment_fallback"))
         capabilities = self.get_effective_provider_capabilities(company_short_name, provider)
+        capabilities = self._apply_model_capability_hints(capabilities, model_capabilities)
 
         if not files:
             return {
@@ -242,6 +244,18 @@ class AttachmentPolicyService:
         merged.setdefault("max_files_per_request", 0)
         return merged
 
+    @staticmethod
+    def _apply_model_capability_hints(capabilities: dict, model_capabilities: dict | None) -> dict:
+        merged = dict(capabilities or {})
+        if not isinstance(model_capabilities, dict):
+            return merged
+
+        model_supports_native_images = model_capabilities.get("supports_native_images")
+        if isinstance(model_supports_native_images, bool):
+            merged["supports_native_images"] = bool(merged.get("supports_native_images")) and model_supports_native_images
+
+        return merged
+
     def _normalize_file_meta(self, file_obj: dict) -> dict:
         if not isinstance(file_obj, dict):
             return {"name": "attachment", "payload": "", "mime_type": "application/octet-stream", "is_image": False}

{iatoolkit-2.28.0 → iatoolkit-2.28.2}/src/iatoolkit/services/configuration_service.py

@@ -111,7 +111,7 @@ class ConfigurationService:
             add_error(section, "Must be a dictionary.")
             return
 
-        allowed_gateway_providers = {"openai", "deepseek", "anthropic", "gemini"}
+        allowed_gateway_providers = {"openai", "deepseek", "anthropic", "gemini", "openrouter"}
         allowed_gateway_modes = {"provider_native"}
         allowed_gateway_vendors = {"cloudflare"}
         allowed_credential_modes = {"provider_key_in_request", "cloudflare_managed"}

{iatoolkit-2.28.0 → iatoolkit-2.28.2}/src/iatoolkit/services/query_service.py

@@ -1422,11 +1422,18 @@ class QueryService:
                 },
             }
             if self.attachment_policy_service:
+                get_capabilities_fn = getattr(self.model_registry, "get_capabilities", None)
+                model_capabilities = (
+                    get_capabilities_fn(effective_model)
+                    if callable(get_capabilities_fn)
+                    else None
+                )
                 attachment_plan = self.attachment_policy_service.build_attachment_plan(
                     company_short_name=company_short_name,
                     provider=provider,
                     files=files or [],
                     policy=effective_attachment_policy,
+                    model_capabilities=model_capabilities,
                 )
             log_stage("attachment_plan")
 

{iatoolkit-2.28.0 → iatoolkit-2.28.2/src/iatoolkit.egg-info}/PKG-INFO

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