动态示例选择器PerryLink/Few-Shot-Selector

Few-Shot-Selector

基于向量相似度动态选择最相关 few-shot 示例的 CLI 工具。

已移植到 dsh-library —— PerryLink DSH 插件家族的一员。

License

English · 简体中文

问题与解决方案

Prompt 中通常使用固定的 few-shot 示例，可能与用户问题不匹配：

- 用户问关于"体育"的问题时，Prompt 里可能是"编程"的例子，导致效果不佳
- 手动维护多套 Prompt 模板既繁琐又难以扩展
- 无法根据用户输入动态调整示例

few-shot-selector 通过向量相似度检索解决该问题：将用户问题与存储的最相关示例匹配，并自动生成 Prompt。

功能简介

few-shot-selector 使用本地 Embedding 模型（all-MiniLM-L6-v2）将问答示例存入本地 ChromaDB。
给定用户问题后，它按向量相似度检索最相似的示例并格式化为 Prompt，使 few-shot 示例始终贴合主题。

特性

- 🚀 基于语义相似度的示例选择
- 💾 轻量级向量数据库（ChromaDB）
- 🔍 本地 Embedding 模型，无需调用 API
- 🎨 Typer + Rich 的现代化 CLI
- 📦 支持添加自定义问答对

快速开始

pip install few-shot-selector

首次使用时初始化数据库
few-shot-selector init

使用方法

检索相似示例（默认 3 条，可用 --num/-n 调整）
few-shot-selector query "如何读取 CSV 文件？"

查看数据库统计信息
few-shot-selector stats

添加自定义问答对（可用 --category/-c 指定类别）
few-shot-selector add "问题" "答案"

项目结构

few-shot-selector/
├── src/few_shot_selector/
│   ├── cli.py               # CLI 接口
│   ├── core.py              # 核心向量检索逻辑
│   ├── utils.py             # 工具函数
│   └── data/qa_examples.json  # QA 示例数据
├── tests/                   # 单元测试
├── pyproject.toml           # Poetry 配置
└── README.md

技术栈

- ChromaDB — 轻量级向量数据库
- Sentence Transformers — 本地 Embedding 模型（all-MiniLM-L6-v2）
- Rich — 终端格式化和样式
- Typer — CLI 框架

开发

git clone https://github.com/PerryLink/few-shot-selector.git
cd few-shot-selector
poetry install

poetry run pytest
poetry run black .
poetry run ruff check .

相关项目

- dsh-library — 本项目被移植进的 DSH 插件
- PerryLink — PerryLink DSH 插件家族

许可证

Apache License 2.0 © 2026 PerryLink