|
18 | 18 | ) |
19 | 19 | from semantic_kernel.processes.kernel_process.kernel_process_state import KernelProcessState |
20 | 20 | from semantic_kernel.processes.kernel_process.kernel_process_step import KernelProcessStep |
21 | | -from semantic_kernel.processes.kernel_process.kernel_process_step_context import KernelProcessStepContext |
22 | 21 | from semantic_kernel.processes.kernel_process.kernel_process_step_info import KernelProcessStepInfo |
23 | 22 | from semantic_kernel.processes.local_runtime.local_kernel_process import start |
24 | 23 | from semantic_kernel.processes.local_runtime.local_message import LocalMessage |
@@ -336,36 +335,11 @@ async def test_handle_message_with_valid_event_id(mock_process_with_output_edges |
336 | 335 | END_PROCESS_ID = "END" |
337 | 336 |
|
338 | 337 |
|
339 | | -@pytest.fixture |
340 | | -def mock_process_with_steps(mock_process, mock_kernel, build_model): |
341 | | - """Fixture to create a LocalProcess with mock steps.""" |
342 | | - local_process = LocalProcess(process=mock_process, kernel=mock_kernel) |
343 | | - |
344 | | - # Create real LocalStep instances |
345 | | - state_1 = MagicMock(spec=KernelProcessState) |
346 | | - state_1.name = "step_1_state" |
347 | | - state_1.id = "step_1_id" |
348 | | - |
349 | | - step_info_1 = MagicMock(spec=KernelProcessStepInfo) |
350 | | - step_info_1.state = state_1 |
351 | | - |
352 | | - state_2 = MagicMock(spec=KernelProcessState) |
353 | | - state_2.name = "step_2_state" |
354 | | - state_2.id = "step_2_id" |
355 | | - |
356 | | - step_info_2 = MagicMock(spec=KernelProcessStepInfo) |
357 | | - step_info_2.state = state_2 |
358 | | - |
359 | | - step_1 = LocalStep(step_info=step_info_1, kernel=mock_kernel, parent_process_id="test_process") |
360 | | - step_2 = LocalStep(step_info=step_info_2, kernel=mock_kernel, parent_process_id="test_process") |
361 | | - |
362 | | - local_process.steps = [step_1, step_2] |
363 | | - return local_process |
364 | | - |
365 | | - |
366 | 338 | @pytest.mark.parametrize("defer_second_b", [False, True]) |
367 | 339 | @pytest.mark.parametrize("fail_first", [False, True]) |
368 | 340 | async def test_messages_to_same_step_do_not_reuse_inputs_during_invocation(defer_second_b, fail_first): |
| 341 | + from semantic_kernel.processes.kernel_process.kernel_process_step_context import KernelProcessStepContext |
| 342 | + |
369 | 343 | pairs = [] |
370 | 344 |
|
371 | 345 | class Producer(KernelProcessStep): |
@@ -403,3 +377,30 @@ async def relay(self, value: int) -> int: |
403 | 377 | pass |
404 | 378 |
|
405 | 379 | assert pairs == [(1, 1), (2, 2)] |
| 380 | + |
| 381 | + |
| 382 | +@pytest.fixture |
| 383 | +def mock_process_with_steps(mock_process, mock_kernel, build_model): |
| 384 | + """Fixture to create a LocalProcess with mock steps.""" |
| 385 | + local_process = LocalProcess(process=mock_process, kernel=mock_kernel) |
| 386 | + |
| 387 | + # Create real LocalStep instances |
| 388 | + state_1 = MagicMock(spec=KernelProcessState) |
| 389 | + state_1.name = "step_1_state" |
| 390 | + state_1.id = "step_1_id" |
| 391 | + |
| 392 | + step_info_1 = MagicMock(spec=KernelProcessStepInfo) |
| 393 | + step_info_1.state = state_1 |
| 394 | + |
| 395 | + state_2 = MagicMock(spec=KernelProcessState) |
| 396 | + state_2.name = "step_2_state" |
| 397 | + state_2.id = "step_2_id" |
| 398 | + |
| 399 | + step_info_2 = MagicMock(spec=KernelProcessStepInfo) |
| 400 | + step_info_2.state = state_2 |
| 401 | + |
| 402 | + step_1 = LocalStep(step_info=step_info_1, kernel=mock_kernel, parent_process_id="test_process") |
| 403 | + step_2 = LocalStep(step_info=step_info_2, kernel=mock_kernel, parent_process_id="test_process") |
| 404 | + |
| 405 | + local_process.steps = [step_1, step_2] |
| 406 | + return local_process |
0 commit comments