La première étape de la création d’une application RAG consiste à préparer l’environnement de travail. Nous commencerons par télécharger les packages que nous utiliserons pour créer l’application :
%pip install -q -U langchain torch transformers sentence-transformers datasets tiktoken upstash_vector
Ensuite, nous importerons les packages et bibliothèques qui seront utilisés :
import torch
from upstash_vector import Vector
from datasets import load_dataset
from tqdm import tqdm, trange
from langchain_community.document_loaders.csv_loader import CSVLoader
from langchain.text_splitter import RecursiveCharacterTextSplitter
from langchain.embeddings import HuggingFaceEmbeddings
from transformers import AutoTokenizer, AutoModelForCausalLM
from transformers import AutoTokenizer, pipeline
from langchain import HuggingFacePipeline
from langchain.chains import RetrievalQA