enoreyes commited on
Commit
160dfa5
·
1 Parent(s): f31c308

Update chain.py

Browse files
Files changed (1) hide show
  1. chain.py +1 -4
chain.py CHANGED
@@ -99,10 +99,7 @@ def get_new_chain1(vectorstore, rephraser_llm, final_output_llm, isFlan) -> Chai
99
  {context}
100
  Based on the above documentation, answer the user's question in markdown. If you can't answer, say "For this topic, I reccomend viewing the docs": {question}"""
101
 
102
- PROMPT = PromptTemplate(template=gpt_template, input_variables=["question", "context"])
103
-
104
- if isFlan:
105
- PROMPT = PromptTemplate(template=flan_template, input_variables=["question", "context"])
106
 
107
  doc_chain = load_qa_chain(
108
  final_output_llm,
 
99
  {context}
100
  Based on the above documentation, answer the user's question in markdown. If you can't answer, say "For this topic, I reccomend viewing the docs": {question}"""
101
 
102
+ PROMPT = PromptTemplate(template=flan_template, input_variables=["question", "context"])
 
 
 
103
 
104
  doc_chain = load_qa_chain(
105
  final_output_llm,