zephyr-7b-beta-rag-demo
/
ingest.py
23 строки · 866.0 Байт
1import os
2from langchain.text_splitter import RecursiveCharacterTextSplitter
3from langchain.vectorstores import Chroma
4from langchain.embeddings import HuggingFaceBgeEmbeddings
5from langchain.document_loaders import PyPDFLoader
6
7model_name = "BAAI/bge-large-en"
8model_kwargs = {'device': 'cpu'}
9encode_kwargs = {'normalize_embeddings': False}
10embeddings = HuggingFaceBgeEmbeddings(
11model_name=model_name,
12model_kwargs=model_kwargs,
13encode_kwargs=encode_kwargs
14)
15
16loader = PyPDFLoader("pet.pdf")
17documents = loader.load()
18text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
19texts = text_splitter.split_documents(documents)
20
21vector_store = Chroma.from_documents(texts, embeddings, collection_metadata={"hnsw:space": "cosine"}, persist_directory="stores/pet_cosine")
22
23print("Vector Store Created.......")