diff --git a/pkgs/development/python-modules/langgraph-checkpoint-duckdb/default.nix b/pkgs/development/python-modules/langgraph-checkpoint-duckdb/default.nix index b40afc694f52..042960f107fe 100644 --- a/pkgs/development/python-modules/langgraph-checkpoint-duckdb/default.nix +++ b/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 = [ diff --git a/pkgs/development/python-modules/langgraph-checkpoint-postgres/default.nix b/pkgs/development/python-modules/langgraph-checkpoint-postgres/default.nix index ce958ca954f2..65c80d782d9e 100644 --- a/pkgs/development/python-modules/langgraph-checkpoint-postgres/default.nix +++ b/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; }; diff --git a/pkgs/development/python-modules/langgraph-checkpoint-sqlite/default.nix b/pkgs/development/python-modules/langgraph-checkpoint-sqlite/default.nix index 9d4df9530640..44b7c4bbaec4 100644 --- a/pkgs/development/python-modules/langgraph-checkpoint-sqlite/default.nix +++ b/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 = [ diff --git a/pkgs/development/python-modules/langgraph-checkpoint/default.nix b/pkgs/development/python-modules/langgraph-checkpoint/default.nix index eccfbe11935c..7940cb7d15ce 100644 --- a/pkgs/development/python-modules/langgraph-checkpoint/default.nix +++ b/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"; diff --git a/pkgs/development/python-modules/langgraph-cli/default.nix b/pkgs/development/python-modules/langgraph-cli/default.nix index 347be2ffcfa3..8560ffcb14b5 100644 --- a/pkgs/development/python-modules/langgraph-cli/default.nix +++ b/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="; }; diff --git a/pkgs/development/python-modules/langgraph-sdk/default.nix b/pkgs/development/python-modules/langgraph-sdk/default.nix index e2fa8f480fd8..696030652f99 100644 --- a/pkgs/development/python-modules/langgraph-sdk/default.nix +++ b/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 = { diff --git a/pkgs/development/python-modules/langgraph/default.nix b/pkgs/development/python-modules/langgraph/default.nix index 3750ff262179..e813a2db8237 100644 --- a/pkgs/development/python-modules/langgraph/default.nix +++ b/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 = ''