langgraph-runtime-pg 0.11.1__py3-none-any.whl

langgraph_runtime_pg/migrations/versions/001_initial_schema.py

@@ -0,0 +1,331 @@
+"""initial schema
+
+Revision ID: 001_initial
+Revises:
+Create Date: 2026-07-25 21:48:52.124561
+"""
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+import sqlalchemy as sa
+from alembic import op
+from sqlalchemy.dialects import postgresql
+
+revision: str = "001_initial"
+down_revision: str | Sequence[str] | None = None
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "assistant_versions",
+        sa.Column("assistant_id", sa.UUID(), nullable=False),
+        sa.Column("version", sa.Integer(), nullable=False),
+        sa.Column("graph_id", sa.String(length=256), nullable=False),
+        sa.Column(
+            "config",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column(
+            "context",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column(
+            "metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column("name", sa.String(length=512), nullable=False),
+        sa.Column("description", sa.Text(), nullable=True),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.PrimaryKeyConstraint("assistant_id", "version"),
+    )
+    op.create_table(
+        "assistants",
+        sa.Column("assistant_id", sa.UUID(), nullable=False),
+        sa.Column("graph_id", sa.String(length=256), nullable=False),
+        sa.Column("name", sa.String(length=512), nullable=False),
+        sa.Column("description", sa.Text(), nullable=True),
+        sa.Column(
+            "config",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column(
+            "context",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column(
+            "metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column("version", sa.Integer(), nullable=False),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.PrimaryKeyConstraint("assistant_id"),
+    )
+    op.create_index("ix_assistants_created_at", "assistants", ["created_at"], unique=False)
+    op.create_index("ix_assistants_graph_id", "assistants", ["graph_id"], unique=False)
+    op.create_index(
+        "ix_assistants_metadata_gin",
+        "assistants",
+        ["metadata"],
+        unique=False,
+        postgresql_using="gin",
+        postgresql_ops={"metadata": "jsonb_path_ops"},
+    )
+    op.create_table(
+        "crons",
+        sa.Column("cron_id", sa.UUID(), nullable=False),
+        sa.Column("assistant_id", sa.UUID(), nullable=False),
+        sa.Column("thread_id", sa.UUID(), nullable=True),
+        sa.Column("schedule", sa.String(length=128), nullable=False),
+        sa.Column(
+            "payload",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column(
+            "metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column("next_run_date", sa.DateTime(timezone=True), nullable=True),
+        sa.Column("end_time", sa.DateTime(timezone=True), nullable=True),
+        sa.Column("user_id", sa.String(length=256), nullable=True),
+        sa.Column("timezone", sa.String(length=64), nullable=True),
+        sa.Column("on_run_completed", sa.String(length=16), nullable=True),
+        sa.Column("enabled", sa.Boolean(), server_default=sa.text("true"), nullable=False),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.PrimaryKeyConstraint("cron_id"),
+    )
+    op.create_index("ix_crons_assistant_id", "crons", ["assistant_id"], unique=False)
+    op.create_index(
+        "ix_crons_enabled_next_run", "crons", ["enabled", "next_run_date"], unique=False
+    )
+    op.create_index(
+        "ix_crons_metadata_gin",
+        "crons",
+        ["metadata"],
+        unique=False,
+        postgresql_using="gin",
+        postgresql_ops={"metadata": "jsonb_path_ops"},
+    )
+    op.create_index("ix_crons_thread_id", "crons", ["thread_id"], unique=False)
+    op.create_table(
+        "retry_counters",
+        sa.Column("run_id", sa.UUID(), nullable=False),
+        sa.Column("count", sa.Integer(), nullable=False),
+        sa.PrimaryKeyConstraint("run_id"),
+    )
+    op.create_table(
+        "runs",
+        sa.Column("run_id", sa.UUID(), nullable=False),
+        sa.Column("thread_id", sa.UUID(), nullable=True),
+        sa.Column("assistant_id", sa.UUID(), nullable=False),
+        sa.Column("status", sa.String(length=32), nullable=False),
+        sa.Column(
+            "metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column(
+            "kwargs",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column("multitask_strategy", sa.String(length=32), nullable=True),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.PrimaryKeyConstraint("run_id"),
+    )
+    op.create_index("ix_runs_assistant_id_status", "runs", ["assistant_id", "status"], unique=False)
+    op.create_index("ix_runs_status_created_at", "runs", ["status", "created_at"], unique=False)
+    op.create_index(
+        "ix_runs_thread_id_created_at", "runs", ["thread_id", "created_at"], unique=False
+    )
+    op.create_index("ix_runs_thread_id_status", "runs", ["thread_id", "status"], unique=False)
+    op.create_index(
+        "uq_runs_one_running_per_thread",
+        "runs",
+        ["thread_id"],
+        unique=True,
+        postgresql_where=sa.text("status = 'running' AND thread_id IS NOT NULL"),
+    )
+    op.create_table(
+        "store_items",
+        sa.Column("prefix", sa.Text(), nullable=False),
+        sa.Column("key", sa.Text(), nullable=False),
+        sa.Column(
+            "value",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column("expires_at", sa.DateTime(timezone=True), nullable=True),
+        sa.PrimaryKeyConstraint("prefix", "key"),
+    )
+    op.create_table(
+        "threads",
+        sa.Column("thread_id", sa.UUID(), nullable=False),
+        sa.Column("status", sa.String(length=32), nullable=False),
+        sa.Column(
+            "metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column(
+            "config",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column("values", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
+        sa.Column(
+            "interrupts",
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False,
+        ),
+        sa.Column("error", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column("state_updated_at", sa.DateTime(timezone=True), nullable=True),
+        sa.PrimaryKeyConstraint("thread_id"),
+    )
+    op.create_index(
+        "ix_threads_metadata_gin",
+        "threads",
+        ["metadata"],
+        unique=False,
+        postgresql_using="gin",
+        postgresql_ops={"metadata": "jsonb_path_ops"},
+    )
+    op.create_index(
+        "ix_threads_status_updated_at", "threads", ["status", "updated_at"], unique=False
+    )
+    op.create_index("ix_threads_updated_at", "threads", ["updated_at"], unique=False)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_index("ix_threads_updated_at", table_name="threads")
+    op.drop_index("ix_threads_status_updated_at", table_name="threads")
+    op.drop_index(
+        "ix_threads_metadata_gin",
+        table_name="threads",
+        postgresql_using="gin",
+        postgresql_ops={"metadata": "jsonb_path_ops"},
+    )
+    op.drop_table("threads")
+    op.drop_table("store_items")
+    op.drop_index(
+        "uq_runs_one_running_per_thread",
+        table_name="runs",
+        postgresql_where=sa.text("status = 'running' AND thread_id IS NOT NULL"),
+    )
+    op.drop_index("ix_runs_thread_id_status", table_name="runs")
+    op.drop_index("ix_runs_thread_id_created_at", table_name="runs")
+    op.drop_index("ix_runs_status_created_at", table_name="runs")
+    op.drop_index("ix_runs_assistant_id_status", table_name="runs")
+    op.drop_table("runs")
+    op.drop_table("retry_counters")
+    op.drop_index("ix_crons_thread_id", table_name="crons")
+    op.drop_index(
+        "ix_crons_metadata_gin",
+        table_name="crons",
+        postgresql_using="gin",
+        postgresql_ops={"metadata": "jsonb_path_ops"},
+    )
+    op.drop_index("ix_crons_enabled_next_run", table_name="crons")
+    op.drop_index("ix_crons_assistant_id", table_name="crons")
+    op.drop_table("crons")
+    op.drop_index(
+        "ix_assistants_metadata_gin",
+        table_name="assistants",
+        postgresql_using="gin",
+        postgresql_ops={"metadata": "jsonb_path_ops"},
+    )
+    op.drop_index("ix_assistants_graph_id", table_name="assistants")
+    op.drop_index("ix_assistants_created_at", table_name="assistants")
+    op.drop_table("assistants")
+    op.drop_table("assistant_versions")
+    # ### end Alembic commands ###

langgraph_runtime_pg/migrations/versions/__init__.py

@@ -0,0 +1 @@
+"""Alembic revision modules."""

langgraph_runtime_pg/models.py

@@ -0,0 +1,196 @@
+"""SQLAlchemy ORM models; schema DDL is applied via Alembic."""
+
+from __future__ import annotations
+
+import uuid
+from datetime import datetime
+
+from sqlalchemy import DateTime, Index, Integer, String, Text, text
+from sqlalchemy.dialects.postgresql import JSONB, UUID as PG_UUID
+from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
+
+
+class Base(DeclarativeBase):
+    pass
+
+
+class AssistantRow(Base):
+    __tablename__ = "assistants"
+    __table_args__ = (
+        Index("ix_assistants_graph_id", "graph_id"),
+        Index("ix_assistants_created_at", "created_at"),
+        Index(
+            "ix_assistants_metadata_gin",
+            "metadata",
+            postgresql_using="gin",
+            postgresql_ops={"metadata": "jsonb_path_ops"},
+        ),
+    )
+
+    assistant_id: Mapped[uuid.UUID] = mapped_column(
+        PG_UUID(as_uuid=True), primary_key=True, default=uuid.uuid4
+    )
+    graph_id: Mapped[str] = mapped_column(String(256), nullable=False)
+    name: Mapped[str] = mapped_column(String(512), nullable=False, default="")
+    description: Mapped[str | None] = mapped_column(Text, nullable=True)
+    config: Mapped[dict] = mapped_column(JSONB, nullable=False, server_default=text("'{}'::jsonb"))
+    context: Mapped[dict] = mapped_column(JSONB, nullable=False, server_default=text("'{}'::jsonb"))
+    metadata_: Mapped[dict] = mapped_column(
+        "metadata", JSONB, nullable=False, server_default=text("'{}'::jsonb")
+    )
+    version: Mapped[int] = mapped_column(Integer, nullable=False, default=1)
+    created_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), nullable=False, server_default=text("now()")
+    )
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), nullable=False, server_default=text("now()")
+    )
+
+
+class AssistantVersionRow(Base):
+    __tablename__ = "assistant_versions"
+
+    assistant_id: Mapped[uuid.UUID] = mapped_column(PG_UUID(as_uuid=True), primary_key=True)
+    version: Mapped[int] = mapped_column(Integer, primary_key=True)
+    graph_id: Mapped[str] = mapped_column(String(256), nullable=False)
+    config: Mapped[dict] = mapped_column(JSONB, nullable=False, server_default=text("'{}'::jsonb"))
+    context: Mapped[dict] = mapped_column(JSONB, nullable=False, server_default=text("'{}'::jsonb"))
+    metadata_: Mapped[dict] = mapped_column(
+        "metadata", JSONB, nullable=False, server_default=text("'{}'::jsonb")
+    )
+    name: Mapped[str] = mapped_column(String(512), nullable=False, default="")
+    description: Mapped[str | None] = mapped_column(Text, nullable=True)
+    created_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), nullable=False, server_default=text("now()")
+    )
+
+
+class ThreadRow(Base):
+    __tablename__ = "threads"
+    __table_args__ = (
+        Index("ix_threads_status_updated_at", "status", "updated_at"),
+        Index("ix_threads_updated_at", "updated_at"),
+        Index(
+            "ix_threads_metadata_gin",
+            "metadata",
+            postgresql_using="gin",
+            postgresql_ops={"metadata": "jsonb_path_ops"},
+        ),
+    )
+
+    thread_id: Mapped[uuid.UUID] = mapped_column(
+        PG_UUID(as_uuid=True), primary_key=True, default=uuid.uuid4
+    )
+    status: Mapped[str] = mapped_column(String(32), nullable=False, default="idle")
+    metadata_: Mapped[dict] = mapped_column(
+        "metadata", JSONB, nullable=False, server_default=text("'{}'::jsonb")
+    )
+    config: Mapped[dict] = mapped_column(JSONB, nullable=False, server_default=text("'{}'::jsonb"))
+    values_: Mapped[dict | None] = mapped_column("values", JSONB, nullable=True)
+    interrupts: Mapped[dict] = mapped_column(
+        JSONB, nullable=False, server_default=text("'{}'::jsonb")
+    )
+    error: Mapped[dict | None] = mapped_column(JSONB, nullable=True)
+    created_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), nullable=False, server_default=text("now()")
+    )
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), nullable=False, server_default=text("now()")
+    )
+    state_updated_at: Mapped[datetime | None] = mapped_column(
+        DateTime(timezone=True), nullable=True
+    )
+
+
+class RunRow(Base):
+    __tablename__ = "runs"
+    __table_args__ = (
+        Index("ix_runs_status_created_at", "status", "created_at"),
+        Index("ix_runs_thread_id_status", "thread_id", "status"),
+        Index("ix_runs_thread_id_created_at", "thread_id", "created_at"),
+        Index("ix_runs_assistant_id_status", "assistant_id", "status"),
+        Index(
+            "uq_runs_one_running_per_thread",
+            "thread_id",
+            unique=True,
+            postgresql_where=text("status = 'running' AND thread_id IS NOT NULL"),
+        ),
+    )
+
+    run_id: Mapped[uuid.UUID] = mapped_column(
+        PG_UUID(as_uuid=True), primary_key=True, default=uuid.uuid4
+    )
+    thread_id: Mapped[uuid.UUID | None] = mapped_column(PG_UUID(as_uuid=True), nullable=True)
+    assistant_id: Mapped[uuid.UUID] = mapped_column(PG_UUID(as_uuid=True), nullable=False)
+    status: Mapped[str] = mapped_column(String(32), nullable=False, default="pending")
+    metadata_: Mapped[dict] = mapped_column(
+        "metadata", JSONB, nullable=False, server_default=text("'{}'::jsonb")
+    )
+    kwargs: Mapped[dict] = mapped_column(JSONB, nullable=False, server_default=text("'{}'::jsonb"))
+    multitask_strategy: Mapped[str | None] = mapped_column(String(32), nullable=True)
+    created_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), nullable=False, server_default=text("now()")
+    )
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), nullable=False, server_default=text("now()")
+    )
+
+
+class CronRow(Base):
+    __tablename__ = "crons"
+    __table_args__ = (
+        Index("ix_crons_enabled_next_run", "enabled", "next_run_date"),
+        Index("ix_crons_assistant_id", "assistant_id"),
+        Index("ix_crons_thread_id", "thread_id"),
+        Index(
+            "ix_crons_metadata_gin",
+            "metadata",
+            postgresql_using="gin",
+            postgresql_ops={"metadata": "jsonb_path_ops"},
+        ),
+    )
+
+    cron_id: Mapped[uuid.UUID] = mapped_column(
+        PG_UUID(as_uuid=True), primary_key=True, default=uuid.uuid4
+    )
+    assistant_id: Mapped[uuid.UUID] = mapped_column(PG_UUID(as_uuid=True), nullable=False)
+    thread_id: Mapped[uuid.UUID | None] = mapped_column(PG_UUID(as_uuid=True), nullable=True)
+    schedule: Mapped[str] = mapped_column(String(128), nullable=False)
+    payload: Mapped[dict] = mapped_column(JSONB, nullable=False, server_default=text("'{}'::jsonb"))
+    metadata_: Mapped[dict] = mapped_column(
+        "metadata", JSONB, nullable=False, server_default=text("'{}'::jsonb")
+    )
+    next_run_date: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
+    end_time: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
+    user_id: Mapped[str | None] = mapped_column(String(256), nullable=True)
+    timezone: Mapped[str | None] = mapped_column(String(64), nullable=True)
+    on_run_completed: Mapped[str | None] = mapped_column(String(16), nullable=True)
+    enabled: Mapped[bool] = mapped_column(nullable=False, default=True, server_default=text("true"))
+    created_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), nullable=False, server_default=text("now()")
+    )
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), nullable=False, server_default=text("now()")
+    )
+
+
+class StoreItemRow(Base):
+    __tablename__ = "store_items"
+
+    prefix: Mapped[str] = mapped_column(Text, primary_key=True)
+    key: Mapped[str] = mapped_column(Text, primary_key=True)
+    value: Mapped[dict] = mapped_column(JSONB, nullable=False, server_default=text("'{}'::jsonb"))
+    created_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), nullable=False, server_default=text("now()")
+    )
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), nullable=False, server_default=text("now()")
+    )
+    expires_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
+
+
+class RetryCounterRow(Base):
+    __tablename__ = "retry_counters"
+
+    run_id: Mapped[uuid.UUID] = mapped_column(PG_UUID(as_uuid=True), primary_key=True)
+    count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)