agent-framework-lib 0.8.12__tar.gz → 0.8.13__tar.gz

{agent_framework_lib-0.8.12/agent_framework_lib.egg-info → agent_framework_lib-0.8.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-framework-lib
-Version: 0.8.12
+Version: 0.8.13
 Summary: A comprehensive Python framework for building and serving conversational AI agents with FastAPI
 Author-email: Sebastian Pavel <sebastian@cinco.ai>, Elliott Girard <elliott.girard@icloud.com>
 Maintainer-email: Sebastian Pavel <sebastian@cinco.ai>
@@ -58,8 +58,9 @@ Requires-Dist: cryptography>=42.0.0
 Requires-Dist: beautifulsoup4>=4.12.0
 Requires-Dist: llama-index>=0.14.16
 Requires-Dist: llama-index-core>=0.14.24
-Requires-Dist: llama-index-workflows>=2.16.0
+Requires-Dist: llama-index-workflows<2.18,>=2.16.0
 Requires-Dist: llama-index-llms-openai>=0.7.10
+Requires-Dist: llama-index-llms-openai-like>=0.8.0
 Requires-Dist: llama-index-llms-anthropic>=0.12.0
 Requires-Dist: llama-index-llms-google-genai>=0.1.0
 Requires-Dist: graphiti-core>=0.24.3
@@ -80,8 +81,9 @@ Requires-Dist: llm-guard>=0.3.0; extra == "security"
 Provides-Extra: llamaindex
 Requires-Dist: llama-index>=0.14.16; extra == "llamaindex"
 Requires-Dist: llama-index-core>=0.14.24; extra == "llamaindex"
-Requires-Dist: llama-index-workflows>=2.16.0; extra == "llamaindex"
+Requires-Dist: llama-index-workflows<2.18,>=2.16.0; extra == "llamaindex"
 Requires-Dist: llama-index-llms-openai>=0.7.10; extra == "llamaindex"
+Requires-Dist: llama-index-llms-openai-like>=0.8.0; extra == "llamaindex"
 Requires-Dist: llama-index-llms-google-genai>=0.1.0; extra == "llamaindex"
 Requires-Dist: llama-index-llms-anthropic>=0.12.0; extra == "llamaindex"
 Provides-Extra: langgraph
@@ -171,7 +173,7 @@ A comprehensive Python framework for building and serving conversational AI agen
 - 🚀 **Quick Setup** - Create agents in 10-15 minutes
 - 🔌 **Easy MCP Integration** - Connect to external tools effortlessly
 - 🎯 **Skills System** - Markdown-based, on-demand capability loading for token optimization
-- 🔄 **Multi-Provider Support** - OpenAI, Anthropic, Gemini
+- 🔄 **Multi-Provider Support** - OpenAI, Anthropic, Gemini, Mistral (switch model mid-session, tool calls converted between providers)
 - 🎯 **Smart Model Routing** - Auto mode selects the best model per query complexity
 - 💾 **Session Management** - Automatic conversation persistence
 - 📁 **File Storage** - Local, S3, MinIO, GCP support
@@ -331,7 +333,7 @@ Pour **déployer** un agent, partez de l'image officielle — Chromium Playwrigh
 sandbox locale déjà en place (sinon l'agent ne peut ni rendre un PNG ni installer quoi que ce soit) :
 
 ```dockerfile
-FROM ghcr.io/komus-ai/agent-framework-runtime:0.8.12
+FROM ghcr.io/komus-ai/agent-framework-runtime:0.8.13
 ```
 
 Détails, publication et visibilité GHCR : [docs/AGENT_RUNTIME_IMAGE.md](docs/AGENT_RUNTIME_IMAGE.md).

{agent_framework_lib-0.8.12 → agent_framework_lib-0.8.13}/README.md

@@ -13,7 +13,7 @@ A comprehensive Python framework for building and serving conversational AI agen
 - 🚀 **Quick Setup** - Create agents in 10-15 minutes
 - 🔌 **Easy MCP Integration** - Connect to external tools effortlessly
 - 🎯 **Skills System** - Markdown-based, on-demand capability loading for token optimization
-- 🔄 **Multi-Provider Support** - OpenAI, Anthropic, Gemini
+- 🔄 **Multi-Provider Support** - OpenAI, Anthropic, Gemini, Mistral (switch model mid-session, tool calls converted between providers)
 - 🎯 **Smart Model Routing** - Auto mode selects the best model per query complexity
 - 💾 **Session Management** - Automatic conversation persistence
 - 📁 **File Storage** - Local, S3, MinIO, GCP support
@@ -173,7 +173,7 @@ Pour **déployer** un agent, partez de l'image officielle — Chromium Playwrigh
 sandbox locale déjà en place (sinon l'agent ne peut ni rendre un PNG ni installer quoi que ce soit) :
 
 ```dockerfile
-FROM ghcr.io/komus-ai/agent-framework-runtime:0.8.12
+FROM ghcr.io/komus-ai/agent-framework-runtime:0.8.13
 ```
 
 Détails, publication et visibilité GHCR : [docs/AGENT_RUNTIME_IMAGE.md](docs/AGENT_RUNTIME_IMAGE.md).

{agent_framework_lib-0.8.12 → agent_framework_lib-0.8.13}/agent_framework/core/agent_interface.py

@@ -1,5 +1,6 @@
 import abc
 import base64
+import logging
 from collections.abc import AsyncGenerator
 from datetime import datetime, timezone
 from typing import Any, Literal, Union
@@ -7,6 +8,9 @@ from typing import Any, Literal, Union
 from pydantic import BaseModel, Field, field_validator, model_validator
 
 
+logger = logging.getLogger(__name__)
+
+
 # Define a type alias for the content part, which can be text or an image dictionary
 # Following OpenAI/AutoGen structure: {"type": "text", "text": "..."} or {"type": "image_url", "image_url": {"url": "..."}}
 # For simplicity in the interface, we accept Dict for images, specific validation happens in server/agent.
@@ -514,8 +518,20 @@ def strip_technical_details(
     return result
 
 
+def _repair_rich_content(text: str) -> str:
+    """Validate and repair rich content blocks in text, returning it unchanged on failure."""
+    try:
+        from ..processing.rich_content_validation import ContentInterceptor
+
+        return ContentInterceptor().process(text)
+    except Exception as e:
+        logger.warning(f"Rich content validation failed: {e}, using original content")
+        return text
+
+
 def consolidate_text_parts(
     parts: list[AgentOutputPartUnion] | list[dict[str, Any]],
+    repair_rich_content: bool = False,
 ) -> list[AgentOutputPartUnion] | list[dict[str, Any]]:
     """Consolidate consecutive TextOutputStreamPart into single TextOutputPart.
 
@@ -532,6 +548,11 @@ def consolidate_text_parts(
 
     Args:
         parts: List of parts to consolidate. Can be Pydantic models or dicts.
+        repair_rich_content: When True, each consolidated text segment goes through the
+            rich content validation (mermaid, chart, tabledata repairs). The frontend
+            renders these parts — not response_text — so repairs must land here to be
+            visible. Leave False for partial text (e.g. cancelled streams) where a
+            block may be unterminated.
 
     Returns:
         A new list with consecutive text streams consolidated into text_output parts,
@@ -560,6 +581,9 @@ def consolidate_text_parts(
                 # This prevents duplication when special blocks are also parsed at the end
                 cleaned_text, special_parts = parse_special_blocks_from_text(combined_text)
 
+                if repair_rich_content:
+                    cleaned_text = _repair_rich_content(cleaned_text)
+
                 # Add cleaned text as TextOutputPart if there's content left
                 if cleaned_text.strip():
                     if result and isinstance(result[0], dict):

{agent_framework_lib-0.8.12 → agent_framework_lib-0.8.13}/agent_framework/core/context_budget.py

@@ -58,6 +58,16 @@ class ModelContextRegistry:
         "gpt-5.5-pro": 400_000,
         "gemini-2.5-pro": 1_048_576,
         "gemini-3.1-pro-preview": 1_048_576,
+        # Mistral — valeurs max_context_length de GET /v1/models, compte Komus, 17/09/2026.
+        "mistral-small-latest": 262_144,
+        "mistral-medium-latest": 262_144,
+        "mistral-large-latest": 262_144,
+        "magistral-small-latest": 262_144,
+        "magistral-medium-latest": 262_144,
+        "codestral-latest": 256_000,
+        "ministral-3b-latest": 131_072,
+        "ministral-8b-latest": 262_144,
+        "ministral-14b-latest": 262_144,
     }
 
     PROVIDER_DEFAULTS: dict[str, int] = {
@@ -68,6 +78,11 @@ class ModelContextRegistry:
         "o1": 128_000,
         "o3": 200_000,
         "o4": 200_000,
+        # Famille Mistral (repli prudent : le plus petit contexte constaté).
+        "mistral": 131_072,
+        "magistral": 131_072,
+        "codestral": 256_000,
+        "ministral": 131_072,
     }
 
     DEFAULT_CONTEXT_WINDOW: int = 128_000

{agent_framework_lib-0.8.12 → agent_framework_lib-0.8.13}/agent_framework/core/loop_detector.py

@@ -93,10 +93,27 @@ class LoopDetector:
             if msg.role != MessageRole.ASSISTANT:
                 continue
 
+            # Les intégrations OpenAI-compatibles (OpenAI, Mistral via OpenAILike) portent le
+            # MÊME appel à la fois dans un ToolCallBlock et dans additional_kwargs : on ne
+            # le compte qu'une fois, depuis les blocks (constaté le 18/09/2026).
+            block_call_ids = {
+                getattr(block, "tool_call_id", None)
+                for block in (getattr(msg, "blocks", None) or [])
+                if type(block).__name__ == "ToolCallBlock"
+            }
+            has_block_calls = bool(block_call_ids)
+
             # Dict-based tool calls from additional_kwargs
             kwargs = msg.additional_kwargs or {}
             for tc in kwargs.get("tool_calls", []):
+                if not isinstance(tc, dict) and callable(getattr(tc, "model_dump", None)):
+                    try:
+                        tc = tc.model_dump()  # objet pydantic du SDK OpenAI
+                    except Exception:  # pragma: no cover - défensif
+                        continue
                 if isinstance(tc, dict):
+                    if has_block_calls and (tc.get("id") in block_call_ids or not tc.get("id")):
+                        continue
                     name = tc.get("name") or (tc.get("function") or {}).get("name", "")
                     arguments = tc.get("arguments") or (tc.get("function") or {}).get(
                         "arguments", {}
@@ -129,8 +146,20 @@ class LoopDetector:
                         if not name:
                             continue
 
+                        # OpenAI / Mistral stockent tool_kwargs comme CHAÎNE JSON
+                        # (function.arguments). La ramener à ``{}`` donnait le même hash à
+                        # tous les appels d'un outil : trois ``shell_exec`` différents
+                        # passaient pour une boucle et l'agent était forcé de conclure au
+                        # milieu d'une skill (bug constaté le 18/09/2026).
+                        if isinstance(tool_kwargs, str):
+                            try:
+                                import json
+
+                                tool_kwargs = json.loads(tool_kwargs) if tool_kwargs.strip() else {}
+                            except (ValueError, TypeError):
+                                tool_kwargs = {"_raw": tool_kwargs}
                         if not isinstance(tool_kwargs, dict):
-                            tool_kwargs = {}
+                            tool_kwargs = {"_raw": str(tool_kwargs)}
 
                         signatures.append(ToolCallSignature.from_tool_call(name, tool_kwargs))