ArturG9 commited on
Commit
472c309
1 Parent(s): eb18342

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -118,8 +118,8 @@ def main():
118
  workflow = StateGraph(GraphState)
119
 
120
  # Define the nodes
121
- workflow.add_node("ask_question", ask_question)
122
- workflow.add_node("retrieve", retrieve) # retrieve
123
  workflow.add_node("grade_documents", grade_documents) # grade documents
124
  workflow.add_node("generate", generate) # generatae
125
  workflow.add_node("web_search", web_search) # web search
 
118
  workflow = StateGraph(GraphState)
119
 
120
  # Define the nodes
121
+ workflow.add_node("ask_question", ask_question(retriever))
122
+ workflow.add_node("retrieve", retrieve(retriever))
123
  workflow.add_node("grade_documents", grade_documents) # grade documents
124
  workflow.add_node("generate", generate) # generatae
125
  workflow.add_node("web_search", web_search) # web search