diff --git a/superduperdb/base/config.py b/superduperdb/base/config.py index 0aaaa8bdb..bf304296b 100644 --- a/superduperdb/base/config.py +++ b/superduperdb/base/config.py @@ -155,6 +155,15 @@ class CronTab(BaseConfig): uri: t.Optional[str] = None +@dc.dataclass +class SchedulerConfig(BaseConfig): + """Describes the configuration for scheduler service + + :param uri: The URI for the scheduler service. + """ + + uri: t.Optional[str] = None # None implies local mode + @dc.dataclass class Compute(BaseConfig): @@ -196,7 +205,7 @@ class Cluster(BaseConfig): vector_search: VectorSearch = dc.field(default_factory=VectorSearch) rest: Rest = dc.field(default_factory=Rest) cdc: CDCConfig = dc.field(default_factory=CDCConfig) - + scheduler: SchedulerConfig = dc.field(default_factory=SchedulerConfig) class LogLevel(str, Enum): """Enumerate log severity level # noqa.""" diff --git a/superduperdb/misc/server.py b/superduperdb/misc/server.py index e658f89a9..476f45295 100644 --- a/superduperdb/misc/server.py +++ b/superduperdb/misc/server.py @@ -42,7 +42,7 @@ def _request_server( elif service == 'vector_search': service_uri = CFG.cluster.vector_search.uri elif service == 'scheduler': - service_uri = 'http://localhost:8181' + service_uri = CFG.cluster.scheduler.uri else: raise NotImplementedError(f'Unknown service {service}')