RAG 시스템 구축

RAG 시스템 구축

구현 예시

from langchain.vectorstores import Chroma
from langchain.embeddings import OpenAIEmbeddings
from langchain.text_splitter import RecursiveCharacterTextSplitter

# 문서 분할
splitter = RecursiveCharacterTextSplitter(chunk_size=1000)
chunks = splitter.split_documents(documents)

# 벡터 저장소
vectorstore = Chroma.from_documents(chunks, OpenAIEmbeddings())

# 검색
results = vectorstore.similarity_search(query, k=3)

벡터 데이터베이스

• Pinecone: 관리형
• Chroma: 오픈소스
• Weaviate: 다기능