python312Packages.langgraph*: 20241206 update (#362648)

pkgs/development/python-modules/langgraph-checkpoint-duckdb/default.nix

@@ -22,7 +22,7 @@ buildPythonPackage rec {
   src = fetchFromGitHub {
     owner = "langchain-ai";
     repo = "langgraph";
-    rev = "refs/tags/checkpointduckdb==${version}";
+    tag = "checkpointduckdb==${version}";
     hash = "sha256-wSrlFBfTcTgyE46uwv9GCyxRT1xVafgWyP2g87KUTAU=";
   };
 
@@ -36,6 +36,9 @@ buildPythonPackage rec {
     langgraph-checkpoint
   ];
 
+  # Checkpoint clients are lagging behind langgraph-checkpoint
+  pythonRelaxDeps = [ "langgraph-checkpoint" ];
+
   pythonImportsCheck = [ "langgraph.checkpoint.duckdb" ];
 
   nativeCheckInputs = [

pkgs/development/python-modules/langgraph-checkpoint-postgres/default.nix

@@ -3,12 +3,13 @@
   buildPythonPackage,
   fetchFromGitHub,
   langgraph-checkpoint,
+  langgraph-sdk,
   orjson,
   psycopg,
   psycopg-pool,
-  langgraph-sdk,
   poetry-core,
   pythonOlder,
+  pgvector,
   postgresql,
   postgresqlTestHook,
   pytestCheckHook,
@@ -18,7 +19,7 @@
 
 buildPythonPackage rec {
   pname = "langgraph-checkpoint-postgres";
-  version = "2.0.2";
+  version = "2.0.8";
   pyproject = true;
 
   disabled = pythonOlder "3.10";
@@ -26,13 +27,14 @@ buildPythonPackage rec {
   src = fetchFromGitHub {
     owner = "langchain-ai";
     repo = "langgraph";
-    rev = "refs/tags/checkpointpostgres==${version}";
-    hash = "sha256-E3gdHWUYelR/xI3Cv3T2uVwsyt+tOkw9MjX7I2RAuiQ=";
+    tag = "checkpointpostgres==${version}";
+    hash = "sha256-yHLkFUp+q/XOt9Y9Dog2Tgs/K2CU7Bfkkucdr9vAKSg=";
   };
 
   postgresqlTestSetupPost = ''
     substituteInPlace tests/conftest.py \
-      --replace-fail "DEFAULT_URI = \"postgres://postgres:postgres@localhost:5441/postgres?sslmode=disable\"" "DEFAULT_URI = \"postgres:///$PGDATABASE\""
+      --replace-fail "DEFAULT_URI = \"postgres://postgres:postgres@localhost:5441/postgres?sslmode=disable\"" "DEFAULT_URI = \"postgres:///$PGDATABASE\"" \
+      --replace-fail "DEFAULT_POSTGRES_URI = \"postgres://postgres:postgres@localhost:5441/\"" "DEFAULT_POSTGRES_URI = \"postgres:///\""
   '';
 
   sourceRoot = "${src.name}/libs/checkpoint-postgres";
@@ -46,19 +48,41 @@ buildPythonPackage rec {
     psycopg-pool
   ];
 
-  pythonRelaxDeps = [ "psycopg-pool" ];
+  pythonRelaxDeps = [
+    "langgraph-checkpoint"
+    "psycopg-pool"
+  ];
 
   doCheck = !(stdenvNoCC.hostPlatform.isDarwin);
 
-  pythonImportsCheck = [ "langgraph.checkpoint.postgres" ];
-
   nativeCheckInputs = [
-    postgresql
-    postgresqlTestHook
     pytest-asyncio
     pytestCheckHook
+    (postgresql.withPackages (p: with p; [ pgvector ]))
+    postgresqlTestHook
   ];
 
+  preCheck = ''
+    export postgresqlTestUserOptions="LOGIN SUPERUSER"
+  '';
+
+  disabledTests = [
+    # psycopg.errors.FeatureNotSupported: extension "vector" is not available
+    # /nix/store/...postgresql-and-plugins-16.4/share/postgresql/extension/vector.control": No such file or directory.
+    "test_embed_with_path"
+    "test_embed_with_path_sync"
+    "test_scores"
+    "test_search_sorting"
+    "test_vector_store_initialization"
+    "test_vector_insert_with_auto_embedding"
+    "test_vector_update_with_embedding"
+    "test_vector_search_with_filters"
+    "test_vector_search_pagination"
+    "test_vector_search_edge_cases"
+  ];
+
+  pythonImportsCheck = [ "langgraph.checkpoint.postgres" ];
+
   passthru = {
     updateScript = langgraph-sdk.updateScript;
   };

pkgs/development/python-modules/langgraph-checkpoint-sqlite/default.nix

@@ -21,7 +21,7 @@ buildPythonPackage rec {
   src = fetchFromGitHub {
     owner = "langchain-ai";
     repo = "langgraph";
-    rev = "refs/tags/checkpointsqlite==${version}";
+    tag = "checkpointsqlite==${version}";
     hash = "sha256-dh+cjcOp6rGFntz82VNfVyetcrQBdBFdXk5xFb0aR5c=";
   };
 
@@ -34,6 +34,9 @@ buildPythonPackage rec {
     langgraph-checkpoint
   ];
 
+  # Checkpoint clients are lagging behind langgraph-checkpoint
+  pythonRelaxDeps = [ "langgraph-checkpoint" ];
+
   pythonImportsCheck = [ "langgraph.checkpoint.sqlite" ];
 
   nativeCheckInputs = [

pkgs/development/python-modules/langgraph-checkpoint/default.nix

@@ -15,7 +15,7 @@
 
 buildPythonPackage rec {
   pname = "langgraph-checkpoint";
-  version = "2.0.2";
+  version = "2.0.8";
   pyproject = true;
 
   disabled = pythonOlder "3.9";
@@ -23,8 +23,8 @@ buildPythonPackage rec {
   src = fetchFromGitHub {
     owner = "langchain-ai";
     repo = "langgraph";
-    rev = "refs/tags/checkpoint==${version}";
-    hash = "sha256-Mjo6NJ6vYb2E7nk0D/2M7jzr39xRvRRhUZE4tP247to=";
+    tag = "checkpoint==${version}";
+    hash = "sha256-obDK6wn+oo8zDQsidogwKTIgT5wuUH/l4y+12cttkd0=";
   };
 
   sourceRoot = "${src.name}/libs/checkpoint";

pkgs/development/python-modules/langgraph-cli/default.nix

@@ -20,7 +20,7 @@ buildPythonPackage rec {
   src = fetchFromGitHub {
     owner = "langchain-ai";
     repo = "langgraph";
-    rev = "refs/tags/cli==${version}";
+    tag = "cli==${version}";
     hash = "sha256-zTBeDJB1Xu/rWsvEC/L4BRzxyh04lPYV7HQNHoJcskk=";
   };
 

pkgs/development/python-modules/langgraph-sdk/default.nix

@@ -17,14 +17,14 @@
 
 buildPythonPackage rec {
   pname = "langgraph-sdk";
-  version = "0.1.35";
+  version = "0.1.43";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "langchain-ai";
     repo = "langgraph";
-    rev = "refs/tags/sdk==${version}";
-    hash = "sha256-HWUGRoe5S0HPfOEbqUnFYLVrHe3SJtk3U8cy1JON050=";
+    tag = "sdk==${version}";
+    hash = "sha256-mG04V36Aa5Df5pUgr+xWej8i2XYx+O/N61sSzxwN9Go=";
   };
 
   sourceRoot = "${src.name}/libs/sdk-py";
@@ -37,6 +37,8 @@ buildPythonPackage rec {
     orjson
   ];
 
+  disabledTests = [ "test_aevaluate_results" ]; # Compares execution time to magic number
+
   pythonImportsCheck = [ "langgraph_sdk" ];
 
   passthru = {

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

@@ -35,14 +35,14 @@
 
 buildPythonPackage rec {
   pname = "langgraph";
-  version = "0.2.43";
+  version = "0.2.56";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "langchain-ai";
     repo = "langgraph";
-    rev = "refs/tags/${version}";
-    hash = "sha256-8xTjWBq6GSl6X2IId3roS3ZNM/h76EGPitS08YQ4e7Y=";
+    tag = version;
+    hash = "sha256-X/IMNEmggu9bSJFUaTohbFYxGZBguf+eFb3ObmQGplk=";
   };
 
   postgresqlTestSetupPost = ''