Fix empty migration (#2978)

This commit is contained in:
William FH
2025-01-09 23:14:02 +00:00
committed by GitHub
parent c6d7c80a99
commit e5b5f9510b
4 changed files with 17 additions and 3 deletions
@@ -58,8 +58,6 @@ MIGRATIONS = [
);""",
"ALTER TABLE checkpoint_blobs ALTER COLUMN blob DROP not null;",
"""
""",
"""
CREATE INDEX CONCURRENTLY IF NOT EXISTS checkpoints_thread_id_idx ON checkpoints(thread_id);
""",
"""
+1 -1
View File
@@ -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"
@@ -1,5 +1,6 @@
# type: ignore
import re
from contextlib import contextmanager
from typing import Any, Optional
from uuid import uuid4
@@ -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()
@@ -1,5 +1,6 @@
# type: ignore
import re
from contextlib import contextmanager
from typing import Any
from uuid import uuid4
@@ -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()