Add workflow payload compression encoder

pyproject.toml

@@ -48,6 +48,9 @@ workflow_payload_offloading = [
 workflow_payload_encryption = [
     "cryptography>=41.0.0,<47.0.0",
 ]
+workflow_payload_compression = [
+    "zstandard>=0.25.0,<0.26",
+]
 
 
 [project.urls]
@@ -69,6 +72,7 @@ dev = [
     "griffe>=1.7.3,<2",
     "authlib>=1.5.2,<2",
     "websockets >=13.0",
+    "zstandard>=0.25.0,<0.26",
 ]
 lint = [
     "ruff>=0.11.10,<0.12",

src/mistralai/extra/exceptions.py

@@ -28,6 +28,10 @@ class WorkflowPayloadEncryptionException(MistralClientException):
     """Workflow payload encryption exception"""
 
 
+class WorkflowPayloadCompressionException(MistralClientException):
+    """Workflow payload compression exception"""
+
+
 class RunException(MistralClientException):
     """Conversation run errors."""
 

src/mistralai/extra/tests/fixtures/workflow_encoding.py

@@ -0,0 +1,26 @@
+from __future__ import annotations
+
+from typing import Any
+
+
+class InMemoryBlobStorage:
+    def __init__(self) -> None:
+        self.blobs: dict[str, bytes] = {}
+
+    async def __aenter__(self) -> "InMemoryBlobStorage":
+        return self
+
+    async def __aexit__(self, *_args: Any) -> None:
+        pass
+
+    async def upload_blob(self, key: str, content: bytes) -> str:
+        self.blobs[key] = content
+        return key
+
+    async def get_blob(self, key: str) -> bytes:
+        return self.blobs[key]
+
+    async def get_blob_properties(self, key: str) -> dict[str, Any] | None:
+        if key not in self.blobs:
+            return None
+        return {"size": len(self.blobs[key]), "last_modified": "test"}