Update rag_engine.py

rag_engine.py

@@ -11,6 +11,8 @@ from langchain.docstore.document import Document
 import pinecone
 import chainlit as cl
 from chainlit.types import AskFileResponse
+from langchain.prompts import PromptTemplate
+
 from dotenv import load_dotenv
 load_dotenv()
 
@@ -112,12 +114,27 @@ async def start():
             chat_memory=message_history,
             return_messages=True,
         )
+        PROMPT = PromptTemplate(
+            template="""Your name is Skandhaar docchat and you are working for Skandhaar org. and your job is to answer the user question from the given context. You are not allowed make an answer and create something that's not there in the context. You strictly follow the context and give extractive answers.
+            Respond for user greetings. If you encounter with out of context questions reply with I'm here to help you with given knowledge source, i can't assist with that.
+            
+            context:{context}
+            
+            question:{question}
+            
+            Answer in the Markdown.
+            
+            """, 
+            input_variables=["context", "question"]
+        )
 
+        chain_type_kwargs = {"prompt": PROMPT}
         chain = RetrievalQA.from_chain_type(
             ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0, streaming=True, openai_api_key=openai_api_key),
             chain_type="stuff",
             retriever=docsearch.as_retriever(),
             return_source_documents=True,
+            chain_type_kwargs=chain_type_kwargs
         )
 
         # Let the user know that the system is ready