mirror of
https://github.com/microsoft/semantic-kernel.git
synced 2026-10-12 02:19:06 +00:00
Evan Mattson
30d2b0802a
### Motivation and Context
SK Python processes have been missing support to be able to serialize
and deserialize JSON state for a process and its steps. This PR brings
in the functionality to allow the developer to do so. The
`getting_started_with_processes` step03 has been update to reflect this
latest functionality. It is possible to dump a JSON state to a file, and
reload the state to continue running the process.
State metadata that handles the version for steps is managed via a
decorator:
```python
@kernel_process_step_metadata("CutFoodStep.V1")
class CutFoodStep(KernelProcessStep):
class Functions(Enum):
ChopFood = "ChopFood"
SliceFood = "SliceFood"
```
If no decorator/state is supplied the step will be built with a default
state version of "v1" which aligns with .Net.
<!-- Thank you for your contribution to the semantic-kernel repo!
Please help reviewers and future users, providing the following
information:
1. Why is this change required?
2. What problem does it solve?
3. What scenario does it contribute to?
4. If it fixes an open issue, please link to the issue here.
-->
### Description
Support state and versioning management for Python processes.
- Update samples to reflect changes.
- Closes #9584
<!-- Describe your changes, the overall approach, the underlying design.
These notes will help understanding how your code works. Thanks! -->
### Contribution Checklist
<!-- Before submitting this PR, please make sure: -->
- [X] The code builds clean without any errors or warnings
- [X] The PR follows the [SK Contribution
Guidelines](https://github.com/microsoft/semantic-kernel/blob/main/CONTRIBUTING.md)
and the [pre-submission formatting
script](https://github.com/microsoft/semantic-kernel/blob/main/CONTRIBUTING.md#development-scripts)
raises no violations
- [X] All unit tests pass, and I have added new tests where possible
- [ ] I didn't break anyone 😄
67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
from collections.abc import Callable
|
|
from typing import TYPE_CHECKING, Any
|
|
|
|
from pydantic import Field
|
|
|
|
from semantic_kernel.processes.kernel_process.kernel_process_edge import KernelProcessEdge
|
|
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.process_state_metadata_utils import kernel_process_to_process_state_metadata
|
|
from semantic_kernel.utils.feature_stage_decorator import experimental
|
|
|
|
if TYPE_CHECKING:
|
|
from semantic_kernel.processes.kernel_process.kernel_process_edge import KernelProcessEdge
|
|
from semantic_kernel.processes.kernel_process.kernel_process_step_state_metadata import KernelProcessStateMetadata
|
|
|
|
|
|
@experimental
|
|
class KernelProcess(KernelProcessStepInfo):
|
|
"""A kernel process."""
|
|
|
|
steps: list[KernelProcessStepInfo] = Field(default_factory=list)
|
|
factories: dict[str, Callable] = Field(default_factory=dict)
|
|
|
|
def __init__(
|
|
self,
|
|
state: KernelProcessState,
|
|
steps: list[KernelProcessStepInfo],
|
|
edges: dict[str, list["KernelProcessEdge"]] | None = None,
|
|
factories: dict[str, Callable] | None = None,
|
|
):
|
|
"""Initialize the kernel process.
|
|
|
|
Args:
|
|
state: The state of the process.
|
|
steps: The steps of the process.
|
|
edges: The edges of the process. Defaults to None.
|
|
factories: The factories of the process. This allows for the creation of
|
|
steps that require complex dependencies that cannot be JSON serialized or deserialized.
|
|
"""
|
|
if not state:
|
|
raise ValueError("state cannot be None")
|
|
if not steps:
|
|
raise ValueError("steps cannot be None")
|
|
if not state.name:
|
|
raise ValueError("state.Name cannot be None")
|
|
|
|
process_steps = []
|
|
process_steps.extend(steps)
|
|
|
|
args: dict[str, Any] = {
|
|
"steps": process_steps,
|
|
"inner_step_type": KernelProcess,
|
|
"state": state,
|
|
"output_edges": edges or {},
|
|
}
|
|
|
|
if factories:
|
|
args["factories"] = factories
|
|
|
|
super().__init__(**args)
|
|
|
|
def to_process_state_metadata(self) -> "KernelProcessStateMetadata":
|
|
"""Converts a kernel process to process state metadata."""
|
|
return kernel_process_to_process_state_metadata(self)
|