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

dbos/cli/_github_init.py

@@ -1,8 +1,9 @@
+import json
 import os
 from base64 import b64decode
-from typing import List, TypedDict
-
-import requests
+from typing import Any, List, TypedDict
+from urllib.error import HTTPError
+from urllib.request import Request, urlopen
 
 DEMO_REPO_API = "https://api.github.com/repos/dbos-inc/dbos-demo-apps"
 PY_DEMO_PATH = "python/"
@@ -34,43 +35,48 @@ class GitHubItem(TypedDict):
     size: int
 
 
-def _fetch_github(url: str) -> requests.Response:
+def _fetch_github(url: str) -> Any:
     headers = {}
     github_token = os.getenv("GITHUB_TOKEN")
     if github_token:
         headers["Authorization"] = f"Bearer {github_token}"
 
-    response = requests.get(url, headers=headers)
+    request = Request(url, headers=headers)
+
+    try:
+        with urlopen(request) as response:
+            data = response.read()
+            return json.loads(data.decode("utf-8"))
+    except HTTPError as e:
+        # Read response headers
+        rate_limit_remaining = e.headers.get("x-ratelimit-remaining")
 
-    if not response.ok:
-        if response.headers.get("x-ratelimit-remaining") == "0":
+        if rate_limit_remaining == "0":
             raise Exception(
                 "Error fetching from GitHub API: rate limit exceeded.\n"
                 "Please wait a few minutes and try again.\n"
                 "To increase the limit, you can create a personal access token and set it in the GITHUB_TOKEN environment variable.\n"
                 "Details: https://docs.github.com/en/rest/using-the-rest-api/rate-limits-for-the-rest-api"
             )
-        elif response.status_code == 401:
+        elif e.code == 401:
             raise Exception(
-                f"Error fetching content from GitHub {url}: {response.status_code} {response.reason}.\n"
+                f"Error fetching content from GitHub {url}: {e.code} {e.reason}.\n"
                 "Please ensure your GITHUB_TOKEN environment variable is set to a valid personal access token."
             )
         raise Exception(
-            f"Error fetching content from GitHub {url}: {response.status_code} {response.reason}"
+            f"Error fetching content from GitHub {url}: {e.code} {e.reason}"
         )
 
-    return response
-
 
 def _fetch_github_tree(tag: str) -> List[GitHubTreeItem]:
-    response = _fetch_github(f"{DEMO_REPO_API}/git/trees/{tag}?recursive=1")
-    tree_data: GitHubTree = response.json()
+    tree_data: GitHubTree = _fetch_github(
+        f"{DEMO_REPO_API}/git/trees/{tag}?recursive=1"
+    )
     return tree_data["tree"]
 
 
 def _fetch_github_item(url: str) -> str:
-    response = _fetch_github(url)
-    item: GitHubItem = response.json()
+    item: GitHubItem = _fetch_github(url)
     return b64decode(item["content"]).decode("utf-8")
 
 

dbos/cli/_template_init.py

@@ -2,10 +2,6 @@ import os
 import shutil
 import typing
 from os import path
-from typing import Any
-
-import tomlkit
-from rich import print
 
 from dbos._dbos_config import _app_name_to_db_name
 
@@ -46,7 +42,7 @@ def _copy_template_dir(src_dir: str, dst_dir: str, ctx: dict[str, str]) -> None:
 
             dst = path.join(dst_root, base if ext == ".dbos" else file)
             if path.exists(dst):
-                print(f"[yellow]File {dst} already exists, skipping[/yellow]")
+                print(f"File {dst} already exists, skipping")
                 continue
 
             if ext == ".dbos":
@@ -62,7 +58,7 @@ def copy_template(src_dir: str, project_name: str, config_mode: bool) -> None:
     package_name = project_name.replace("-", "_")
     default_migration_section = """database:
   migrate:
-    - alembic upgrade head
+    - python3 migrations/create_table.py
 """
     ctx = {
         "project_name": project_name,
@@ -89,18 +85,11 @@ def copy_template(src_dir: str, project_name: str, config_mode: bool) -> None:
 
 def get_project_name() -> typing.Union[str, None]:
     name = None
+
     try:
-        with open("pyproject.toml", "rb") as file:
-            pyproj = typing.cast(dict[str, Any], tomlkit.load(file))
-            name = typing.cast(str, pyproj["project"]["name"])
+        _, parent = path.split(path.abspath("."))
+        name = parent
     except:
         pass
 
-    if name == None:
-        try:
-            _, parent = path.split(path.abspath("."))
-            name = parent
-        except:
-            pass
-
     return name

dbos/cli/cli.py

@@ -1,3 +1,4 @@
+import json
 import os
 import platform
 import signal
@@ -5,14 +6,10 @@ import subprocess
 import time
 import typing
 from os import path
-from typing import Any, Optional, Tuple
+from typing import Annotated, Any, List, Optional, Tuple
 
-import jsonpickle  # type: ignore
 import sqlalchemy as sa
 import typer
-from rich import print as richprint
-from rich.prompt import IntPrompt
-from typing_extensions import Annotated, List
 
 from dbos._context import SetWorkflowID
 from dbos._debug import debug_workflow, parse_start_command
@@ -34,9 +31,14 @@ from ..cli._github_init import create_template_from_github
 from ._template_init import copy_template, get_project_name, get_templates_directory
 
 
+class DefaultEncoder(json.JSONEncoder):
+    def default(self, obj: Any) -> str:
+        return str(obj)
+
+
 def _get_db_url(
     *, system_database_url: Optional[str], application_database_url: Optional[str]
-) -> Tuple[str, str]:
+) -> Tuple[str, str | None]:
     """
     Get the database URL to use for the DBOS application.
     Order of precedence:
@@ -201,7 +203,7 @@ def init(
                 path.join(templates_dir, template), project_name, config_mode=config
             )
     except Exception as e:
-        richprint(f"[red]{e}[/red]")
+        print(e)
 
 
 def _resolve_project_name_and_template(
@@ -222,27 +224,21 @@ def _resolve_project_name_and_template(
         if template not in templates:
             raise Exception(f"Template {template} not found in {templates_dir}")
     else:
-        richprint("\n[bold]Available templates:[/bold]")
+        print("\nAvailable templates:")
         for idx, template_name in enumerate(templates, 1):
-            richprint(f"  {idx}. {template_name}")
+            print(f"  {idx}. {template_name}")
         while True:
             try:
-                choice = IntPrompt.ask(
-                    "\nSelect template number",
-                    show_choices=False,
-                    show_default=False,
-                )
+                choice = int(input("\nSelect template number: "))
                 if 1 <= choice <= len(templates):
                     template = templates[choice - 1]
                     break
                 else:
-                    richprint(
-                        "[red]Invalid selection. Please choose a number from the list.[/red]"
-                    )
+                    print("Invalid selection. Please choose a number from the list.")
             except (KeyboardInterrupt, EOFError):
                 raise typer.Abort()
             except ValueError:
-                richprint("[red]Please enter a valid number.[/red]")
+                print("Please enter a valid number.")
 
     if template in git_templates:
         if project_name is None:
@@ -298,7 +294,8 @@ def migrate(
     )
 
     typer.echo(f"Starting DBOS migrations")
-    typer.echo(f"Application database: {sa.make_url(application_database_url)}")
+    if application_database_url:
+        typer.echo(f"Application database: {sa.make_url(application_database_url)}")
     typer.echo(f"System database: {sa.make_url(system_database_url)}")
 
     # First, run DBOS migrations on the system database and the application database
@@ -309,9 +306,10 @@ def migrate(
 
     # Next, assign permissions on the DBOS schema to the application role, if any
     if application_role:
-        grant_dbos_schema_permissions(
-            database_url=application_database_url, role_name=application_role
-        )
+        if application_database_url:
+            grant_dbos_schema_permissions(
+                database_url=application_database_url, role_name=application_role
+            )
         grant_dbos_schema_permissions(
             database_url=system_database_url, role_name=application_role
         )
@@ -499,7 +497,7 @@ def list(
         app_version=appversion,
         name=name,
     )
-    print(jsonpickle.encode(workflows, unpicklable=False))
+    print(json.dumps([w.__dict__ for w in workflows], cls=DefaultEncoder))
 
 
 @workflow.command(help="Retrieve the status of a workflow")
@@ -531,7 +529,7 @@ def get(
         system_database_url=system_database_url,
     )
     status = client.retrieve_workflow(workflow_id=workflow_id).get_status()
-    print(jsonpickle.encode(status, unpicklable=False))
+    print(json.dumps(status.__dict__, cls=DefaultEncoder))
 
 
 @workflow.command(help="List the steps of a workflow")
@@ -562,12 +560,8 @@ def steps(
         application_database_url=application_database_url,
         system_database_url=system_database_url,
     )
-    print(
-        jsonpickle.encode(
-            client.list_workflow_steps(workflow_id=workflow_id),
-            unpicklable=False,
-        )
-    )
+    steps = client.list_workflow_steps(workflow_id=workflow_id)
+    print(json.dumps(steps, cls=DefaultEncoder))
 
 
 @workflow.command(
@@ -665,7 +659,7 @@ def restart(
         system_database_url=system_database_url,
     )
     status = client.fork_workflow(workflow_id=workflow_id, start_step=1).get_status()
-    print(jsonpickle.encode(status, unpicklable=False))
+    print(json.dumps(status.__dict__, cls=DefaultEncoder))
 
 
 @workflow.command(
@@ -736,7 +730,7 @@ def fork(
             start_step=step,
             application_version=application_version,
         ).get_status()
-    print(jsonpickle.encode(status, unpicklable=False))
+    print(json.dumps(status.__dict__, cls=DefaultEncoder))
 
 
 @queue.command(name="list", help="List enqueued functions for your application")
@@ -836,7 +830,7 @@ def list_queue(
         status=status,
         name=name,
     )
-    print(jsonpickle.encode(workflows, unpicklable=False))
+    print(json.dumps([w.__dict__ for w in workflows], cls=DefaultEncoder))
 
 
 if __name__ == "__main__":

dbos/cli/migration.py

@@ -1,3 +1,5 @@
+from typing import Optional
+
 import sqlalchemy as sa
 import typer
 
@@ -5,7 +7,9 @@ from dbos._app_db import ApplicationDatabase
 from dbos._sys_db import SystemDatabase
 
 
-def migrate_dbos_databases(app_database_url: str, system_database_url: str) -> None:
+def migrate_dbos_databases(
+    app_database_url: Optional[str], system_database_url: str
+) -> None:
     app_db = None
     sys_db = None
     try:
@@ -17,16 +21,17 @@ def migrate_dbos_databases(app_database_url: str, system_database_url: str) -> N
                 "pool_size": 2,
             },
         )
-        app_db = ApplicationDatabase.create(
-            database_url=app_database_url,
-            engine_kwargs={
-                "pool_timeout": 30,
-                "max_overflow": 0,
-                "pool_size": 2,
-            },
-        )
         sys_db.run_migrations()
-        app_db.run_migrations()
+        if app_database_url:
+            app_db = ApplicationDatabase.create(
+                database_url=app_database_url,
+                engine_kwargs={
+                    "pool_timeout": 30,
+                    "max_overflow": 0,
+                    "pool_size": 2,
+                },
+            )
+            app_db.run_migrations()
     except Exception as e:
         typer.echo(f"DBOS migrations failed: {e}")
         raise typer.Exit(code=1)

{dbos-1.14.0a9.dist-info → dbos-1.15.0a2.dist-info}/METADATA

@@ -1,28 +1,20 @@
 Metadata-Version: 2.1
 Name: dbos
-Version: 1.14.0a9
+Version: 1.15.0a2
 Summary: Ultra-lightweight durable execution in Python
 Author-Email: "DBOS, Inc." <contact@dbos.dev>
 License: MIT
-Requires-Python: >=3.9
+Requires-Python: >=3.10
 Requires-Dist: pyyaml>=6.0.2
-Requires-Dist: jsonschema>=4.23.0
-Requires-Dist: alembic>=1.13.3
-Requires-Dist: typing-extensions>=4.12.2; python_version < "3.10"
-Requires-Dist: typer>=0.12.5
-Requires-Dist: jsonpickle>=3.3.0
-Requires-Dist: opentelemetry-api>=1.27.0
-Requires-Dist: opentelemetry-sdk>=1.27.0
-Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.27.0
 Requires-Dist: python-dateutil>=2.9.0.post0
-Requires-Dist: fastapi[standard]>=0.115.2
-Requires-Dist: tomlkit>=0.13.2
 Requires-Dist: psycopg[binary]>=3.1
-Requires-Dist: docker>=7.1.0
-Requires-Dist: cryptography>=43.0.3
-Requires-Dist: rich>=13.9.4
-Requires-Dist: pyjwt>=2.10.1
 Requires-Dist: websockets>=14.0
+Requires-Dist: typer-slim>=0.17.4
+Requires-Dist: sqlalchemy>=2.0.43
+Provides-Extra: otel
+Requires-Dist: opentelemetry-api>=1.37.0; extra == "otel"
+Requires-Dist: opentelemetry-sdk>=1.37.0; extra == "otel"
+Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.37.0; extra == "otel"
 Description-Content-Type: text/markdown
 
 

dbos-1.15.0a2.dist-info/RECORD

@@ -0,0 +1,59 @@
+dbos-1.15.0a2.dist-info/METADATA,sha256=STEFFiAPM8uZbNbNRPz81v65hcKlFKT1hiL52v7yjo8,13021
+dbos-1.15.0a2.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+dbos-1.15.0a2.dist-info/entry_points.txt,sha256=_QOQ3tVfEjtjBlr1jS4sHqHya9lI2aIEIWkz8dqYp14,58
+dbos-1.15.0a2.dist-info/licenses/LICENSE,sha256=VGZit_a5-kdw9WT6fY5jxAWVwGQzgLFyPWrcVVUhVNU,1067
+dbos/__init__.py,sha256=pT4BuNLDCrIQX27vQG8NlfxX6PZRU7r9miq4thJTszU,982
+dbos/__main__.py,sha256=G7Exn-MhGrVJVDbgNlpzhfh8WMX_72t3_oJaFT9Lmt8,653
+dbos/_admin_server.py,sha256=e8ELhcDWqR3_PNobnNgUvLGh5lzZq0yFSF6dvtzoQRI,16267
+dbos/_app_db.py,sha256=GsV-uYU0QsChWwQDxnrh8_iiZ_zMQB-bsP2jPGIe2aM,16094
+dbos/_classproperty.py,sha256=f0X-_BySzn3yFDRKB2JpCbLYQ9tLwt1XftfshvY7CBs,626
+dbos/_client.py,sha256=zfFQhj_mf4NS85Nlf1xbXwMQVX_mu3UaU7sb8uE5peM,18794
+dbos/_conductor/conductor.py,sha256=3E_hL3c9g9yWqKZkvI6KA0-ZzPMPRo06TOzT1esMiek,24114
+dbos/_conductor/protocol.py,sha256=q3rgLxINFtWFigdOONc-4gX4vn66UmMlJQD6Kj8LnL4,7420
+dbos/_context.py,sha256=cJDxVbswTLXKE5MV4Hmg6gpIX3Dd5mBTG-4lmofWP9E,27668
+dbos/_core.py,sha256=13DNN_fpSIs42NquV80XsHV7yKwY_adKP03h_xhXok4,50493
+dbos/_croniter.py,sha256=XHAyUyibs_59sJQfSNWkP7rqQY6_XrlfuuCxk4jYqek,47559
+dbos/_dbos.py,sha256=ZrNrFUoND2t8UorqPLjeDvfOP73AasOXqpeuSS6Rz7E,57836
+dbos/_dbos_config.py,sha256=sWXd9RuWGmhkd7j2SraxDWQir_-F2p0SIqGO61ILeyk,25391
+dbos/_debouncer.py,sha256=VmGq1_ZIQ79fnH14LEhdoqxKWp6rlEwzsUwumwAMgTQ,15095
+dbos/_debug.py,sha256=0MfgNqutCUhI4PEmmra9x7f3DiFE_0nscfUCHdLimEY,1415
+dbos/_docker_pg_helper.py,sha256=xySum4hTA8TVMBODoG19u4cXQAB1vCock-jwM2pnmSI,7791
+dbos/_error.py,sha256=GwO0Ng4d4iB52brY09-Ss6Cz_V28Xc0D0cRCzZ6XmNM,8688
+dbos/_event_loop.py,sha256=cvaFN9-II3MsHEOq8QoICc_8qSKrjikMlLfuhC3Y8Dk,2923
+dbos/_fastapi.py,sha256=toYYfbe2aui2aHw0021PoXi2dKlI6NzO3M3pHB0dHOk,3421
+dbos/_flask.py,sha256=Npnakt-a3W5OykONFRkDRnumaDhTQmA0NPdUCGRYKXE,1652
+dbos/_kafka.py,sha256=Gm4fHWl7gYb-i5BMvwNwm5Km3z8zQpseqdMgqgFjlGI,4252
+dbos/_kafka_message.py,sha256=NYvOXNG3Qn7bghn1pv3fg4Pbs86ILZGcK4IB-MLUNu0,409
+dbos/_logger.py,sha256=djnCp147QoQ1iG9Bt3Uz8RyGaXGmi6gebccXsrA6Cps,4660
+dbos/_migration.py,sha256=LgxWPtXqRRwjvS5CrSvQ81B_UzLvRNWd4fnQ_Wo-gek,9507
+dbos/_outcome.py,sha256=7HvosMfEHTh1U5P6xok7kFTGLwa2lPaul0YApb3UnN4,8191
+dbos/_queue.py,sha256=0kJTPwXy3nZ4Epzt-lHky9M9S4L31645drPGFR8fIJY,4854
+dbos/_recovery.py,sha256=K-wlFhdf4yGRm6cUzyhcTjQUS0xp2T5rdNMLiiBErYg,2882
+dbos/_registrations.py,sha256=bEOntObnWaBylnebr5ZpcX2hk7OVLDd1z4BvW4_y3zA,7380
+dbos/_roles.py,sha256=kCuhhg8XLtrHCgKgm44I0abIRTGHltf88OwjEKAUggk,2317
+dbos/_scheduler.py,sha256=CWeGVfl9h51VXfxt80y5Da_5pE8SPty_AYkfpJkkMxQ,2117
+dbos/_schemas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dbos/_schemas/application_database.py,sha256=SypAS9l9EsaBHFn9FR8jmnqt01M74d9AF1AMa4m2hhI,1040
+dbos/_schemas/system_database.py,sha256=-dAKk-_Y3vzbpLT4ei-sIrBQgFyQiwPj1enZb1TYc8I,4943
+dbos/_serialization.py,sha256=GLgWLtHpvk7nSHyXukVQLE1ASNA3CJBtfF8w6iflBDw,3590
+dbos/_sys_db.py,sha256=SspVk-wYmE6xZLuyYQUclwh_AMjnkDXcog5g5WmYn7c,83036
+dbos/_sys_db_postgres.py,sha256=CcvxWzoByEvCZ2P_P-KNBRcyJ_8vSpCjtHBRmc7l5hI,7324
+dbos/_sys_db_sqlite.py,sha256=xT9l-czMhLmfuu5UcnBzAyUxSFgzt3XtEWx9t_D8mZs,7361
+dbos/_templates/dbos-db-starter/README.md,sha256=GhxhBj42wjTt1fWEtwNriHbJuKb66Vzu89G4pxNHw2g,930
+dbos/_templates/dbos-db-starter/__package/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dbos/_templates/dbos-db-starter/__package/main.py.dbos,sha256=aQnBPSSQpkB8ERfhf7gB7P9tsU6OPKhZscfeh0yiaD8,2702
+dbos/_templates/dbos-db-starter/__package/schema.py,sha256=7Z27JGC8yy7Z44cbVXIREYxtUhU4JVkLCp5Q7UahVQ0,260
+dbos/_templates/dbos-db-starter/dbos-config.yaml.dbos,sha256=0wPktElM7kMB3OPHTXw4xBk9bgGKMqOHrrr7x_R23Z8,446
+dbos/_templates/dbos-db-starter/migrations/create_table.py.dbos,sha256=pVm2Q0AsxS8pg85llbrXFD6jMccMqGjhGRjTEvS-hXk,942
+dbos/_templates/dbos-db-starter/start_postgres_docker.py,sha256=lQVLlYO5YkhGPEgPqwGc7Y8uDKse9HsWv5fynJEFJHM,1681
+dbos/_tracer.py,sha256=PHbD7iTEkHk7z4B9hc-wPgi2dPTeI1rhZgLI33TQEeM,3786
+dbos/_utils.py,sha256=ZdoM1MDbHnlJrh31zfhp3iX62bAxK1kyvMwXnltC_84,1779
+dbos/_workflow_commands.py,sha256=k-i1bCfNrux43BHLT8wQ-l-MVZX3D6LGZLH7-uuiDRo,4951
+dbos/cli/_github_init.py,sha256=R_94Fnn40CAmPy-zM00lwHi0ndyfv57TmIooADjmag4,3378
+dbos/cli/_template_init.py,sha256=AltKk256VocgvxLpuTxpjJyACrdHFjbGoqYhHzeLae4,2649
+dbos/cli/cli.py,sha256=8fn8hseZWWseJiJMo21_mWYfMqgM2y7l_3UbMP0YNMI,26724
+dbos/cli/migration.py,sha256=vaYxHy0k5KgEuoOQUl6R9oxKv4V5nKKpaVhRbkLDXpo,3440
+dbos/dbos-config.schema.json,sha256=LyUT1DOTaAwOP6suxQGS5KemVIqXGPyu_q7Hbo0neA8,6192
+dbos/py.typed,sha256=QfzXT1Ktfk3Rj84akygc7_42z0lRpCq0Ilh8OXI6Zas,44
+version/__init__.py,sha256=L4sNxecRuqdtSFdpUGX3TtBi9KL3k7YsZVIvv-fv9-A,1678
+dbos-1.15.0a2.dist-info/RECORD,,

dbos/_alembic_migrations/env.py

@@ -1,62 +0,0 @@
-from alembic import context
-from sqlalchemy import engine_from_config, pool
-
-# this is the Alembic Config object, which provides
-# access to the values within the .ini file in use.
-config = context.config
-
-# add your model's MetaData object here
-# for 'autogenerate' support
-target_metadata = None
-
-
-def run_migrations_offline() -> None:
-    """
-    Run migrations in 'offline' mode.
-
-    This configures the context with just a URL
-    and not an Engine, though an Engine is acceptable
-    here as well.  By skipping the Engine creation
-    we don't even need a DBAPI to be available.
-
-    Calls to context.execute() here emit the given string to the
-    script output.
-    """
-
-    url = config.get_main_option("sqlalchemy.url")
-    context.configure(
-        url=url,
-        target_metadata=target_metadata,
-        literal_binds=True,
-        dialect_opts={"paramstyle": "named"},
-    )
-
-    with context.begin_transaction():
-        context.run_migrations()
-
-
-def run_migrations_online() -> None:
-    """
-    Run migrations in 'online' mode.
-
-    In this scenario we need to create an Engine
-    and associate a connection with the context.
-    """
-
-    connectable = engine_from_config(
-        config.get_section(config.config_ini_section, {}),
-        prefix="sqlalchemy.",
-        poolclass=pool.NullPool,
-    )
-
-    with connectable.connect() as connection:
-        context.configure(connection=connection, target_metadata=target_metadata)
-
-        with context.begin_transaction():
-            context.run_migrations()
-
-
-if context.is_offline_mode():
-    run_migrations_offline()
-else:
-    run_migrations_online()

dbos/_alembic_migrations/script.py.mako

@@ -1,26 +0,0 @@
-"""${message}
-
-Revision ID: ${up_revision}
-Revises: ${down_revision | comma,n}
-Create Date: ${create_date}
-
-"""
-from typing import Sequence, Union
-
-from alembic import op
-import sqlalchemy as sa
-${imports if imports else ""}
-
-# revision identifiers, used by Alembic.
-revision: str = ${repr(up_revision)}
-down_revision: Union[str, None] = ${repr(down_revision)}
-branch_labels: Union[str, Sequence[str], None] = ${repr(branch_labels)}
-depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
-
-
-def upgrade() -> None:
-    ${upgrades if upgrades else "pass"}
-
-
-def downgrade() -> None:
-    ${downgrades if downgrades else "pass"}

dbos/_alembic_migrations/versions/01ce9f07bd10_streaming.py

@@ -1,42 +0,0 @@
-"""streaming
-
-Revision ID: 01ce9f07bd10
-Revises: d994145b47b6
-Create Date: 2025-08-05 10:20:46.424975
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "01ce9f07bd10"
-down_revision: Union[str, None] = "d994145b47b6"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    # Create streams table
-    op.create_table(
-        "streams",
-        sa.Column("workflow_uuid", sa.Text(), nullable=False),
-        sa.Column("key", sa.Text(), nullable=False),
-        sa.Column("value", sa.Text(), nullable=False),
-        sa.Column("offset", sa.Integer(), nullable=False),
-        sa.ForeignKeyConstraint(
-            ["workflow_uuid"],
-            ["dbos.workflow_status.workflow_uuid"],
-            onupdate="CASCADE",
-            ondelete="CASCADE",
-        ),
-        sa.PrimaryKeyConstraint("workflow_uuid", "key", "offset"),
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    # Drop streams table
-    op.drop_table("streams", schema="dbos")

dbos/_alembic_migrations/versions/04ca4f231047_workflow_queues_executor_id.py

@@ -1,34 +0,0 @@
-"""workflow_queues_executor_id
-
-Revision ID: 04ca4f231047
-Revises: d76646551a6c
-Create Date: 2025-01-15 15:05:08.043190
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "04ca4f231047"
-down_revision: Union[str, None] = "d76646551a6c"
-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(
-            "executor_id",
-            sa.Text(),
-            nullable=True,
-        ),
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("workflow_queue", "executor_id", schema="dbos")

dbos/_alembic_migrations/versions/27ac6900c6ad_add_queue_dedup.py

@@ -1,45 +0,0 @@
-"""add queue dedup
-
-Revision ID: 27ac6900c6ad
-Revises: 83f3732ae8e7
-Create Date: 2025-04-23 16:18:48.530047
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "27ac6900c6ad"
-down_revision: Union[str, None] = "83f3732ae8e7"
-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(
-            "deduplication_id",
-            sa.Text(),
-            nullable=True,
-        ),
-        schema="dbos",
-    )
-
-    # Unique constraint for queue_name, deduplication_id
-    op.create_unique_constraint(
-        "uq_workflow_queue_name_dedup_id",
-        "workflow_queue",
-        ["queue_name", "deduplication_id"],
-        schema="dbos",
-    )
-
-
-def downgrade() -> None:
-    op.drop_constraint(
-        "uq_workflow_queue_name_dedup_id", "workflow_queue", schema="dbos"
-    )
-    op.drop_column("workflow_queue", "deduplication_id", schema="dbos")

dbos/_alembic_migrations/versions/471b60d64126_dbos_migrations.py

@@ -1,35 +0,0 @@
-"""dbos_migrations
-
-Revision ID: 471b60d64126
-Revises: 01ce9f07bd10
-Create Date: 2025-08-21 14:22:31.455266
-
-"""
-
-from typing import Sequence, Union
-
-import sqlalchemy as sa
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision: str = "471b60d64126"
-down_revision: Union[str, None] = "01ce9f07bd10"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-def upgrade() -> None:
-    # Create dbos_migrations table
-    op.create_table(
-        "dbos_migrations",
-        sa.Column("version", sa.BigInteger(), nullable=False),
-        sa.PrimaryKeyConstraint("version"),
-        schema="dbos",
-    )
-
-    # Insert initial version 1
-    op.execute("INSERT INTO dbos.dbos_migrations (version) VALUES (1)")
-
-
-def downgrade() -> None:
-    op.drop_table("dbos_migrations", schema="dbos")