카테고리 없음

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도 만들어보자

 

 

 

질의해보자

반응형