Update app.py
Browse files
app.py
CHANGED
|
@@ -19,15 +19,17 @@ class AdvancedPdfChatbot:
|
|
| 19 |
os.environ["OPENAI_API_KEY"] = openai_api_key
|
| 20 |
self.embeddings = OpenAIEmbeddings()
|
| 21 |
self.text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
|
| 22 |
-
self.llm = ChatOpenAI(temperature=0,model_name='gpt-4o-mini')
|
| 23 |
|
| 24 |
self.memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
|
| 25 |
self.qa_chain = None
|
| 26 |
self.pdf_path = None
|
| 27 |
self.template = """
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
|
|
|
|
|
|
| 31 |
"""
|
| 32 |
self.prompt = PromptTemplate(template=self.template, input_variables=["context", "question"])
|
| 33 |
|
|
|
|
| 19 |
os.environ["OPENAI_API_KEY"] = openai_api_key
|
| 20 |
self.embeddings = OpenAIEmbeddings()
|
| 21 |
self.text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
|
| 22 |
+
self.llm = ChatOpenAI(temperature=0.1,model_name='gpt-4o-mini',max_tokens=3000, top_p = 0.1)
|
| 23 |
|
| 24 |
self.memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
|
| 25 |
self.qa_chain = None
|
| 26 |
self.pdf_path = None
|
| 27 |
self.template = """
|
| 28 |
+
I want you to act as a chat assistant for research more similar to ChatGPT itself, specifically focusing on knowledge retrieval. Provide detailed responses with great attention to context understanding and structured format. Be specific and detail-oriented in your responses, ensuring accuracy and depth in information provided.
|
| 29 |
+
Note: If you dont know the answer from the given knowledge base say you dont know it, **dont hallucinate**.
|
| 30 |
+
Context: {context}
|
| 31 |
+
Question: {question}
|
| 32 |
+
Answer:
|
| 33 |
"""
|
| 34 |
self.prompt = PromptTemplate(template=self.template, input_variables=["context", "question"])
|
| 35 |
|