1. Learn
    2. /
  2. 课程
    3. /
  3. 使用 Pinecone 的向量数据库与 Embeddings
Connected

道练习

构建检索函数

在检索增强生成（RAG）流程中，一个关键步骤是从数据库中检索数据。在本练习中，您将设计一个名为 retrieve() 的自定义函数，它将在课程最后一个练习中执行这一关键过程。

说明

100 XP
  • 使用您的 API 密钥初始化 Pinecone 客户端（OpenAI 客户端已作为 client 提供）。
  • 定义函数 retrieve，它接收 4 个参数：query、top_k、namespace 和 emb_model。
  • 使用 emb_model 参数对输入的 query 进行嵌入。
  • 检索与 query_emb 最相似的 top_k 个向量，并返回其元数据；同时将函数参数中提供的 namespace 用作命名空间。