mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Stephen ToubandGitHub
456e7d1b65
- Moved InProcessRuntime type into abstractions package and deleted InProcess package. - Moved several members of IAgentRuntime to be extension methods instead, e.g. multiple GetActorAsync overloads. - Added synchronous RegisterMessageHandler overloads and used them to avoid unnecessary async usage at call sites. - Removed unnecessary surface area from InProcessRuntime, e.g. StopAsync, RunUntilIdleAsync, etc. - Fixed spin loop in InProcessRuntime that would consume an entire core for the duration of the orchestration's operation. - Removed a bunch of allocation from InProcessRuntime. - Made a runtime optional for orchestrations, defaulting to using a temporary InProcessRuntime if none is provided. - Removed custom delegate types from orchestrations. - Consolidated namespaces. - Used records to simplify message classes. - Tweaked naming on AgentActor to make purpose of protected methods more clear. - Removed invocation in AgentActor.InvokeAsync of empty update / isFinal parameter. - Changed OrchestrationHandoffs to avoid needing to pass in agents duplicatively. - Made various extension methods, such as those on OrchestrationHandoffsExtensions, into instance methods.
60 lines
1.8 KiB
C#
60 lines
1.8 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using System.Threading.Tasks;
|
|
using Microsoft.Extensions.AI.Agents;
|
|
|
|
namespace Microsoft.Agents.Orchestration.UnitTest;
|
|
|
|
/// <summary>
|
|
/// Tests for the <see cref="SequentialOrchestration"/> class.
|
|
/// </summary>
|
|
public class SequentialOrchestrationTests
|
|
{
|
|
[Fact]
|
|
public async Task SequentialOrchestrationWithSingleAgentAsync()
|
|
{
|
|
// Arrange
|
|
MockAgent mockAgent1 = MockAgent.CreateWithResponse(2, "xyz");
|
|
|
|
// Act: Create and execute the orchestration
|
|
string response = await ExecuteOrchestrationAsync(mockAgent1);
|
|
|
|
// Assert
|
|
Assert.Equal(1, mockAgent1.InvokeCount);
|
|
Assert.Equal("xyz", response);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task SequentialOrchestrationWithMultipleAgentsAsync()
|
|
{
|
|
// Arrange
|
|
MockAgent mockAgent1 = MockAgent.CreateWithResponse(1, "abc");
|
|
MockAgent mockAgent2 = MockAgent.CreateWithResponse(2, "xyz");
|
|
MockAgent mockAgent3 = MockAgent.CreateWithResponse(3, "lmn");
|
|
|
|
// Act: Create and execute the orchestration
|
|
string response = await ExecuteOrchestrationAsync(mockAgent1, mockAgent2, mockAgent3);
|
|
|
|
// Assert
|
|
Assert.Equal(1, mockAgent1.InvokeCount);
|
|
Assert.Equal(1, mockAgent2.InvokeCount);
|
|
Assert.Equal(1, mockAgent3.InvokeCount);
|
|
Assert.Equal("lmn", response);
|
|
}
|
|
|
|
private static async Task<string> ExecuteOrchestrationAsync(params Agent[] mockAgents)
|
|
{
|
|
// Act
|
|
SequentialOrchestration orchestration = new(mockAgents);
|
|
|
|
const string InitialInput = "123";
|
|
OrchestrationResult<string> result = await orchestration.InvokeAsync(InitialInput);
|
|
|
|
// Assert
|
|
Assert.NotNull(result);
|
|
|
|
// Act
|
|
return await result.Task;
|
|
}
|
|
}
|