iatoolkit 2.43.0__tar.gz → 2.44.1__tar.gz

{iatoolkit-2.43.0/src/iatoolkit.egg-info → iatoolkit-2.44.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 2.43.0
+Version: 2.44.1
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT
@@ -31,7 +31,7 @@ Requires-Dist: google-auth-httplib2==0.2.0
 Requires-Dist: google-auth-oauthlib==1.2.1
 Requires-Dist: google-cloud-core==2.4.1
 Requires-Dist: google-cloud-storage==3.0.0
-Requires-Dist: google-crc32c==1.6.0
+Requires-Dist: google-crc32c==1.8.0
 Requires-Dist: google-genai==1.57.0
 Requires-Dist: google-resumable-media==2.7.2
 Requires-Dist: googleapis-common-protos==1.66.0
@@ -88,6 +88,11 @@ Requires-Dist: Werkzeug==3.1.3
 Requires-Dist: pyjwt[crypto]>=2.8.0
 Provides-Extra: docling
 Requires-Dist: docling==2.72.0; extra == "docling"
+Provides-Extra: ocr
+Requires-Dist: pytesseract==0.3.13; extra == "ocr"
+Provides-Extra: inference
+Requires-Dist: torch==2.10.0; extra == "inference"
+Requires-Dist: diffusers==0.39.0; extra == "inference"
 Dynamic: license-file
 
 # 🧠 IAToolkit — Open-Source Framework for Real-World AI Assistants

iatoolkit-2.44.1/pyproject.toml

@@ -0,0 +1,64 @@
+[build-system]
+requires = ["setuptools>=68", "wheel", "build"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "iatoolkit"
+dynamic = ["version", "dependencies"]
+requires-python = ">=3.12"
+description = "IAToolkit"
+readme = "readme.md"
+license = "MIT"
+authors = [{ name = "Fernando Libedinsky" }]
+
+# Capabilities that carry a heavy dependency and that most deployments never
+# turn on. Each of these was already being installed by hand into whatever
+# environment happened to need it, which is why a clean install could not run
+# the whole suite: nothing declared them, so nothing could install them on
+# purpose. Declared here, `pip install iatoolkit[ocr]` is a decision instead of
+# a memory.
+[project.optional-dependencies]
+# Parsing a document with the Docling pipeline rather than the basic provider.
+docling = [
+    "docling==2.72.0",
+]
+# Reading a scanned PDF. This extra is NOT enough to do OCR: pytesseract only
+# shells out to the tesseract binary, which pip cannot install. All three of
+# these have to be true, and `_get_tesseract_status()` checks them in order:
+#
+#   1. TESSERACT_ENABLED=true      — off by default, so this extra alone
+#                                    changes nothing at runtime
+#   2. this extra installed        — the only gate pip covers
+#   3. apt-get install tesseract-ocr tesseract-ocr-spa
+#                                  — the binary, and the Spanish data pack,
+#                                    because the call is image_to_string(lang="spa")
+#
+# Render's native Python runtime ships neither of the two system packages.
+# Installed in CI because the two OCR assertions in test_document_service.py
+# fake gates 1 and 3 but import pytesseract for real.
+ocr = [
+    "pytesseract==0.3.13",
+]
+# Running a model locally in `infra/inference_handler`, which imports both of
+# these at module level. Nothing in the source loads that module today — only
+# its own test does, by importlib — so treat these pins as untested: no code
+# path has exercised diffusers at any version.
+inference = [
+    "torch==2.10.0",
+    "diffusers==0.39.0",
+]
+
+[tool.setuptools]
+package-dir = {"" = "src"}
+
+[tool.setuptools.packages.find]
+where = ["src"]
+exclude = ["tests*", "*/tests*"]
+
+
+[tool.setuptools.dynamic]
+version = { attr = "iatoolkit.__version__" }
+dependencies = { file = ["requirements.txt"] }
+
+[tool.setuptools.package-data]
+iatoolkit = ["templates/**/*", "static/**/*", "locales/*.yaml", "config/*.yaml", "config/system_prompts/*.prompt"]

{iatoolkit-2.43.0 → iatoolkit-2.44.1}/requirements.txt

@@ -21,7 +21,9 @@ google-auth-httplib2==0.2.0
 google-auth-oauthlib==1.2.1
 google-cloud-core==2.4.1
 google-cloud-storage==3.0.0
-google-crc32c==1.6.0
+# 1.6.0 publishes no cp313 wheel. This was the only pin in the whole
+# dependency set that Python 3.13 required moving.
+google-crc32c==1.8.0
 google-genai==1.57.0
 google-resumable-media==2.7.2
 googleapis-common-protos==1.66.0

{iatoolkit-2.43.0 → iatoolkit-2.44.1}/src/iatoolkit/__init__.py

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

{iatoolkit-2.43.0 → iatoolkit-2.44.1}/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.43.0 → iatoolkit-2.44.1}/src/iatoolkit/infra/llm_providers/openrouter_adapter.py

@@ -16,6 +16,15 @@ 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.
+    _PDF_FILE_PARSER_PLUGIN = [{"id": "file-parser", "pdf": {"engine": "mistral-ocr"}}]
+
     def __init__(self, openrouter_client):
         super().__init__(openai_compatible_client=openrouter_client, provider_label="OpenRouter")
 
@@ -58,5 +67,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.43.0 → iatoolkit-2.44.1}/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.43.0 → iatoolkit-2.44.1/src/iatoolkit.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 2.43.0
+Version: 2.44.1
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT
@@ -31,7 +31,7 @@ Requires-Dist: google-auth-httplib2==0.2.0
 Requires-Dist: google-auth-oauthlib==1.2.1
 Requires-Dist: google-cloud-core==2.4.1
 Requires-Dist: google-cloud-storage==3.0.0
-Requires-Dist: google-crc32c==1.6.0
+Requires-Dist: google-crc32c==1.8.0
 Requires-Dist: google-genai==1.57.0
 Requires-Dist: google-resumable-media==2.7.2
 Requires-Dist: googleapis-common-protos==1.66.0
@@ -88,6 +88,11 @@ Requires-Dist: Werkzeug==3.1.3
 Requires-Dist: pyjwt[crypto]>=2.8.0
 Provides-Extra: docling
 Requires-Dist: docling==2.72.0; extra == "docling"
+Provides-Extra: ocr
+Requires-Dist: pytesseract==0.3.13; extra == "ocr"
+Provides-Extra: inference
+Requires-Dist: torch==2.10.0; extra == "inference"
+Requires-Dist: diffusers==0.39.0; extra == "inference"
 Dynamic: license-file
 
 # 🧠 IAToolkit — Open-Source Framework for Real-World AI Assistants

{iatoolkit-2.43.0 → iatoolkit-2.44.1}/src/iatoolkit.egg-info/requires.txt

@@ -21,7 +21,7 @@ google-auth-httplib2==0.2.0
 google-auth-oauthlib==1.2.1
 google-cloud-core==2.4.1
 google-cloud-storage==3.0.0
-google-crc32c==1.6.0
+google-crc32c==1.8.0
 google-genai==1.57.0
 google-resumable-media==2.7.2
 googleapis-common-protos==1.66.0
@@ -79,3 +79,10 @@ pyjwt[crypto]>=2.8.0
 
 [docling]
 docling==2.72.0
+
+[inference]
+torch==2.10.0
+diffusers==0.39.0
+
+[ocr]
+pytesseract==0.3.13

iatoolkit-2.43.0/pyproject.toml

@@ -1,32 +0,0 @@
-[build-system]
-requires = ["setuptools>=68", "wheel", "build"]
-build-backend = "setuptools.build_meta"
-
-[project]
-name = "iatoolkit"
-dynamic = ["version", "dependencies"]
-requires-python = ">=3.12"
-description = "IAToolkit"
-readme = "readme.md"
-license = "MIT"
-authors = [{ name = "Fernando Libedinsky" }]
-
-[project.optional-dependencies]
-docling = [
-    "docling==2.72.0",
-]
-
-[tool.setuptools]
-package-dir = {"" = "src"}
-
-[tool.setuptools.packages.find]
-where = ["src"]
-exclude = ["tests*", "*/tests*"]
-
-
-[tool.setuptools.dynamic]
-version = { attr = "iatoolkit.__version__" }
-dependencies = { file = ["requirements.txt"] }
-
-[tool.setuptools.package-data]
-iatoolkit = ["templates/**/*", "static/**/*", "locales/*.yaml", "config/*.yaml", "config/system_prompts/*.prompt"]