Load every enabled built-in workload in the worker by default

This commit is contained in:
Emil
2026-08-02 17:47:47 +03:00
parent 5d738e0a14
commit 700a96a259
+12 -3
View File
@@ -61,7 +61,9 @@ def _definitions_from_environment() -> dict[str, WorkloadDefinition]:
"""Load definitions from ``SCIMESH_WORKLOAD_ALLOWLIST`` or the built-ins.
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
@@ -82,10 +84,17 @@ def _definitions_from_environment() -> dict[str, WorkloadDefinition]:
if not definitions:
raise ValueError("workload_allowlist discovered no workloads")
return definitions
from scimesh.workloads.search import similarity_search_sdk_definition
from scimesh.workloads.library import default_sdk_registry
registry = default_sdk_registry()
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
}