-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathupload_func.py
More file actions
37 lines (28 loc) · 1.17 KB
/
upload_func.py
File metadata and controls
37 lines (28 loc) · 1.17 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
from pypdf import PdfReader
from langchain.text_splitter import RecursiveCharacterTextSplitter
from langchain.embeddings.sentence_transformer import SentenceTransformerEmbeddings
import pinecone
from langchain.vectorstores import Pinecone
#**********Functions to help you load documents to PINECONE***********
def read_pdf_data(pdf_file):
pdf_page = PdfReader(pdf_file)
text = ""
for page in pdf_page.pages:
text += page.extract_text()
return text
def split_data(text):
text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=50)
docs = text_splitter.split_text(text)
docs_chunks =text_splitter.create_documents(docs)
return docs_chunks
def create_embeddings_load_data():
embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
return embeddings
def push_to_pinecone(pinecone_apikey,pinecone_environment,pinecone_index_name,embeddings,docs):
pinecone.init(
api_key=pinecone_apikey,
environment=pinecone_environment
)
index_name = pinecone_index_name
index = Pinecone.from_documents(docs, embeddings, index_name=index_name)
return index