Nazhar commited on
Commit
b521dd5
·
verified ·
1 Parent(s): c523ff7

Create ragfile2.py

Browse files
Files changed (1) hide show
  1. ragfile2.py +26 -0
ragfile2.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from langchain.indexes import VectorstoreIndexCreator
2
+ from langchain.chains import RetrievalQA
3
+ from langchain.llms import OpenAI
4
+ from langchain_community.document_loaders import CSVLoader
5
+ from langchain_text_splitters import RecursiveCharacterTextSplitter
6
+ # from langchain_community.text_splitter import RecursiveCharacterTextSplitter
7
+ from langchain_community.embeddings import HuggingFaceInstructEmbeddings
8
+ from langchain_community.vectorstores import FAISS
9
+ from langchain_community.llms import HuggingFaceHub
10
+ from langchain_core.prompts import PromptTemplate
11
+
12
+
13
+ loader = CSVLoader("Events_SameDay.csv",encoding='iso-8859-1')
14
+
15
+ # Create an index using the loaded documents
16
+ index_creator = VectorstoreIndexCreator()
17
+ docsearch = index_creator.from_loaders([loader])
18
+
19
+ repo_id = "mistralai/Mistral-7B-Instruct-v0.1"
20
+ llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={"temperature": 0.1, "max_new_tokens": 1024})
21
+
22
+ # Create a question-answering chain using the index
23
+ chain = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=docsearch.vectorstore.as_retriever(), input_key="question")
24
+
25
+ question = """Retreive latest news on OPEC? """
26
+ result = chain({"question": question})