Conductor Key in DBOSConfig (#474)

Author: kraftp

dbos/_dbos.py

@@ -335,6 +335,8 @@ def __init__(
         self._background_threads: List[threading.Thread] = []
         self.conductor_url: Optional[str] = conductor_url
         self.conductor_key: Optional[str] = conductor_key
+        if config.get("conductor_key"):
+            self.conductor_key = config.get("conductor_key")
         self.conductor_websocket: Optional[ConductorWebsocket] = None
         self._background_event_loop: BackgroundEventLoop = BackgroundEventLoop()
         self._active_workflows_set: set[str] = set()

dbos/_dbos_config.py

@@ -36,6 +36,7 @@ class DBOSConfig(TypedDict, total=False):
         dbos_system_schema (str): Schema name for DBOS system tables. Defaults to "dbos".
         enable_otlp (bool): If True, enable built-in DBOS OTLP tracing and logging.
         system_database_engine (sa.Engine): A custom system database engine. If provided, DBOS will not create an engine but use this instead.
+        conductor_key (str): An API key for DBOS Conductor. Pass this in to connect your process to Conductor.
     """
 
     name: str
@@ -55,6 +56,7 @@ class DBOSConfig(TypedDict, total=False):
     dbos_system_schema: Optional[str]
     enable_otlp: Optional[bool]
     system_database_engine: Optional[sa.Engine]
+    conductor_key: Optional[str]
 
 
 class RuntimeConfig(TypedDict, total=False):