From c593e4de546554a9805e120d2022cb7dc6342dad Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Sun, 5 Jul 2026 14:23:50 +0000
Subject: [PATCH] python3Packages.mistralai: 2.4.4 -> 2.5.2

Diff: https://github.com/mistralai/client-python/compare/v2.4.4...v2.5.2

Changelog:
https://github.com/mistralai/client-python/blob/v2.5.2/RELEASES.md
---
 .../python-modules/mistralai/default.nix      | 26 ++++++++++++++-----
 1 file changed, 19 insertions(+), 7 deletions(-)

diff --git a/pkgs/development/python-modules/mistralai/default.nix b/pkgs/development/python-modules/mistralai/default.nix
index 7fc33720d0a2..35b41c8926a4 100644
--- a/pkgs/development/python-modules/mistralai/default.nix
+++ b/pkgs/development/python-modules/mistralai/default.nix
@@ -22,6 +22,8 @@
   mcp,
   google-auth,
   requests,
+  websockets,
+  opentelemetry-exporter-otlp-proto-http,
 
   # tests
   opentelemetry-sdk,
@@ -30,7 +32,7 @@
 
 buildPythonPackage (finalAttrs: {
   pname = "mistralai";
-  version = "2.4.4";
+  version = "2.5.2";
   pyproject = true;
   __structuredAttrs = true;
 
@@ -38,7 +40,7 @@ buildPythonPackage (finalAttrs: {
     owner = "mistralai";
     repo = "client-python";
     tag = "v${finalAttrs.version}";
-    hash = "sha256-e2G/bqU45hCfcPEhj4zf5X7EKjs/zWhxCkn9J0u4F7U=";
+    hash = "sha256-skcg4fa9WUNqQs3rqfwVovpQ65hCXqhDDyeJa7kAwNA=";
   };
 
   preBuild = ''
@@ -73,20 +75,30 @@ buildPythonPackage (finalAttrs: {
       google-auth
       requests
     ];
+    realtime = [
+      websockets
+    ];
+    telemetry = [
+      opentelemetry-sdk
+      opentelemetry-exporter-otlp-proto-http
+    ];
   };
 
   pythonImportsCheck = [ "mistralai" ];
 
   nativeCheckInputs = [
-    opentelemetry-sdk
     pytestCheckHook
   ]
   ++ finalAttrs.passthru.optional-dependencies.agents
-  ++ finalAttrs.passthru.optional-dependencies.gcp;
+  ++ finalAttrs.passthru.optional-dependencies.gcp
+  ++ finalAttrs.passthru.optional-dependencies.realtime
+  ++ finalAttrs.passthru.optional-dependencies.telemetry;
 
-  disabledTests = [
-    # AssertionError: <Response [200 OK]> is not an instance of <class 'mistralai.extra.observability.otel.TracedResponse'>
-    "TestOtelTracing"
+  disabledTestPaths = [
+    # ModuleNotFoundError: No module named 'opentelemetry.instrumentation'
+    "src/mistralai/extra/tests/test_otel_tracing.py"
+    # ModuleNotFoundError: No module named 'msgpack'
+    "src/mistralai/extra/tests/test_workflow_encoding.py"
   ];
 
   meta = {