Replace legacy distributed protocol with SDK-built workloads

2026-08-01 23:57:41 +03:00
parent 96169086f0
commit 19fbb8e926
34 changed files with 3009 additions and 1986 deletions
@@ -0,0 +1,68 @@
+"""Built-in workload library: default registry and runtime.
+
+This module is workload code, not SDK framework code. It composes the
+installed SciMesh workload packages (``search``, ``graph``, ``descriptors``)
+with the SDK registry and runtime. External workload libraries can follow the
+same pattern with their own packages and entry points.
+"""
+
+from __future__ import annotations
+
+import os
+import platform
+
+from scimesh.sdk.identity import SDK_API_VERSION
+from scimesh.sdk.registry import WorkloadRegistry
+from scimesh.sdk.resources import ResourceInventory
+from scimesh.sdk.runtime import RuntimeCapabilities
+
+from .descriptors import descriptor_batch_sdk_definition
+from .environment import current_environment_digest
+from .graph import similarity_graph_sdk_definition
+from .search import similarity_search_sdk_definition
+
+__all__ = [
+    "default_sdk_registry",
+    "default_sdk_runtime",
+]
+
+
+def default_sdk_registry(*, shard_rows: int = 10_000) -> WorkloadRegistry:
+    """Registry of every built-in SDK-built workload, all enabled."""
+    registry = WorkloadRegistry()
+    registry.register(
+        similarity_search_sdk_definition(shard_rows=shard_rows).definition(),
+        enabled=True,
+    )
+    registry.register(
+        similarity_graph_sdk_definition().definition(),
+        enabled=True,
+    )
+    registry.register(
+        descriptor_batch_sdk_definition(shard_rows=shard_rows).definition(),
+        enabled=True,
+    )
+    return registry
+
+
+def default_sdk_runtime() -> RuntimeCapabilities:
+    """Runtime advertising the built-in workloads' capabilities and inventory."""
+    architecture = platform.machine().lower() or "unknown"
+    return RuntimeCapabilities(
+        sdk_api_version=SDK_API_VERSION,
+        protocol_version="1.0.0",
+        profiles=("core-batch-v1",),
+        features={"artifact-collections": "1.0.0", "exact-verifier": "1.0.0"},
+        workload_capabilities=(
+            "similarity-search",
+            "similarity-graph",
+            "descriptor-batch",
+        ),
+        inventory=ResourceInventory(
+            cpu_cores=max(os.cpu_count() or 1, 1),
+            memory_mb=4096,
+            scratch_mb=4096,
+            architecture=architecture,
+            environment_digests=(current_environment_digest(),),
+        ),
+    )