mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
westey
b12ff578af
* Add a StateBag to AgentSession and pass Agent and AgentSession to AIContextProvider and ChatHistoryProviders * Convert all AIContextProviders to use the statebag * Update InMemoryChatHistoryProvider to use StateBag * Update Comsos and Workflow ChatHistoryProviders * Update 3rd party chat history storage sample. * Remove serialize method from providers * Replacing provider factories with properties * Remove Providers from Session and flatten state bag serialization * Update samples to use getservice on agent * Updated additional session types to serialize statebag * Fix regression * Address PR comments * Address PR comments. * Fix formatting * Fix unit tests * Remove InMemoryAgentSession since it is not required anymore. * Address PR comments * Convert sessions for A2AAgent, ChatClientAgent, CopilotStudioAgent and GithubCopilotAgent to use regular json serialization. * Fix durable agent session jso usgae * Add jso to InMemory and Workflow ChatHistoryProviders * Update InMemoryChatHistoryProvider to use an options class for it's many optional settings. * Apply suggestions from code review Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> * Address PR feedback * Fix verification bug. * Improve state bag thread safety * Address PR comments and fix unit tests * Address PR comments * Fix unit test --------- Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
53 lines
2.8 KiB
C#
53 lines
2.8 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
// This sample shows how to use a chat history reducer to keep the context within model size limits.
|
|
// Any implementation of Microsoft.Extensions.AI.IChatReducer can be used to customize how the chat history is reduced.
|
|
// NOTE: this feature is only supported where the chat history is stored locally, such as with OpenAI Chat Completion.
|
|
// Where the chat history is stored server side, such as with Azure Foundry Agents, the service must manage the chat history size.
|
|
|
|
using Azure.AI.OpenAI;
|
|
using Azure.Identity;
|
|
using Microsoft.Agents.AI;
|
|
using Microsoft.Extensions.AI;
|
|
using OpenAI.Chat;
|
|
using ChatMessage = Microsoft.Extensions.AI.ChatMessage;
|
|
|
|
var endpoint = Environment.GetEnvironmentVariable("AZURE_OPENAI_ENDPOINT") ?? throw new InvalidOperationException("AZURE_OPENAI_ENDPOINT is not set.");
|
|
var deploymentName = Environment.GetEnvironmentVariable("AZURE_OPENAI_DEPLOYMENT_NAME") ?? "gpt-4o-mini";
|
|
|
|
// Construct the agent, and provide a factory to create an in-memory chat message store with a reducer that keeps only the last 2 non-system messages.
|
|
AIAgent agent = new AzureOpenAIClient(
|
|
new Uri(endpoint),
|
|
new AzureCliCredential())
|
|
.GetChatClient(deploymentName)
|
|
.AsAIAgent(new ChatClientAgentOptions
|
|
{
|
|
ChatOptions = new() { Instructions = "You are good at telling jokes." },
|
|
Name = "Joker",
|
|
ChatHistoryProvider = new InMemoryChatHistoryProvider(new() { ChatReducer = new MessageCountingChatReducer(2) })
|
|
});
|
|
|
|
AgentSession session = await agent.CreateSessionAsync();
|
|
|
|
// Invoke the agent and output the text result.
|
|
Console.WriteLine(await agent.RunAsync("Tell me a joke about a pirate.", session));
|
|
|
|
// Get the chat history to see how many messages are stored.
|
|
// We can use the ChatHistoryProvider, that is also used by the agent, to read the
|
|
// chat history from the session state, and see how the reducer is affecting the stored messages.
|
|
var provider = agent.GetService<InMemoryChatHistoryProvider>();
|
|
List<ChatMessage>? chatHistory = provider?.GetMessages(session);
|
|
Console.WriteLine($"\nChat history has {chatHistory?.Count} messages.\n");
|
|
|
|
// Invoke the agent a few more times.
|
|
Console.WriteLine(await agent.RunAsync("Tell me a joke about a robot.", session));
|
|
Console.WriteLine($"\nChat history has {chatHistory?.Count} messages.\n");
|
|
Console.WriteLine(await agent.RunAsync("Tell me a joke about a lemur.", session));
|
|
Console.WriteLine($"\nChat history has {chatHistory?.Count} messages.\n");
|
|
|
|
// At this point, the chat history has exceeded the limit and the original message will not exist anymore,
|
|
// so asking a follow up question about it will not work as expected.
|
|
Console.WriteLine(await agent.RunAsync("Tell me the joke about the pirate again, but add emojis and use the voice of a parrot.", session));
|
|
|
|
Console.WriteLine($"\nChat history has {chatHistory?.Count} messages.\n");
|