PromptLab：AI 查询增强引擎

PromptLab 将基本用户查询转换为针对 AI 系统的优化提示。它自动检测内容类型（文章、电子邮件、技术解释、创意写作），应用定制模板，并验证增强的提示是否保持了原始意图。

🔍 概述

PromptLab 构建于模块化架构之上，采用基于 YAML 的模板系统，使任何人无需编码知识即可创建和管理提示模板。该系统最终通过更结构化的输入产生更高质量的 AI 响应。

🏗️ 架构

PromptLab 由三个主要组件组成：

1. 模板系统 (prompt_templates.yaml) - 针对不同内容类型的结构化模板
2. MCP 服务器 (promptlab_server.py) - 通过带有 LangGraph 工作流的标准化协议提供模板
3. 处理客户端 (promptlab_client.py) - 处理用户查询的轻量级客户端

工作流程

1. 查询输入：用户提交自然语言查询
2. 分类：系统确定内容类型（文章、电子邮件等）
3. 参数提取：识别关键参数（主题、受众等）
4. 模板应用：检索并填充相应的模板
5. 验证：根据原始意图检查增强的提示
6. 调整：自动进行任何必要的改进
7. 响应生成：优化的提示产生高质量的响应

📋 功能

• 内容类型检测 - 自动将用户查询分类为文章、电子邮件、技术或创意写作请求
• 参数提取 - 智能提取关键参数，如主题、收件人和受众级别
• 模板库 - 针对常见内容类型的预配置模板，带有结构化指导
• 验证系统 - 确保增强的提示保持原始用户意图
• 反馈循环 - 当验证识别出不一致时，调整提示
• 模块化设计 - MCP 服务器可以插入任何 LLM 系统
• 非技术管理 - 无需编码知识即可更新模板

🚀 快速开始

前提条件

• Python 3.8+
• 依赖项：
  • mcp[cli]
  • langchain-openai
  • langgraph>=0.0.20
  • python-dotenv
  • pyyaml

安装

# 克隆存储库
git clone https://github.com/iRahulPandey/PromptLab.git
cd PromptLab

# 安装依赖项
pip install -r requirements.txt

# 设置环境变量
cp .env
# 编辑 .env 以添加您的 OpenAI API 密钥

用法

1. 首先运行服务器：

# 服务器从 prompt_templates.yaml 加载模板
python promptlab_server.py

2. 使用您的查询运行客户端：

python promptlab_client.py "写一篇关于气候变化的文章"

3. 系统将输出：
   • 原始查询
   • 分类的内容类型
   • 增强的提示
   • 验证结果
   • 最终响应

📝 模板系统

模板在 prompt_templates.yaml 中使用结构化格式定义：

templates:
  essay_prompt:
    description: "生成用于撰写文章的优化提示模板。"
    template: |
      撰写一篇结构良好的关于 {topic} 的文章，包括：
      - 一个引人入胜的介绍，提供背景信息并陈述您的论点
      ...
    parameters:
      - name: topic
        type: string
        description: 文章的主题
        required: true

添加新模板

1. 打开 prompt_templates.yaml
2. 按照现有格式添加新模板
3. 定义参数和转换
4. 在服务器端定义一个工具并加载模板
5. 服务器将在重启时自动加载新模板

🛠️ 高级配置

环境变量

• TEMPLATES_FILE - 模板 YAML 文件的路径（默认：prompt_templates.yaml）
• OPENAI_API_KEY - 您的 OpenAI API 密钥，用于 LLM 访问
• MODEL_NAME - 要使用的 OpenAI 模型（默认：gpt-3.5-turbo）
• PERSONA_SERVER_SCRIPT - 服务器脚本的路径（默认：promptlab_server.py）

自定义转换

模板可以包括动态调整参数的转换：

transformations:
  - name: formality
    value: "如果 recipient_type.lower() 在 ['boss', 'client'] 中，则为 'formal'，否则为 'semi-formal'"

📊 示例输出

输入查询

"为我的教授写一些关于可再生能源的东西"

增强的提示

撰写一篇结构良好的关于可再生能源的文章，包括：
- 一个引人入胜的介绍，提供背景信息并陈述您的论点
- 2-3 个正文段落，每个段落都有一个清晰的主题句和支持性证据
- 段落之间有逻辑过渡，引导读者
- 一个总结您的主要观点并提供最终想法的结论

这篇文章应该具有信息性、论证充分并展示批判性思维。

📄 许可证

MIT 许可证 - 有关详细信息，请参阅 LICENSE 文件

🤝 贡献

欢迎贡献！请随时提交 Pull Request。

1. Fork 存储库
2. 创建您的功能分支 (git checkout -b feature/amazing-feature)
3. 提交您的更改 (git commit -m 'Add some amazing feature')
4. 将分支推送到远程仓库 (git push origin feature/amazing-feature)
5. 打开一个 Pull Request