sarahec 2024-12-06 19:39:10 +00:00 committed by Sarah Clark
parent 2f0cfddd95
commit 3535716d80

View File

@ -3,12 +3,13 @@
buildPythonPackage, buildPythonPackage,
fetchFromGitHub, fetchFromGitHub,
langgraph-checkpoint, langgraph-checkpoint,
langgraph-sdk,
orjson, orjson,
psycopg, psycopg,
psycopg-pool, psycopg-pool,
langgraph-sdk,
poetry-core, poetry-core,
pythonOlder, pythonOlder,
pgvector,
postgresql, postgresql,
postgresqlTestHook, postgresqlTestHook,
pytestCheckHook, pytestCheckHook,
@ -18,7 +19,7 @@
buildPythonPackage rec { buildPythonPackage rec {
pname = "langgraph-checkpoint-postgres"; pname = "langgraph-checkpoint-postgres";
version = "2.0.2"; version = "2.0.8";
pyproject = true; pyproject = true;
disabled = pythonOlder "3.10"; disabled = pythonOlder "3.10";
@ -27,12 +28,13 @@ buildPythonPackage rec {
owner = "langchain-ai"; owner = "langchain-ai";
repo = "langgraph"; repo = "langgraph";
rev = "refs/tags/checkpointpostgres==${version}"; rev = "refs/tags/checkpointpostgres==${version}";
hash = "sha256-E3gdHWUYelR/xI3Cv3T2uVwsyt+tOkw9MjX7I2RAuiQ="; hash = "sha256-yHLkFUp+q/XOt9Y9Dog2Tgs/K2CU7Bfkkucdr9vAKSg=";
}; };
postgresqlTestSetupPost = '' postgresqlTestSetupPost = ''
substituteInPlace tests/conftest.py \ 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"; sourceRoot = "${src.name}/libs/checkpoint-postgres";
@ -50,15 +52,34 @@ buildPythonPackage rec {
doCheck = !(stdenvNoCC.hostPlatform.isDarwin); doCheck = !(stdenvNoCC.hostPlatform.isDarwin);
pythonImportsCheck = [ "langgraph.checkpoint.postgres" ];
nativeCheckInputs = [ nativeCheckInputs = [
postgresql
postgresqlTestHook
pytest-asyncio pytest-asyncio
pytestCheckHook 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 = { passthru = {
updateScript = langgraph-sdk.updateScript; updateScript = langgraph-sdk.updateScript;
}; };