Upload midterm.py

midterm.py

@@ -21,7 +21,7 @@ OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
 openai.api_key = OPENAI_API_KEY
 
 # Load PDF
-loader = PyMuPDFLoader("AirBnB10Q.pdf")
+loader = PyMuPDFLoader("./AirBnB10Q.pdf")
 documents = loader.load()
 
 # Split Document
@@ -46,7 +46,10 @@ retriever = qdrant_vector_store.as_retriever()
 # Pull LangChain QA Prompt Template
 retrieval_qa_prompt = hub.pull("langchain-ai/retrieval-qa-chat")
 
-template = """Answer the question based only on the following context. If you cannot answer the question with the context, please respond with 'I don't know':
+template = """You are a helpful assistant. Use only the context available in the file. The file is a PDF and is a company filing submitted to the SEC.
+        Pages include company information and detailed reports about financial performance. Pages contain tables, where some key information is found.
+        Table columns include name, title, and shares owned. Do not make up any information that is not in the file. 
+        Use the context provided in the file to answer the questions. Explain your answer by describing how you arrived at the answer:
 
 Context:
 {context}