ArturG9 commited on
Commit
4e84954
1 Parent(s): 845722a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -120,7 +120,7 @@ def main():
120
  workflow.add_node("ask_question", lambda state: ask_question(state, retriever))
121
  workflow.add_node("retrieve", lambda state: retrieve(state, retriever))
122
  workflow.add_node("grade_documents", lambda state: grade_documents(state, retrieval_grader_grader(llm) )) # grade documents
123
- workflow.add_node("generate", generate) # generatae
124
  workflow.add_node("web_search", web_search) # web search
125
  workflow.add_node("transform_query", lambda state: transform_query(state,create_question_rewriter(llm) ))
126
 
 
120
  workflow.add_node("ask_question", lambda state: ask_question(state, retriever))
121
  workflow.add_node("retrieve", lambda state: retrieve(state, retriever))
122
  workflow.add_node("grade_documents", lambda state: grade_documents(state, retrieval_grader_grader(llm) )) # grade documents
123
+ workflow.add_node("generate", lambda state: generate(state,QA_chain(llm) )) # generatae
124
  workflow.add_node("web_search", web_search) # web search
125
  workflow.add_node("transform_query", lambda state: transform_query(state,create_question_rewriter(llm) ))
126