1. 학습
  2. /
  3. 강의
  4. /
  5. Pinecone으로 배우는 임베딩용 벡터 데이터베이스

Connected

연습 문제

검색 함수 만들기

Retrieval Augmented Generation(RAG) 워크플로에서 핵심 단계는 데이터베이스에서 데이터를 검색하는 일입니다. 이 연습에서는 마지막 연습 문제에서 사용할 중요한 과정을 수행하는 retrieve()라는 사용자 지정 함수를 설계해 보겠습니다.

지침

100 XP
  • Pinecone 클라이언트를 API 키로 초기화하세요(OpenAI 클라이언트는 client로 제공됩니다).
  • 네 개의 매개변수 query, top_k, namespace, emb_model을 받는 함수 retrieve를 정의하세요.
  • emb_model 인자를 사용해 입력 query를 임베딩하세요.
  • 함수에 전달된 namespace를 지정하여, 메타데이터와 함께 query_emb와 유사한 벡터 top_k개를 검색하세요.