diff --git a/pkgs/development/python-modules/langgraph/default.nix b/pkgs/development/python-modules/langgraph/default.nix index 712e90a602c1..cb442e8919f4 100644 --- a/pkgs/development/python-modules/langgraph/default.nix +++ b/pkgs/development/python-modules/langgraph/default.nix @@ -1,43 +1,48 @@ { lib, buildPythonPackage, + fetchFromGitHub, + + # build-system + poetry-core, + + # dependencies + langchain-core, + langgraph-checkpoint, + + # tests aiosqlite, dataclasses-json, - fetchFromGitHub, grandalf, httpx, - langchain-core, - langgraph-sdk, - langgraph-checkpoint, langgraph-checkpoint-postgres, langgraph-checkpoint-sqlite, - psycopg, langsmith, - poetry-core, + psycopg, pydantic, pytest-asyncio, pytest-mock, pytest-repeat, pytest-xdist, pytestCheckHook, - pythonOlder, syrupy, postgresql, postgresqlTestHook, + + # passthru + langgraph-sdk, }: buildPythonPackage rec { pname = "langgraph"; - version = "0.2.4"; + version = "0.2.19"; pyproject = true; - disabled = pythonOlder "3.9"; - src = fetchFromGitHub { owner = "langchain-ai"; repo = "langgraph"; rev = "refs/tags/${version}"; - hash = "sha256-jUBaWXrHCXAph8EGEJnH7lbKIyjQ8oPt4eDMyIkbURo="; + hash = "sha256-qJIZAHftIKyWK0A/MjilalmmB8b8E7JtLnFn156hE08="; }; postgresqlTestSetupPost = '' @@ -65,6 +70,7 @@ buildPythonPackage rec { langgraph-checkpoint-sqlite langsmith psycopg + psycopg.pool pydantic pytest-asyncio pytest-mock @@ -93,6 +99,12 @@ buildPythonPackage rec { "test_remove_message_via_state_update" ]; + disabledTestPaths = [ + # psycopg.errors.InsufficientPrivilege: permission denied to create database + "tests/test_pregel_async.py" + "tests/test_pregel.py" + ]; + passthru = { updateScript = langgraph-sdk.updateScript; };