mirjam-m commited on
Commit
13ded9e
·
1 Parent(s): b3c88ed

graph test

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -112,7 +112,9 @@ class BasicAgent:
112
  print(f"[log_question] Agent received question: {state['question']}...")
113
  state["messages"] = [SystemMessage(content=PROMPT)]
114
  state["messages"].append(HumanMessage(content="Question: " + state["question"]))
115
- return {}
 
 
116
 
117
  def final_answer(self, state: AnswerState) -> Dict[str, Any]:
118
  print(f"[final_answer] Agent returning answer: {state['answer']}")
 
112
  print(f"[log_question] Agent received question: {state['question']}...")
113
  state["messages"] = [SystemMessage(content=PROMPT)]
114
  state["messages"].append(HumanMessage(content="Question: " + state["question"]))
115
+ return {
116
+ "messages": state["messages"],
117
+ }
118
 
119
  def final_answer(self, state: AnswerState) -> Dict[str, Any]:
120
  print(f"[final_answer] Agent returning answer: {state['answer']}")