use super::registry::PromptRegistry;
use super::template::{PromptTemplate, PromptVariable};
pub fn general_assistant() -> PromptTemplate {
PromptTemplate::new("general-assistant")
.with_name("通用助手")
.with_description("通用 AI 助手系统提示")
.with_content(
"你是一个乐于助人的 AI 助手。请以清晰、准确、专业的方式回答用户的问题。\n\n\
回答时请注意：\n\
1. 保持回答简洁明了\n\
2. 如果不确定，请诚实说明\n\
3. 必要时提供示例说明",
)
.with_tag("system")
.with_tag("general")
}
pub fn role_assistant() -> PromptTemplate {
PromptTemplate::new("role-assistant")
.with_name("角色助手")
.with_description("可自定义角色的助手模板")
.with_content(
"你是一个专业的{role}。你的专长是{expertise}。\n\n\
在回答问题时，请：\n\
1. 运用你的专业知识\n\
2. 提供有见地的分析\n\
3. 给出可操作的建议",
)
.with_variable(
PromptVariable::new("role").with_description("角色名称，如：软件工程师、数据分析师"),
)
.with_variable(
PromptVariable::new("expertise")
.with_description("专业领域")
.with_default("解决问题和提供帮助"),
)
.with_tag("system")
.with_tag("role")
}
pub fn code_review() -> PromptTemplate {
PromptTemplate::new("code-review")
.with_name("代码审查")
.with_description("专业代码审查模板")
.with_content(
"请作为一个资深的{language}开发者，审查以下代码：\n\n\
```{language}\n{code}\n```\n\n\
请从以下方面进行审查：\n\
1. **代码质量**：可读性、命名规范、代码结构\n\
2. **潜在问题**：Bug、边界情况、错误处理\n\
3. **性能**：效率问题、优化建议\n\
4. **安全性**：安全漏洞、敏感信息处理\n\
5. **最佳实践**：设计模式、惯用写法\n\n\
请提供具体的改进建议和示例代码。",
)
.with_variable(
PromptVariable::new("language")
.with_description("编程语言")
.with_default("代码"),
)
.with_variable(PromptVariable::new("code").with_description("要审查的代码"))
.with_tag("code")
.with_tag("review")
}
pub fn code_explain() -> PromptTemplate {
PromptTemplate::new("code-explain")
.with_name("代码解释")
.with_description("解释代码功能和原理")
.with_content(
"请详细解释以下{language}代码的功能和工作原理：\n\n\
```{language}\n{code}\n```\n\n\
请包含：\n\
1. **功能概述**：代码的主要用途\n\
2. **逐行/逐块解释**：关键部分的详细说明\n\
3. **使用示例**：如何调用或使用这段代码\n\
4. **注意事项**：使用时需要注意的问题",
)
.with_variable(
PromptVariable::new("language")
.with_description("编程语言")
.with_default("代码"),
)
.with_variable(PromptVariable::new("code").with_description("要解释的代码"))
.with_tag("code")
.with_tag("explain")
}
pub fn code_generate() -> PromptTemplate {
PromptTemplate::new("code-generate")
.with_name("代码生成")
.with_description("根据需求生成代码")
.with_content(
"请使用 {language} 编写代码实现以下功能：\n\n\
**需求描述**：\n{requirement}\n\n\
**要求**：\n\
1. 代码应该简洁、高效、可读性强\n\
2. 包含必要的注释说明\n\
3. 考虑边界情况和错误处理\n\
4. 遵循 {language} 的最佳实践",
)
.with_variable(PromptVariable::new("language").with_description("编程语言"))
.with_variable(PromptVariable::new("requirement").with_description("功能需求描述"))
.with_tag("code")
.with_tag("generate")
}
pub fn code_refactor() -> PromptTemplate {
PromptTemplate::new("code-refactor")
.with_name("代码重构")
.with_description("重构和优化代码")
.with_content(
"请重构以下{language}代码，{goal}：\n\n\
```{language}\n{code}\n```\n\n\
重构时请：\n\
1. 保持功能不变\n\
2. 提高代码质量\n\
3. 解释你的改动及原因\n\
4. 提供重构后的完整代码",
)
.with_variable(
PromptVariable::new("language")
.with_description("编程语言")
.with_default("代码"),
)
.with_variable(PromptVariable::new("code").with_description("要重构的代码"))
.with_variable(
PromptVariable::new("goal")
.with_description("重构目标")
.with_default("使其更加清晰、高效"),
)
.with_tag("code")
.with_tag("refactor")
}
pub fn code_test() -> PromptTemplate {
PromptTemplate::new("code-test")
.with_name("测试生成")
.with_description("为代码生成单元测试")
.with_content(
"请为以下{language}代码编写单元测试：\n\n\
```{language}\n{code}\n```\n\n\
测试要求：\n\
1. 覆盖主要功能路径\n\
2. 包含边界情况测试\n\
3. 包含错误情况测试\n\
4. 使用 {test_framework} 测试框架",
)
.with_variable(PromptVariable::new("language").with_description("编程语言"))
.with_variable(PromptVariable::new("code").with_description("要测试的代码"))
.with_variable(
PromptVariable::new("test_framework")
.with_description("测试框架")
.with_default("标准"),
)
.with_tag("code")
.with_tag("test")
}
pub fn tech_doc() -> PromptTemplate {
PromptTemplate::new("tech-doc")
.with_name("技术文档")
.with_description("撰写技术文档")
.with_content(
"请为 {subject} 撰写技术文档。\n\n\
文档应包含：\n\
1. **概述**：简要介绍主题\n\
2. **安装/配置**：如何开始使用\n\
3. **使用指南**：基本用法和示例\n\
4. **API 参考**：详细接口说明（如适用）\n\
5. **常见问题**：FAQ 和故障排除\n\n\
目标受众：{audience}",
)
.with_variable(PromptVariable::new("subject").with_description("文档主题"))
.with_variable(
PromptVariable::new("audience")
.with_description("目标读者")
.with_default("开发者"),
)
.with_tag("doc")
.with_tag("writing")
}
pub fn summarize() -> PromptTemplate {
PromptTemplate::new("summarize")
.with_name("内容总结")
.with_description("总结长文本内容")
.with_content(
"请总结以下内容：\n\n{content}\n\n\
总结要求：\n\
1. 长度约 {length}\n\
2. 保留关键信息和主要观点\n\
3. 使用清晰的结构组织\n\
4. 语言简练准确",
)
.with_variable(PromptVariable::new("content").with_description("要总结的内容"))
.with_variable(
PromptVariable::new("length")
.with_description("目标长度")
.with_default("200-300字"),
)
.with_tag("writing")
.with_tag("summary")
}
pub fn translate() -> PromptTemplate {
PromptTemplate::new("translate")
.with_name("翻译")
.with_description("翻译文本")
.with_content(
"请将以下内容从{source_lang}翻译成{target_lang}：\n\n\
{content}\n\n\
翻译要求：\n\
1. 保持原文含义准确\n\
2. 符合目标语言的表达习惯\n\
3. 保持专业术语的准确性\n\
4. 保持原文的语气和风格",
)
.with_variable(
PromptVariable::new("source_lang")
.with_description("源语言")
.with_default("英文"),
)
.with_variable(
PromptVariable::new("target_lang")
.with_description("目标语言")
.with_default("中文"),
)
.with_variable(PromptVariable::new("content").with_description("要翻译的内容"))
.with_tag("writing")
.with_tag("translation")
}
pub fn analyze() -> PromptTemplate {
PromptTemplate::new("analyze")
.with_name("问题分析")
.with_description("分析问题并给出解决方案")
.with_content(
"请分析以下问题并给出解决方案：\n\n\
**问题描述**：\n{problem}\n\n\
**上下文**：\n{context}\n\n\
请提供：\n\
1. **问题分析**：问题的根本原因\n\
2. **解决方案**：可行的解决方法\n\
3. **优劣分析**：各方案的优缺点\n\
4. **推荐方案**：最佳建议及理由",
)
.with_variable(PromptVariable::new("problem").with_description("问题描述"))
.with_variable(
PromptVariable::new("context")
.with_description("相关背景信息")
.with_default("无额外上下文"),
)
.with_tag("analysis")
.with_tag("problem-solving")
}
pub fn compare() -> PromptTemplate {
PromptTemplate::new("compare")
.with_name("对比分析")
.with_description("对比多个选项")
.with_content(
"请对比分析以下选项：\n\n\
{options}\n\n\
对比维度：{dimensions}\n\n\
请提供：\n\
1. **详细对比表格**\n\
2. **各选项优缺点**\n\
3. **适用场景分析**\n\
4. **推荐建议**",
)
.with_variable(PromptVariable::new("options").with_description("要对比的选项列表"))
.with_variable(
PromptVariable::new("dimensions")
.with_description("对比维度")
.with_default("功能、性能、易用性、成本"),
)
.with_tag("analysis")
.with_tag("comparison")
}
pub fn react_system() -> PromptTemplate {
PromptTemplate::new("react-system")
.with_name("ReAct 系统提示")
.with_description("ReAct Agent 的系统提示")
.with_content(
"你是一个使用 ReAct（Reasoning + Acting）方法解决问题的 AI Agent。\n\n\
你可以使用以下工具：\n{tools}\n\n\
解决问题时，请遵循以下步骤：\n\n\
1. **Thought（思考）**：分析当前情况，决定下一步行动\n\
2. **Action（行动）**：选择合适的工具并执行\n\
3. **Observation（观察）**：分析工具返回的结果\n\
4. 重复上述步骤直到得到答案\n\
5. **Final Answer（最终答案）**：给出完整的回答\n\n\
格式要求：\n\
- Thought: 你的思考过程\n\
- Action: 工具名称[参数]\n\
- Observation: 工具返回的结果\n\
- Final Answer: 你的最终回答",
)
.with_variable(PromptVariable::new("tools").with_description("可用工具列表"))
.with_tag("react")
.with_tag("agent")
}
pub fn react_task() -> PromptTemplate {
PromptTemplate::new("react-task")
.with_name("ReAct 任务")
.with_description("ReAct Agent 的任务模板")
.with_content("请完成以下任务：\n\n{task}\n\n开始你的推理和行动：")
.with_variable(PromptVariable::new("task").with_description("任务描述"))
.with_tag("react")
.with_tag("agent")
}
pub fn debater() -> PromptTemplate {
PromptTemplate::new("debater")
.with_name("辩论者")
.with_description("辩论模式中的辩论者角色")
.with_content(
"你是辩论中的{position}方。\n\n\
辩题：{topic}\n\n\
前述观点：\n{previous}\n\n\
请从你的立场出发，提供有力的论点。注意：\n\
1. 基于事实和逻辑论证\n\
2. 回应对方的观点\n\
3. 保持专业和理性",
)
.with_variable(
PromptVariable::new("position")
.with_description("辩论立场")
.with_enum(vec!["正".to_string(), "反".to_string()]),
)
.with_variable(PromptVariable::new("topic").with_description("辩论话题"))
.with_variable(
PromptVariable::new("previous")
.with_description("之前的辩论内容")
.with_default("这是辩论的开始"),
)
.with_tag("multi-agent")
.with_tag("debate")
}
pub fn supervisor() -> PromptTemplate {
PromptTemplate::new("supervisor")
.with_name("监督者")
.with_description("监督模式中的监督者角色")
.with_content(
"你是一个团队监督者，负责评估团队成员的工作成果。\n\n\
**任务**：{task}\n\n\
**团队成员的回答**：\n{responses}\n\n\
请：\n\
1. 评估每个回答的质量\n\
2. 指出各自的优点和不足\n\
3. 综合最佳内容给出最终答案\n\
4. 给出改进建议",
)
.with_variable(PromptVariable::new("task").with_description("任务描述"))
.with_variable(PromptVariable::new("responses").with_description("团队成员的回答"))
.with_tag("multi-agent")
.with_tag("supervisor")
}
pub fn aggregator() -> PromptTemplate {
PromptTemplate::new("aggregator")
.with_name("聚合者")
.with_description("并行模式中的结果聚合角色")
.with_content(
"多个 Agent 已经分别处理了以下任务：\n\n\
**原始任务**：{task}\n\n\
**各 Agent 的结果**：\n{results}\n\n\
请综合以上结果：\n\
1. 找出共同点和差异点\n\
2. 整合各方优势\n\
3. 形成一个完整、准确的最终答案",
)
.with_variable(PromptVariable::new("task").with_description("原始任务"))
.with_variable(PromptVariable::new("results").with_description("各 Agent 的结果"))
.with_tag("multi-agent")
.with_tag("aggregation")
}
pub fn create_preset_registry() -> PromptRegistry {
let mut registry = PromptRegistry::new();
registry.register(general_assistant());
registry.register(role_assistant());
registry.register(code_review());
registry.register(code_explain());
registry.register(code_generate());
registry.register(code_refactor());
registry.register(code_test());
registry.register(tech_doc());
registry.register(summarize());
registry.register(translate());
registry.register(analyze());
registry.register(compare());
registry.register(react_system());
registry.register(react_task());
registry.register(debater());
registry.register(supervisor());
registry.register(aggregator());
registry
}
pub fn load_presets(registry: &mut PromptRegistry) {
let presets = create_preset_registry();
registry.merge(presets);
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_preset_templates() {
let registry = create_preset_registry();
assert!(registry.contains("general-assistant"));
assert!(registry.contains("code-review"));
assert!(registry.contains("react-system"));
assert!(registry.contains("supervisor"));
let code_templates = registry.find_by_tag("code");
assert!(code_templates.len() >= 4);
let agent_templates = registry.find_by_tag("agent");
assert!(agent_templates.len() >= 2);
}
#[test]
fn test_code_review_template() {
let template = code_review();
let result = template
.render(&[
("language", "rust"),
("code", "fn main() { info!(\"Hello\"); }"),
])
.unwrap();
assert!(result.contains("rust"));
assert!(result.contains("fn main"));
}
#[test]
fn test_role_assistant_template() {
let template = role_assistant();
let result = template.render(&[("role", "数据分析师")]).unwrap();
assert!(result.contains("数据分析师"));
assert!(result.contains("解决问题和提供帮助")); }
#[test]
fn test_react_system_template() {
let template = react_system();
let tools = "1. search: 搜索网页\n2. calculator: 计算器";
let result = template.render(&[("tools", tools)]).unwrap();
assert!(result.contains("ReAct"));
assert!(result.contains("search"));
assert!(result.contains("Thought"));
}
#[test]
fn test_translate_template() {
let template = translate();
let result = template.render(&[("content", "Hello, World!")]).unwrap();
assert!(result.contains("Hello, World!"));
assert!(result.contains("英文")); assert!(result.contains("中文")); }
#[test]
fn test_load_presets() {
let mut registry = PromptRegistry::new();
assert!(registry.is_empty());
load_presets(&mut registry);
assert!(!registry.is_empty());
assert!(registry.len() >= 15);
}
}