From 105e61ab039581a23893f73cc9f1cb4579cd5240 Mon Sep 17 00:00:00 2001
From: Sarah Clark <seclark@nextquestion.net>
Date: Fri, 6 Dec 2024 12:08:20 -0800
Subject: [PATCH] langgraph-checkpoint-*: Relax dependency on
 langgraph-checkpoint

---
 .../python-modules/langgraph-checkpoint-duckdb/default.nix | 3 +++
 .../langgraph-checkpoint-postgres/default.nix              | 7 +++++--
 .../python-modules/langgraph-checkpoint-sqlite/default.nix | 3 +++
 3 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/pkgs/development/python-modules/langgraph-checkpoint-duckdb/default.nix b/pkgs/development/python-modules/langgraph-checkpoint-duckdb/default.nix
index b40afc694f52..2c3b68f52b67 100644
--- a/pkgs/development/python-modules/langgraph-checkpoint-duckdb/default.nix
+++ b/pkgs/development/python-modules/langgraph-checkpoint-duckdb/default.nix
@@ -36,6 +36,9 @@ buildPythonPackage rec {
     langgraph-checkpoint
   ];
 
+  # Checkpoint clients are lagging behind langgraph-checkpoint
+  pythonRelaxDeps = [ "langgraph-checkpoint" ];
+
   pythonImportsCheck = [ "langgraph.checkpoint.duckdb" ];
 
   nativeCheckInputs = [
diff --git a/pkgs/development/python-modules/langgraph-checkpoint-postgres/default.nix b/pkgs/development/python-modules/langgraph-checkpoint-postgres/default.nix
index 28a72cc7855f..a2021ba1fdf3 100644
--- a/pkgs/development/python-modules/langgraph-checkpoint-postgres/default.nix
+++ b/pkgs/development/python-modules/langgraph-checkpoint-postgres/default.nix
@@ -48,7 +48,10 @@ buildPythonPackage rec {
     psycopg-pool
   ];
 
-  pythonRelaxDeps = [ "psycopg-pool" ];
+  pythonRelaxDeps = [
+    "langgraph-checkpoint"
+    "psycopg-pool"
+  ];
 
   doCheck = !(stdenvNoCC.hostPlatform.isDarwin);
 
@@ -60,7 +63,7 @@ buildPythonPackage rec {
   ];
 
   preCheck = ''
-    export postgresqlTestUserOptions = "LOGIN SUPERUSER"
+    export postgresqlTestUserOptions="LOGIN SUPERUSER"
   '';
 
   disabledTests = [
diff --git a/pkgs/development/python-modules/langgraph-checkpoint-sqlite/default.nix b/pkgs/development/python-modules/langgraph-checkpoint-sqlite/default.nix
index 9d4df9530640..2f973160521c 100644
--- a/pkgs/development/python-modules/langgraph-checkpoint-sqlite/default.nix
+++ b/pkgs/development/python-modules/langgraph-checkpoint-sqlite/default.nix
@@ -34,6 +34,9 @@ buildPythonPackage rec {
     langgraph-checkpoint
   ];
 
+  # Checkpoint clients are lagging behind langgraph-checkpoint
+  pythonRelaxDeps = [ "langgraph-checkpoint" ];
+
   pythonImportsCheck = [ "langgraph.checkpoint.sqlite" ];
 
   nativeCheckInputs = [