徐涛
932a06f512
- 定义 16 个 features + 4 个快捷组合，default = ["full"] 保持向后兼容 - 12 个重型依赖 optional 化（tokio/reqwest/rusqlite 等） - 全模块 #[cfg(feature)] 门控注入（llm/tools/memory/agent/engine） - 将 LlmProvider trait 及关联类型移出 provider 模块归属 llm（ADR-1） - 为 session.rs bundle() 方法添加 engine feature 门控 - 更新 12 个内部文件 + 4 个示例文件的 import 路径 - 向后兼容：provider.rs 保留 pub use 重导出老路径
61 lines
2.7 KiB
Rust
61 lines
2.7 KiB
Rust
//! quick_start —— 30 行最小可运行示例，展示 Agent / BaseTool / Builder / Session 四层抽象。
|
|
//!
|
|
//! 运行：`cargo run --example quick_start`（离线，零配置）
|
|
|
|
use std::sync::Arc;
|
|
use agcore::agent::{Agent, AgentBuilder, AgentSession};
|
|
use agcore::llm::hooks::HookExecutor;
|
|
use agcore::llm::mock::MockProvider;
|
|
use agcore::llm::LlmProvider;
|
|
use agcore::llm::types::{Usage, message::{ContentBlock, Message}, response_v2::{MessageResponse, StopReason}};
|
|
use agcore::tools::{BaseTool, ToolContext, ToolError, ToolRegistry};
|
|
use async_trait::async_trait;
|
|
use serde_json::{Value, json};
|
|
|
|
struct Greeter;
|
|
impl Agent for Greeter {
|
|
fn name(&self) -> &str { "greeter" }
|
|
fn system_prompt(&self) -> Option<&str> { Some("中文助手，先调用 echo 工具，再总结。") }
|
|
}
|
|
|
|
struct EchoTool;
|
|
#[async_trait]
|
|
impl BaseTool for EchoTool {
|
|
fn name(&self) -> &str { "echo" }
|
|
fn description(&self) -> &str { "回显文本" }
|
|
fn parameters(&self) -> Value {
|
|
json!({"type":"object","properties":{"text":{"type":"string"}},"required":["text"]})
|
|
}
|
|
async fn execute(&self, args: Value, _: &ToolContext<'_>) -> Result<Value, ToolError> {
|
|
let text = args.get("text").and_then(|v| v.as_str())
|
|
.ok_or_else(|| ToolError::InvalidArguments("text".into(), "需要 string 类型的 text 参数".into()))?;
|
|
Ok(json!({"echoed": format!("收到: {text}")}))
|
|
}
|
|
}
|
|
|
|
fn resp(content: Vec<ContentBlock>, stop: StopReason, u: (u32, u32)) -> MessageResponse {
|
|
MessageResponse { id: String::new(), model: "mock".into(), message: Message::Assistant { content },
|
|
usage: Usage::from_input_output(u.0, u.1), stop_reason: stop, extra: Default::default() }
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
let mut registry = ToolRegistry::new();
|
|
registry.register(Arc::new(EchoTool)).unwrap();
|
|
let provider: Arc<dyn LlmProvider> = Arc::new(MockProvider::new(vec![
|
|
resp(vec![ContentBlock::ToolUse { id: "c1".into(), name: "echo".into(),
|
|
input: json!({"text": "你好"}) }], StopReason::ToolUse, (5, 8)),
|
|
resp(vec![ContentBlock::Text { text: "EchoTool 已收到您的消息并完成回传。".into() }],
|
|
StopReason::Stop, (8, 16)),
|
|
]));
|
|
let bundle = Arc::new(AgentBuilder::new()
|
|
.provider(provider).tool_registry(Arc::new(registry))
|
|
.hook_executor(Arc::new(HookExecutor::new())).build().unwrap());
|
|
let mut session = AgentSession::new(Arc::new(Greeter), "qs", bundle);
|
|
let resp = session.submit_turn("你好").await.unwrap();
|
|
let text = resp.text();
|
|
println!("LLM: {text}");
|
|
assert!(text.contains("收到"), "响应应包含'收到'字样: {text}");
|
|
println!("\n✓ quick_start 完成");
|
|
}
|