重排序角色
重排序模型角色
“重排序模型”(Reranking model)经过专门训练,可以接收两段文本(通常是用户的问题和文档),并返回一个 0 到 1 之间的相关性分数,以评估该文档在回答问题时的有用程度。重排序模型通常比 LLM 小得多,且相比之下速度极快且成本更低。
在 Continue 中,重排序器通过
rerank 角色指定,并由代码库感知功能使用,以便在向量搜索后筛选出最相关的代码片段。推荐的重排序模型
如需查看包括开源和闭源选项在内的所有重排序模型的比较,请参阅我们的综合模型推荐。
如果您可以使用任何模型,我们推荐 Voyage AI 的
rerank-2,它与下文列出的其他重排序模型选项一起提供。Voyage AI
Voyage AI 凭借其
rerank-2 模型提供了最适合代码的重排序模型。从此处获取 API 密钥后,您可以按如下方式配置重排序器:name: My Config
version: 0.0.1
schema: v1
models:
- uses: voyageai/rerank-2
Cohere
点击此处查看 Cohere 重排序器的文档。
name: My Config
version: 0.0.1
schema: v1
models:
- name: Cohere Reranker
provider: cohere
model: rerank-english-v3.0
apiKey: <YOUR_COHERE_API_KEY>
roles:
- rerank
LLM
如果您仅能访问单一 LLM,也可以将其用作重排序器。除非确实必要,否则不建议这样做,因为它不仅成本昂贵,而且在准确性上也不及上述专门针对此任务训练的模型。请注意,如果您使用的是本地模型(例如通过 Ollama),此方法将无法工作,因为需要进行过多的并行请求。
name: My Config
version: 0.0.1
schema: v1
models:
- name: LLM Reranker
provider: openai
model: gpt-4o
roles:
- rerank
文本嵌入推理 (TEI)
Hugging Face Text Embeddings Inference 使您能够托管自己的重排序器端点。您可以按如下方式配置重排序器:
name: My Config
version: 0.0.1
schema: v1
models:
- name: Huggingface-tei Reranker
provider: huggingface-tei
model: tei
apiBase: http://:8080
apiKey: <YOUR_TEI_API_KEY>
roles:
- rerank