Update key_extract.py

key_extract.py

@@ -1,6 +1,7 @@
 from langchain.prompts.prompt import PromptTemplate
 from langchain_community.llms import HuggingFaceEndpoint
-from langchain.chains import LLMChain
+#from langchain.chains import LLMChain
+from langchain_core.output_parsers import StrOutputParser
 from qa_txt import llm
 # llm = HuggingFaceEndpoint(
 #             repo_id='mistralai/Mixtral-8x7B-Instruct-v0.1',
@@ -18,6 +19,6 @@ template = """
         AI Assistant:"""
 
 prompt = PromptTemplate(input_variables=['input'], template = template)
-chain = LLMChain(llm=llm, prompt=prompt)
+chain = prompt | llm | StrOutputParser()