徐涛
5baa170508
- README 添加 feature 组合表 + 模块级 features 清单 + 升级指南 - 18 个 example 顶部添加 Required features 注释 - roadmap.md 和 roadmap-v0.3.2.md 同步 Phase 26-27 完成状态 - cargo fmt 全量格式化（修复预存格式问题，CI format job 可通过）
131 lines
4.2 KiB
Rust
131 lines
4.2 KiB
Rust
//! agent_session_demo —— Agent 装配 + 会话链路 + SessionMemory 桥接。
|
||
//! Required features: cargo run --example agent_session_demo --features "agent"
|
||
//!
|
||
//! 演示：
|
||
//! 1. 实现 `Agent` trait（定义角色 + system prompt）
|
||
//! 2. 用 `MockProvider` 预设响应（离线可跑）
|
||
//! 3. `AgentBuilder` 装配 `RuntimeBundle`
|
||
//! 4. `AgentSession::submit_turn` 跑多轮对话
|
||
//! 5. `SessionMemory` 读写 + snapshot 输出
|
||
//! 6. 跨 session 数据隔离验证
|
||
//!
|
||
//! 运行：`cargo run --example agent_session_demo`
|
||
|
||
use std::sync::Arc;
|
||
|
||
use agcore::agent::{Agent, AgentBuilder, AgentSession};
|
||
use agcore::llm::hooks::HookExecutor;
|
||
use agcore::llm::mock::MockProvider;
|
||
use agcore::llm::types::Usage;
|
||
use agcore::llm::types::message::{ContentBlock, Message};
|
||
use agcore::llm::types::response_v2::{MessageResponse, StopReason};
|
||
use agcore::tools::ToolRegistry;
|
||
|
||
/// 计算器角色 Agent。
|
||
struct CalculatorAgent;
|
||
|
||
impl Agent for CalculatorAgent {
|
||
fn name(&self) -> &str {
|
||
"calculator"
|
||
}
|
||
fn system_prompt(&self) -> Option<&str> {
|
||
Some("你是一个简洁的计算器助手，每轮回答一句话。")
|
||
}
|
||
}
|
||
|
||
/// 构造预设的纯文本 Assistant 响应。
|
||
fn assistant_text(text: &str) -> MessageResponse {
|
||
MessageResponse {
|
||
id: String::new(),
|
||
model: String::new(),
|
||
message: Message::Assistant {
|
||
content: vec![ContentBlock::Text { text: text.into() }],
|
||
},
|
||
usage: Usage::from_input_output(8, 4),
|
||
stop_reason: StopReason::Stop,
|
||
extra: Default::default(),
|
||
}
|
||
}
|
||
|
||
#[tokio::main]
|
||
async fn main() {
|
||
// 1. MockProvider：预设三轮响应（无须 API key 即可离线运行）
|
||
let provider = Arc::new(MockProvider::new(vec![
|
||
assistant_text("1 + 1 = 2"),
|
||
assistant_text("2 + 2 = 4"),
|
||
assistant_text("会话即将结束。"),
|
||
]));
|
||
|
||
// 2. AgentBuilder 装配 RuntimeBundle（必填：provider / tool_registry / hook_executor）
|
||
let bundle = Arc::new(
|
||
AgentBuilder::new()
|
||
.provider(provider)
|
||
.tool_registry(Arc::new(ToolRegistry::new()))
|
||
.hook_executor(Arc::new(HookExecutor::new()))
|
||
.build()
|
||
.expect("RuntimeBundle 装配失败"),
|
||
);
|
||
|
||
// 3. 创建会话
|
||
let agent: Arc<dyn Agent> = Arc::new(CalculatorAgent);
|
||
let mut session = AgentSession::new(agent, "demo-session", bundle.clone());
|
||
assert_eq!(session.turn_index(), 0);
|
||
|
||
// 4. 提交第一轮
|
||
println!("=== 提交第 1 轮 ===");
|
||
let resp = session
|
||
.submit_turn("1+1=?")
|
||
.await
|
||
.expect("submit_turn 失败");
|
||
println!("LLM: {}", resp.text());
|
||
session
|
||
.set_session_data("last_q", "1+1=?")
|
||
.await
|
||
.expect("set_session_data 失败");
|
||
session
|
||
.set_session_data("last_a", resp.text())
|
||
.await
|
||
.expect("set_session_data 失败");
|
||
assert_eq!(session.turn_index(), 1);
|
||
|
||
// 5. 提交第二轮
|
||
println!("\n=== 提交第 2 轮 ===");
|
||
let resp = session.submit_turn("再加一次 2+2=?").await.unwrap();
|
||
println!("LLM: {}", resp.text());
|
||
assert_eq!(session.turn_index(), 2);
|
||
|
||
// 6. 验证 SessionMemory 读取
|
||
println!("\n=== Session Memory 读取 ===");
|
||
println!(
|
||
"last_q = {:?}",
|
||
session.get_session_data("last_q").await.unwrap()
|
||
);
|
||
println!(
|
||
"last_a = {:?}",
|
||
session.get_session_data("last_a").await.unwrap()
|
||
);
|
||
|
||
// 7. Snapshot 格式化输出
|
||
println!("\n=== Session Memory Snapshot ===");
|
||
println!("{}", session.session_memory().snapshot().await.unwrap());
|
||
|
||
// 8. 跨 session 数据隔离验证
|
||
println!("=== 数据隔离验证 ===");
|
||
let other = AgentSession::new(Arc::new(CalculatorAgent), "other-session", bundle);
|
||
assert!(
|
||
other.get_session_data("last_q").await.unwrap().is_none(),
|
||
"新会话不应看到旧 session 的 last_q"
|
||
);
|
||
println!("新会话 last_q = None ✓");
|
||
|
||
// 9. 用量累计验证
|
||
println!("\n=== 用量累计 ===");
|
||
let total = session.usage().total();
|
||
println!(
|
||
"prompt={}, completion={}, total={}",
|
||
total.prompt_tokens, total.completion_tokens, total.total_tokens
|
||
);
|
||
|
||
println!("\n✓ agent_session_demo 完成");
|
||
}
|