Merge pull request #40 from a180285/main

change checkpoint_ns to 255 (temp fix)
tjni · Dec 21, 2024 · b54e96c · b54e96c
2 parents 8345e91 + 7d04b8f
commit b54e96c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,2 +1,3 @@
 .mypy_cache/
 __pycache__/
+.idea/
diff --git a/langgraph/checkpoint/mysql/__init__.py b/langgraph/checkpoint/mysql/__init__.py
@@ -119,6 +119,7 @@ def setup(self) -> None:
             ):
                 cur.execute(migration)
                 cur.execute(f"INSERT INTO checkpoint_migrations (v) VALUES ({v})")
+                cur.execute("COMMIT")
 
     def list(
         self,

diff --git a/langgraph/checkpoint/mysql/base.py b/langgraph/checkpoint/mysql/base.py
@@ -69,6 +69,9 @@
     """
     CREATE INDEX checkpoints_checkpoint_id_idx ON checkpoints (checkpoint_id);
     """,
+    "ALTER TABLE checkpoints MODIFY COLUMN `checkpoint_ns` VARCHAR(255) NOT NULL DEFAULT '';",
+    "ALTER TABLE checkpoint_blobs MODIFY COLUMN `checkpoint_ns` VARCHAR(255) NOT NULL DEFAULT '';",
+    "ALTER TABLE checkpoint_writes MODIFY COLUMN `checkpoint_ns` VARCHAR(255) NOT NULL DEFAULT '';",
 ]
 
 SELECT_SQL = f"""