dbos 0.23.0a12__tar.gz → 0.23.0a13__tar.gz

{dbos-0.23.0a12 → dbos-0.23.0a13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbos
-Version: 0.23.0a12
+Version: 0.23.0a13
 Summary: Ultra-lightweight durable execution in Python
 Author-Email: "DBOS, Inc." <contact@dbos.dev>
 License: MIT

{dbos-0.23.0a12 → dbos-0.23.0a13}/dbos/_conductor/conductor.py

@@ -8,7 +8,7 @@ from websockets.sync.client import connect
 from websockets.sync.connection import Connection
 
 from dbos._utils import GlobalParams
-from dbos._workflow_commands import list_queued_workflows, list_workflows
+from dbos._workflow_commands import get_workflow, list_queued_workflows, list_workflows
 
 from . import protocol as p
 
@@ -175,6 +175,25 @@ class ConductorWebsocket(threading.Thread):
                                 )
                             )
                             websocket.send(list_queued_workflows_response.to_json())
+                        elif type == p.MessageType.GET_WORKFLOW:
+                            get_workflow_message = p.GetWorkflowRequest.from_json(
+                                message
+                            )
+                            info = get_workflow(
+                                self.dbos._sys_db,
+                                get_workflow_message.workflow_id,
+                                getRequest=False,
+                            )
+                            get_workflow_response = p.GetWorkflowResponse(
+                                type=p.MessageType.GET_WORKFLOW,
+                                request_id=base_message.request_id,
+                                output=(
+                                    p.WorkflowsOutput.from_workflow_information(info)
+                                    if info is not None
+                                    else None
+                                ),
+                            )
+                            websocket.send(get_workflow_response.to_json())
                         else:
                             self.dbos.logger.warning(f"Unexpected message type: {type}")
             except ConnectionClosedOK:

{dbos-0.23.0a12 → dbos-0.23.0a13}/dbos/_conductor/protocol.py

@@ -14,6 +14,7 @@ class MessageType(str, Enum):
     LIST_QUEUED_WORKFLOWS = "list_queued_workflows"
     RESUME = "resume"
     RESTART = "restart"
+    GET_WORKFLOW = "get_workflow"
 
 
 T = TypeVar("T", bound="BaseMessage")
@@ -184,3 +185,13 @@ class ListQueuedWorkflowsRequest(BaseMessage):
 @dataclass
 class ListQueuedWorkflowsResponse(BaseMessage):
     output: List[WorkflowsOutput]
+
+
+@dataclass
+class GetWorkflowRequest(BaseMessage):
+    workflow_id: str
+
+
+@dataclass
+class GetWorkflowResponse(BaseMessage):
+    output: Optional[WorkflowsOutput]

{dbos-0.23.0a12 → dbos-0.23.0a13}/pyproject.toml

@@ -28,7 +28,7 @@ dependencies = [
 ]
 requires-python = ">=3.9"
 readme = "README.md"
-version = "0.23.0a12"
+version = "0.23.0a13"
 
 [project.license]
 text = "MIT"