mistral-vibe: 2.4.2 -> 2.5.0 (#500483)

Author: NickCao

pkgs/by-name/mi/mistral-vibe/package.nix

@@ -12,14 +12,14 @@
 
 python3Packages.buildPythonApplication (finalAttrs: {
   pname = "mistral-vibe";
-  version = "2.4.2";
+  version = "2.5.0";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "mistralai";
     repo = "mistral-vibe";
     tag = "v${finalAttrs.version}";
-    hash = "sha256-r/9kMhkoLfj9oEifFun/bpIQYEouqm9YEiWZVk07+S8=";
+    hash = "sha256-5su0Qfg3M+Yq4pkptDOJhvM8VFGCaOLeeDijeFeywP4=";
   };
 
   build-system = with python3Packages; [
@@ -58,12 +58,14 @@ python3Packages.buildPythonApplication (finalAttrs: {
     pyyaml
     requests
     rich
+    sounddevice
     textual
     textual-speedups
     tomli-w
     tree-sitter
     tree-sitter-bash
     watchfiles
+    websockets
     zstandard
   ];
 
@@ -81,7 +83,29 @@ python3Packages.buildPythonApplication (finalAttrs: {
   ];
   versionCheckKeepEnvironment = [ "HOME" ];
 
-  disabledTests = lib.optionals stdenv.hostPlatform.isDarwin [
+  disabledTests = [
+    # Fail in the sandbox
+    # vibe.core.audio_recorder.audio_recorder_port.NoAudioInputDeviceError: No audio input device available
+    "test_audio_stream_yields_chunks"
+    "test_auto_stops_after_max_duration"
+    "test_buffer_mode_audio_stream_yields_nothing"
+    "test_can_record_multiple_times"
+    "test_cancel_discards_audio"
+    "test_manual_stop_prevents_on_expire"
+    "test_on_expire_receives_audio"
+    "test_peak_clamps_to_one"
+    "test_peak_updates_from_callback"
+    "test_silent_audio_has_zero_peak"
+    "test_start_sets_recording_state"
+    "test_start_when_already_recording_raises"
+    "test_stop_from_event_loop_does_not_block"
+    "test_stop_returns_empty_data_in_stream_mode"
+    "test_stop_returns_positive_duration"
+    "test_stop_returns_valid_wav"
+    "test_stop_without_drain_returns_promptly"
+    "test_stream_audio_does_not_leak_into_buffer_recording"
+  ]
+  ++ lib.optionals stdenv.hostPlatform.isDarwin [
     # AssertionError
     "test_rebuilds_index_when_mass_change_threshold_is_exceeded"
     "test_updates_index_incrementally_by_default"

pkgs/development/python-modules/lmstudio/default.nix

@@ -0,0 +1,94 @@
+{
+  lib,
+  buildPythonPackage,
+  fetchFromGitHub,
+
+  # build-system
+  pdm-backend,
+
+  # dependencies
+  anyio,
+  httpx,
+  httpx-ws,
+  msgspec,
+  typing-extensions,
+
+  # tests
+  pytest-asyncio,
+  pytestCheckHook,
+}:
+
+buildPythonPackage (finalAttrs: {
+  pname = "lmstudio";
+  version = "1.5.0";
+  pyproject = true;
+
+  src = fetchFromGitHub {
+    owner = "lmstudio-ai";
+    repo = "lmstudio-python";
+    tag = finalAttrs.version;
+    hash = "sha256-3LOoCWoQ7eXRPXHRio9Rtle07HcV3ZrWkrtVVY6mvfI=";
+  };
+
+  build-system = [
+    pdm-backend
+  ];
+
+  dependencies = [
+    anyio
+    httpx
+    httpx-ws
+    msgspec
+    typing-extensions
+  ];
+
+  pythonImportsCheck = [ "lmstudio" ];
+
+  nativeCheckInputs = [
+    pytest-asyncio
+    pytestCheckHook
+  ];
+
+  disabledTests = [
+    # lmstudio.LMStudioRuntimeError: Local API host port is not yet resolved.
+    "test_session_disconnected_async"
+  ];
+
+  disabledTestPaths = [
+    # ModuleNotFoundError: No module named 'pytest_subtests'
+    "tests/async/test_model_catalog_async.py"
+    "tests/sync/test_model_catalog_sync.py"
+    "tests/test_inference.py"
+
+    # lmstudio.LMStudioRuntimeError: Local API host port is not yet resolved.
+    "tests/async/test_embedding_async.py"
+    "tests/async/test_images_async.py"
+    "tests/async/test_inference_async.py"
+    "tests/async/test_llm_async.py"
+    "tests/async/test_model_handles_async.py"
+    "tests/async/test_repository_async.py"
+    "tests/async/test_sdk_bypass_async.py"
+    "tests/sync/test_embedding_sync.py"
+    "tests/sync/test_images_sync.py"
+    "tests/sync/test_inference_sync.py"
+    "tests/sync/test_llm_sync.py"
+    "tests/sync/test_model_handles_sync.py"
+    "tests/sync/test_repository_sync.py"
+    "tests/sync/test_sdk_bypass_sync.py"
+    "tests/test_convenience_api.py"
+    "tests/test_logging.py"
+    "tests/test_plugin_examples.py"
+    "tests/test_sessions.py"
+    "tests/test_timeouts.py"
+  ];
+
+  __darwinAllowLocalNetworking = true;
+
+  meta = {
+    description = "LM Studio Python SDK";
+    homepage = "https://github.com/lmstudio-ai/lmstudio-python";
+    changelog = "https://github.com/lmstudio-ai/lmstudio-python/releases/tag/${finalAttrs.src.tag}";
+    license = lib.licenses.mit;
+    maintainers = with lib.maintainers; [ GaetanLepage ];
+  };
+})

pkgs/development/python-modules/mistralai/default.nix

@@ -9,13 +9,10 @@
   # dependencies
   eval-type-backport,
   httpx,
-  invoke,
   opentelemetry-api,
-  opentelemetry-exporter-otlp-proto-http,
-  opentelemetry-sdk,
+  opentelemetry-semantic-conventions,
   pydantic,
   python-dateutil,
-  pyyaml,
   typing-inspection,
 
   # optional-dependencies
@@ -26,19 +23,20 @@
   requests,
 
   # tests
+  opentelemetry-sdk,
   pytestCheckHook,
 }:
 
 buildPythonPackage (finalAttrs: {
   pname = "mistralai";
-  version = "1.12.3";
+  version = "2.0.4";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "mistralai";
     repo = "client-python";
     tag = "v${finalAttrs.version}";
-    hash = "sha256-DaCAO4/DoskFuRgNcoLWBuRdnxXfYhsgY/WZu/s1wk0=";
+    hash = "sha256-SLPLj9rp8TTOSE3ldobBFU1+MpffzH1Bpshw+7LLUvU=";
   };
 
   preBuild = ''
@@ -50,18 +48,15 @@ buildPythonPackage (finalAttrs: {
   ];
 
   pythonRelaxDeps = [
-    "opentelemetry-exporter-otlp-proto-http"
+    "opentelemetry-semantic-conventions"
   ];
   dependencies = [
     eval-type-backport
     httpx
-    invoke
     opentelemetry-api
-    opentelemetry-exporter-otlp-proto-http
-    opentelemetry-sdk
+    opentelemetry-semantic-conventions
     pydantic
     python-dateutil
-    pyyaml
     typing-inspection
   ];
 
@@ -80,7 +75,15 @@ buildPythonPackage (finalAttrs: {
   pythonImportsCheck = [ "mistralai" ];
 
   nativeCheckInputs = [
+    opentelemetry-sdk
     pytestCheckHook
+  ]
+  ++ finalAttrs.passthru.optional-dependencies.agents
+  ++ finalAttrs.passthru.optional-dependencies.gcp;
+
+  disabledTests = [
+    # AssertionError: <Response [200 OK]> is not an instance of <class 'mistralai.extra.observability.otel.TracedResponse'>
+    "TestOtelTracing"
   ];
 
   meta = {