diff --git a/pkgs/development/python-modules/langgraph-cli/default.nix b/pkgs/development/python-modules/langgraph-cli/default.nix
index 9fbad3120837..ea74852e47d4 100644
--- a/pkgs/development/python-modules/langgraph-cli/default.nix
+++ b/pkgs/development/python-modules/langgraph-cli/default.nix
@@ -3,16 +3,22 @@
   buildPythonPackage,
   click,
   fetchFromGitHub,
-  nix-update-script,
   poetry-core,
+
+  # for update script
+  langgraph-sdk,
+
+  # testing
   pytest-asyncio,
   pytestCheckHook,
+  docker-compose,
+
   pythonOlder,
 }:
 
 buildPythonPackage rec {
   pname = "langgraph-cli";
-  version = "0.1.52";
+  version = "0.1.71";
   pyproject = true;
 
   disabled = pythonOlder "3.10";
@@ -21,7 +27,7 @@ buildPythonPackage rec {
     owner = "langchain-ai";
     repo = "langgraph";
     tag = "cli==${version}";
-    hash = "sha256-zTBeDJB1Xu/rWsvEC/L4BRzxyh04lPYV7HQNHoJcskk=";
+    hash = "sha256-bTW+je4wuoR0YX5T1wdAee4w/T2jMTQybLLpCxouJxA=";
   };
 
   sourceRoot = "${src.name}/libs/cli";
@@ -33,6 +39,7 @@ buildPythonPackage rec {
   nativeCheckInputs = [
     pytest-asyncio
     pytestCheckHook
+    docker-compose
   ];
 
   pytestFlagsArray = [ "tests/unit_tests" ];
@@ -48,18 +55,15 @@ buildPythonPackage rec {
     "test_config_to_compose_end_to_end"
     "test_config_to_compose_simple_config"
     "test_config_to_compose_watch"
+    # Tests exit value, needs to happen in a passthru test
+    "test_dockerfile_command_with_docker_compose"
   ];
 
-  passthru.updateScript = nix-update-script {
-    extraArgs = [
-      "--version-regex"
-      "cli==(.*)"
-    ];
+  passthru = {
+    inherit (langgraph-sdk) updateScript;
+    skipBulkUpdate = true; # Broken, see https://github.com/NixOS/nixpkgs/issues/379898
   };
 
-  # multiple tags confuse the bulk updater
-  passthru.skipBulkUpdate = true;
-
   meta = {
     description = "Official CLI for LangGraph API";
     homepage = "https://github.com/langchain-ai/langgraph/libs/cli";