RuaZhou commited on
Commit
9d41a19
·
verified ·
1 Parent(s): 9f73cd3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -135,8 +135,8 @@ class BasicAgent:
135
  print(f"Agent received question (first 50 chars): {question[:50]}...")
136
  messages = [HumanMessage(content=f"{question}")]
137
 
138
- messages = self.agent.invoke({"messages": messages})
139
- answer = f"{messages[-1].content}"
140
  print(f"Agent returning answer: {answer}")
141
  return answer
142
 
 
135
  print(f"Agent received question (first 50 chars): {question[:50]}...")
136
  messages = [HumanMessage(content=f"{question}")]
137
 
138
+ state = self.agent.invoke({"messages": messages,"input_file": None})
139
+ answer = state["messages"][-1].content
140
  print(f"Agent returning answer: {answer}")
141
  return answer
142