dbos 1.14.0a9__py3-none-any.whl → 1.15.0a2__py3-none-any.whl

dbos/_alembic_migrations/versions/50f3227f0b4b_fix_job_queue.py

@@ -1,35 +0,0 @@
-"""
-Fix job queue PK.
-
-Revision ID: 50f3227f0b4b
-Revises: eab0cc1d9a14
-Create Date: 2024-09-25 14:03:53.308068
-
-"""
-
-from typing import Sequence, Union
-
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "50f3227f0b4b"
-down_revision: Union[str, None] = "eab0cc1d9a14"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.drop_constraint("job_queue_pkey", "job_queue", schema="dbos", type_="primary")
-
-    op.create_primary_key(
-        "job_queue_pkey", "job_queue", ["workflow_uuid"], schema="dbos"
-    )
-
-
-def downgrade() -> None:
-    # Reverting the changes
-    op.drop_constraint("job_queue_pkey", "job_queue", schema="dbos", type_="primary")
-
-    op.create_primary_key(
-        "job_queue_pkey", "job_queue", ["created_at_epoch_ms"], schema="dbos"
-    )

dbos/_alembic_migrations/versions/5c361fc04708_added_system_tables.py

@@ -1,193 +0,0 @@
-"""
-Add system tables.
-
-Revision ID: 5c361fc04708
-Revises:
-Create Date: 2024-07-21 13:06:13.724602
-# mypy: allow-untyped-defs, allow-untyped-calls
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "5c361fc04708"
-down_revision: Union[str, None] = None
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.execute(sa.schema.CreateSchema(name="dbos", if_not_exists=True))
-    op.execute('CREATE EXTENSION IF NOT EXISTS "uuid-ossp";')
-
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.create_table(
-        "scheduler_state",
-        sa.Column("workflow_fn_name", sa.Text(), nullable=False),
-        sa.Column("last_run_time", sa.BigInteger(), nullable=False),
-        sa.PrimaryKeyConstraint("workflow_fn_name"),
-        schema="dbos",
-    )
-    op.create_table(
-        "workflow_status",
-        sa.Column("workflow_uuid", sa.Text(), nullable=False),
-        sa.Column("status", sa.Text(), nullable=True),
-        sa.Column("name", sa.Text(), nullable=True),
-        sa.Column("authenticated_user", sa.Text(), nullable=True),
-        sa.Column("assumed_role", sa.Text(), nullable=True),
-        sa.Column("authenticated_roles", sa.Text(), nullable=True),
-        sa.Column("request", sa.Text(), nullable=True),
-        sa.Column("output", sa.Text(), nullable=True),
-        sa.Column("error", sa.Text(), nullable=True),
-        sa.Column("executor_id", sa.Text(), nullable=True),
-        sa.Column(
-            "created_at",
-            sa.BigInteger(),
-            server_default=sa.text(
-                "(EXTRACT(epoch FROM now()) * 1000::numeric)::bigint"
-            ),
-            nullable=False,
-        ),
-        sa.Column(
-            "updated_at",
-            sa.BigInteger(),
-            server_default=sa.text(
-                "(EXTRACT(epoch FROM now()) * 1000::numeric)::bigint"
-            ),
-            nullable=False,
-        ),
-        sa.Column("application_version", sa.Text(), nullable=True),
-        sa.Column("application_id", sa.Text(), nullable=True),
-        sa.Column(
-            "class_name",
-            sa.String(length=255),
-            server_default=sa.text("NULL"),
-            nullable=True,
-        ),
-        sa.Column(
-            "config_name",
-            sa.String(length=255),
-            server_default=sa.text("NULL"),
-            nullable=True,
-        ),
-        sa.Column(
-            "recovery_attempts",
-            sa.BigInteger(),
-            server_default=sa.text("'0'::bigint"),
-            nullable=True,
-        ),
-        sa.PrimaryKeyConstraint("workflow_uuid"),
-        schema="dbos",
-    )
-    op.create_index(
-        "workflow_status_created_at_index",
-        "workflow_status",
-        ["created_at"],
-        unique=False,
-        schema="dbos",
-    )
-    op.create_index(
-        "workflow_status_executor_id_index",
-        "workflow_status",
-        ["executor_id"],
-        unique=False,
-        schema="dbos",
-    )
-    op.create_table(
-        "notifications",
-        sa.Column("destination_uuid", sa.Text(), nullable=False),
-        sa.Column("topic", sa.Text(), nullable=True),
-        sa.Column("message", sa.Text(), nullable=False),
-        sa.Column(
-            "created_at_epoch_ms",
-            sa.BigInteger(),
-            server_default=sa.text(
-                "(EXTRACT(epoch FROM now()) * 1000::numeric)::bigint"
-            ),
-            nullable=False,
-        ),
-        sa.Column(
-            "message_uuid",
-            sa.Text(),
-            server_default=sa.text("uuid_generate_v4()"),
-            nullable=False,
-        ),
-        sa.ForeignKeyConstraint(
-            ["destination_uuid"],
-            ["dbos.workflow_status.workflow_uuid"],
-            onupdate="CASCADE",
-            ondelete="CASCADE",
-        ),
-        schema="dbos",
-    )
-    op.create_index(
-        "idx_workflow_topic",
-        "notifications",
-        ["destination_uuid", "topic"],
-        unique=False,
-        schema="dbos",
-    )
-    op.create_table(
-        "operation_outputs",
-        sa.Column("workflow_uuid", sa.Text(), nullable=False),
-        sa.Column("function_id", sa.Integer(), nullable=False),
-        sa.Column("output", sa.Text(), nullable=True),
-        sa.Column("error", sa.Text(), nullable=True),
-        sa.ForeignKeyConstraint(
-            ["workflow_uuid"],
-            ["dbos.workflow_status.workflow_uuid"],
-            onupdate="CASCADE",
-            ondelete="CASCADE",
-        ),
-        sa.PrimaryKeyConstraint("workflow_uuid", "function_id"),
-        schema="dbos",
-    )
-    op.create_table(
-        "workflow_events",
-        sa.Column("workflow_uuid", sa.Text(), nullable=False),
-        sa.Column("key", sa.Text(), nullable=False),
-        sa.Column("value", sa.Text(), nullable=False),
-        sa.ForeignKeyConstraint(
-            ["workflow_uuid"],
-            ["dbos.workflow_status.workflow_uuid"],
-            onupdate="CASCADE",
-            ondelete="CASCADE",
-        ),
-        sa.PrimaryKeyConstraint("workflow_uuid", "key"),
-        schema="dbos",
-    )
-    op.create_table(
-        "workflow_inputs",
-        sa.Column("workflow_uuid", sa.Text(), nullable=False),
-        sa.Column("inputs", sa.Text(), nullable=False),
-        sa.ForeignKeyConstraint(
-            ["workflow_uuid"],
-            ["dbos.workflow_status.workflow_uuid"],
-            onupdate="CASCADE",
-            ondelete="CASCADE",
-        ),
-        sa.PrimaryKeyConstraint("workflow_uuid"),
-        schema="dbos",
-    )
-    # ### end Alembic commands ###
-
-
-def downgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.drop_table("workflow_inputs", schema="dbos")
-    op.drop_table("workflow_events", schema="dbos")
-    op.drop_table("operation_outputs", schema="dbos")
-    op.drop_index("idx_workflow_topic", table_name="notifications", schema="dbos")
-    op.drop_table("notifications", schema="dbos")
-    op.drop_index(
-        "workflow_status_executor_id_index", table_name="workflow_status", schema="dbos"
-    )
-    op.drop_index(
-        "workflow_status_created_at_index", table_name="workflow_status", schema="dbos"
-    )
-    op.drop_table("workflow_status", schema="dbos")
-    op.drop_table("scheduler_state", schema="dbos")
-    # ### end Alembic commands ###

dbos/_alembic_migrations/versions/66478e1b95e5_consolidate_queues.py

@@ -1,71 +0,0 @@
-"""consolidate_queues
-
-Revision ID: 66478e1b95e5
-Revises: 933e86bdac6a
-Create Date: 2025-05-21 10:14:25.674613
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "66478e1b95e5"
-down_revision: Union[str, None] = "933e86bdac6a"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    # Add new columns to workflow_status table
-    op.add_column(
-        "workflow_status",
-        sa.Column("started_at_epoch_ms", sa.BigInteger(), nullable=True),
-        schema="dbos",
-    )
-
-    op.add_column(
-        "workflow_status",
-        sa.Column("deduplication_id", sa.Text(), nullable=True),
-        schema="dbos",
-    )
-
-    op.add_column(
-        "workflow_status",
-        sa.Column(
-            "priority", sa.Integer(), nullable=False, server_default=sa.text("'0'::int")
-        ),
-        schema="dbos",
-    )
-
-    # Add unique constraint for deduplication_id
-    op.create_unique_constraint(
-        "uq_workflow_status_queue_name_dedup_id",
-        "workflow_status",
-        ["queue_name", "deduplication_id"],
-        schema="dbos",
-    )
-
-    # Add index on status field
-    op.create_index(
-        "workflow_status_status_index", "workflow_status", ["status"], schema="dbos"
-    )
-
-
-def downgrade() -> None:
-    # Drop indexes
-    op.drop_index(
-        "workflow_status_status_index", table_name="workflow_status", schema="dbos"
-    )
-
-    # Drop unique constraint
-    op.drop_constraint(
-        "uq_workflow_status_queue_name_dedup_id", "workflow_status", schema="dbos"
-    )
-
-    # Drop columns
-    op.drop_column("workflow_status", "priority", schema="dbos")
-    op.drop_column("workflow_status", "deduplication_id", schema="dbos")
-    op.drop_column("workflow_status", "started_at_epoch_ms", schema="dbos")

dbos/_alembic_migrations/versions/83f3732ae8e7_workflow_timeout.py

@@ -1,44 +0,0 @@
-"""workflow_timeout
-
-Revision ID: 83f3732ae8e7
-Revises: f4b9b32ba814
-Create Date: 2025-04-16 17:05:36.642395
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "83f3732ae8e7"
-down_revision: Union[str, None] = "f4b9b32ba814"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "workflow_status",
-        sa.Column(
-            "workflow_timeout_ms",
-            sa.BigInteger(),
-            nullable=True,
-        ),
-        schema="dbos",
-    )
-    op.add_column(
-        "workflow_status",
-        sa.Column(
-            "workflow_deadline_epoch_ms",
-            sa.BigInteger(),
-            nullable=True,
-        ),
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("workflow_status", "workflow_deadline_epoch_ms", schema="dbos")
-    op.drop_column("workflow_status", "workflow_timeout_ms", schema="dbos")

dbos/_alembic_migrations/versions/933e86bdac6a_add_queue_priority.py

@@ -1,35 +0,0 @@
-"""add queue priority
-
-Revision ID: 933e86bdac6a
-Revises: 27ac6900c6ad
-Create Date: 2025-04-25 18:17:40.462737
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "933e86bdac6a"
-down_revision: Union[str, None] = "27ac6900c6ad"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "workflow_queue",
-        sa.Column(
-            "priority",
-            sa.Integer(),
-            nullable=False,
-            server_default=sa.text("'0'::int"),
-        ),
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("workflow_queue", "priority", schema="dbos")

dbos/_alembic_migrations/versions/a3b18ad34abe_added_triggers.py

@@ -1,72 +0,0 @@
-"""
-Add triggers.
-
-Revision ID: a3b18ad34abe
-Revises: 5c361fc04708
-Create Date: 2024-07-21 13:07:09.814989
-"""
-
-from typing import Sequence, Union
-
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "a3b18ad34abe"
-down_revision: Union[str, None] = "5c361fc04708"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-
-    op.execute(
-        """
-      CREATE OR REPLACE FUNCTION dbos.notifications_function() RETURNS TRIGGER AS $$
-      DECLARE
-          payload text := NEW.destination_uuid || '::' || NEW.topic;
-      BEGIN
-          PERFORM pg_notify('dbos_notifications_channel', payload);
-          RETURN NEW;
-      END;
-      $$ LANGUAGE plpgsql;
-               """
-    )
-    op.execute(
-        """
-      CREATE TRIGGER dbos_notifications_trigger
-      AFTER INSERT ON dbos.notifications
-      FOR EACH ROW EXECUTE FUNCTION dbos.notifications_function();
-               """
-    )
-    op.execute(
-        """
-      CREATE OR REPLACE FUNCTION dbos.workflow_events_function() RETURNS TRIGGER AS $$
-      DECLARE
-          payload text := NEW.workflow_uuid || '::' || NEW.key;
-      BEGIN
-          PERFORM pg_notify('dbos_workflow_events_channel', payload);
-          RETURN NEW;
-      END;
-      $$ LANGUAGE plpgsql;
-               """
-    )
-    op.execute(
-        """
-    CREATE TRIGGER dbos_workflow_events_trigger
-      AFTER INSERT ON dbos.workflow_events
-      FOR EACH ROW EXECUTE FUNCTION dbos.workflow_events_function();
-               """
-    )
-    # ### end Alembic commands ###
-
-
-def downgrade() -> None:
-    op.execute(
-        "DROP TRIGGER IF EXISTS dbos_notifications_trigger ON dbos.notifications;"
-    )
-    op.execute("DROP FUNCTION IF EXISTS dbos.notifications_function;")
-    op.execute(
-        "DROP TRIGGER IF EXISTS dbos_workflow_events_trigger ON dbos.workflow_events;"
-    )
-    op.execute("DROP FUNCTION IF EXISTS dbos.workflow_events_function;")
-    # ### end Alembic commands ###

dbos/_alembic_migrations/versions/d76646551a6b_job_queue_limiter.py

@@ -1,43 +0,0 @@
-"""
-Adjust workflow queue to add columns for rate limiter.
-
-Revision ID: d76646551a6b
-Revises: 50f3227f0b4b
-Create Date: 2024-09-25 14:48:10.218015
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "d76646551a6b"
-down_revision: Union[str, None] = "50f3227f0b4b"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "job_queue",
-        sa.Column(
-            "started_at_epoch_ms",
-            sa.BigInteger(),
-        ),
-        schema="dbos",
-    )
-    op.add_column(
-        "job_queue",
-        sa.Column(
-            "completed_at_epoch_ms",
-            sa.BigInteger(),
-        ),
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("job_queue", "started_at_epoch_ms", schema="dbos")
-    op.drop_column("job_queue", "completed_at_epoch_ms", schema="dbos")

dbos/_alembic_migrations/versions/d76646551a6c_workflow_queue.py

@@ -1,28 +0,0 @@
-"""workflow_queue
-
-Revision ID: d76646551a6c
-Revises: d76646551a6b
-Create Date: 2024-09-27 12:00:00.0
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "d76646551a6c"
-down_revision: Union[str, None] = "d76646551a6b"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.rename_table("job_queue", "workflow_queue", schema="dbos")
-    op.execute("CREATE VIEW dbos.job_queue AS SELECT * FROM dbos.workflow_queue;")
-
-
-def downgrade() -> None:
-    op.execute("DROP VIEW dbos.job_queue;")
-    op.rename_table("workflow_queue", "job_queue", schema="dbos")

dbos/_alembic_migrations/versions/d994145b47b6_consolidate_inputs.py

@@ -1,30 +0,0 @@
-"""consolidate_inputs
-
-Revision ID: d994145b47b6
-Revises: 66478e1b95e5
-Create Date: 2025-05-23 08:09:15.515009
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "d994145b47b6"
-down_revision: Union[str, None] = "66478e1b95e5"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "workflow_status",
-        sa.Column("inputs", sa.Text(), nullable=True),
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("workflow_status", "inputs", schema="dbos")

dbos/_alembic_migrations/versions/eab0cc1d9a14_job_queue.py

@@ -1,56 +0,0 @@
-"""
-Add workflow queue table.
-
-Revision ID: eab0cc1d9a14
-Revises: a3b18ad34abe
-Create Date: 2024-09-13 14:50:00.531294
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "eab0cc1d9a14"
-down_revision: Union[str, None] = "a3b18ad34abe"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.create_table(
-        "job_queue",
-        sa.Column("workflow_uuid", sa.Text(), nullable=False),
-        sa.Column("queue_name", sa.Text(), nullable=False),
-        sa.Column(
-            "created_at_epoch_ms",
-            sa.BigInteger(),
-            server_default=sa.text(
-                "(EXTRACT(epoch FROM now()) * 1000::numeric)::bigint"
-            ),
-            nullable=False,
-            primary_key=True,
-        ),
-        sa.ForeignKeyConstraint(
-            ["workflow_uuid"],
-            ["dbos.workflow_status.workflow_uuid"],
-            onupdate="CASCADE",
-            ondelete="CASCADE",
-        ),
-        schema="dbos",
-    )
-    op.add_column(
-        "workflow_status",
-        sa.Column(
-            "queue_name",
-            sa.Text(),
-        ),
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    op.drop_table("job_queue", schema="dbos")
-    op.drop_column("workflow_status", "queue_name", schema="dbos")

dbos/_alembic_migrations/versions/f4b9b32ba814_functionname_childid_op_outputs.py

@@ -1,46 +0,0 @@
-"""functionname_childid_op_outputs
-
-Revision ID: f4b9b32ba814
-Revises: 04ca4f231047
-Create Date: 2025-03-21 14:32:43.091074
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "f4b9b32ba814"
-down_revision: Union[str, None] = "04ca4f231047"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "operation_outputs",
-        sa.Column(
-            "function_name",
-            sa.Text(),
-            nullable=False,
-            server_default="",
-        ),
-        schema="dbos",
-    )
-
-    op.add_column(
-        "operation_outputs",
-        sa.Column(
-            "child_workflow_id",
-            sa.Text(),
-            nullable=True,
-        ),
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("operation_outputs", "function_name", schema="dbos")
-    op.drop_column("operation_outputs", "child_workflow_id", schema="dbos")