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.
59 lines
2.7 KiB
C#
59 lines
2.7 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using System;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace Microsoft.Extensions.AI.Agents.Runtime;
|
|
|
|
/// <summary>
|
|
/// Extension methods for <see cref="IAgentRuntime"/>.
|
|
/// </summary>
|
|
internal static class AgentRuntimeExtensions
|
|
{
|
|
/// <summary>
|
|
/// Sends a message to the specified agent.
|
|
/// </summary>
|
|
public static ValueTask PublishMessageAsync(this IAgentRuntime runtime, object message, ActorType agentType, CancellationToken cancellationToken = default) =>
|
|
runtime.PublishMessageAsync(message, new TopicId(agentType.Name), sender: null, messageId: null, cancellationToken);
|
|
|
|
/// <summary>
|
|
/// Registers an agent factory for the specified agent type and associates it with the runtime.
|
|
/// </summary>
|
|
/// <param name="runtime">The runtime targeted for registration.</param>
|
|
/// <param name="agentType">The type of agent to register.</param>
|
|
/// <param name="factoryFunc">The factory function for creating the agent.</param>
|
|
/// <returns>The registered agent type.</returns>
|
|
public static async ValueTask<ActorType> RegisterOrchestrationAgentAsync(this IAgentRuntime runtime, ActorType agentType, Func<ActorId, IAgentRuntime, ValueTask<IRuntimeActor>> factoryFunc)
|
|
{
|
|
ActorType registeredType = await runtime.RegisterActorFactoryAsync(agentType, factoryFunc).ConfigureAwait(false);
|
|
|
|
// Subscribe agent to its own unique topic
|
|
await runtime.SubscribeAsync(new(registeredType.Name)).ConfigureAwait(false);
|
|
|
|
return registeredType;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Subscribes the specified agent type to its own dedicated topic.
|
|
/// </summary>
|
|
/// <param name="runtime">The runtime for managing the subscription.</param>
|
|
/// <param name="agentType">The agent type to subscribe.</param>
|
|
public static Task SubscribeAsync(this IAgentRuntime runtime, ActorType agentType) =>
|
|
runtime.AddSubscriptionAsync(new TypeSubscription(agentType.Name, agentType)).AsTask();
|
|
|
|
/// <summary>
|
|
/// Subscribes the specified agent type to the provided topics.
|
|
/// </summary>
|
|
/// <param name="runtime">The runtime for managing the subscription.</param>
|
|
/// <param name="agentType">The agent type to subscribe.</param>
|
|
/// <param name="topics">A variable list of topics for subscription.</param>
|
|
public static async Task SubscribeAsync(this IAgentRuntime runtime, ActorType agentType, params TopicId[] topics)
|
|
{
|
|
for (int index = 0; index < topics.Length; ++index)
|
|
{
|
|
await runtime.AddSubscriptionAsync(new TypeSubscription(topics[index].Type, agentType)).ConfigureAwait(false);
|
|
}
|
|
}
|
|
}
|