chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,150 @@
|
||||
using System.Diagnostics.CodeAnalysis;
|
||||
using System.Runtime.CompilerServices;
|
||||
using System.Text.Json;
|
||||
using Microsoft.Agents.AI;
|
||||
using Microsoft.Extensions.AI;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Microsoft.Extensions.Logging.Abstractions;
|
||||
|
||||
// AgentConfigAgent — the /agent-config demo.
|
||||
//
|
||||
// Reads three forwarded properties — tone, expertise, responseLength — from
|
||||
// the AG-UI shared-state payload (attached as `ag_ui_state` on
|
||||
// ChatClientAgentRunOptions.AdditionalProperties, matching the convention
|
||||
// already used by SharedStateAgent) and builds a dynamic system prompt per
|
||||
// turn.
|
||||
//
|
||||
// The frontend <CopilotKitProvider agent="agent-config-demo" />'s
|
||||
// useAgent().setState(...) call pushes the typed config into shared state;
|
||||
// this agent reads it on every run and prepends a system message that adapts
|
||||
// the inner ChatClientAgent's behavior. Missing / unrecognized values fall
|
||||
// back to the documented defaults — the agent never throws on malformed
|
||||
// config, so a misbehaving frontend can't kill the demo.
|
||||
[SuppressMessage("Performance", "CA1812:Avoid uninstantiated internal classes", Justification = "Instantiated by SalesAgentFactory")]
|
||||
internal sealed class AgentConfigAgent : DelegatingAIAgent
|
||||
{
|
||||
private static readonly HashSet<string> ValidTones = new(StringComparer.Ordinal)
|
||||
{
|
||||
"professional",
|
||||
"casual",
|
||||
"enthusiastic",
|
||||
};
|
||||
|
||||
private static readonly HashSet<string> ValidExpertise = new(StringComparer.Ordinal)
|
||||
{
|
||||
"beginner",
|
||||
"intermediate",
|
||||
"expert",
|
||||
};
|
||||
|
||||
private static readonly HashSet<string> ValidResponseLengths = new(StringComparer.Ordinal)
|
||||
{
|
||||
"concise",
|
||||
"detailed",
|
||||
};
|
||||
|
||||
private const string DefaultTone = "professional";
|
||||
private const string DefaultExpertise = "intermediate";
|
||||
private const string DefaultResponseLength = "concise";
|
||||
|
||||
private readonly ILogger<AgentConfigAgent> _logger;
|
||||
|
||||
public AgentConfigAgent(AIAgent innerAgent, ILogger<AgentConfigAgent>? logger = null)
|
||||
: base(innerAgent)
|
||||
{
|
||||
ArgumentNullException.ThrowIfNull(innerAgent);
|
||||
_logger = logger ?? NullLogger<AgentConfigAgent>.Instance;
|
||||
}
|
||||
|
||||
public override Task<AgentRunResponse> RunAsync(IEnumerable<ChatMessage> messages, AgentThread? thread = null, AgentRunOptions? options = null, CancellationToken cancellationToken = default)
|
||||
{
|
||||
return RunStreamingAsync(messages, thread, options, cancellationToken).ToAgentRunResponseAsync(cancellationToken);
|
||||
}
|
||||
|
||||
public override async IAsyncEnumerable<AgentRunResponseUpdate> RunStreamingAsync(
|
||||
IEnumerable<ChatMessage> messages,
|
||||
AgentThread? thread = null,
|
||||
AgentRunOptions? options = null,
|
||||
[EnumeratorCancellation] CancellationToken cancellationToken = default)
|
||||
{
|
||||
ArgumentNullException.ThrowIfNull(messages);
|
||||
|
||||
// Materialize up-front so we can both inspect it (to read the state)
|
||||
// and forward it to the inner agent without re-enumerating a
|
||||
// single-use iterator.
|
||||
var messageList = messages as IReadOnlyList<ChatMessage> ?? messages.ToList();
|
||||
|
||||
var (tone, expertise, responseLength) = ReadConfig(options);
|
||||
var systemPrompt = BuildSystemPrompt(tone, expertise, responseLength);
|
||||
|
||||
_logger.LogInformation(
|
||||
"AgentConfigAgent: tone={Tone}, expertise={Expertise}, responseLength={ResponseLength}",
|
||||
tone, expertise, responseLength);
|
||||
|
||||
var systemMessage = new ChatMessage(ChatRole.System, systemPrompt);
|
||||
var augmentedMessages = new List<ChatMessage>(messageList.Count + 1) { systemMessage };
|
||||
augmentedMessages.AddRange(messageList);
|
||||
|
||||
await foreach (var update in InnerAgent.RunStreamingAsync(augmentedMessages, thread, options, cancellationToken).ConfigureAwait(false))
|
||||
{
|
||||
yield return update;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Reads the forwarded config triple from the AG-UI shared-state payload
|
||||
/// attached to the run options. Any missing / unrecognized value falls
|
||||
/// back to the corresponding default constant. Never throws.
|
||||
/// </summary>
|
||||
internal static (string Tone, string Expertise, string ResponseLength) ReadConfig(AgentRunOptions? options)
|
||||
{
|
||||
if (options is not ChatClientAgentRunOptions { ChatOptions.AdditionalProperties: { } properties } ||
|
||||
!properties.TryGetValue("ag_ui_state", out JsonElement state) ||
|
||||
state.ValueKind != JsonValueKind.Object)
|
||||
{
|
||||
return (DefaultTone, DefaultExpertise, DefaultResponseLength);
|
||||
}
|
||||
|
||||
var tone = ReadStringProperty(state, "tone", ValidTones, DefaultTone);
|
||||
var expertise = ReadStringProperty(state, "expertise", ValidExpertise, DefaultExpertise);
|
||||
var responseLength = ReadStringProperty(state, "responseLength", ValidResponseLengths, DefaultResponseLength);
|
||||
|
||||
return (tone, expertise, responseLength);
|
||||
}
|
||||
|
||||
private static string ReadStringProperty(JsonElement state, string name, HashSet<string> valid, string defaultValue)
|
||||
{
|
||||
if (!state.TryGetProperty(name, out var element) || element.ValueKind != JsonValueKind.String)
|
||||
{
|
||||
return defaultValue;
|
||||
}
|
||||
var value = element.GetString();
|
||||
return value is not null && valid.Contains(value) ? value : defaultValue;
|
||||
}
|
||||
|
||||
internal static string BuildSystemPrompt(string tone, string expertise, string responseLength)
|
||||
{
|
||||
var toneRule = tone switch
|
||||
{
|
||||
"casual" => "Use friendly, conversational language. Contractions OK. Light humor welcome.",
|
||||
"enthusiastic" => "Use upbeat, energetic language. Exclamation points OK. Emoji OK.",
|
||||
_ => "Use neutral, precise language. No emoji. Short sentences.",
|
||||
};
|
||||
var expertiseRule = expertise switch
|
||||
{
|
||||
"beginner" => "Assume no prior knowledge. Define jargon. Use analogies.",
|
||||
"expert" => "Assume technical fluency. Use precise terminology. Skip basics.",
|
||||
_ => "Assume common terms are understood; explain specialized terms.",
|
||||
};
|
||||
var lengthRule = responseLength switch
|
||||
{
|
||||
"detailed" => "Respond in multiple paragraphs with examples where relevant.",
|
||||
_ => "Respond in 1-3 sentences.",
|
||||
};
|
||||
|
||||
return "You are a helpful assistant.\n\n" +
|
||||
$"Tone: {toneRule}\n" +
|
||||
$"Expertise level: {expertiseRule}\n" +
|
||||
$"Response length: {lengthRule}";
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user