wehub-resource-sync
69 lines
3.0 KiB
Python
69 lines
3.0 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
|
|
from collections.abc import MutableSequence, Sequence
|
|
from typing import Literal
|
|
|
|
from pydantic import Field
|
|
|
|
from semantic_kernel.processes.dapr_runtime.dapr_step_info import DaprStepInfo
|
|
from semantic_kernel.processes.kernel_process.kernel_process import KernelProcess
|
|
from semantic_kernel.processes.kernel_process.kernel_process_state import KernelProcessState
|
|
from semantic_kernel.processes.kernel_process.kernel_process_step_info import KernelProcessStepInfo
|
|
from semantic_kernel.processes.step_utils import DEFAULT_ALLOWED_MODULE_PREFIXES
|
|
from semantic_kernel.utils.feature_stage_decorator import experimental
|
|
|
|
|
|
@experimental
|
|
class DaprProcessInfo(DaprStepInfo):
|
|
"""A Dapr process info."""
|
|
|
|
type: Literal["DaprProcessInfo"] = "DaprProcessInfo" # type: ignore
|
|
steps: MutableSequence["DaprStepInfo | DaprProcessInfo"] = Field(default_factory=list)
|
|
|
|
def to_kernel_process(
|
|
self, allowed_module_prefixes: Sequence[str] | None = DEFAULT_ALLOWED_MODULE_PREFIXES
|
|
) -> KernelProcess:
|
|
"""Converts the Dapr process info to a kernel process.
|
|
|
|
Args:
|
|
allowed_module_prefixes: Sequence of module prefixes that are allowed
|
|
for step class loading. Defaults to DEFAULT_ALLOWED_MODULE_PREFIXES
|
|
("semantic_kernel.",). Pass None to disable the allowlist and allow
|
|
any module (not recommended for production). An empty sequence blocks
|
|
all modules.
|
|
"""
|
|
if not isinstance(self.state, KernelProcessState):
|
|
raise ValueError("State must be a kernel process state")
|
|
|
|
steps: list[KernelProcessStepInfo] = []
|
|
for step in self.steps:
|
|
if isinstance(step, DaprProcessInfo):
|
|
steps.append(step.to_kernel_process(allowed_module_prefixes=allowed_module_prefixes))
|
|
else:
|
|
steps.append(step.to_kernel_process_step_info(allowed_module_prefixes=allowed_module_prefixes))
|
|
|
|
return KernelProcess(state=self.state, steps=steps, edges=self.edges)
|
|
|
|
@classmethod
|
|
def from_kernel_process(cls, kernel_process: KernelProcess) -> "DaprProcessInfo":
|
|
"""Creates a Dapr process info from a kernel process."""
|
|
if kernel_process is None:
|
|
raise ValueError("Kernel process must be provided")
|
|
|
|
dapr_step_info = DaprStepInfo.from_kernel_step_info(kernel_process)
|
|
dapr_steps: MutableSequence[DaprProcessInfo | DaprStepInfo] = []
|
|
|
|
for step in kernel_process.steps:
|
|
if isinstance(step, KernelProcess):
|
|
dapr_steps.append(DaprProcessInfo.from_kernel_process(step))
|
|
else:
|
|
dapr_steps.append(DaprStepInfo.from_kernel_step_info(step))
|
|
|
|
return DaprProcessInfo(
|
|
inner_step_python_type=dapr_step_info.inner_step_python_type,
|
|
state=dapr_step_info.state,
|
|
edges={key: list(value) for key, value in dapr_step_info.edges.items()},
|
|
steps=dapr_steps,
|
|
)
|