Cohere 重新排序器
Cohere 是一家加拿大初创公司,提供自然语言处理模型,帮助公司改善人机交互。
本笔记本展示了如何在检索器中使用 Cohere 的重新排序端点。这基于 ContextualCompressionRetriever 中的思想。
%pip install --upgrade --quiet cohere
%pip install --upgrade --quiet faiss
# OR (depending on Python version)
%pip install --upgrade --quiet faiss-cpu
# get a new token: https://dashboard.cohere.ai/
import getpass
import os
if "COHERE_API_KEY" not in os.environ:
os.environ["COHERE_API_KEY"] = getpass.getpass("Cohere API Key:")
# Helper function for printing docs
def pretty_print_docs(docs):
print(
f"\n{'-' * 100}\n".join(
[f"Document {i+1}:\n\n" + d.page_content for i, d in enumerate(docs)]
)
)