1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
{
lib,
stdenv,
buildPythonPackage,
fetchFromGitHub,
# build-system
poetry-core,
# dependencies
langchain-core,
langgraph-checkpoint,
# tests
aiosqlite,
dataclasses-json,
grandalf,
httpx,
langgraph-checkpoint-postgres,
langgraph-checkpoint-sqlite,
langsmith,
psycopg,
pydantic,
pytest-asyncio,
pytest-mock,
pytest-repeat,
pytest-xdist,
pytestCheckHook,
syrupy,
postgresql,
postgresqlTestHook,
# passthru
langgraph-sdk,
}:
buildPythonPackage rec {
pname = "langgraph";
version = "0.2.21";
pyproject = true;
src = fetchFromGitHub {
owner = "langchain-ai";
repo = "langgraph";
rev = "refs/tags/${version}";
hash = "sha256-1Ch2V85omAKnXK9rMihNtyjIoOvmVUm8Dbdo5GBoik4=";
};
postgresqlTestSetupPost = ''
substituteInPlace tests/conftest.py \
--replace-fail "DEFAULT_POSTGRES_URI = \"postgres://postgres:postgres@localhost:5442/\"" "DEFAULT_POSTGRES_URI = \"postgres:///$PGDATABASE\""
'';
sourceRoot = "${src.name}/libs/langgraph";
build-system = [ poetry-core ];
dependencies = [
langchain-core
langgraph-checkpoint
];
pythonImportsCheck = [ "langgraph" ];
# postgresql doesn't play nicely with the darwin sandbox:
# FATAL: could not create shared memory segment: Operation not permitted
doCheck = !stdenv.hostPlatform.isDarwin;
nativeCheckInputs = [
aiosqlite
dataclasses-json
grandalf
httpx
langgraph-checkpoint-postgres
langgraph-checkpoint-sqlite
langsmith
psycopg
psycopg.pool
pydantic
pytest-asyncio
pytest-mock
pytest-repeat
pytest-xdist
pytestCheckHook
syrupy
postgresql
postgresqlTestHook
];
disabledTests = [
"test_doesnt_warn_valid_schema" # test is flaky due to pydantic error on the exception
# Disabling tests that requires to create new random databases
"test_cancel_graph_astream"
"test_cancel_graph_astream_events_v2"
"test_channel_values"
"test_fork_always_re_runs_nodes"
"test_interruption_without_state_updates"
"test_interruption_without_state_updates_async"
"test_invoke_two_processes_in_out_interrupt"
"test_nested_graph_interrupts"
"test_no_modifier_async"
"test_no_modifier"
"test_pending_writes_resume"
"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;
};
meta = {
description = "Build resilient language agents as graphs";
homepage = "https://github.com/langchain-ai/langgraph";
changelog = "https://github.com/langchain-ai/langgraph/releases/tag/${version}";
license = lib.licenses.mit;
maintainers = with lib.maintainers; [ sarahec ];
};
}
|
