■ Chroma 클래스의 생성자에서 collection_name/embedding_function/persist_directory 인자를 사용해 메모리 벡터 저장소를 만드는 방법을 보여준다.
※ OPENAI_API_KEY 환경 변수 값은 .env 파일에 정의한다.
▶ main.py
1 2 3 4 5 6 7 8 9 10 11 |
from dotenv import load_dotenv from langchain_openai.embeddings import OpenAIEmbeddings from langchain_chroma import Chroma load_dotenv() openAIEmbeddings = OpenAIEmbeddings() chroma = Chroma(collection_name = "test", embedding_function = openAIEmbeddings, persist_directory = None) |
※ pip install python-dotenv langchain_openai langchain_chroma 명령을 실행했다.