exposed all proxies

Author: Andrei Neagu

services/dynamic-scheduler/src/simcore_service_dynamic_scheduler/services/generic_scheduler/__init__.py

@@ -1,5 +1,9 @@
 from ._core import cancel_operation, start_operation
-from ._deferred_runner import get_step_group_proxy, get_step_store_proxy
+from ._deferred_runner import (
+    get_opration_context_proxy,
+    get_step_group_proxy,
+    get_step_store_proxy,
+)
 from ._lifespan import get_generic_scheduler_lifespans
 from ._models import (
     OperationName,
@@ -14,15 +18,17 @@
     ParallelStepGroup,
     SingleStepGroup,
 )
-from ._store import StepGroupProxy, StepStoreProxy
+from ._store import OperationContextProxy, StepGroupProxy, StepStoreProxy
 
 __all__: tuple[str, ...] = (
     "BaseStep",
     "cancel_operation",
     "get_generic_scheduler_lifespans",
+    "get_opration_context_proxy",
     "get_step_group_proxy",
     "get_step_store_proxy",
     "Operation",
+    "OperationContextProxy",
     "OperationName",
     "OperationRegistry",
     "ParallelStepGroup",

services/dynamic-scheduler/src/simcore_service_dynamic_scheduler/services/generic_scheduler/_deferred_runner.py

@@ -65,7 +65,7 @@ def get_step_group_proxy(context: DeferredContext) -> StepGroupProxy:
     )
 
 
-def _get_opration_context_proxy(context: DeferredContext) -> OperationContextProxy:
+def get_opration_context_proxy(context: DeferredContext) -> OperationContextProxy:
     app: FastAPI = context["app"]
     schedule_id: ScheduleId = context["schedule_id"]
     operation_name: OperationName = context["operation_name"]
@@ -174,7 +174,7 @@ async def run(cls, context: DeferredContext) -> None:
 
         step = _get_step(context)
 
-        opration_context_proxy = _get_opration_context_proxy(context)
+        opration_context_proxy = get_opration_context_proxy(context)
 
         if is_creating:
             required_context = await opration_context_proxy.get_required_context(

services/dynamic-scheduler/src/simcore_service_dynamic_scheduler/services/generic_scheduler/_operation.py

@@ -60,8 +60,8 @@ def get_create_provides_context_keys(cls) -> set[str]:
     async def get_create_retries(cls, context: DeferredContext) -> int:
         """
         [optional] amount of retires in case of creation
-        HINT: you can use `get_step_group_proxy(context)` and `get_step_store_proxy(context)`
-        to implement custom retry strategy
+        HINT: you can use `get_opration_context_proxy()`, `get_step_group_proxy(context)`
+            and `get_step_store_proxy(context)` to implement custom retry strategy
         """
         assert context  # nosec
         return _DEFAULT_STEP_RETRIES
@@ -72,8 +72,8 @@ async def get_create_wait_between_attempts(
     ) -> timedelta:
         """
         [optional] wait time between retires case of creation
-        HINT: you can use `get_step_group_proxy(context)` and `get_step_store_proxy(context)`
-        to implement custom retry strategy
+        HINT: you can use `get_opration_context_proxy()`, `get_step_group_proxy(context)`
+            and `get_step_store_proxy(context)` to implement custom retry strategy
         """
         assert context  # nosec
         return _DEFAULT_STEP_TIMEOUT
@@ -120,8 +120,8 @@ def get_revert_provides_context_keys(cls) -> set[str]:
     async def get_revert_retries(cls, context: DeferredContext) -> int:
         """
         [optional] amount of retires in case of failure
-        HINT: you can use `get_step_group_proxy(context)` and `get_step_store_proxy(context)`
-        to implement custom retry strategy
+        HINT: you can use `get_opration_context_proxy()`, `get_step_group_proxy(context)`
+            and `get_step_store_proxy(context)` to implement custom retry strategy
         """
         assert context  # nosec
         return _DEFAULT_STEP_RETRIES
@@ -132,8 +132,8 @@ async def get_revert_wait_between_attempts(
     ) -> timedelta:
         """
         [optional] timeout between retires in case of failure
-        HINT: you can use `get_step_group_proxy(context)` and `get_step_store_proxy(context)`
-        to implement custom retry strategy
+        HINT: you can use `get_opration_context_proxy()`, `get_step_group_proxy(context)`
+            and `get_step_store_proxy(context)` to implement custom retry strategy
         """
         assert context  # nosec
         return _DEFAULT_STEP_TIMEOUT