ArturG9 commited on
Commit
a9412b7
·
verified ·
1 Parent(s): a3ea4fc

Update functions.py

Browse files
Files changed (1) hide show
  1. functions.py +1 -1
functions.py CHANGED
@@ -568,7 +568,7 @@ def QA_chain(llm):
568
  )
569
 
570
 
571
- rag_chain = LLMChain(prompt=prompt, llm=llm, output_parser=StrOutputParser())
572
 
573
 
574
  return rag_chain
 
568
  )
569
 
570
 
571
+ rag_chain = prompt | llm | StrOutputParser()
572
 
573
 
574
  return rag_chain