iatoolkit 2.45.0__tar.gz → 2.45.2__tar.gz

{iatoolkit-2.45.0/src/iatoolkit.egg-info → iatoolkit-2.45.2}/PKG-INFO

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

{iatoolkit-2.45.0 → iatoolkit-2.45.2}/src/iatoolkit/__init__.py

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

{iatoolkit-2.45.0 → iatoolkit-2.45.2}/src/iatoolkit/infra/llm_providers/openai_compatible_chat_adapter.py

@@ -43,6 +43,8 @@ class OpenAICompatibleChatAdapter:
     #: what DeepseekAdapter does.
     supports_reasoning_effort = False
     supports_reasoning_content_messages = False
+    supports_reasoning_details_messages = False
+    reasoning_content_message_field = "reasoning_content"
     supports_metadata = False
     supports_parallel_tool_calls = False
     supports_thinking = False
@@ -375,10 +377,11 @@ class OpenAICompatibleChatAdapter:
                 message["tool_calls"] = tool_calls
                 seen_tool_call_ids.update(self._extract_tool_call_ids(tool_calls))
 
-            if self.supports_reasoning_content_messages:
-                reasoning_content = str(item.get("reasoning_content") or "").strip()
-                if reasoning_content:
-                    message["reasoning_content"] = reasoning_content
+            self._attach_reasoning_to_message(
+                message,
+                reasoning_content=item.get("reasoning_content"),
+                reasoning_details=item.get("reasoning_details"),
+            )
 
             tool_call_id = str(item.get("tool_call_id") or "").strip()
             if role == "tool" and tool_call_id:
@@ -677,7 +680,7 @@ class OpenAICompatibleChatAdapter:
             total_tokens=getattr(getattr(response, "usage", None), "total_tokens", 0) or 0,
         )
 
-        reasoning_content = getattr(message, "reasoning_content", "") or ""
+        reasoning_content, reasoning_details = self._extract_reasoning_from_message(message)
 
         tool_calls_out: List[ToolCall] = []
         content_parts: List[Dict] = []
@@ -701,6 +704,7 @@ class OpenAICompatibleChatAdapter:
                 message=message,
                 normalized_tool_calls=normalized_tool_calls,
                 reasoning_content=reasoning_content,
+                reasoning_details=reasoning_details,
             )
 
             for tc in tool_calls:
@@ -738,7 +742,9 @@ class OpenAICompatibleChatAdapter:
             output=tool_calls_out,
             usage=usage,
             reasoning_content=reasoning_content,
+            reasoning_details=reasoning_details,
             content_parts=content_parts,
+            provider_metadata=self._extract_provider_metadata(response),
         )
 
     # Backward-compatible alias retained while the old DeepSeek-specific name
@@ -752,6 +758,7 @@ class OpenAICompatibleChatAdapter:
         message: Any,
         normalized_tool_calls: List[Dict[str, Any]],
         reasoning_content: str,
+        reasoning_details: Any = None,
     ) -> None:
         if not normalized_tool_calls:
             return
@@ -761,8 +768,11 @@ class OpenAICompatibleChatAdapter:
             "content": getattr(message, "content", "") or "",
             "tool_calls": normalized_tool_calls,
         }
-        if self.supports_reasoning_content_messages and reasoning_content:
-            assistant_message["reasoning_content"] = reasoning_content
+        self._attach_reasoning_to_message(
+            assistant_message,
+            reasoning_content=reasoning_content,
+            reasoning_details=reasoning_details,
+        )
 
         group_id = str(getattr(response, "id", "") or normalized_tool_calls[0]["id"]).strip()
         pending_entry = {
@@ -774,3 +784,55 @@ class OpenAICompatibleChatAdapter:
         with self._pending_assistant_tool_messages_lock:
             for tool_call in normalized_tool_calls:
                 self._pending_assistant_tool_messages[tool_call["id"]] = pending_entry
+
+    def _extract_reasoning_from_message(self, message: Any) -> tuple[str, Any]:
+        reasoning_content = self._coerce_optional_text(
+            self._get_message_value(message, "reasoning_content")
+        )
+        reasoning_details = self._get_message_value(message, "reasoning_details")
+
+        if not reasoning_content:
+            reasoning_content = self._coerce_optional_text(
+                self._get_message_value(message, "reasoning")
+            )
+
+        return reasoning_content, reasoning_details
+
+    def _attach_reasoning_to_message(
+        self,
+        message: Dict[str, Any],
+        reasoning_content: Any = None,
+        reasoning_details: Any = None,
+    ) -> None:
+        if self.supports_reasoning_details_messages and reasoning_details:
+            message["reasoning_details"] = reasoning_details
+            return
+
+        if not self.supports_reasoning_content_messages:
+            return
+
+        normalized_reasoning = self._coerce_optional_text(reasoning_content)
+        if normalized_reasoning:
+            message[self.reasoning_content_message_field] = normalized_reasoning
+
+    @staticmethod
+    def _get_message_value(message: Any, key: str) -> Any:
+        if isinstance(message, dict):
+            return message.get(key)
+
+        value = getattr(message, key, None)
+        if type(value).__module__.startswith("unittest.mock"):
+            return None
+        return value
+
+    @staticmethod
+    def _coerce_optional_text(value: Any) -> str:
+        if value is None:
+            return ""
+        if type(value).__module__.startswith("unittest.mock"):
+            return ""
+        return str(value or "").strip()
+
+    def _extract_provider_metadata(self, response: Any) -> Dict[str, Any]:
+        _ = response
+        return {}

iatoolkit-2.45.2/src/iatoolkit/infra/llm_providers/openrouter_adapter.py

@@ -0,0 +1,155 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from typing import Any, Dict
+
+from iatoolkit.infra.llm_providers.openai_compatible_chat_adapter import OpenAICompatibleChatAdapter
+
+
+class OpenRouterAdapter(OpenAICompatibleChatAdapter):
+    """OpenRouter-specific adapter built on top of the shared chat.completions core."""
+
+    supports_multimodal = True
+    supports_reasoning = False
+    supports_reasoning_content_messages = True
+    supports_reasoning_details_messages = True
+    reasoning_content_message_field = "reasoning"
+    supports_metadata = True
+    supports_parallel_tool_calls = True
+
+    def __init__(self, openrouter_client):
+        super().__init__(openai_compatible_client=openrouter_client, provider_label="OpenRouter")
+
+    def _extend_call_kwargs(self, call_kwargs: Dict[str, Any], kwargs: Dict[str, Any]) -> None:
+        text = kwargs.get("text") or {}
+        verbosity = text.get("verbosity") if isinstance(text, dict) else None
+        if verbosity:
+            call_kwargs["verbosity"] = verbosity
+
+        passthrough_keys = (
+            "service_tier",
+            "temperature",
+            "top_p",
+            "max_tokens",
+            "max_completion_tokens",
+            "seed",
+            "stop",
+            "stream",
+            "stream_options",
+            "modalities",
+            "user",
+        )
+        for key in passthrough_keys:
+            if kwargs.get(key) is not None:
+                call_kwargs[key] = kwargs.get(key)
+
+        extra_body = dict(call_kwargs.get("extra_body") or {})
+
+        reasoning_payload = self._build_reasoning_payload(kwargs.get("reasoning"), kwargs)
+        if reasoning_payload:
+            extra_body["reasoning"] = reasoning_payload
+
+        vendor_specific_keys = (
+            "models",
+            "provider",
+            "plugins",
+            "session_id",
+        )
+        for key in vendor_specific_keys:
+            if kwargs.get(key) is not None:
+                extra_body[key] = kwargs.get(key)
+
+        if extra_body:
+            call_kwargs["extra_body"] = extra_body
+
+    def _extract_provider_metadata(self, response: Any) -> Dict[str, Any]:
+        raw_metadata = self._get_message_value(response, "openrouter_metadata")
+        metadata = self._normalize_metadata_object(raw_metadata)
+
+        provider_name = self._extract_provider_name(metadata)
+        if not provider_name:
+            provider_name = self._coerce_optional_text(
+                self._get_message_value(response, "provider_name")
+            )
+
+        provider_metadata: Dict[str, Any] = {"provider": "openrouter"}
+        if provider_name:
+            provider_metadata["provider_name"] = provider_name
+        if metadata:
+            provider_metadata["openrouter"] = metadata
+
+        return provider_metadata
+
+    @classmethod
+    def _extract_provider_name(cls, metadata: Dict[str, Any]) -> str:
+        for key in ("provider_name", "provider", "upstream_provider", "selected_provider"):
+            value = metadata.get(key)
+            if isinstance(value, dict):
+                nested = cls._extract_provider_name(value)
+                if nested:
+                    return nested
+            normalized = cls._coerce_optional_text(value)
+            if normalized:
+                return normalized
+
+        for value in metadata.values():
+            if isinstance(value, dict):
+                nested = cls._extract_provider_name(value)
+                if nested:
+                    return nested
+
+        return ""
+
+    @classmethod
+    def _normalize_metadata_object(cls, value: Any) -> Dict[str, Any]:
+        if value is None or type(value).__module__.startswith("unittest.mock"):
+            return {}
+
+        if isinstance(value, dict):
+            return cls._sanitize_metadata_dict(value)
+
+        model_dump = getattr(value, "model_dump", None)
+        if callable(model_dump):
+            try:
+                dumped = model_dump()
+                if isinstance(dumped, dict):
+                    return cls._sanitize_metadata_dict(dumped)
+            except Exception:
+                return {}
+
+        raw_dict = getattr(value, "__dict__", None)
+        if isinstance(raw_dict, dict):
+            return cls._sanitize_metadata_dict({
+                key: item
+                for key, item in raw_dict.items()
+                if not key.startswith("_") and not callable(item)
+            })
+
+        return {}
+
+    @classmethod
+    def _sanitize_metadata_dict(cls, metadata: Dict[str, Any]) -> Dict[str, Any]:
+        sanitized: Dict[str, Any] = {}
+        for key, value in dict(metadata or {}).items():
+            normalized_key = cls._coerce_optional_text(key)
+            if not normalized_key:
+                continue
+
+            if isinstance(value, dict):
+                sanitized_value = cls._sanitize_metadata_dict(value)
+            elif isinstance(value, list):
+                sanitized_value = [
+                    cls._sanitize_metadata_dict(item) if isinstance(item, dict) else item
+                    for item in value
+                    if not type(item).__module__.startswith("unittest.mock")
+                ]
+            elif type(value).__module__.startswith("unittest.mock"):
+                continue
+            else:
+                sanitized_value = value
+
+            sanitized[normalized_key] = sanitized_value
+
+        return sanitized

{iatoolkit-2.45.0 → iatoolkit-2.45.2}/src/iatoolkit/infra/llm_response.py

@@ -33,10 +33,12 @@ class LLMResponse:
     output: List[ToolCall]  # lista de tool calls
     usage: Usage
     reasoning_content: str = None # campo opcional para Chain of Thought
+    reasoning_details: Any = None # provider-native reasoning blocks, when available
 
     # ordered list of content blocks (text and image mixed)
     # Example: [{"type": "text", "text": "..."}, {"type": "image", "source": {"type": "base64", "data": "..."}}]
     content_parts: List[Dict] = None
+    provider_metadata: Dict[str, Any] = None
 
     def __post_init__(self):
         """Asegura que output sea una lista"""
@@ -56,3 +58,5 @@ class LLMResponse:
                     "text": self.output_text
                 })
 
+        if self.provider_metadata is None:
+            self.provider_metadata = {}

{iatoolkit-2.45.0 → iatoolkit-2.45.2}/src/iatoolkit/services/llm_client_service.py

@@ -758,7 +758,16 @@ class llmClient:
         if reasoning_content:
             assistant_message["reasoning_content"] = reasoning_content
 
-        has_assistant_payload = bool(assistant_message.get("content")) or bool(tool_calls) or bool(reasoning_content)
+        reasoning_details = getattr(response, "reasoning_details", None)
+        if reasoning_details:
+            assistant_message["reasoning_details"] = reasoning_details
+
+        has_assistant_payload = (
+            bool(assistant_message.get("content"))
+            or bool(tool_calls)
+            or bool(reasoning_content)
+            or bool(reasoning_details)
+        )
         if not has_assistant_payload:
             return []
 
@@ -1246,15 +1255,58 @@ class llmClient:
             "output_tokens": response.usage.output_tokens,
             "total_tokens": response.usage.total_tokens
         }
+        provider_stats = self._build_provider_stats(getattr(response, "provider_metadata", None))
+        if provider_stats:
+            stats_dict.update(provider_stats)
         return stats_dict
 
-    def add_stats(self, stats1: dict, stats2: dict) -> dict:
+    @staticmethod
+    def _build_provider_stats(provider_metadata: dict | None) -> dict:
+        if not isinstance(provider_metadata, dict) or not provider_metadata:
+            return {}
+
+        provider = str(provider_metadata.get("provider") or "").strip()
+        routed_provider = str(provider_metadata.get("provider_name") or "").strip()
+
+        stats: dict = {}
+        if provider:
+            stats["provider"] = provider
+        if routed_provider:
+            stats["routed_provider"] = routed_provider
+
+        openrouter_metadata = provider_metadata.get("openrouter")
+        if isinstance(openrouter_metadata, dict) and openrouter_metadata:
+            stats["openrouter"] = openrouter_metadata
+
+        provider_call = {}
+        if provider:
+            provider_call["provider"] = provider
+        if routed_provider:
+            provider_call["routed_provider"] = routed_provider
+        if provider_call:
+            stats["provider_calls"] = [provider_call]
+
+        return stats
+
+    @staticmethod
+    def add_stats(stats1: dict, stats2: dict) -> dict:
+        stats1 = dict(stats1 or {})
+        stats2 = dict(stats2 or {})
         stats_dict = {
-            "model": stats1.get('model', ''),
+            **stats1,
+            **stats2,
+            "model": stats2.get('model') or stats1.get('model', ''),
             "input_tokens": (stats1.get('input_tokens') or 0) + (stats2.get('input_tokens') or 0),
             "output_tokens": (stats1.get('output_tokens') or 0) + (stats2.get('output_tokens') or 0),
             "total_tokens": (stats1.get('total_tokens') or 0) + (stats2.get('total_tokens') or 0),
         }
+        provider_calls = []
+        for stats in (stats1, stats2):
+            for provider_call in stats.get("provider_calls") or []:
+                if isinstance(provider_call, dict) and provider_call not in provider_calls:
+                    provider_calls.append(provider_call)
+        if provider_calls:
+            stats_dict["provider_calls"] = provider_calls
         return stats_dict
 
 

{iatoolkit-2.45.0 → iatoolkit-2.45.2/src/iatoolkit.egg-info}/PKG-INFO

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

iatoolkit-2.45.0/src/iatoolkit/infra/llm_providers/openrouter_adapter.py

@@ -1,100 +0,0 @@
-# Copyright (c) 2024 Fernando Libedinsky
-# Product: IAToolkit
-#
-# IAToolkit is open source software.
-
-from typing import Any, Dict
-
-from iatoolkit.infra.llm_providers.openai_compatible_chat_adapter import OpenAICompatibleChatAdapter
-
-
-class OpenRouterAdapter(OpenAICompatibleChatAdapter):
-    """OpenRouter-specific adapter built on top of the shared chat.completions core."""
-
-    supports_multimodal = True
-    supports_reasoning = False
-    supports_metadata = True
-    supports_parallel_tool_calls = True
-
-    # Fixed at "native" by policy: a PDF attachment must reach the routed model
-    # as-is, for it to read with its own vision/multimodal capability - never
-    # silently rewritten into extracted text by OpenRouter's OCR/parsing
-    # pipeline. This only applies when nothing else already requested a plugin
-    # (see _extend_call_kwargs below); an explicit `plugins` kwarg always wins.
-    _PDF_FILE_PARSER_PLUGIN = [{"id": "file-parser", "pdf": {"engine": "native"}}]
-
-    # Providers observed returning HTTP 400 on `type: "file"` content parts
-    # even when routed to a model whose aggregate catalog entry lists "file"
-    # as a supported input modality (OpenRouter's per-endpoint support lags
-    # the model-level declaration). Excluded via `provider.ignore` whenever
-    # we send a native file attachment, so routing skips them instead of
-    # failing the request.
-    _FILE_UNSUPPORTED_PROVIDERS = ("parasail", "akashml", "alibaba", "chutes", "reka")
-
-    def __init__(self, openrouter_client):
-        super().__init__(openai_compatible_client=openrouter_client, provider_label="OpenRouter")
-
-    def _extend_call_kwargs(self, call_kwargs: Dict[str, Any], kwargs: Dict[str, Any]) -> None:
-        text = kwargs.get("text") or {}
-        verbosity = text.get("verbosity") if isinstance(text, dict) else None
-        if verbosity:
-            call_kwargs["verbosity"] = verbosity
-
-        passthrough_keys = (
-            "service_tier",
-            "temperature",
-            "top_p",
-            "max_tokens",
-            "max_completion_tokens",
-            "seed",
-            "stop",
-            "stream",
-            "stream_options",
-            "modalities",
-            "user",
-        )
-        for key in passthrough_keys:
-            if kwargs.get(key) is not None:
-                call_kwargs[key] = kwargs.get(key)
-
-        extra_body = dict(call_kwargs.get("extra_body") or {})
-
-        reasoning_payload = self._build_reasoning_payload(kwargs.get("reasoning"), kwargs)
-        if reasoning_payload:
-            extra_body["reasoning"] = reasoning_payload
-
-        vendor_specific_keys = (
-            "models",
-            "provider",
-            "plugins",
-            "session_id",
-        )
-        for key in vendor_specific_keys:
-            if kwargs.get(key) is not None:
-                extra_body[key] = kwargs.get(key)
-
-        if self._has_pdf_attachment(kwargs.get("attachments")):
-            if "plugins" not in extra_body:
-                extra_body["plugins"] = self._PDF_FILE_PARSER_PLUGIN
-
-            provider_opts = dict(extra_body.get("provider") or {})
-            ignore = list(provider_opts.get("ignore") or [])
-            for provider_slug in self._FILE_UNSUPPORTED_PROVIDERS:
-                if provider_slug not in ignore:
-                    ignore.append(provider_slug)
-            provider_opts["ignore"] = ignore
-            extra_body["provider"] = provider_opts
-
-        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