Update greg_funcs.py (#3)

- Update greg_funcs.py (aa445c467a585214eebdeb43dd5d10d1743a2b01)


Co-authored-by: Greg Hayworth <gfhayworth@users.noreply.huggingface.co>

greg_funcs.py

@@ -6,11 +6,9 @@ from datasets import load_dataset
 
 from langchain.llms import OpenAI
 from langchain.docstore.document import Document
-
+from langchain.chains.question_answering import load_qa_chain
 from langchain.chains.qa_with_sources import load_qa_with_sources_chain
-
-
-
+from langchain.prompts import PromptTemplate
 
 
 """# import models"""
@@ -64,8 +62,17 @@ def get_text_fmt(qry, passages = mypassages, doc_embedding=mycorpus_embeddings):
         prediction_text.append(result)
     return prediction_text
 
-
-chain_qa = load_qa_with_sources_chain(OpenAI(temperature=0), chain_type="stuff")
+template = """You are a friendly AI assistant for the insurance company Humana. Given the following extracted parts of a long document and a question, create a succinct final answer. 
+If you don't know the answer, just say that you don't know. Don't try to make up an answer.
+If the question is not about Humana, politely inform them that you are tuned to only answer questions about Humana.
+QUESTION: {question}
+=========
+{context}
+=========
+FINAL ANSWER:"""
+PROMPT = PromptTemplate(template=template, input_variables=["context", "question"])
+
+chain_qa = load_qa_chain(OpenAI(temperature=0), chain_type="stuff", prompt=PROMPT)
 
 
 def get_llm_response(message):