mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
7dccf3a07b
* Update message source code to match python. * Apply suggestion from @Copilot Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> * Address PR comment * Move setting of source information to extension method * Add underscore for attribution key to indicate internal usage * Stick to version 102 of the SDK since 103 is causing issues. * Revert global.json change * Fix unit test --------- Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
53 lines
2.7 KiB
C#
53 lines
2.7 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Microsoft.Extensions.AI;
|
|
|
|
namespace Microsoft.Agents.AI;
|
|
|
|
/// <summary>
|
|
/// Contains extension methods for the <see cref="ChatHistoryProvider"/> class.
|
|
/// </summary>
|
|
public static class ChatHistoryProviderExtensions
|
|
{
|
|
/// <summary>
|
|
/// Adds message filtering to an existing <see cref="ChatHistoryProvider"/>, so that messages passed to the <see cref="ChatHistoryProvider"/> and messages
|
|
/// provided by the <see cref="ChatHistoryProvider"/> can be filtered, updated or replaced.
|
|
/// </summary>
|
|
/// <param name="provider">The <see cref="ChatHistoryProvider"/> to add the message filter to.</param>
|
|
/// <param name="invokingMessagesFilter">An optional filter function to apply to messages produced by the <see cref="ChatHistoryProvider"/>. If null, no filter is applied at this
|
|
/// stage.</param>
|
|
/// <param name="invokedMessagesFilter">An optional filter function to apply to the invoked context messages before they are passed to the <see cref="ChatHistoryProvider"/>. If null, no
|
|
/// filter is applied at this stage.</param>
|
|
/// <returns>The <see cref="ChatHistoryProvider"/> with filtering applied.</returns>
|
|
public static ChatHistoryProvider WithMessageFilters(
|
|
this ChatHistoryProvider provider,
|
|
Func<IEnumerable<ChatMessage>, IEnumerable<ChatMessage>>? invokingMessagesFilter = null,
|
|
Func<ChatHistoryProvider.InvokedContext, ChatHistoryProvider.InvokedContext>? invokedMessagesFilter = null)
|
|
{
|
|
return new ChatHistoryProviderMessageFilter(
|
|
innerProvider: provider,
|
|
invokingMessagesFilter: invokingMessagesFilter,
|
|
invokedMessagesFilter: invokedMessagesFilter);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Decorates the provided <see cref="ChatHistoryProvider"/> so that it does not add
|
|
/// messages with <see cref="AgentRequestMessageSourceType.AIContextProvider"/> to chat history.
|
|
/// </summary>
|
|
/// <param name="provider">The <see cref="ChatHistoryProvider"/> to add the message filter to.</param>
|
|
/// <returns>A new <see cref="ChatHistoryProvider"/> instance that filters out <see cref="AIContextProvider"/> messages so they do not get added.</returns>
|
|
public static ChatHistoryProvider WithAIContextProviderMessageRemoval(this ChatHistoryProvider provider)
|
|
{
|
|
return new ChatHistoryProviderMessageFilter(
|
|
innerProvider: provider,
|
|
invokedMessagesFilter: (ctx) =>
|
|
{
|
|
ctx.RequestMessages = ctx.RequestMessages.Where(x => x.GetAgentRequestMessageSourceType() != AgentRequestMessageSourceType.AIContextProvider);
|
|
return ctx;
|
|
});
|
|
}
|
|
}
|