惯性聚合 高效追踪和阅读你感兴趣的博客、新闻、科技资讯
阅读原文 在惯性聚合中打开

推荐订阅源

U
Unit 42
C
Cybersecurity and Infrastructure Security Agency CISA
Exploit-DB.com RSS Feed
Exploit-DB.com RSS Feed
Know Your Adversary
Know Your Adversary
S
Securelist
I
Intezer
AWS News Blog
AWS News Blog
L
LINUX DO - 热门话题
P
Privacy International News Feed
Recent Announcements
Recent Announcements
cs.CL updates on arXiv.org
cs.CL updates on arXiv.org
博客园 - 聂微东
Threat Intelligence Blog | Flashpoint
Threat Intelligence Blog | Flashpoint
Attack and Defense Labs
Attack and Defense Labs
N
News and Events Feed by Topic
The GitHub Blog
The GitHub Blog
C
Cyber Attacks, Cyber Crime and Cyber Security
Schneier on Security
Schneier on Security
N
Netflix TechBlog - Medium
爱范儿
爱范儿
B
Blog
CTFtime.org: upcoming CTF events
CTFtime.org: upcoming CTF events
cs.CV updates on arXiv.org
cs.CV updates on arXiv.org
C
CERT Recently Published Vulnerability Notes
Hacker News: Ask HN
Hacker News: Ask HN
Google DeepMind News
Google DeepMind News
Engineering at Meta
Engineering at Meta
Blog — PlanetScale
Blog — PlanetScale
WordPress大学
WordPress大学
S
Secure Thoughts
K
Kaspersky official blog
N
News | PayPal Newsroom
O
OpenAI News
Last Week in AI
Last Week in AI
C
Check Point Blog
D
Darknet – Hacking Tools, Hacker News & Cyber Security
Cyberwarzone
Cyberwarzone
Application and Cybersecurity Blog
Application and Cybersecurity Blog
T
Tor Project blog
大猫的无限游戏
大猫的无限游戏
Vercel News
Vercel News
D
Docker
Hugging Face - Blog
Hugging Face - Blog
T
Threat Research - Cisco Blogs
Cisco Talos Blog
Cisco Talos Blog
The Register - Security
The Register - Security
博客园 - 司徒正美
Martin Fowler
Martin Fowler
人人都是产品经理
人人都是产品经理
P
Palo Alto Networks Blog

博客园_首页

Plist 二进制格式 Milvus 和 PGVector,哪个更好? OpenClaw 已过时?在 VS Code 中运行 Hermes Agent! 第30篇文章:一个大三计科生的自白 Manim如何在数学公式中完美显示中文? Docker 部署 RocketMQ 5 并发编程核心概念辨析 C#事务处理最佳实践:别再让“主表存了、明细丢了”的破事发生 CLI 是什么?为什么大厂突然集体卷命令行? 【从0到1构建一个ClaudeAgent】协作-自主Agent UIImageView 设置图片不生效的原因排查 最小二乘问题详解20:无先验约束下的增量式SFM自由网平差 痞子衡嵌入式:大话双核i.MXRT1180之XIP应用里借助MU实现可靠Flash IAP的方法 AI Chat 封装, SemanticKerne.AiProvider.Unified 已发布 Windows下右键编辑js文件无法打开记事本——在注册表中使用环境变量 在后台服务中使用 Scoped 服务,为什么总是报错? H200 安装驱动并使用sglang启动模型 wireshark 抓包Trap上报告警内容 我用 AI 辅助开发了一系列小工具(2):图片压缩工具 [A Primer On MC and CC] 2.1 Memory Consistency 1 - 指令重排序和 SC 模型 Oracle数据库SCN推进技术详解与实践指南 玩转控件:封装个带图片的Label控件 Claude Code 4.7 真正该升级的不是模型,而是你的工作流 前端小白一句话,AI 帮我做了个颜值拉满的桌面媒体播放器。当代码不再是门槛,一句话编程就是现实。 5. WorkBuddy: 小龙虾的灵魂三件套,让你的小龙虾不只是工具 SQLite 分片方案实战:三种分片策略的深度对比 告别简陋 UI!一款基于 Fluent Design 和基于 WinUI 的开源免费、现代化的 Avalonia UI 控件库 关于二进制排列组合枚举的总结 AI开发-python-LangGraph框架(3-27-LangGraph从零实现大模型智能决策工作流) ElasticSearch主分片和副本分片概念详解 【002】HTTPS 粗解:证书、TLS 握手与对后端配置的影响 Hermes Agent 一周暴涨五万 Star,但我劝你别急着追 明明连接的是Redis的DB0,为什么能查到DB3的数据? 【从0到1构建一个ClaudeAgent】协作-Agent团队 熟悉电子元器件之后,电子小白下一步该怎么走? MAF快速入门(23)通过C#类定义Skills .NET 高级开发 | 手写一个对象映射框架 FastAPI数据库ORM怎么选?我肝了三个Demo后,终于不再纠结了 mysqldump 参数拾遗:在遗忘与铭记之间 C# .NET 周刊|2026年3月5期 Claude code入门 - 陈彦斌 一文学习入门 ThingsBoard 开源物联网平台 GitHub 热门项目 | 2026年04月16日 如何为GIT设置全局勾子,为每次提交追加信息 Number.isFinite和isFinite与isNaN()和Number.isNaN的区别 PortSwigger SQL注入LAB2 推荐一个测试人必备的Skills,从功能到性能全搞定(附详细实操和安装下载方式) 筑基期:掌握Odoo基础核心知识点02(Odoo XML 开发方式详解) GLM模型这么火,咱们用vllm也咧一个呗! 深入理解 AbortController:从底层原理到跨语言设计哲学 字符串学习笔记 多租户系统框架的基础模块设计和分析设计 Apache SeaTunnel Zeta 为什么能做到“又快又稳”? AI开发-python-LangGraph框架(3-26-LangGraph基本概念及第一个简单样例) Vue 3 组件通信,别只会用 Props 和 Emits 了,这几个狠活儿你得看看 ElasticSearch7.X版本配置密码 用Manim实现动态交点计算--从一个动点问题说起 团结引擎+Addressable+Instant Game打包抖音小游戏 function call 实战:让 LLM 自动判断 pod 异常、调用日志工具并完成故障分析 bubseek —— 让 Agent 的足迹,变成团队的洞察 通过 C# 读取并导出 PDF 书签 如何用 GitHub Actions 实现 Steam 自动化发布 【从0到1构建一个ClaudeAgent】并发-后台任务 .NET 高级开发 | 定制 ASP.NET Core 框架 电子小白:什么是运算放大器(运放) zero2Agent:面向大厂面试的 Agent 工程教程,从概念到生产的完整学习路线 堆上的ORW HC32F460 USB CDC通信异常:非对齐访问异常排查 20260413-Hyperbridge 攻击事件:发生在默克尔山上的验证绕过 那些喊着AI 要淘汰你的人,正在靠你的焦虑赚大钱! 深度学习进阶(八)Swin Transformer 最小二乘问题详解19:带先验约束的增量式SFM优化与实现 SnapTranslate 3.0 正式发布:全局划词翻译 + 完整英语学习闭环,一站式搞定查词、记词、复习 工作的意义、工作的困难认知再思考 .NET + AI 进阶实战:基于类的技能开发 - 打造可治理的 Agent 能力模块 【从0到1构建一个ClaudeAgent】规划与协调-技能 上周热点回顾(4.6-4.12) 电子小白的工具三件套:面包板、杜邦线、万能板 单表五亿数据的查询优化 | Mysql、StarRocks 2. WorkBuddy:从“我是谁”到“帮我干活” C# 如何减少代码运行时间:7 个实战技巧 基于HelixToolkit.SharpDX 渲染3D模型 - 笺上知微 从零开始的双臂具身VLA起源及现阶段发展综述 - SkyXZ 记对 xonsh shell 的使用, 脚本编写, 迁移及调优 - pluvium27 受够了Vibe Coding的失控?换个起点,让AI事半功倍 从开始配置漏洞环境到漏洞复现流程 - 難しい 关于10年工作经验的程序员对OpenClaw的实战经验分享以及看法 - 虚无境 Any metadata 的内存布局 C# .NET 周刊|2026年3月2期 - InCerry 我帮你测过了,测试圈排名第二的 Skill 依然很牛逼 Skill Discovery | 无监督技能发现的经典工作总结 - MoonOut 上下文工程是什么?过时了么?一文讲明白! - 一枫说码 开了 TUN 模式还是直连?90% 的人都踩过这个坑 AScript扩展多种脚本语言 - rockey627 AI 学习笔记:Agent 的记忆机制 你能被装进一个文件里吗?——7 万人把同事"蒸馏"成了 AI - 我没有三颗心脏 Claude Code 通关手册(七):给 AI 装上技能包——Skills 完全指南 - 暮色之狐 在浏览器中快速编辑代码:VSCode Web 集成实践 - Newbe36524 蒸馏自己 skill?基于 Deepseek 的蒸馏器,丐版蒸馏方式,简单便捷 - To_Carpe_Diem Spring AI Aliababa和AgentScope,哪个更好? - 苏三说技术
[MAF预定义的AIContextProvider-03]ChatHistoryMemoryProvider——赋予Agent从经验中学习的能力
Artech · 2026-06-20 · via 博客园_首页

LLM具有固化的知识,而且针对LLM的调用是完全无状态,永远只做一锤子买卖。但是交给Agent的任务基本上不可能一蹴而就,而且还希望Agent具有学习进化的能力。所以你会发现,很多的Harness手段的目的就是为了弥合两者之间的鸿沟。解决这个问题的基本的前提是:需要赋予Agent记忆。短期记忆赋予Agent在同一个语境下进行多轮对话的能力,对于MAF来说,就是Session。长期记忆实现了跨Session的信息共享,其共享范围可以针对用户(比如了解用户的偏好)、针对Agent(比如了解Agent的能力和经验)或者针对整个系统(比如了解系统的环境和历史)。作为AIContextProvider的派生类,ChatHistoryMemoryProvider使我们可以基于不同的Scope将对话历史记录下来,并在后续的对话中将这些历史记录作为上下文提供给Agent,从而赋予Agent从经验中学习的能力。

1. 让Agent记住用户的偏好

在正式介绍ChatHistoryMemoryProvider的设计和实现原理之前,我们先来看一个简单的例子,来感受一下ChatHistoryMemoryProvider的作用。我们利用如下的代码创建了一个用来帮助我们点外卖的Agent。注册的两个工具GetMenuPlaceOrder分别用来获取菜单和下单。我们赋予Agent自主选择菜品和直接下单的权力,所以我们希望Agent能够在为不同用户提供服务的过程中,能够记住每个用户的口味偏好,并且在后续的点餐过程中能够基于这些偏好来推荐菜品。

using Azure;
using dotenv.net;
using Microsoft.Agents.AI;
using Microsoft.Extensions.AI;
using Microsoft.SemanticKernel.Connectors.InMemory;
using OpenAI;
using System.ComponentModel;

DotEnv.Load();

var model = Environment.GetEnvironmentVariable("MODEL")!;
var apiKey = Environment.GetEnvironmentVariable("API_KEY")!;
var endpoint = Environment.GetEnvironmentVariable("OPENAI_URL")!;

var openAIClient= new OpenAIClient(
     credential: new AzureKeyCredential(apiKey),
     options: new OpenAIClientOptions { Endpoint = new Uri(endpoint) });

var embeddingGenerator = openAIClient
    .GetEmbeddingClient(model: "text-embedding-3-small")
    .AsIEmbeddingGenerator()
    .AsBuilder()
    .Use((texts, options, next, ct) =>
    {
        var sanitizedTexts = texts.Select(t => string.IsNullOrWhiteSpace(t) ? " " : t);
        return next.GenerateAsync(sanitizedTexts, options, ct);
    })
    .Build(); ;

var memoryProvider = new ChatHistoryMemoryProvider(
    vectorStore: new InMemoryVectorStore(options: new InMemoryVectorStoreOptions { EmbeddingGenerator = embeddingGenerator }),
    collectionName: "user_preference",
    vectorDimensions: 1536,
    stateInitializer: InitializeMemoryState);

AITool[] tools = [AIFunctionFactory.Create(GetMenu, "GetMenu"), AIFunctionFactory.Create(PlaceOrder, "PlaceOrder")];
var agent = openAIClient
    .GetChatClient(model:model)
    .AsIChatClient()
    .AsAIAgent(options: new ChatClientAgentOptions
    {
        Name= "delivery-order",
        AIContextProviders = [memoryProvider],       
        ChatOptions = new ChatOptions {
            Instructions = """
            你是一个贴心外卖点餐助手。
            用户授权自己选择菜品和数量的权力,无需用户确认。但下单数量务必控制在三份以内。
            点餐时既要考虑用户的口味偏好,也要考虑菜品多样性,以及尽可能与上次订餐有所不同。
            """,
            Tools = tools }
    } );

var session = await agent.CreateSessionAsync();
session.StateBag.SetValue("user_id", "Alice");
var response = await agent.RunAsync("帮我点一份外卖,一荤一素,我不能吃辣",session);
Console.WriteLine($"""
        {new string('-', 30)} 来自 Alice 的订单 {new string('-', 30)}
        {response}

        """);

await OrderDelivery("Alice");
await OrderDelivery("Alice");
await OrderDelivery("Bob");
await OrderDelivery("Bob");

async Task OrderDelivery(string userName)
{
    var session = await agent.CreateSessionAsync();
    session.StateBag.SetValue("user_id", userName);
    var response = await agent.RunAsync("帮我点一份外卖", session);
    Console.WriteLine($"""
        {new string('-', 30)} 来自 {userName} 的订单 {new string('-', 30)}
        {response}

        """);
}

static ChatHistoryMemoryProvider.State InitializeMemoryState(AgentSession? session)
{
    if (session is not ChatClientAgentSession chatSession)
    {
        throw new InvalidOperationException("Session is not of type ChatClientAgentSession.");
    }

    if (chatSession.StateBag?.TryGetValue<string>("user_id", out var userId) != true)
    {
        throw new InvalidOperationException("User ID not found in session state.");
    }

    var scope = new ChatHistoryMemoryProviderScope { UserId = userId };
    return new ChatHistoryMemoryProvider.State(storageScope: scope,searchScope: scope);
}


[Description("提取外卖菜单")]
static string[] GetMenu()
{
    return ["辣椒炒肉", "剁椒鱼头", "番茄炒蛋", "清蒸鲈鱼", "清炒菜心","酸辣土豆丝","西芹百合"];
}

[Description("外卖下单")]
static IReadOnlyList<OrderItem> PlaceOrder(params KeyValuePair<string, int>[] orderItems)
{
    return [.. orderItems.Select(item => new OrderItem(item.Key, item.Value))];
}

public readonly record struct OrderItem(string DishName, int Quantity);

ChatHistoryMemoryProvider利用一个向量数据库来存储对话历史,简单起见,我们使用了一个基于内存的向量数据库InMemoryVectorStoreInMemoryVectorStore需要一个IEmbeddingGenerator对象为输入的文本生成嵌入向量,所以我们调用OpenAIClientGetEmbeddingClient方法得到了一个针对"text-embedding-3-small"模型的EmbeddingClient,并将其转换成了IEmbeddingGenerator对象。创建ChatHistoryMemoryProvider除了提供作为存储的向量数据库之外,我们还提供了存储记忆的集合名称(user_preference)和嵌入向量的维度(1536)。

ChatHistoryMemoryProvider存储的记忆来源于对话历史,但是需要针对需要以不同的Scope进行存储。由于我们希望利用记忆了解用户的口味偏好,所以我们选择了以用户为Scope来存储记忆,并要求Session中必需包含用户的ID。创建ChatHistoryMemoryProvider提供的第四个参数是一个StateInitializer类型的委托,它的作用就是根据当前的Session来初始化作为状态的ChatHistoryMemoryProvider.State对象,而ChatHistoryMemoryProvider.State对象中包含了检索和存储记忆所需要的Scope信息。该参数对应的方法为InitializeMemoryState,我们将两种Scope都设置为UserId。

我们通过注册两个工具(GetMenuPlaceOrder)和ChatHistoryMemoryProvider将Agent创建出来后,就可以开始点餐了。我们定义了辅助方法OrderDelivery来为指定的用户点餐,并且每次调用都创建了一个新的Session来屏蔽短期记忆的干扰。我们第一次直接调用Agent以Alice的名义点餐,并且告诉Agent:“帮我点一份外卖,一荤一素,我不能吃辣”。后续则通过调用OrderDelivery方法来为Alice和Bob点餐。整个程序会生成如下的输出:

------------------------------ 来自 Alice 的订单 ------------------------------
已经帮您下单 ✅

🥩 荤菜:清蒸鲈鱼 ×1(清淡不辣,营养又鲜美)
🥬 素菜:西芹百合 ×1(清爽可口,搭配均衡)

这份搭配清淡健康、完全不辣,也兼顾营养和口味层次。祝您用餐愉快 🍽️😊

------------------------------ 来自 Alice 的订单 ------------------------------
已经帮您下单 ✅

🍅 荤菜:番茄炒蛋 ×1(酸甜开胃,不辣又下饭)
🥬 素菜:清炒菜心 ×1(清爽鲜嫩,清淡健康)

这次给您换了搭配,依旧保持不辣口味,同时清爽均衡。祝您用餐愉快 🍽️😊

------------------------------ 来自 Alice 的订单 ------------------------------
已经帮您下单 ✅

🍅 荤菜:番茄炒蛋 ×1(酸甜开胃,不辣又下饭)
🥬 素菜:清炒菜心 ×1(清爽脆嫩,营养均衡)

这次给您换了搭配,依旧是一荤一素、清淡不辣,也和上次有所不同。祝您用餐愉快 🍽️😊

------------------------------ 来自 Bob 的订单 ------------------------------
已经帮您下单啦 ✅

🛒 本次为您搭配了:
- 辣椒炒肉 ×1(香辣下饭)
- 清蒸鲈鱼 ×1(清淡鲜美,营养均衡)
- 清炒菜心 ×1(清爽解腻)

荤素搭配、口味有层次,吃起来不会腻~
祝您用餐愉快 🍚😋

------------------------------ 来自 Bob 的订单 ------------------------------
已经帮您下单啦 ✅

🛒 本次为您搭配了:
- 剁椒鱼头 ×1(鲜辣开胃,和上次不同口味)
- 西芹百合 ×1(清爽脆口,解腻健康)
- 番茄炒蛋 ×1(酸甜开胃,经典下饭)

这次偏鲜辣+清爽搭配,和上次的清蒸鲈鱼、辣椒炒肉组合有所变化,口味更有层次~

祝您用餐愉快 🍚😋

上面的输出显示了5次点餐的结果。可以看到,Alice的三次点餐都没有辣椒炒肉和剁椒鱼头,因为她不能吃辣;Bob的两次点餐都包含了辣椒炒肉和剁椒鱼头,因为对话中并没有涉及Bob的口味偏好。但是这个Agent还是有一个问题,我通过系统指令说明尽量尽可能与上次订餐有所不同,但是在实际输出中,Alice的三次点餐中有两次的搭配是相同的,我将会在后面介绍背后的原因和改进方案。

2. 检索和存储的Scope

ChatHistoryMemoryProvider从本质上也属于RAG的一种实现方式,它根据当前对话从向量数据库中检索相关的历史对话记录,并将这些记录作为上下文提供给LLM来生成响应。这里存储的对话历史可不是针对某个Session,甚至不是某个Agent(多Agent可共享同一个向量数据库),所以在检索的时候必需指定适当的Scope以保证检索结果的相关性。既然检索的时候需要限定范围,存储的时候自然也需要指定针对Scope的维度,而且还得保证检索的Scope维度是存储Scope的子集。

ChatHistoryMemoryProvider将所需的检索和存储Scope封装到内嵌的State类中。所谓的Scope通过ChatHistoryMemoryProviderScope类表示,四个属性ApplicationIdAgentIdSessionIdUserId分别代表了Scope针对应用、Agent、Session和用户的四个维度。

public sealed class ChatHistoryMemoryProvider : MessageAIContextProvider, IDisposable
{
	public sealed class State
	{
		public ChatHistoryMemoryProviderScope StorageScope { get; }
		public ChatHistoryMemoryProviderScope SearchScope { get; }
		public State(
            ChatHistoryMemoryProviderScope storageScope, 
            ChatHistoryMemoryProviderScope? searchScope = null);
	}

}
public sealed class ChatHistoryMemoryProviderScope
{
    public string? ApplicationId { get; set; }
    public string? AgentId { get; set; }
    public string? SessionId { get; set; }
    public string? UserId { get; set; }
}

Socpe四个维度的值来源于Session状态,构造ChatHistoryMemoryProvider的时候需要提供一个StateInitializer类型的委托来根据当前的Session来初始化State对象。构造函数会根据此委托对象,结合由ChatHistoryMemoryProviderOptions指定的键(如果没有指定,则使用ChatHistoryMemoryProvider的类名)创建一个ProviderSessionState<State>对象在Session中维护这个State对象。对于上面点餐的例子来说,我们将UserId作为Scope的维度,所以StateInitializer委托对象的实现逻辑就是从Session状态中获取UserId的值,并将其分别赋值给StorageScopeSearchScopeUserId属性。

public sealed class ChatHistoryMemoryProvider : MessageAIContextProvider, IDisposable
{
    private readonly ProviderSessionState<State> _sessionState;
    private IReadOnlyList<string>? _stateKeys;
    public override IReadOnlyList<string> StateKeys => _stateKeys ??= [_sessionState.StateKey];

    public ChatHistoryMemoryProvider(
        VectorStore vectorStore,
        string collectionName,
        int vectorDimensions,
        Func<AgentSession?, State> stateInitializer,
        ChatHistoryMemoryProviderOptions? options = null,
        ILoggerFactory? loggerFactory = null)
     : base(options?.SearchInputMessageFilter, options?.StorageInputRequestMessageFilter, options?.StorageInputResponseMessageFilter)
    {
         _sessionState = new ProviderSessionState<State>(
            stateInitializer,
            options?.StateKey ?? GetType().Name,
            AgentJsonUtilities.DefaultOptions);
    }    
}

3. ChatHistoryMemoryProviderOptions

由于ChatHistoryMemoryProvider本质上也是RAG的一种实现方式,所以其实现原理与TextSearchProvider非常类似,它们的配置选项类型也有很多共同的成员。ChatHistoryMemoryProvider的上下文检索方法也通过ChatHistoryMemoryProviderOptionsSearchTime属性分两种:一种在调用LLM之前,主动检索对应Scope维度历史对话记录并将这些记录作为上下文提供给LLM;另一种是注册检索工具,并利用工具描述和系统提示词指导LLM在最终作答之前调用它来获取上下文信息。对于第二种SearchTime的取值,我们需要提供FunctionToolNameFunctionToolDescription属性来指定供LLM调用的工具的名称和描述信息。当检索内容被用作LLM上下文时,ContextPrompt属性可以用来指定对应提示词的前缀。

public sealed class ChatHistoryMemoryProviderOptions
{
	public enum SearchBehavior
	{
		BeforeAIInvoke,
		OnDemandFunctionCalling
	}
	public SearchBehavior SearchTime { get; set; } = SearchBehavior.BeforeAIInvoke;
	public string? FunctionToolName { get; set; }
	public string? FunctionToolDescription { get; set; }
    public string? ContextPrompt { get; set; }
}

SearchTime的默认值为BeforeAIInvoke。如果没有对FunctionToolNameFunctionToolDescriptionContextPrompt进行配置,ChatHistoryMemoryProvider会使用如下的默认值:

  • FunctionToolName: Search
  • FunctionToolDescription: Allows searching for related previous chat history to help answer the user question.
  • ContextPrompt: ## Memories\nConsider the following memories when answering user questions:

ChatHistoryMemoryProviderOptions还提供了如下的配置选项, MaxResults用来指定每次检索返回的最大结果数量;EnableSensitiveTelemetryDataRedactor用来控制是否以及如何对包含敏感数据的消息进行脱敏处理;StateKey用来指定在Session中维护ChatHistoryMemoryProvider.State对象所使用的键;SearchInputMessageFilterStorageInputRequestMessageFilterStorageInputResponseMessageFilter分别用来指定在检索输入消息、存储请求消息和存储响应消息时需要应用的过滤器。

public sealed class ChatHistoryMemoryProviderOptions
{	
	public int? MaxResults { get; set; }
	public bool EnableSensitiveTelemetryData { get; set; }
	public Redactor? Redactor { get; set; }
	public string? StateKey { get; set; }

	public Func<IEnumerable<ChatMessage>, IEnumerable<ChatMessage>>? SearchInputMessageFilter { get; set; }
	public Func<IEnumerable<ChatMessage>, IEnumerable<ChatMessage>>? StorageInputRequestMessageFilter { get; set; }
	public Func<IEnumerable<ChatMessage>, IEnumerable<ChatMessage>>? StorageInputResponseMessageFilter { get; set; }
}

4. 基于对话历史的检索

ChatHistoryMemoryProviderTextSearchProvider针对上下文的检索基本一致,甚至还要简单一些,因为TextSearchProvider用于检索的查询文本来源于整个Session的对话历史,但是ChatHistoryMemoryProvider用于检索的查询文本来源于当前输入消息。具体的实现体现在重写的ProvideAIContextAsyncProvideMessagesAsync方法中。具体的检索实现在私有方法SearchTextAsync中,userQuestion参数由LLM提供,SearchScope参数来源于State对象,SearchTextAsync方法会根据这两个参数检索向量数据库,最后将检索到的结果格式化成返回的文本。

public sealed class ChatHistoryMemoryProvider : MessageAIContextProvider, IDisposable
{
    protected override async ValueTask<AIContext> ProvideAIContextAsync(
        AIContextProvider.InvokingContext context, 
        CancellationToken cancellationToken = default);
    protected override async ValueTask<IEnumerable<ChatMessage>> ProvideMessagesAsync(
        InvokingContext context, 
        CancellationToken cancellationToken = default);
    private async Task<string> SearchTextAsync(
        string userQuestion, 
        ChatHistoryMemoryProviderScope searchScope, 
        CancellationToken cancellationToken = default)
}

如果SearchTime被设置成OnDemandFunctionCalling,它会将上下文检索逻辑实现在一个工具中,并封装在返回的AIConetxt中。如果SearchTime被设置成BeforeAIInvoke,它会在ProvideMessagesAsync方法中直接进行上下文检索,具体的流程为:

  • 从Session状态中获取State对象,并从State对象中获取SearchScope;
  • 从当前输入的一个或者多个消息中提取文本内容,并拼接成查询文本;
  • 调用SearchTextAsync方法进行检索,获取检索结果文本;
  • 将检索结果文本与ContextPrompt进行拼接,构造成一条新的ChatMessage,它将被添加到当前的消息列表中,并作为上下文提供给LLM。

5. 查看工具和生成的消息

ChatHistoryMemoryProvider注册的工具和根据检索内容生成的ChatMessage是整个ChatHistoryMemoryProvider的核心产出,所以我们有必要来看看它会生成一个怎样的工具,以及根据检索内容生成的ChatMessage具有怎样的内容。为此我们定义了如下这个AIContextTrackingProvider,如代码所示:这是一个自定义的AIContextProvider类,它重写了InvokingCoreAsync方法,在方法中我们打印了当前调用的AIContext中的消息列表和工具列表。

class AIContextTrackingProvider : AIContextProvider
{
    protected override ValueTask<AIContext> InvokingCoreAsync(InvokingContext context, CancellationToken cancellationToken = default)
    {
        var index = 1;
        foreach (var message in context.AIContext?.Messages!)
        {
            Console.WriteLine($"""
               {new string('-', 20)} Message {index++} {new string('-', 20)}
               Role: {message.Role}
               Text:                

               {message.Text}
               """);
        }

        var function = context.AIContext?.Tools?.SingleOrDefault(tool => tool.Name == "Search") as AIFunction;
        if (function is not null)
        {
            Console.WriteLine($"""

                {new string('-', 20)} Tool `{function.Name}` {new string('-', 20)}
                Description: 
                {function.Description}

                JsonSchema: 
                {JsonSerializer.Serialize(function.JsonSchema, new JsonSerializerOptions { WriteIndented = true })}
                """);
        }

        Console.WriteLine("\n\n");
        return base.InvokingCoreAsync(context, cancellationToken);
    }
}

5.1 查看根据检索结果生成的消息

我们将这个AIContextTrackingProvider应用到我们开篇演示的实例中。如代码所示,我们在创建Agent的时候将AIContextTrackingProvider添加到AIContextProviders列表中,并放在ChatHistoryMemoryProvider的后面,这样就可以保证在Agent调用LLM之前,AIContextTrackingProvider能够看到ChatHistoryMemoryProvider添加的消息和工具了。然后我们针对同一个用户Alice,采用不同的Session两次调用Agent来点餐。

...
var memoryProvider = new ChatHistoryMemoryProvider(
    vectorStore: new InMemoryVectorStore(options: new InMemoryVectorStoreOptions { EmbeddingGenerator = embeddingGenerator }),
    collectionName: "chat_history_memory",
    vectorDimensions: 1536,
    stateInitializer: InitializeMemoryState);
var trackingProvider = new AIContextTrackingProvider();

AITool[] tools = [AIFunctionFactory.Create(GetMenu, "GetMenu"), AIFunctionFactory.Create(PlaceOrder, "PlaceOrder")];
var agent = openAIClient
    .GetChatClient(model:model)
    .AsIChatClient()
    .AsAIAgent(options: new ChatClientAgentOptions
    {
        Name= "delivery-order",
        AIContextProviders = [memoryProvider, trackingProvider],       
        ChatOptions = new ChatOptions {
            Instructions = """
            你是一个贴心外卖点餐助手。
            用户授权自己选择菜品和数量的权力,无需用户确认。但下单数量务必控制在三份以内。
            点餐时既要考虑用户的口味偏好,也要考虑菜品多样性,以及尽可能与上次订餐有所不同。
            """,
            Tools = tools }
    } );

var session = await agent.CreateSessionAsync();
session.StateBag.SetValue("user_id", "Alice");
await agent.RunAsync("帮我点一份外卖,一荤一素,我不能吃辣",session);

session = await agent.CreateSessionAsync();
session.StateBag.SetValue("user_id", "Alice");
await agent.RunAsync("帮我点一份外卖", session);
...

输出:

-------------------- Message 1 --------------------
Role: user
Text:

帮我点一份外卖,一荤一素,我不能吃辣

-------------------- Message 1 --------------------
Role: user
Text:

帮我点一份外卖
-------------------- Message 2 --------------------
Role: user
Text:

## Memories
Consider the following memories when answering user questions:
帮我点一份外卖,一荤一素,我不能吃辣
已经帮您下单 ✅

🧂 荤菜:清蒸鲈鱼 ×1(清淡不辣)
🥬 素菜:清炒菜心 ×1(爽口清香)

整体搭配清淡健康,完全不辣,营养也比较均衡。
祝您用餐愉快 🍽️

从输出结果可以看出,第二次调用会出现两个ChatMessage,第二个ChatMessage的内容正好包含第一次调用的请求和响应。

5.2 查看注册的工具

为了查看ChatHistoryMemoryProvider注册的工具,我们需要将SearchTime设置成OnDemandFunctionCalling,这样它才会将检索逻辑封装成一个工具来供LLM调用。我们在创建ChatHistoryMemoryProvider的时候通过ChatHistoryMemoryProviderOptions来设置这个选项,如代码所示:

var memoryProvider = new ChatHistoryMemoryProvider(
    vectorStore: new InMemoryVectorStore(options: new InMemoryVectorStoreOptions { 
        EmbeddingGenerator = embeddingGenerator }),
    collectionName: "chat_history_memory",
    vectorDimensions: 1536,
    stateInitializer: InitializeMemoryState,
    options: new ChatHistoryMemoryProviderOptions { 
        SearchTime = ChatHistoryMemoryProviderOptions.SearchBehavior.OnDemandFunctionCalling }
);

输出:

-------------------- Message 1 --------------------
Role: user
Text:

帮我点一份外卖,一荤一素,我不能吃辣

-------------------- Tool `Search` --------------------
Description:
Allows searching for related previous chat history to help answer the user question.

JsonSchema:
{
  "type": "object",
  "properties": {
    "userQuestion": {
      "type": "string"
    }
  },
  "required": [
    "userQuestion"
  ]
}

从输出结果可以看出,ChatHistoryMemoryProvider注册了一个名为Search的工具,工具函数具有一个字符串类型的参数userQuestion,表示作为上下文检索查询文本的用户问题。

6. 对话实例存储

ChatHistoryMemoryProvider针对对话历史存储的实现在重写的StoreAIContextAsync方法中。它的实现流程为:从InvokedContext提取出本轮调用新产生的请求和响应消息,并提取它们的文本内容并拼接成一个字符串作为存储文本;从Session状态中获取State对象,并从State对象中获取StorageScope;然后针对StorageScope的维度对生成的文本内容进行存储即可。

public sealed class ChatHistoryMemoryProvider : MessageAIContextProvider, IDisposable
{
    protected override async ValueTask StoreAIContextAsync(
        InvokedContext context, 
        CancellationToken cancellationToken = default)
}

7. 如何解决订单重复问题?

在上面的点餐例子中,我们利用ChatHistoryMemoryProvider赋予了Agent记忆用户口味偏好的能力,从而使得Agent能够基于这些偏好来推荐菜品。但是我们发现系统指令尽可能与上次订餐有所不同,并没有生效。在了解了ChatHistoryMemoryProvider的实现原理之后,我们可以回答这个问题了:虽然ChatHistoryMemoryProvider存储了对话历史,但是它进行上下文检索的时候总是使用当前输入消息中的文本作为查询文本来进行检索的,所以检索过程并不会考虑系统指令的存在。我们看看将这个指令添加到请求消息中会不会有帮助。

为了验证这个想法,我们创建了一个新的AdditionalMessageProvider,它会在每次调用时都往上下文中添加一条新的消息,我们将系统指令作为这条消息的内容,这样就可以保证系统指令会被ChatHistoryMemoryProvider用来进行上下文检索了。如下代码所示:

class AdditionalMessageProvider(string content) : MessageAIContextProvider
{
    protected override ValueTask<IEnumerable<ChatMessage>> ProvideMessagesAsync(InvokingContext context, CancellationToken cancellationToken = default)
    {
        var aiContent = new TextContent(content);
        return ValueTask.FromResult<IEnumerable<ChatMessage>>([new ChatMessage { Role =  ChatRole.User, Contents = [aiContent] }]);
    }
}

然后我们将这个AdditionalMessageProvider添加到Agent的AIContextProviders列表中,并放在ChatHistoryMemoryProvider的后面,这样就可以保证在Agent调用LLM之前,ChatHistoryMemoryProvider能够看到这个额外的消息了。如下代码所示:

var agent = openAIClient
    .GetChatClient(model:model)
    .AsIChatClient()
    .AsAIAgent(options: new ChatClientAgentOptions
    {
        Name= "delivery-order",
        AIContextProviders = [
            memoryProvider, 
            new AdditionalMessageProvider("点餐时既要考虑用户的口味偏好,也要考虑菜品多样性,以及尽可能与上次订餐有所不同")],       
        ChatOptions = new ChatOptions {
            Instructions = """
            你是一个贴心外卖点餐助手。
            用户授权自己选择菜品和数量的权力,无需用户确认。但下单数量务必控制在三份以内。
            点餐时既要考虑用户的口味偏好,也要考虑菜品多样性,以及尽可能与上次订餐有所不同。
            """,
            Tools = tools }
    } );

var session = await agent.CreateSessionAsync();
session.StateBag.SetValue("user_id", "Alice");
var response = await agent.RunAsync("帮我点一份外卖,一荤一素,我不能吃辣",session);
Console.WriteLine($"""
        {new string('-', 30)} 来自 "Alice" 的订单 {new string('-', 30)}
        {response}

        """);

await OrderDelivery("Alice");
await OrderDelivery("Alice");
await OrderDelivery("Alice");
await OrderDelivery("Alice");

输出:

------------------------------ 来自 "Alice" 的订单 ------------------------------
已经帮您搭配好啦 ✅

🥢 **清蒸鲈鱼** ×1(荤菜,清淡不辣,鲜嫩爽口)
🥬 **西芹百合** ×1(素菜,清爽解腻,口感清甜)

整体口味清淡、不含辣椒,营养搭配均衡,一荤一素刚刚好 👍
祝您用餐愉快!如果下次想换换口味,也可以告诉我偏好~

------------------------------ 来自 Alice 的订单 ------------------------------
已经帮您重新搭配好啦 ✅(避开上次的清蒸鲈鱼和西芹百合,也不含辣)

🍅 **番茄炒蛋** ×1(酸甜开胃,经典家常荤素结合)
🥬 **清炒菜心** ×1(清爽解腻,口感脆嫩)

这次换了不同搭配,依然清淡不辣,营养均衡 👍
祝您用餐愉快!下次想试试别的风味也可以告诉我~

------------------------------ 来自 Alice 的订单 ------------------------------
已经为您搭配好新的餐品啦 ✅(这次和上次不同,更换了菜品,保证多样性)

🍅 **番茄炒蛋** ×1(酸甜开胃,家常经典,不辣)
🥬 **清炒菜心** ×1(清爽脆嫩,清淡健康)

整体依然是不辣搭配,清爽不油腻,同时和上次的鲈鱼、西芹百合不同,换了新口味 👍

祝您用餐愉快!下次如果想吃肉类或者想要更丰富一点,也可以告诉我~

------------------------------ 来自 Alice 的订单 ------------------------------
已为你下单 ✅

🥢 清蒸鲈鱼 ×1(清淡鲜美,不辣,优质蛋白)
🥬 西芹百合 ×1(清爽解腻,搭配均衡)

考虑到你不能吃辣,这次特意避开了辣椒类和酸辣口味;同时选择了一荤一素,营养搭配也比较清爽,不会太油腻。

祝你用餐愉快 🍽️

------------------------------ 来自 Alice 的订单 ------------------------------
已经帮您下单完成 ✅

本次为您搭配了三道菜,兼顾口味和营养,也保证菜品多样性:
- 🌶️ **剁椒鱼头** ×1(鲜辣开胃,主菜担当)
- 🥬 **清炒菜心** ×1(清爽解腻,均衡搭配)
- 🍅 **番茄炒蛋** ×1(经典家常,酸甜适口)

祝您用餐愉快!如果有特别想吃的口味(比如清淡一点、重辣一点或者想吃肉类为主),下次告诉我,我会帮您更精准搭配~ 🍽️

从输出结果可以看出,这样的解决方案还是有效果的。但是由于它无法精确地定位上次点餐,依然还是会出现两次点餐内容相同的情况。