Files
opro_demo/README.md
leehwui 65cdcf29dc refactor: replace OPRO with simple iterative refinement
Major changes:
- Remove fake OPRO evaluation (no more fake 0.5 scores)
- Add simple refinement based on user selection
- New endpoint: POST /opro/refine (selected + rejected instructions)
- Update prompt generation to focus on comprehensive coverage instead of style variety
- All generated instructions now start with role definition (你是一个...)
- Update README to reflect new approach and API endpoints

Technical details:
- Added refine_based_on_selection() in prompt_utils.py
- Added refine_instruction_candidates() in user_prompt_optimizer.py
- Added OPRORefineReq model and /opro/refine endpoint in api.py
- Updated frontend handleContinueOptimize() to use new refinement flow
- Changed prompt requirements from 'different styles' to 'comprehensive coverage'
- Added role definition requirement as first item in all prompt templates
2025-12-08 09:43:20 +08:00

13 KiB
Raw Blame History

System Prompt Generator

功能概述

这是一个基于大语言模型的系统提示词System Prompt生成和迭代优化工具。通过简单的任务描述自动生成高质量的系统指令并支持基于用户选择的迭代改进。

核心功能

  • 智能指令生成:根据任务描述自动生成多个高质量的系统指令候选
  • 迭代式改进:基于用户选择的指令生成改进版本,避免被拒绝的方向
  • 角色定义格式:所有生成的指令都以角色定义开头(如"你是一个..."),符合最佳实践
  • 智能候选选择:通过语义聚类和多样性选择,从大量候选中筛选出最具代表性的指令
  • 会话管理:支持多个任务的并行管理和历史记录
  • 全面覆盖要求:生成的指令全面覆盖任务的所有要求和细节,而非仅追求风格多样性

用户界面

  • 现代化聊天界面:类似 Google Gemini 的简洁设计
  • 侧边栏会话管理:可折叠的侧边栏,支持多会话切换
  • 实时生成反馈:每轮生成 5 个候选指令,用户可选择继续优化或复制使用
  • 模型选择:支持在界面中选择不同的 LLM 模型

核心特性

1. 简单直观的工作流程

不同于复杂的 OPRO 算法(需要测试用例和自动评估),本工具采用简单直观的迭代改进方式:

  • 初始生成:输入任务描述 → 生成 5 个全面的系统指令候选
  • 迭代改进:选择喜欢的指令 → 生成基于该指令的改进版本,同时避免被拒绝的方向
  • 无需评分:不需要测试用例或性能评分,完全基于用户偏好进行改进

2. 高质量指令生成

  • 角色定义格式:所有指令以"你是一个..."开头,符合系统提示词最佳实践
  • 全面覆盖要求:生成的指令全面覆盖任务的所有要求和细节
  • 清晰可执行:指令清晰、具体、可执行,包含必要的行为规范和输出格式
  • 简体中文:所有生成的指令使用简体中文

3. 性能优化

  • 候选池大小优化:生成 10 个候选,通过聚类选择 5 个最具多样性的
  • 智能聚类选择:使用 AgglomerativeClustering 从候选池中选择最具代表性的指令
  • 嵌入服务回退Xinference → Ollama 自动回退机制,确保服务可用性

4. API 架构

  • 核心端点
    • POST /opro/create - 创建新任务
    • POST /opro/generate_and_evaluate - 生成初始候选
    • POST /opro/refine - 基于用户选择进行迭代改进
    • GET /opro/sessions - 获取所有会话
    • GET /opro/runs - 获取所有任务
  • 会话管理:支持多会话、多任务的并行管理
  • 向后兼容:保留原有查询重写功能,标记为 opro-legacy

5. 前端界面

  • Gemini 风格设计:简洁的白色/灰色配色,圆角设计,微妙的阴影效果
  • 可折叠侧边栏:默认折叠,支持会话列表管理
  • 多行输入框:支持多行文本输入,底部工具栏包含模型选择器
  • 候选指令卡片:每个候选显示编号和内容,提供"继续优化"和"复制"按钮
  • 简体中文界面:所有 UI 文本和生成的指令均使用简体中文

快速开始

环境要求

  • Python ≥ 3.10(推荐使用 conda 虚拟环境)
  • Ollama 本地服务及模型(如 qwen3:8bqwen3-embedding:4b
  • 可选Xinference embedding 服务

安装依赖

# 创建 conda 环境(推荐)
conda create -n opro python=3.10
conda activate opro

# 安装 Python 依赖
pip install fastapi uvicorn requests numpy scikit-learn pydantic

启动 Ollama 服务

# 确保 Ollama 已安装并运行
ollama serve

# 拉取所需模型
ollama pull qwen3:8b
ollama pull qwen3-embedding:4b

启动应用

# 启动后端服务
uvicorn _qwen_xinference_demo.api:app --host 127.0.0.1 --port 8010

# 或使用 0.0.0.0 允许外部访问
uvicorn _qwen_xinference_demo.api:app --host 0.0.0.0 --port 8010

访问界面

使用示例

  1. 创建新会话:在界面点击"新建会话"或侧边栏的 + 按钮
  2. 输入任务描述:例如"帮我写一个专业的营销文案生成助手"
  3. 查看候选指令:系统生成 5 个全面的系统指令,每个都以角色定义开头
  4. 选择并改进:点击喜欢的指令上的"继续优化"按钮,生成基于该指令的改进版本
  5. 复制使用:点击"复制"按钮将指令复制到剪贴板,用于你的应用中

配置说明

配置文件:config.py

关键配置项

# Ollama 服务配置
OLLAMA_HOST = "http://127.0.0.1:11434"
DEFAULT_CHAT_MODEL = "qwen3:8b"
DEFAULT_EMBED_MODEL = "qwen3-embedding:4b"

# 生成参数
GENERATION_POOL_SIZE = 10  # 生成候选池大小生成10个聚类选择5个
TOP_K = 5                   # 返回给用户的候选数量
CLUSTER_DISTANCE_THRESHOLD = 0.15  # 聚类距离阈值

# Xinference 配置(可选)
XINFERENCE_EMBED_URL = "http://127.0.0.1:9997/models/bge-base-zh/embed"

项目结构

.
├── _qwen_xinference_demo/
│   ├── api.py                          # FastAPI 主应用
│   └── opro/
│       ├── user_prompt_optimizer.py    # OPRO 核心逻辑
│       ├── prompt_utils.py             # 元提示生成
│       ├── session_state.py            # 会话状态管理
│       ├── ollama_client.py            # Ollama 客户端
│       └── xinference_client.py        # Xinference 客户端
├── frontend/
│   ├── opro.html                       # OPRO 优化界面
│   └── index.html                      # 传统三栏界面
├── examples/
│   ├── opro_demo.py                    # OPRO 功能演示
│   └── client_demo.py                  # API 调用示例
├── config.py                           # 全局配置
├── API.md                              # API 文档
└── README.md                           # 本文件

API 端点

会话管理

  • POST /opro/session/create - 创建新会话
  • GET /opro/sessions - 获取所有会话
  • GET /opro/session/{session_id} - 获取会话详情

任务管理

  • POST /opro/create - 在会话中创建新任务
    • 请求体:{"session_id": "xxx", "task_description": "任务描述", "model_name": "qwen3:8b"}
    • 返回:{"run_id": "xxx", "task_description": "...", "iteration": 0}

指令生成

  • POST /opro/generate_and_evaluate - 生成初始候选指令

    • 请求体:{"run_id": "xxx", "top_k": 5, "pool_size": 10}
    • 返回:{"candidates": [{"instruction": "...", "score": null}, ...]}
  • POST /opro/refine - 基于用户选择进行迭代改进

    • 请求体:{"run_id": "xxx", "selected_instruction": "用户选择的指令", "rejected_instructions": ["被拒绝的指令1", "被拒绝的指令2"]}
    • 返回:{"candidates": [{"instruction": "...", "score": null}, ...], "iteration": 1}

任务查询

  • GET /opro/runs - 获取所有任务
  • GET /opro/run/{run_id} - 获取任务详情

传统端点(向后兼容)

  • POST /query - 查询重写(首轮)
  • POST /select - 选择候选并回答
  • POST /reject - 拒绝并重新生成
  • POST /message - 聊天消息

通用端点

  • GET /health - 健康检查
  • GET /version - 版本信息
  • GET /models - 可用模型列表
  • POST /set_model - 设置模型

详细 API 文档请访问:http://127.0.0.1:8010/docs

工作原理

初始生成流程

  1. 用户输入任务描述(如"帮我写一个专业的营销文案生成助手"
  2. 系统使用 LLM 生成 10 个候选指令
  3. 通过语义嵌入和聚类算法选择 5 个最具多样性的候选
  4. 所有候选都以角色定义开头,全面覆盖任务要求

迭代改进流程

  1. 用户选择喜欢的指令(如候选 #3
  2. 系统记录被拒绝的指令(候选 #1, #2, #4, #5
  3. 向 LLM 发送改进请求:"基于选中的指令生成改进版本,避免被拒绝指令的方向"
  4. 生成新的 10 个候选,聚类选择 5 个返回
  5. 用户可以继续迭代或复制使用

与 OPRO 的区别

OPRO原始算法

  • 需要测试用例(如数学题的正确答案)
  • 自动评分(如准确率 0.73, 0.81
  • 基于性能轨迹优化
  • 适用于有明确评估标准的任务

本工具(简单迭代改进)

  • 不需要测试用例
  • 不需要自动评分
  • 基于用户偏好改进
  • 适用于任意通用任务

常见问题

1. 无法连接 Ollama 服务

确保 Ollama 服务正在运行:

ollama serve

检查配置文件中的 OLLAMA_HOST 是否正确。

2. 模型不可用

通过 /models 端点查看可用模型列表,使用 /set_model 切换模型。

3. 生成速度慢

  • 调整 GENERATION_POOL_SIZE 减少候选数量(如改为 6返回 3 个)
  • 使用更小的模型(如 qwen3:4b
  • 确保 Ollama 使用 GPU 加速

4. 生成的指令质量不高

  • 提供更详细的任务描述
  • 多次迭代改进,选择最好的继续优化
  • 尝试不同的模型

5. 界面显示异常

硬刷新浏览器缓存:

  • Mac: Cmd + Shift + R
  • Windows/Linux: Ctrl + Shift + R

原始 README点击展开
  • 项目简介

  • OPRO Prompt Optimizer面向提示优化的交互式系统支持多轮拒选/再生成、语义聚类去重与 TopK 代表选择。

  • 后端 FastAPI 提供 REST 接口,前端三栏 UI 便于会话管理与候选挑选。

  • 架构概览

  • Frontend /ui/ → POST /query 首轮候选 → POST /select 选择并回答 → POST /reject 再生成 → POST /query_from_message 基于最近消息优化 → POST /message 聊天

  • OPRO 流程:指令构造 → Qwen 批量生成 → EmbeddingXinference→Ollama 回退)→ 聚类去重 → TopK

  • 核心实现位置: _qwen_xinference_demo/opro/user_prompt_optimizer.py:45-54 (候选生成)、 _qwen_xinference_demo/opro/xinference_client.py:7-28 embedding 回退)

  • 环境与依赖

  • Python ≥ 3.10(建议使用 conda 虚拟环境)

  • 必需Ollama 本地服务与模型(如 qwen3:8b qwen3-embedding:4b

  • 可选Xinference embedding 服务( http://127.0.0.1:9997/models/bge-base-zh/embed

  • Python 依赖: fastapi 、 uvicorn 、 requests 、 numpy 、 scikit-learn 、 pydantic

  • 安装与启动

  • 安装依赖

    • pip install fastapi uvicorn requests numpy scikit-learn pydantic
  • 启动后端服务

    • uvicorn _qwen_xinference_demo.api:app --host 0.0.0.0 --port 8010
  • 访问页面

  • 配置

  • 文件: config.py

  • 关键项

    • APP_TITLE 、 APP_DESCRIPTION 、 APP_VERSION 、 APP_CONTACT (应用元信息,见 _qwen_xinference_demo/api.py:14-26
    • OLLAMA_HOST 、 OLLAMA_GENERATE_URL 、 OLLAMA_TAGS_URL Ollama 端点)
    • DEFAULT_CHAT_MODEL 、 DEFAULT_EMBED_MODEL (默认模型,用于 _qwen_xinference_demo/opro/ollama_client.py:4-7 与 _qwen_xinference_demo/opro/xinference_client.py:1-6,20-21
    • XINFERENCE_EMBED_URL (优先 embedding 端点)
    • TOP_K 、 CLUSTER_DISTANCE_THRESHOLD (候选选择参数,引用 _qwen_xinference_demo/opro/user_prompt_optimizer.py:19,45
  • 统一响应与错误处理

  • 成功: {"success": true, "data": {...}}

  • 失败: {"success": false, "error": "...", "error_code": "..."} ,状态码保持 HTTP 值

  • 应用级异常: AppException(status_code, detail, error_code) _qwen_xinference_demo/api.py:23-39

  • 示例:会话不存在抛出 SESSION_NOT_FOUND Ollama 调用异常抛出 OLLAMA_ERROR

  • API 与示例

  • 全量端点与示例:见 API.md

  • 健康与版本

    • GET /health 返回 {status, version} _qwen_xinference_demo/api.py:129-134
    • GET /version 返回 {version} _qwen_xinference_demo/api.py:135-138
  • 示例脚本

    • 入口: examples/client_demo.py
    • 功能:健康检查 → 创建会话 → 选择候选 → 继续优化 → 聊天 → 会话列表
  • 目录结构

  • /_qwen_xinference_demo/api.py FastAPI 主应用与路由

  • /_qwen_xinference_demo/opro/user_prompt_optimizer.py OPRO 候选生成与聚类选择

  • /_qwen_xinference_demo/opro/xinference_client.py EmbeddingXinference→Ollama 回退)

  • /_qwen_xinference_demo/opro/ollama_client.py Ollama 调用与模型列表

  • /_qwen_xinference_demo/opro/session_state.py :会话态管理

  • /frontend/index.html :三栏 UI 页面

  • /API.md :接口文档

  • /examples/client_demo.py :示例调用脚本

  • /config.py :全局配置

  • 常见问题

  • 无法访问 /ui/react :使用 /ui/ React 示例仅作演示入口 _qwen_xinference_demo/api.py:133-144

  • 模型不可用: /models 查看列表并通过 /set_model 应用;错误返回 MODEL_NOT_AVAILABLE

  • 第二轮无相关候选:使用 POST /query_from_message 基于最近消息再生候选 _qwen_xinference_demo/api.py:193-206

  • 立即回答诉求:用 POST /answer 先答后给候选 _qwen_xinference_demo/api.py:211-219

  • 端口与地址访问差异:在启动命令中明确 --host 0.0.0.0 --port 8010 ,本地浏览器建议访问 127.0.0.1