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: is not an instance of - "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 = {