Skip to content

Commit

Permalink
Fix empty migration (#2978)
Browse files Browse the repository at this point in the history
  • Loading branch information
hinthornw authored Jan 9, 2025
1 parent c6d7c80 commit e5b5f95
Show file tree
Hide file tree
Showing 4 changed files with 17 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -58,8 +58,6 @@
);""",
"ALTER TABLE checkpoint_blobs ALTER COLUMN blob DROP not null;",
"""
""",
"""
CREATE INDEX CONCURRENTLY IF NOT EXISTS checkpoints_thread_id_idx ON checkpoints(thread_id);
""",
"""
Expand Down
2 changes: 1 addition & 1 deletion libs/checkpoint-postgres/pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[tool.poetry]
name = "langgraph-checkpoint-postgres"
version = "2.0.9"
version = "2.0.10"
description = "Library with a Postgres implementation of LangGraph checkpoint saver."
authors = []
license = "MIT"
Expand Down
8 changes: 8 additions & 0 deletions libs/checkpoint-postgres/tests/test_store.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
# type: ignore

import re
from contextlib import contextmanager
from typing import Any, Optional
from uuid import uuid4
Expand Down Expand Up @@ -782,3 +783,10 @@ def test_scores(

assert len(results) == 1
assert results[0].score == pytest.approx(similarities[0], abs=1e-3)


def test_nonnull_migrations() -> None:
_leading_comment_remover = re.compile(r"^/\*.*?\*/")
for migration in PostgresStore.MIGRATIONS:
statement = _leading_comment_remover.sub("", migration).split()[0]
assert statement.strip()
8 changes: 8 additions & 0 deletions libs/checkpoint-postgres/tests/test_sync.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
# type: ignore

import re
from contextlib import contextmanager
from typing import Any
from uuid import uuid4
Expand Down Expand Up @@ -238,3 +239,10 @@ def test_null_chars(saver_name: str, test_data) -> None:
list(saver.list(None, filter={"my_key": "abc"}))[0].metadata["my_key"]
== "abc"
)


def test_nonnull_migrations() -> None:
_leading_comment_remover = re.compile(r"^/\*.*?\*/")
for migration in PostgresSaver.MIGRATIONS:
statement = _leading_comment_remover.sub("", migration).split()[0]
assert statement.strip()

0 comments on commit e5b5f95

Please sign in to comment.