From 925d4992275e40b610d2371d39ecb581f57a9ddd Mon Sep 17 00:00:00 2001
From: Sarah Clark <seclark@nextquestion.net>
Date: Fri, 6 Sep 2024 17:17:42 -0700
Subject: [PATCH] 
 python312Packages.langgraph-{sdk,checkpoint,checkpoint-sqlite,checkpoint-postgres}:
 Sept. 6, 2024 updates

---
 .../langgraph-checkpoint-postgres/default.nix         | 11 +++++++++--
 .../langgraph-checkpoint-sqlite/default.nix           |  2 +-
 .../python-modules/langgraph-checkpoint/default.nix   |  9 +++++++--
 .../python-modules/langgraph-sdk/default.nix          |  3 +++
 4 files changed, 20 insertions(+), 5 deletions(-)

diff --git a/pkgs/development/python-modules/langgraph-checkpoint-postgres/default.nix b/pkgs/development/python-modules/langgraph-checkpoint-postgres/default.nix
index a3e9d63d6213..adcd0856f027 100644
--- a/pkgs/development/python-modules/langgraph-checkpoint-postgres/default.nix
+++ b/pkgs/development/python-modules/langgraph-checkpoint-postgres/default.nix
@@ -5,6 +5,7 @@
   langgraph-checkpoint,
   orjson,
   psycopg,
+  psycopg-pool,
   langgraph-sdk,
   poetry-core,
   pythonOlder,
@@ -12,11 +13,12 @@
   postgresqlTestHook,
   pytestCheckHook,
   pytest-asyncio,
+  stdenvNoCC,
 }:
 
 buildPythonPackage rec {
   pname = "langgraph-checkpoint-postgres";
-  version = "1.0.3";
+  version = "1.0.6";
   pyproject = true;
 
   disabled = pythonOlder "3.10";
@@ -25,7 +27,7 @@ buildPythonPackage rec {
     owner = "langchain-ai";
     repo = "langgraph";
     rev = "refs/tags/checkpointpostgres==${version}";
-    hash = "sha256-U7Bymo+Nj82kwjxN33W2MT10jv+lioZUxIKUt8Yxh/s=";
+    hash = "sha256-F9sgZQQBFs5hDUsaR5BI9ERve9L8LTUvEKOgyz5ioqY=";
   };
 
   postgresqlTestSetupPost = ''
@@ -41,8 +43,13 @@ buildPythonPackage rec {
     langgraph-checkpoint
     orjson
     psycopg
+    psycopg-pool
   ];
 
+  pythonRelaxDeps = [ "psycopg-pool" ];
+
+  doCheck = !(stdenvNoCC.isDarwin);
+
   pythonImportsCheck = [ "langgraph.checkpoint.postgres" ];
 
   nativeCheckInputs = [
diff --git a/pkgs/development/python-modules/langgraph-checkpoint-sqlite/default.nix b/pkgs/development/python-modules/langgraph-checkpoint-sqlite/default.nix
index 323faf5ffb1b..b8bf9df2eaa9 100644
--- a/pkgs/development/python-modules/langgraph-checkpoint-sqlite/default.nix
+++ b/pkgs/development/python-modules/langgraph-checkpoint-sqlite/default.nix
@@ -13,7 +13,7 @@
 
 buildPythonPackage rec {
   pname = "langgraph-checkpoint-sqlite";
-  version = "1.0.0";
+  version = "1.0.2";
   pyproject = true;
 
   disabled = pythonOlder "3.9";
diff --git a/pkgs/development/python-modules/langgraph-checkpoint/default.nix b/pkgs/development/python-modules/langgraph-checkpoint/default.nix
index 77b3ba34826e..82323a80488c 100644
--- a/pkgs/development/python-modules/langgraph-checkpoint/default.nix
+++ b/pkgs/development/python-modules/langgraph-checkpoint/default.nix
@@ -4,6 +4,7 @@
   dataclasses-json,
   fetchFromGitHub,
   langchain-core,
+  langgraph-sdk,
   poetry-core,
   pytest-asyncio,
   pytestCheckHook,
@@ -12,7 +13,7 @@
 
 buildPythonPackage rec {
   pname = "langgraph-checkpoint";
-  version = "1.0.3";
+  version = "1.0.9";
   pyproject = true;
 
   disabled = pythonOlder "3.9";
@@ -21,7 +22,7 @@ buildPythonPackage rec {
     owner = "langchain-ai";
     repo = "langgraph";
     rev = "refs/tags/checkpoint==${version}";
-    hash = "sha256-5JP9f2uHNo71btQ96sBPlS7JPqo35C3VEMeHN1cJSro=";
+    hash = "sha256-3gm+L67pPAKpY1kqnX1lPnca40KoBVZdRZ1Cy6D0dzU=";
   };
 
   sourceRoot = "${src.name}/libs/checkpoint";
@@ -38,6 +39,10 @@ buildPythonPackage rec {
     pytestCheckHook
   ];
 
+  passthru = {
+    updateScript = langgraph-sdk.updateScript;
+  };
+
   meta = {
     changelog = "https://github.com/langchain-ai/langgraph/releases/tag/checkpoint==${version}";
     description = "Library with base interfaces for LangGraph checkpoint savers";
diff --git a/pkgs/development/python-modules/langgraph-sdk/default.nix b/pkgs/development/python-modules/langgraph-sdk/default.nix
index 528e29c6bf3b..710433dfe134 100644
--- a/pkgs/development/python-modules/langgraph-sdk/default.nix
+++ b/pkgs/development/python-modules/langgraph-sdk/default.nix
@@ -48,6 +48,9 @@ buildPythonPackage rec {
       set -eu -o pipefail
       nix-update --commit --version-regex '(.*)' python3Packages.langgraph
       nix-update --commit --version-regex 'sdk==(.*)' python3Packages.langgraph-sdk
+      nix-update --commit --version-regex 'checkpoint==(.*)' python3Packages.langgraph-checkpoint
+      nix-update --commit --version-regex 'checkpointpostgres==(.*)' python3Packages.langgraph-checkpoint-postgres
+      nix-update --commit --version-regex 'checkpointsqlite==(.*)' python3Packages.langgraph-checkpoint-sqlite
     '';
   };