Update functions.py

functions.py

@@ -60,40 +60,44 @@ margin-bottom: 2.5rem">{}</div> """
 
 #Stuff Chain Type Prompt template
 
-# def load_prompt()
+@st.cacche_resource
+def load_prompt()
 
-#     system_template="""Use only the following pieces of earnings context to answer the users question thoroughly.
-#     Do not use any information not provided in the context and remember you are a finance expert.
-#     If you don't know the answer, just say that you don't know, don't try to make up an answer.
-#     ALWAYS return a "SOURCES" part in your answer.
-#     The "SOURCES" part should be a reference to the source of the document from which you got your answer.
+    system_template="""Use only the following pieces of earnings context to answer the users question accurately.
+    Do not use any information not provided in the earnings context and remember you are a to speak like a finance expert.
+    If you don't know the answer, just say 'There is no relevant answer in the given earnings call transcript', 
+    don't try to make up an answer.
     
-#     Remember, do not reference any information not given in the context. 
+    ALWAYS return a "SOURCES" part in your answer.
+    The "SOURCES" part should be a reference to the source of the document from which you got your answer.
     
-#     Follow the below format when answering:
+    Remember, do not reference any information not given in the context.
+
+    If the answer is not available in the given context just say 'There is no relevant answer in the given earnings call transcript'
     
-#     Question: [question here]
-#     Helpful Answer: [answer here]
-#     SOURCES: xyz
+    Follow the below format when answering:
     
-#     If there is no sources found please return the below:
+    Question: [question here]
+    Helpful Answer: [answer here]
+    SOURCES: xyz
     
-#     ```
-#     The answer is: foo
-#     SOURCES: Please refer to references section
-#     ```
+    If there is no sources found please return the below:
     
-#     Begin!
-#     ----------------
-#     {context}"""
+    ```
+    The answer is: foo
+    SOURCES: Please refer to references section
+    ```
+    Begin!
+    ----------------
+    {context}"""
     
-#     messages = [
-#         SystemMessagePromptTemplate.from_template(system_template),
-#         HumanMessagePromptTemplate.from_template("{question}")
-#     ]
-#     prompt = ChatPromptTemplate.from_messages(messages)
+    messages = [
+        SystemMessagePromptTemplate.from_template(system_template),
+        HumanMessagePromptTemplate.from_template("{question}")
+    ]
+    prompt = ChatPromptTemplate.from_messages(messages)
 
-#     return prompt
+    return prompt
 
 ###################### Functions #######################################################################################
 
@@ -205,6 +209,7 @@ def embed_text(query,title,embedding_model,_docsearch):
 
     chain = ConversationalRetrievalChain.from_llm(chat_llm, 
                                                   retriever= _docsearch.as_retriever(),
+                                                  qa_prompt = load_prompt(),
                                                   return_source_documents=True)
     
     answer = chain({"question": question, "chat_history": chat_history})