From 583552a0583081865f64a0df0eccdf6064d13cc2 Mon Sep 17 00:00:00 2001
From: Pol Dellaiera <pol.dellaiera@protonmail.com>
Date: Sun, 18 Aug 2024 10:10:07 +0200
Subject: [PATCH] python312Packages.langgraph: 0.1.9 -> 0.2.4

---
 .../python-modules/langgraph/default.nix      | 44 ++++++++++++++++---
 1 file changed, 38 insertions(+), 6 deletions(-)

diff --git a/pkgs/development/python-modules/langgraph/default.nix b/pkgs/development/python-modules/langgraph/default.nix
index 7a0c6756f419..712e90a602c1 100644
--- a/pkgs/development/python-modules/langgraph/default.nix
+++ b/pkgs/development/python-modules/langgraph/default.nix
@@ -8,36 +8,51 @@
   httpx,
   langchain-core,
   langgraph-sdk,
+  langgraph-checkpoint,
+  langgraph-checkpoint-postgres,
+  langgraph-checkpoint-sqlite,
+  psycopg,
   langsmith,
   poetry-core,
   pydantic,
   pytest-asyncio,
   pytest-mock,
+  pytest-repeat,
   pytest-xdist,
   pytestCheckHook,
   pythonOlder,
   syrupy,
+  postgresql,
+  postgresqlTestHook,
 }:
 
 buildPythonPackage rec {
   pname = "langgraph";
-  version = "0.1.9";
+  version = "0.2.4";
   pyproject = true;
 
-  disabled = pythonOlder "3.10";
+  disabled = pythonOlder "3.9";
 
   src = fetchFromGitHub {
     owner = "langchain-ai";
     repo = "langgraph";
     rev = "refs/tags/${version}";
-    hash = "sha256-sBjSfKzcILkHgvo8g/NHC+/yUjQSyZB/8xaSCY3rPDs=";
+    hash = "sha256-jUBaWXrHCXAph8EGEJnH7lbKIyjQ8oPt4eDMyIkbURo=";
   };
 
+  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 ];
+  dependencies = [
+    langchain-core
+    langgraph-checkpoint
+  ];
 
   pythonImportsCheck = [ "langgraph" ];
 
@@ -46,19 +61,36 @@ buildPythonPackage rec {
     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
   ];
 
-  pytestFlagsArray = [ "--snapshot-update" ];
-
   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"
   ];
 
   passthru = {