Add distributed similarity search

2026-07-24 14:38:11 +03:00
parent 0bef7604fd
commit 0f3a2d92d8
10 changed files with 749 additions and 47 deletions
@@ -12,17 +12,25 @@ from .models import (
    FinalResult,
    PlannedTask,
 )
-from .registry import DistributedWorkloadRegistry, PlanningService, WorkloadDescription
+from .registry import (
+    DistributedWorkloadRegistry,
+    PlanningService,
+    WorkloadDescription,
+    default_distributed_registry,
+)
+from .similarity_search import SimilaritySearchDistributedWorkload
 from .workload import DistributedWorkload

 __all__ = [
    "ArtifactReference",
    "CompletedPartial",
+    "default_distributed_registry",
    "DistributedPlan",
    "DistributedWorkload",
    "DistributedWorkloadRegistry",
    "FinalResult",
    "PlannedTask",
    "PlanningService",
+    "SimilaritySearchDistributedWorkload",
    "WorkloadDescription",
 ]