■ FAISS 클래스의 생성자에서 embedding_function/index/docstore/index_to_docstore_id 인자를 사용해 FAISS 객체를 만드는 방법을 보여준다.
※ OPENAI_API_KEY 환경 변수 값은 .env 파일에 정의한다.
▶ 예제 코드 (PY)
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
import faiss from dotenv import load_dotenv from langchain_openai import OpenAIEmbeddings from langchain_community.docstore.in_memory import InMemoryDocstore from langchain_community.vectorstores import FAISS load_dotenv() openAIEmbeddings = OpenAIEmbeddings() indexFlatL2 = faiss.IndexFlatL2(1536) inMemoryDocstore = InMemoryDocstore() faiss = FAISS(embedding_function = openAIEmbeddings, index = indexFlatL2, docstore = inMemoryDocstore, index_to_docstore_id = {}) |
※ pip install python-dotenv langchain-community langchain-openai faiss-gpu 명령을 실행했다.