iatoolkit 2.44.0__tar.gz → 2.44.2__tar.gz

{iatoolkit-2.44.0/src/iatoolkit.egg-info → iatoolkit-2.44.2}/PKG-INFO

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

{iatoolkit-2.44.0 → iatoolkit-2.44.2}/src/iatoolkit/__init__.py

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

{iatoolkit-2.44.0 → iatoolkit-2.44.2}/src/iatoolkit/infra/llm_providers/openai_compatible_chat_adapter.py

@@ -4,6 +4,7 @@
 # IAToolkit is open source software.
 
 import base64
+import json
 import logging
 import mimetypes
 import threading
@@ -592,11 +593,41 @@ class OpenAICompatibleChatAdapter:
             return dict(text["response_format"])
         return None
 
+    @staticmethod
+    def _safe_dump_response(response: Any) -> str:
+        try:
+            if hasattr(response, "model_dump"):
+                return json.dumps(response.model_dump(), default=str)
+        except Exception:
+            pass
+        return str(response)
+
+    @staticmethod
+    def _extract_error_reason(raw_error: Any) -> Optional[str]:
+        if not raw_error:
+            return None
+        if isinstance(raw_error, dict):
+            message = raw_error.get("message")
+        else:
+            message = getattr(raw_error, "message", None)
+        return str(message) if message else str(raw_error)
+
     def _map_chat_completion_response(self, response: Any) -> LLMResponse:
         if not response.choices:
+            raw_error = getattr(response, "error", None)
+            logging.error(
+                "[%sAdapter] Response has no choices. error=%s raw_response=%s",
+                self.provider_label,
+                raw_error,
+                self._safe_dump_response(response),
+            )
+            reason = self._extract_error_reason(raw_error)
+            message = f"{self.provider_label} response has no choices."
+            if reason:
+                message = f"{message} Reason: {reason}"
             raise IAToolkitException(
                 IAToolkitException.ErrorType.LLM_ERROR,
-                f"{self.provider_label} response has no choices."
+                message,
             )
 
         choice = response.choices[0]

{iatoolkit-2.44.0 → iatoolkit-2.44.2}/src/iatoolkit/infra/llm_providers/openrouter_adapter.py

@@ -16,6 +16,22 @@ class OpenRouterAdapter(OpenAICompatibleChatAdapter):
     supports_metadata = True
     supports_parallel_tool_calls = True
 
+    # llm_capabilities.yaml marks `application/pdf` as natively supported for
+    # every OpenRouter model, but that flag isn't verified per-model the way
+    # native images are (see AttachmentPolicyService._get_openrouter_native_image_error).
+    # Models with no native PDF support (e.g. some Qwen models) reject the raw
+    # file part and OpenRouter answers with no `choices`. Always routing PDFs
+    # through OpenRouter's own `file-parser` plugin sidesteps that per-model gap
+    # instead of trying to track which routed model does or doesn't accept PDFs.
+    #
+    # engine="native": let the routed model read the PDF with its own
+    # multimodal capability, falling back to OpenRouter's own extraction only
+    # when the model doesn't support that. engine="mistral-ocr" used to be the
+    # default here, but it forces every PDF - regardless of model support -
+    # through OpenRouter's shared OCR pipeline, which is what produced the
+    # "document parsing engine is currently rate limited" failures.
+    _PDF_FILE_PARSER_PLUGIN = [{"id": "file-parser", "pdf": {"engine": "native"}}]
+
     def __init__(self, openrouter_client):
         super().__init__(openai_compatible_client=openrouter_client, provider_label="OpenRouter")
 
@@ -58,5 +74,19 @@ class OpenRouterAdapter(OpenAICompatibleChatAdapter):
             if kwargs.get(key) is not None:
                 extra_body[key] = kwargs.get(key)
 
+        if "plugins" not in extra_body and self._has_pdf_attachment(kwargs.get("attachments")):
+            extra_body["plugins"] = self._PDF_FILE_PARSER_PLUGIN
+
         if extra_body:
             call_kwargs["extra_body"] = extra_body
+
+    @staticmethod
+    def _has_pdf_attachment(attachments: Any) -> bool:
+        for attachment in attachments or []:
+            if not isinstance(attachment, dict):
+                continue
+            mime_type = str(attachment.get("mime_type") or attachment.get("type") or "").strip().lower()
+            filename = str(attachment.get("name") or attachment.get("filename") or "").strip().lower()
+            if mime_type == "application/pdf" or filename.endswith(".pdf"):
+                return True
+        return False

{iatoolkit-2.44.0 → iatoolkit-2.44.2}/src/iatoolkit/services/warmup_service.py

@@ -92,7 +92,7 @@ class WarmupService:
                     model_type=model_type,
                     suppress_error_logging=True,
                 )
-                logging.info(
+                logging.debug(
                     "🔥 Startup warm-up woke endpoint='%s' via company='%s' profile=%s trigger='%s'.",
                     host, company_short_name, label, trigger,
                 )

{iatoolkit-2.44.0 → iatoolkit-2.44.2/src/iatoolkit.egg-info}/PKG-INFO

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