카테고리 없음
wikipedia에서 가져온 정보 ChromaDB에 저장하고 RAG 실습
worldforest
2024. 8. 7. 00:40
타일러 선생님이 생각나는 비주얼
믿고 가봅니다.
https://colab.research.google.com/drive/1Zpg8WKVm1wQ4lkJgyJ37ZKwQNwWkvXGg?usp=sharing
Untitled0.ipynb
Colab notebook
colab.research.google.com
유튜브보고 따라했고 에러나는 것들 바로바로 고치면서 진행!
from langchain.document_loaders import WikipediaLoader
from langchain.text_splitter import RecursiveCharacterTextSplitter
embedding해야되는데 openai 설치 안 했어서 설치했고
openai api key 입력해야 계정연결되니까 그것도 설정해주고
import getpass
import os
os.environ["OPENAI_API_KEY"] = getpass.getpass("OpenAI API Key:")
embedding은 이걸로 사용할거라고 설정하고
embeddings = OpenAIEmbeddings()
pip install chromadb
pip install tiktoken
ChromaDB에 임베딩한 파일 저장!
이제 질의할건데 prompt template도 만들어보자
질의해보자
반응형