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

View File

@ -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";
@ -27,12 +28,13 @@ buildPythonPackage rec {
owner = "langchain-ai";
repo = "langgraph";
rev = "refs/tags/checkpointpostgres==${version}";
hash = "sha256-E3gdHWUYelR/xI3Cv3T2uVwsyt+tOkw9MjX7I2RAuiQ=";
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";
@ -50,15 +52,34 @@ buildPythonPackage rec {
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;
};