dbos 0.28.0a4__py3-none-any.whl → 0.28.0a7__py3-none-any.whl

dbos/_admin_server.py

@@ -66,11 +66,11 @@ class AdminRequestHandler(BaseHTTPRequestHandler):
         elif self.path == _deactivate_path:
             if not AdminRequestHandler.is_deactivated:
                 dbos_logger.info(
-                    f"Deactivating DBOS executor {GlobalParams.executor_id} with version {GlobalParams.app_version}. This executor will complete existing workflows but will not start new workflows."
+                    f"Deactivating DBOS executor {GlobalParams.executor_id} with version {GlobalParams.app_version}. This executor will complete existing workflows but will not create new workflows."
                 )
                 AdminRequestHandler.is_deactivated = True
-            # Stop all scheduled workflows, queues, and kafka loops
-            for event in self.dbos.stop_events:
+            # Stop all event receivers (scheduler and Kafka threads)
+            for event in self.dbos.poller_stop_events:
                 event.set()
             self.send_response(200)
             self._end_headers()

dbos/_client.py

@@ -70,7 +70,7 @@ class WorkflowHandleClientPolling(Generic[R]):
         return res
 
     def get_status(self) -> WorkflowStatus:
-        status = get_workflow(self._sys_db, self.workflow_id, False)
+        status = get_workflow(self._sys_db, self.workflow_id)
         if status is None:
             raise DBOSNonExistentWorkflowError(self.workflow_id)
         return status
@@ -92,9 +92,7 @@ class WorkflowHandleClientAsyncPolling(Generic[R]):
         return res
 
     async def get_status(self) -> WorkflowStatus:
-        status = await asyncio.to_thread(
-            get_workflow, self._sys_db, self.workflow_id, False
-        )
+        status = await asyncio.to_thread(get_workflow, self._sys_db, self.workflow_id)
         if status is None:
             raise DBOSNonExistentWorkflowError(self.workflow_id)
         return status
@@ -141,7 +139,6 @@ class DBOSClient:
             "authenticated_user": None,
             "assumed_role": None,
             "authenticated_roles": None,
-            "request": None,
             "output": None,
             "error": None,
             "created_at": None,
@@ -181,13 +178,13 @@ class DBOSClient:
         return WorkflowHandleClientAsyncPolling[R](workflow_id, self._sys_db)
 
     def retrieve_workflow(self, workflow_id: str) -> WorkflowHandle[R]:
-        status = get_workflow(self._sys_db, workflow_id, False)
+        status = get_workflow(self._sys_db, workflow_id)
         if status is None:
             raise DBOSNonExistentWorkflowError(workflow_id)
         return WorkflowHandleClientPolling[R](workflow_id, self._sys_db)
 
     async def retrieve_workflow_async(self, workflow_id: str) -> WorkflowHandleAsync[R]:
-        status = asyncio.to_thread(get_workflow, self._sys_db, workflow_id, False)
+        status = asyncio.to_thread(get_workflow, self._sys_db, workflow_id)
         if status is None:
             raise DBOSNonExistentWorkflowError(workflow_id)
         return WorkflowHandleClientAsyncPolling[R](workflow_id, self._sys_db)
@@ -210,7 +207,6 @@ class DBOSClient:
             "authenticated_user": None,
             "assumed_role": None,
             "authenticated_roles": None,
-            "request": None,
             "output": None,
             "error": None,
             "created_at": None,

dbos/_conductor/conductor.py

@@ -2,6 +2,7 @@ import socket
 import threading
 import time
 import traceback
+from importlib.metadata import version
 from typing import TYPE_CHECKING, Optional
 
 from websockets import ConnectionClosed, ConnectionClosedOK, InvalidStatus
@@ -21,6 +22,9 @@ from . import protocol as p
 if TYPE_CHECKING:
     from dbos import DBOS
 
+ws_version = version("websockets")
+use_keepalive = ws_version < "15.0"
+
 
 class ConductorWebsocket(threading.Thread):
 
@@ -35,14 +39,64 @@ class ConductorWebsocket(threading.Thread):
         self.url = (
             conductor_url.rstrip("/") + f"/websocket/{self.app_name}/{conductor_key}"
         )
+        # TODO: once we can upgrade to websockets>=15.0, we can always use the built-in keepalive
+        self.ping_interval = 20  # Time between pings in seconds
+        self.ping_timeout = 15  # Time to wait for a pong response in seconds
+        self.keepalive_thread: Optional[threading.Thread] = None
+        self.pong_event: Optional[threading.Event] = None
+        self.last_ping_time = -1.0
+
+        self.dbos.logger.debug(
+            f"Connecting to conductor at {self.url} using websockets version {ws_version}"
+        )
+
+    def keepalive(self) -> None:
+        self.dbos.logger.debug("Starting keepalive thread")
+        while not self.evt.is_set():
+            if self.websocket is None or self.websocket.close_code is not None:
+                time.sleep(1)
+                continue
+            try:
+                self.last_ping_time = time.time()
+                self.pong_event = self.websocket.ping()
+                self.dbos.logger.debug("> Sent ping to conductor")
+                pong_result = self.pong_event.wait(self.ping_timeout)
+                elapsed_time = time.time() - self.last_ping_time
+                if not pong_result:
+                    self.dbos.logger.warning(
+                        f"Failed to receive pong from conductor after {elapsed_time:.2f} seconds. Reconnecting."
+                    )
+                    self.websocket.close()
+                    continue
+                # Wait for the next ping interval
+                self.dbos.logger.debug(
+                    f"< Received pong from conductor after {elapsed_time:.2f} seconds"
+                )
+                wait_time = self.ping_interval - elapsed_time
+                self.evt.wait(max(0, wait_time))
+            except ConnectionClosed:
+                # The main loop will try to reconnect
+                self.dbos.logger.debug("Connection to conductor closed.")
+            except Exception as e:
+                self.dbos.logger.warning(f"Failed to send ping to conductor: {e}.")
+                self.websocket.close()
 
     def run(self) -> None:
         while not self.evt.is_set():
             try:
                 with connect(
-                    self.url, open_timeout=5, logger=self.dbos.logger
+                    self.url,
+                    open_timeout=5,
+                    close_timeout=5,
+                    logger=self.dbos.logger,
                 ) as websocket:
                     self.websocket = websocket
+                    if use_keepalive and self.keepalive_thread is None:
+                        self.keepalive_thread = threading.Thread(
+                            target=self.keepalive,
+                            daemon=True,
+                        )
+                        self.keepalive_thread.start()
                     while not self.evt.is_set():
                         message = websocket.recv()
                         if not isinstance(message, str):
@@ -143,7 +197,6 @@ class ConductorWebsocket(threading.Thread):
                                     start_time=body["start_time"],
                                     end_time=body["end_time"],
                                     status=body["status"],
-                                    request=False,
                                     app_version=body["application_version"],
                                     name=body["workflow_name"],
                                     limit=body["limit"],
@@ -176,7 +229,6 @@ class ConductorWebsocket(threading.Thread):
                                     start_time=q_body["start_time"],
                                     end_time=q_body["end_time"],
                                     status=q_body["status"],
-                                    request=False,
                                     name=q_body["workflow_name"],
                                     limit=q_body["limit"],
                                     offset=q_body["offset"],
@@ -206,9 +258,7 @@ class ConductorWebsocket(threading.Thread):
                             info = None
                             try:
                                 info = get_workflow(
-                                    self.dbos._sys_db,
-                                    get_workflow_message.workflow_id,
-                                    get_request=False,
+                                    self.dbos._sys_db, get_workflow_message.workflow_id
                                 )
                             except Exception as e:
                                 error_message = f"Exception encountered when getting workflow {get_workflow_message.workflow_id}: {traceback.format_exc()}"
@@ -287,8 +337,15 @@ class ConductorWebsocket(threading.Thread):
                             # Still need to send a response to the conductor
                             websocket.send(unknown_message.to_json())
             except ConnectionClosedOK:
-                self.dbos.logger.info("Conductor connection terminated")
-                break
+                if self.evt.is_set():
+                    self.dbos.logger.info("Conductor connection terminated")
+                    break
+                # Otherwise, we are trying to reconnect
+                self.dbos.logger.warning(
+                    "Connection to conductor lost. Reconnecting..."
+                )
+                time.sleep(1)
+                continue
             except ConnectionClosed as e:
                 self.dbos.logger.warning(
                     f"Connection to conductor lost. Reconnecting: {e}"
@@ -309,3 +366,9 @@ class ConductorWebsocket(threading.Thread):
                 )
                 time.sleep(1)
                 continue
+
+        # Wait for the keepalive thread to finish
+        if self.keepalive_thread is not None:
+            if self.pong_event is not None:
+                self.pong_event.set()
+            self.keepalive_thread.join()

dbos/_conductor/protocol.py

@@ -149,7 +149,7 @@ class WorkflowsOutput:
         inputs_str = str(info.input) if info.input is not None else None
         outputs_str = str(info.output) if info.output is not None else None
         error_str = str(info.error) if info.error is not None else None
-        request_str = str(info.request) if info.request is not None else None
+        request_str = None
         roles_str = (
             str(info.authenticated_roles)
             if info.authenticated_roles is not None

dbos/_context.py

@@ -16,7 +16,6 @@ from sqlalchemy.orm import Session
 from dbos._utils import GlobalParams
 
 from ._logger import dbos_logger
-from ._request import Request
 from ._tracer import dbos_tracer
 
 
@@ -76,8 +75,6 @@ class DBOSContext:
 
         self.logger = dbos_logger
 
-        self.request: Optional["Request"] = None
-
         self.id_assigned_for_next_workflow: str = ""
         self.is_within_set_workflow_id_block: bool = False
 
@@ -120,7 +117,6 @@ class DBOSContext:
             if self.authenticated_roles is not None
             else None
         )
-        rv.request = self.request
         rv.assumed_role = self.assumed_role
         return rv
 

dbos/_core.py

@@ -266,9 +266,6 @@ def _init_workflow(
         "app_id": ctx.app_id,
         "app_version": GlobalParams.app_version,
         "executor_id": ctx.executor_id,
-        "request": (
-            _serialization.serialize(ctx.request) if ctx.request is not None else None
-        ),
         "recovery_attempts": None,
         "authenticated_user": ctx.authenticated_user,
         "authenticated_roles": (
@@ -296,7 +293,7 @@ def _init_workflow(
 
     if workflow_deadline_epoch_ms is not None:
         evt = threading.Event()
-        dbos.stop_events.append(evt)
+        dbos.background_thread_stop_events.append(evt)
 
         def timeout_func() -> None:
             try:
@@ -443,10 +440,6 @@ def execute_workflow_by_id(dbos: "DBOS", workflow_id: str) -> "WorkflowHandle[An
         )
     with DBOSContextEnsure():
         ctx = assert_current_dbos_context()
-        request = status["request"]
-        ctx.request = (
-            _serialization.deserialize(request) if request is not None else None
-        )
         # If this function belongs to a configured class, add that class instance as its first argument
         if status["config_name"] is not None:
             config_name = status["config_name"]

dbos/_dbos.py

@@ -71,14 +71,11 @@ from ._tracer import DBOSTracer, dbos_tracer
 if TYPE_CHECKING:
     from fastapi import FastAPI
     from ._kafka import _KafkaConsumerWorkflow
-    from ._request import Request
     from flask import Flask
 
 from sqlalchemy import URL
 from sqlalchemy.orm import Session
 
-from ._request import Request
-
 if sys.version_info < (3, 10):
     from typing_extensions import ParamSpec
 else:
@@ -197,7 +194,7 @@ class DBOSRegistry:
         self, evt: threading.Event, func: Callable[..., Any], *args: Any, **kwargs: Any
     ) -> None:
         if self.dbos and self.dbos._launched:
-            self.dbos.stop_events.append(evt)
+            self.dbos.poller_stop_events.append(evt)
             self.dbos._executor.submit(func, *args, **kwargs)
         else:
             self.pollers.append((evt, func, args, kwargs))
@@ -247,7 +244,7 @@ class DBOS:
     2. Starting workflow functions
     3. Retrieving workflow status information
     4. Interacting with workflows via events and messages
-    5. Accessing context, including the current user, request, SQL session, logger, and tracer
+    5. Accessing context, including the current user, SQL session, logger, and tracer
 
     """
 
@@ -330,7 +327,10 @@ class DBOS:
         self._registry: DBOSRegistry = _get_or_create_dbos_registry()
         self._registry.dbos = self
         self._admin_server_field: Optional[AdminServer] = None
-        self.stop_events: List[threading.Event] = []
+        # Stop internal background threads (queue thread, timeout threads, etc.)
+        self.background_thread_stop_events: List[threading.Event] = []
+        # Stop pollers (event receivers) that can create new workflows (scheduler, Kafka)
+        self.poller_stop_events: List[threading.Event] = []
         self.fastapi: Optional["FastAPI"] = fastapi
         self.flask: Optional["Flask"] = flask
         self._executor_field: Optional[ThreadPoolExecutor] = None
@@ -502,7 +502,7 @@ class DBOS:
 
             # Start the queue thread
             evt = threading.Event()
-            self.stop_events.append(evt)
+            self.background_thread_stop_events.append(evt)
             bg_queue_thread = threading.Thread(
                 target=queue_thread, args=(evt, self), daemon=True
             )
@@ -515,7 +515,7 @@ class DBOS:
                     dbos_domain = os.environ.get("DBOS_DOMAIN", "cloud.dbos.dev")
                     self.conductor_url = f"wss://{dbos_domain}/conductor/v1alpha1"
                 evt = threading.Event()
-                self.stop_events.append(evt)
+                self.background_thread_stop_events.append(evt)
                 self.conductor_websocket = ConductorWebsocket(
                     self,
                     conductor_url=self.conductor_url,
@@ -527,7 +527,7 @@ class DBOS:
 
             # Grab any pollers that were deferred and start them
             for evt, func, args, kwargs in self._registry.pollers:
-                self.stop_events.append(evt)
+                self.poller_stop_events.append(evt)
                 poller_thread = threading.Thread(
                     target=func, args=args, kwargs=kwargs, daemon=True
                 )
@@ -583,7 +583,9 @@ class DBOS:
 
     def _destroy(self) -> None:
         self._initialized = False
-        for event in self.stop_events:
+        for event in self.poller_stop_events:
+            event.set()
+        for event in self.background_thread_stop_events:
             event.set()
         self._background_event_loop.stop()
         if self._sys_db_field is not None:
@@ -760,7 +762,7 @@ class DBOS:
         """Return the status of a workflow execution."""
 
         def fn() -> Optional[WorkflowStatus]:
-            return get_workflow(_get_dbos_instance()._sys_db, workflow_id, True)
+            return get_workflow(_get_dbos_instance()._sys_db, workflow_id)
 
         return _get_dbos_instance()._sys_db.call_function_as_step(fn, "DBOS.getStatus")
 
@@ -1156,12 +1158,6 @@ class DBOS:
         assert span
         return span
 
-    @classproperty
-    def request(cls) -> Optional["Request"]:
-        """Return the HTTP `Request`, if any, associated with the current context."""
-        ctx = assert_current_dbos_context()
-        return ctx.request
-
     @classproperty
     def authenticated_user(cls) -> Optional[str]:
         """Return the current authenticated user, if any, associated with the current context."""

dbos/_fastapi.py

@@ -7,15 +7,9 @@ from fastapi.responses import JSONResponse
 from starlette.types import ASGIApp, Receive, Scope, Send
 
 from . import DBOS
-from ._context import (
-    EnterDBOSHandler,
-    OperationType,
-    SetWorkflowID,
-    TracedAttributes,
-    assert_current_dbos_context,
-)
+from ._context import EnterDBOSHandler, OperationType, SetWorkflowID, TracedAttributes
 from ._error import DBOSException
-from ._request import Address, Request, request_id_header
+from ._utils import request_id_header
 
 
 def _get_or_generate_request_id(request: FastAPIRequest) -> str:
@@ -26,19 +20,6 @@ def _get_or_generate_request_id(request: FastAPIRequest) -> str:
         return str(uuid.uuid4())
 
 
-def _make_request(request: FastAPIRequest) -> Request:
-    return Request(
-        headers=request.headers,
-        path_params=request.path_params,
-        query_params=request.query_params,
-        url=str(request.url),
-        base_url=str(request.base_url),
-        client=Address(*request.client) if request.client is not None else None,
-        cookies=request.cookies,
-        method=request.method,
-    )
-
-
 async def _dbos_error_handler(request: FastAPIRequest, gexc: Exception) -> JSONResponse:
     exc: DBOSException = cast(DBOSException, gexc)
     status_code = 500
@@ -96,8 +77,6 @@ def setup_fastapi_middleware(app: FastAPI, dbos: DBOS) -> None:
             "operationType": OperationType.HANDLER.value,
         }
         with EnterDBOSHandler(attributes):
-            ctx = assert_current_dbos_context()
-            ctx.request = _make_request(request)
             workflow_id = request.headers.get("dbos-idempotency-key")
             if workflow_id is not None:
                 # Set the workflow ID for the handler

dbos/_flask.py

@@ -2,17 +2,11 @@ import uuid
 from typing import Any
 from urllib.parse import urlparse
 
-from flask import Flask, request
+from flask import Flask
 from werkzeug.wrappers import Request as WRequest
 
-from ._context import (
-    EnterDBOSHandler,
-    OperationType,
-    SetWorkflowID,
-    TracedAttributes,
-    assert_current_dbos_context,
-)
-from ._request import Address, Request, request_id_header
+from ._context import EnterDBOSHandler, OperationType, SetWorkflowID, TracedAttributes
+from ._utils import request_id_header
 
 
 class FlaskMiddleware:
@@ -32,8 +26,6 @@ class FlaskMiddleware:
             "operationType": OperationType.HANDLER.value,
         }
         with EnterDBOSHandler(attributes):
-            ctx = assert_current_dbos_context()
-            ctx.request = _make_request(request)
             workflow_id = request.headers.get("dbos-idempotency-key")
             if workflow_id is not None:
                 # Set the workflow ID for the handler
@@ -52,31 +44,5 @@ def _get_or_generate_request_id(request: WRequest) -> str:
         return str(uuid.uuid4())
 
 
-def _make_request(request: WRequest) -> Request:
-    parsed_url = urlparse(request.url)
-    base_url = f"{parsed_url.scheme}://{parsed_url.netloc}"
-
-    client = None
-    if request.remote_addr:
-        hostname = request.remote_addr
-        port = request.environ.get("REMOTE_PORT")
-        if port:
-            client = Address(hostname=hostname, port=int(port))
-        else:
-            # If port is not available, use 0 as a placeholder
-            client = Address(hostname=hostname, port=0)
-
-    return Request(
-        headers=dict(request.headers),
-        path_params={},
-        query_params=dict(request.args),
-        url=request.url,
-        base_url=base_url,
-        client=client,
-        cookies=dict(request.cookies),
-        method=request.method,
-    )
-
-
 def setup_flask_middleware(app: Flask) -> None:
     app.wsgi_app = FlaskMiddleware(app.wsgi_app)  # type: ignore

dbos/_recovery.py

@@ -29,7 +29,7 @@ def startup_recovery_thread(
 ) -> None:
     """Attempt to recover local pending workflows on startup using a background thread."""
     stop_event = threading.Event()
-    dbos.stop_events.append(stop_event)
+    dbos.background_thread_stop_events.append(stop_event)
     while not stop_event.is_set() and len(pending_workflows) > 0:
         try:
             for pending_workflow in list(pending_workflows):

dbos/_schemas/system_database.py

@@ -29,7 +29,6 @@ class SystemSchema:
         Column("authenticated_user", Text, nullable=True),
         Column("assumed_role", Text, nullable=True),
         Column("authenticated_roles", Text, nullable=True),
-        Column("request", Text, nullable=True),
         Column("output", Text, nullable=True),
         Column("error", Text, nullable=True),
         Column("executor_id", Text, nullable=True),

dbos/_sys_db.py

@@ -106,8 +106,6 @@ class WorkflowStatus:
     app_id: Optional[str]
     # The number of times this workflow's execution has been attempted
     recovery_attempts: Optional[int]
-    # The HTTP request that triggered the workflow, if known
-    request: Optional[str]
 
 
 class WorkflowStatusInternal(TypedDict):
@@ -120,7 +118,6 @@ class WorkflowStatusInternal(TypedDict):
     assumed_role: Optional[str]
     authenticated_roles: Optional[str]  # JSON list of roles
     output: Optional[str]  # JSON (jsonpickle)
-    request: Optional[str]  # JSON (jsonpickle)
     error: Optional[str]  # JSON (jsonpickle)
     created_at: Optional[int]  # Unix epoch timestamp in ms
     updated_at: Optional[int]  # Unix epoch timestamp in ms
@@ -372,7 +369,6 @@ class SystemDatabase:
                 executor_id=status["executor_id"],
                 application_version=status["app_version"],
                 application_id=status["app_id"],
-                request=status["request"],
                 authenticated_user=status["authenticated_user"],
                 authenticated_roles=status["authenticated_roles"],
                 assumed_role=status["assumed_role"],
@@ -480,7 +476,6 @@ class SystemDatabase:
                 executor_id=status["executor_id"],
                 application_version=status["app_version"],
                 application_id=status["app_id"],
-                request=status["request"],
                 authenticated_user=status["authenticated_user"],
                 authenticated_roles=status["authenticated_roles"],
                 assumed_role=status["assumed_role"],
@@ -628,7 +623,6 @@ class SystemDatabase:
                         else status["app_version"]
                     ),
                     application_id=status["app_id"],
-                    request=status["request"],
                     authenticated_user=status["authenticated_user"],
                     authenticated_roles=status["authenticated_roles"],
                     assumed_role=status["assumed_role"],
@@ -690,7 +684,6 @@ class SystemDatabase:
                 sa.select(
                     SystemSchema.workflow_status.c.status,
                     SystemSchema.workflow_status.c.name,
-                    SystemSchema.workflow_status.c.request,
                     SystemSchema.workflow_status.c.recovery_attempts,
                     SystemSchema.workflow_status.c.config_name,
                     SystemSchema.workflow_status.c.class_name,
@@ -715,21 +708,20 @@ class SystemDatabase:
                 "error": None,
                 "status": row[0],
                 "name": row[1],
-                "request": row[2],
-                "recovery_attempts": row[3],
-                "config_name": row[4],
-                "class_name": row[5],
-                "authenticated_user": row[6],
-                "authenticated_roles": row[7],
-                "assumed_role": row[8],
-                "queue_name": row[9],
-                "executor_id": row[10],
-                "created_at": row[11],
-                "updated_at": row[12],
-                "app_version": row[13],
-                "app_id": row[14],
-                "workflow_deadline_epoch_ms": row[15],
-                "workflow_timeout_ms": row[16],
+                "recovery_attempts": row[2],
+                "config_name": row[3],
+                "class_name": row[4],
+                "authenticated_user": row[5],
+                "authenticated_roles": row[6],
+                "assumed_role": row[7],
+                "queue_name": row[8],
+                "executor_id": row[9],
+                "created_at": row[10],
+                "updated_at": row[11],
+                "app_version": row[12],
+                "app_id": row[13],
+                "workflow_deadline_epoch_ms": row[14],
+                "workflow_timeout_ms": row[15],
             }
             return status
 
@@ -805,9 +797,7 @@ class SystemDatabase:
             )
             return inputs
 
-    def get_workflows(
-        self, input: GetWorkflowsInput, get_request: bool = False
-    ) -> List[WorkflowStatus]:
+    def get_workflows(self, input: GetWorkflowsInput) -> List[WorkflowStatus]:
         """
         Retrieve a list of workflows result and inputs based on the input criteria. The result is a list of external-facing workflow status objects.
         """
@@ -815,7 +805,6 @@ class SystemDatabase:
             SystemSchema.workflow_status.c.workflow_uuid,
             SystemSchema.workflow_status.c.status,
             SystemSchema.workflow_status.c.name,
-            SystemSchema.workflow_status.c.request,
             SystemSchema.workflow_status.c.recovery_attempts,
             SystemSchema.workflow_status.c.config_name,
             SystemSchema.workflow_status.c.class_name,
@@ -888,27 +877,26 @@ class SystemDatabase:
             info.workflow_id = row[0]
             info.status = row[1]
             info.name = row[2]
-            info.request = row[3] if get_request else None
-            info.recovery_attempts = row[4]
-            info.config_name = row[5]
-            info.class_name = row[6]
-            info.authenticated_user = row[7]
+            info.recovery_attempts = row[3]
+            info.config_name = row[4]
+            info.class_name = row[5]
+            info.authenticated_user = row[6]
             info.authenticated_roles = (
-                json.loads(row[8]) if row[8] is not None else None
+                json.loads(row[7]) if row[7] is not None else None
             )
-            info.assumed_role = row[9]
-            info.queue_name = row[10]
-            info.executor_id = row[11]
-            info.created_at = row[12]
-            info.updated_at = row[13]
-            info.app_version = row[14]
-            info.app_id = row[15]
+            info.assumed_role = row[8]
+            info.queue_name = row[9]
+            info.executor_id = row[10]
+            info.created_at = row[11]
+            info.updated_at = row[12]
+            info.app_version = row[13]
+            info.app_id = row[14]
 
             inputs, output, exception = _serialization.safe_deserialize(
                 info.workflow_id,
-                serialized_input=row[16],
-                serialized_output=row[17],
-                serialized_exception=row[18],
+                serialized_input=row[15],
+                serialized_output=row[16],
+                serialized_exception=row[17],
             )
             info.input = inputs
             info.output = output
@@ -918,7 +906,7 @@ class SystemDatabase:
         return infos
 
     def get_queued_workflows(
-        self, input: GetQueuedWorkflowsInput, get_request: bool = False
+        self, input: GetQueuedWorkflowsInput
     ) -> List[WorkflowStatus]:
         """
         Retrieve a list of queued workflows result and inputs based on the input criteria. The result is a list of external-facing workflow status objects.
@@ -927,7 +915,6 @@ class SystemDatabase:
             SystemSchema.workflow_status.c.workflow_uuid,
             SystemSchema.workflow_status.c.status,
             SystemSchema.workflow_status.c.name,
-            SystemSchema.workflow_status.c.request,
             SystemSchema.workflow_status.c.recovery_attempts,
             SystemSchema.workflow_status.c.config_name,
             SystemSchema.workflow_status.c.class_name,
@@ -996,27 +983,26 @@ class SystemDatabase:
             info.workflow_id = row[0]
             info.status = row[1]
             info.name = row[2]
-            info.request = row[3] if get_request else None
-            info.recovery_attempts = row[4]
-            info.config_name = row[5]
-            info.class_name = row[6]
-            info.authenticated_user = row[7]
+            info.recovery_attempts = row[3]
+            info.config_name = row[4]
+            info.class_name = row[5]
+            info.authenticated_user = row[6]
             info.authenticated_roles = (
-                json.loads(row[8]) if row[8] is not None else None
+                json.loads(row[7]) if row[7] is not None else None
             )
-            info.assumed_role = row[9]
-            info.queue_name = row[10]
-            info.executor_id = row[11]
-            info.created_at = row[12]
-            info.updated_at = row[13]
-            info.app_version = row[14]
-            info.app_id = row[15]
+            info.assumed_role = row[8]
+            info.queue_name = row[9]
+            info.executor_id = row[10]
+            info.created_at = row[11]
+            info.updated_at = row[12]
+            info.app_version = row[13]
+            info.app_id = row[14]
 
             inputs, output, exception = _serialization.safe_deserialize(
                 info.workflow_id,
-                serialized_input=row[16],
-                serialized_output=row[17],
-                serialized_exception=row[18],
+                serialized_input=row[15],
+                serialized_output=row[16],
+                serialized_exception=row[17],
             )
             info.input = inputs
             info.output = output

dbos/_utils.py

@@ -3,6 +3,8 @@ import os
 
 INTERNAL_QUEUE_NAME = "_dbos_internal_queue"
 
+request_id_header = "x-request-id"
+
 
 class GlobalParams:
     app_version: str = os.environ.get("DBOS__APPVERSION", "")

dbos/_workflow_commands.py

@@ -27,7 +27,6 @@ def list_workflows(
     limit: Optional[int] = None,
     offset: Optional[int] = None,
     sort_desc: bool = False,
-    request: bool = False,
     workflow_id_prefix: Optional[str] = None,
 ) -> List[WorkflowStatus]:
     input = GetWorkflowsInput()
@@ -43,7 +42,7 @@ def list_workflows(
     input.sort_desc = sort_desc
     input.workflow_id_prefix = workflow_id_prefix
 
-    infos: List[WorkflowStatus] = sys_db.get_workflows(input, request)
+    infos: List[WorkflowStatus] = sys_db.get_workflows(input)
 
     return infos
 
@@ -59,7 +58,6 @@ def list_queued_workflows(
     limit: Optional[int] = None,
     offset: Optional[int] = None,
     sort_desc: bool = False,
-    request: bool = False,
 ) -> List[WorkflowStatus]:
     input: GetQueuedWorkflowsInput = {
         "queue_name": queue_name,
@@ -72,17 +70,15 @@ def list_queued_workflows(
         "sort_desc": sort_desc,
     }
 
-    infos: List[WorkflowStatus] = sys_db.get_queued_workflows(input, request)
+    infos: List[WorkflowStatus] = sys_db.get_queued_workflows(input)
     return infos
 
 
-def get_workflow(
-    sys_db: SystemDatabase, workflow_id: str, get_request: bool
-) -> Optional[WorkflowStatus]:
+def get_workflow(sys_db: SystemDatabase, workflow_id: str) -> Optional[WorkflowStatus]:
     input = GetWorkflowsInput()
     input.workflow_ids = [workflow_id]
 
-    infos: List[WorkflowStatus] = sys_db.get_workflows(input, get_request)
+    infos: List[WorkflowStatus] = sys_db.get_workflows(input)
     if not infos:
         return None
 

{dbos-0.28.0a4.dist-info → dbos-0.28.0a7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbos
-Version: 0.28.0a4
+Version: 0.28.0a7
 Summary: Ultra-lightweight durable execution in Python
 Author-Email: "DBOS, Inc." <contact@dbos.dev>
 License: MIT
@@ -22,7 +22,7 @@ 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>=15.0
+Requires-Dist: websockets>=14.0
 Description-Content-Type: text/markdown
 
 

{dbos-0.28.0a4.dist-info → dbos-0.28.0a7.dist-info}/RECORD

@@ -1,26 +1,26 @@
-dbos-0.28.0a4.dist-info/METADATA,sha256=KyMx9FTut8a9dhtuvbD5FingAQayckrfPUj4wjrGCZQ,13268
-dbos-0.28.0a4.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
-dbos-0.28.0a4.dist-info/entry_points.txt,sha256=_QOQ3tVfEjtjBlr1jS4sHqHya9lI2aIEIWkz8dqYp14,58
-dbos-0.28.0a4.dist-info/licenses/LICENSE,sha256=VGZit_a5-kdw9WT6fY5jxAWVwGQzgLFyPWrcVVUhVNU,1067
+dbos-0.28.0a7.dist-info/METADATA,sha256=qUA4yCy56WYU15e1V8V2Pb_xNM9ccKpov9vvnfMGYYM,13268
+dbos-0.28.0a7.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
+dbos-0.28.0a7.dist-info/entry_points.txt,sha256=_QOQ3tVfEjtjBlr1jS4sHqHya9lI2aIEIWkz8dqYp14,58
+dbos-0.28.0a7.dist-info/licenses/LICENSE,sha256=VGZit_a5-kdw9WT6fY5jxAWVwGQzgLFyPWrcVVUhVNU,1067
 dbos/__init__.py,sha256=-FdBlOlr-f2tY__C23J4v22MoCAXqcDN_-zXsJXdoZ0,1005
 dbos/__main__.py,sha256=G7Exn-MhGrVJVDbgNlpzhfh8WMX_72t3_oJaFT9Lmt8,653
-dbos/_admin_server.py,sha256=NG0JWQQer9kEslPNAA0dBv-O262sjarz7ZSlv8yird0,9053
+dbos/_admin_server.py,sha256=CM02jyC9H21fM7Pjn1BhPxNwAOV7CXmMJd0SdaNq8dQ,9062
 dbos/_app_db.py,sha256=3j8_5-MlSDY0otLRszFE-GfenU6JC20fcfSL-drSNYk,11800
 dbos/_classproperty.py,sha256=f0X-_BySzn3yFDRKB2JpCbLYQ9tLwt1XftfshvY7CBs,626
-dbos/_client.py,sha256=Cn-feSnOcuDNnBfP0UBvb3xUWSmaLuoSc8CKQAoDwP8,13931
-dbos/_conductor/conductor.py,sha256=HYzVL29IMMrs2Mnms_7cHJynCnmmEN5SDQOMjzn3UoU,16840
-dbos/_conductor/protocol.py,sha256=zEKIuOQdIaSduNqfZKpo8PSD9_1oNpKIPnBNCu3RUyE,6681
-dbos/_context.py,sha256=fX_lWKx2_bHNTigVkuDtjDfX7BIeCsdDAQ1u47A2_Fo,24590
-dbos/_core.py,sha256=1hDRqEmm77GUJVY1xCEzYsUP_HzLGtflI_v3TCxkQuI,48569
+dbos/_client.py,sha256=mGrricoU6437QM4SWV-6Vm806AgVru8ygKgGgK1LZGA,13823
+dbos/_conductor/conductor.py,sha256=L8va6dGae0BlH43AHWAAW8_z04QUYPWPhjUK64sn3P4,19692
+dbos/_conductor/protocol.py,sha256=jwX8ZjmAIlXu1vw9R3b5PfHSNdwofeYOKj8rkfAFVg0,6630
+dbos/_context.py,sha256=Ly1CXF1nWxICQgIpDZSaONGlz1yERBs63gqmR-yqCzM,24476
+dbos/_core.py,sha256=UDpSgRA9m_YuViNXR9tVgNFLC-zxKZPxjlkj2a-Kj00,48317
 dbos/_croniter.py,sha256=XHAyUyibs_59sJQfSNWkP7rqQY6_XrlfuuCxk4jYqek,47559
-dbos/_dbos.py,sha256=f0SnnWL-koLgohj8IaV6l5-tKDtnKpJgAYyel-SwYaE,48346
+dbos/_dbos.py,sha256=nne4oaIEC0tR2OV766y7mjTLjGaqKz7EnuY-T2CbTtc,48431
 dbos/_dbos_config.py,sha256=L0Z0OOB5FoPM9g-joZqXGeJnlxWQsEUtgPtgtg9Uf48,21732
 dbos/_debug.py,sha256=MNlQVZ6TscGCRQeEEL0VE8Uignvr6dPeDDDefS3xgIE,1823
 dbos/_docker_pg_helper.py,sha256=NmcgqmR5rQA_4igfeqh8ugNT2z3YmoOvuep_MEtxTiY,5854
 dbos/_error.py,sha256=EN4eVBjMT3k7O7hfqJl6mIf4sxWPsiAOM086yhcGH_g,8012
 dbos/_event_loop.py,sha256=NmaLbEQFfEK36S_0KhVD39YdYrGce3qSKCTJ-5RqKQ0,2136
-dbos/_fastapi.py,sha256=PhaKftbApHnjtYEOw0EYna_3K0cmz__J9of7mRJWzu4,3704
-dbos/_flask.py,sha256=DZKUZR5-xOzPI7tYZ53r2PvvHVoAb8SYwLzMVFsVfjI,2608
+dbos/_fastapi.py,sha256=m4SL3H9P-NBQ_ZrbFxAWMOqNyIi3HGEn2ODR7xAK038,3118
+dbos/_flask.py,sha256=Npnakt-a3W5OykONFRkDRnumaDhTQmA0NPdUCGRYKXE,1652
 dbos/_kafka.py,sha256=pz0xZ9F3X9Ky1k-VSbeF3tfPhP3UPr3lUUhUfE41__U,4198
 dbos/_kafka_message.py,sha256=NYvOXNG3Qn7bghn1pv3fg4Pbs86ILZGcK4IB-MLUNu0,409
 dbos/_logger.py,sha256=qv2srteCF2rSRjCK1VGOck3ieIkwUe9Lvbv60mJc16E,4069
@@ -39,16 +39,15 @@ dbos/_migrations/versions/eab0cc1d9a14_job_queue.py,sha256=uvhFOtqbBreCePhAxZfIT
 dbos/_migrations/versions/f4b9b32ba814_functionname_childid_op_outputs.py,sha256=m90Lc5YH0ZISSq1MyxND6oq3RZrZKrIqEsZtwJ1jWxA,1049
 dbos/_outcome.py,sha256=EXxBg4jXCVJsByDQ1VOCIedmbeq_03S6d-p1vqQrLFU,6810
 dbos/_queue.py,sha256=aKCGahWBGJOLOv5PCOOId96Va3YQ4ICuHWXy-eQXohE,3526
-dbos/_recovery.py,sha256=98Py7icfytyIELJ54gIsdvmURBvTb0HmWaxEAuYL0dc,2546
+dbos/_recovery.py,sha256=jVMexjfCCNopzyn8gVQzJCmGJaP9G3C1EFaoCQ_Nh7g,2564
 dbos/_registrations.py,sha256=EZzG3ZfYmWA2bHX2hpnSIQ3PTi3-cXsvbcmXjyOusMk,7302
-dbos/_request.py,sha256=cX1B3Atlh160phgS35gF1VEEV4pD126c9F3BDgBmxZU,929
 dbos/_roles.py,sha256=iOsgmIAf1XVzxs3gYWdGRe1B880YfOw5fpU7Jwx8_A8,2271
 dbos/_scheduler.py,sha256=SR1oRZRcVzYsj-JauV2LA8JtwTkt8mru7qf6H1AzQ1U,2027
 dbos/_schemas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dbos/_schemas/application_database.py,sha256=SypAS9l9EsaBHFn9FR8jmnqt01M74d9AF1AMa4m2hhI,1040
-dbos/_schemas/system_database.py,sha256=VVjMUfxPwFMdZPBazNkO00ufG6SgL9OCybLcwC9CiVk,5883
+dbos/_schemas/system_database.py,sha256=3Z0L72bOgHnusK1hBaETWU9RfiLBP0QnS-fdu41i0yY,5835
 dbos/_serialization.py,sha256=bWuwhXSQcGmiazvhJHA5gwhrRWxtmFmcCFQSDJnqqkU,3666
-dbos/_sys_db.py,sha256=r9a3RA9hUogRJ9v3gNol_CjTFKjRhIO9aFaXXLc7G_Y,82820
+dbos/_sys_db.py,sha256=GjRTrN64jUMNUnEhLSkqGMiwJWOZhB9WFeuihJs35aM,82163
 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,sha256=nJMN3ZD2lmwg4Dcgmiwqc-tQGuCJuJal2Xl85iA277U,2453
@@ -60,12 +59,12 @@ dbos/_templates/dbos-db-starter/migrations/script.py.mako,sha256=MEqL-2qATlST9TA
 dbos/_templates/dbos-db-starter/migrations/versions/2024_07_31_180642_init.py,sha256=MpS7LGaJS0CpvsjhfDkp9EJqvMvVCjRPfUp4c0aE2ys,941
 dbos/_templates/dbos-db-starter/start_postgres_docker.py,sha256=lQVLlYO5YkhGPEgPqwGc7Y8uDKse9HsWv5fynJEFJHM,1681
 dbos/_tracer.py,sha256=yN6GRDKu_1p-EqtQLNarMocPfga2ZuqpzStzzSPYhzo,2732
-dbos/_utils.py,sha256=pu8qvhZ__44LRCxO4xSb6OuyTuIV4DP-yWT22p2IuVM,477
-dbos/_workflow_commands.py,sha256=7_f8-w0MbS1gqC5v68EwzbUtomVM0lLebozpHxXmRYg,3982
+dbos/_utils.py,sha256=UbpMYRBSyvJqdXeWAnfSw8xXM1R1mfnyl1oTunhEjJM,513
+dbos/_workflow_commands.py,sha256=2E8FRUv_nLYkpBTwfhh_ELhySYpMrm8qGB9J44g6DSE,3872
 dbos/cli/_github_init.py,sha256=Y_bDF9gfO2jB1id4FV5h1oIxEJRWyqVjhb7bNEa5nQ0,3224
 dbos/cli/_template_init.py,sha256=-WW3kbq0W_Tq4WbMqb1UGJG3xvJb3woEY5VspG95Srk,2857
 dbos/cli/cli.py,sha256=gXKELYAK9_CTejQ-WbNEIqnEYByJndXHDYSX4naFg8g,19106
 dbos/dbos-config.schema.json,sha256=8KcwJb_sQc4-6tQG2TLmjE_nratfrQa0qVLl9XPsvWE,6367
 dbos/py.typed,sha256=QfzXT1Ktfk3Rj84akygc7_42z0lRpCq0Ilh8OXI6Zas,44
 version/__init__.py,sha256=L4sNxecRuqdtSFdpUGX3TtBi9KL3k7YsZVIvv-fv9-A,1678
-dbos-0.28.0a4.dist-info/RECORD,,
+dbos-0.28.0a7.dist-info/RECORD,,

dbos/_request.py

@@ -1,35 +0,0 @@
-from dataclasses import dataclass
-from typing import Any, Mapping, NamedTuple, Optional
-
-request_id_header = "x-request-id"
-
-
-class Address(NamedTuple):
-    hostname: str
-    port: int
-
-
-@dataclass
-class Request:
-    """
-    Serializable HTTP Request object.
-
-    Attributes:
-        base_url(str): Base of URL requested, as in application code
-        client(Optional[Address]): HTTP Client
-        cookies(Mapping[str, str]): HTTP Cookies
-        headers(Mapping[str, str]): HTTP headers
-        method(str): HTTP verb
-        path_params(Mapping[str, Any]): Parameters extracted from URL path sections
-        query_params(Mapping[str, str]): URL query string parameters
-        url(str): Full URL accessed
-    """
-
-    headers: Mapping[str, str]
-    path_params: Mapping[str, Any]
-    query_params: Mapping[str, str]
-    url: str
-    base_url: str
-    client: Optional[Address]
-    cookies: Mapping[str, str]
-    method: str