Load every enabled built-in workload in the worker by default
This commit is contained in:
@@ -61,7 +61,9 @@ def _definitions_from_environment() -> dict[str, WorkloadDefinition]:
|
|||||||
"""Load definitions from ``SCIMESH_WORKLOAD_ALLOWLIST`` or the built-ins.
|
"""Load definitions from ``SCIMESH_WORKLOAD_ALLOWLIST`` or the built-ins.
|
||||||
|
|
||||||
The environment-driven discovery mirrors the former worker configuration;
|
The environment-driven discovery mirrors the former worker configuration;
|
||||||
the Go agent passes the allowlist through unchanged.
|
the Go agent passes the allowlist through unchanged. Without an allowlist
|
||||||
|
the worker serves every enabled built-in SDK workload of the installed
|
||||||
|
package — the same library the coordinator embeds as its catalog.
|
||||||
"""
|
"""
|
||||||
import os
|
import os
|
||||||
|
|
||||||
@@ -82,10 +84,17 @@ def _definitions_from_environment() -> dict[str, WorkloadDefinition]:
|
|||||||
if not definitions:
|
if not definitions:
|
||||||
raise ValueError("workload_allowlist discovered no workloads")
|
raise ValueError("workload_allowlist discovered no workloads")
|
||||||
return definitions
|
return definitions
|
||||||
from scimesh.workloads.search import similarity_search_sdk_definition
|
from scimesh.workloads.library import default_sdk_registry
|
||||||
|
|
||||||
|
registry = default_sdk_registry()
|
||||||
return {
|
return {
|
||||||
"similarity-search": similarity_search_sdk_definition().definition(),
|
description.workload.name: registry.require(
|
||||||
|
description.workload.name,
|
||||||
|
description.workload.version,
|
||||||
|
description.package_digest,
|
||||||
|
)[0]
|
||||||
|
for description in registry.descriptions()
|
||||||
|
if description.enabled
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user