kenqia commited on
Commit
e57e3bd
·
verified ·
1 Parent(s): dbe065b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -83,12 +83,12 @@ class BasicAgent:
83
  builder.add_edge("final_process", END)
84
  self.react_graph = builder.compile()
85
 
86
- def __call__(self, question: str) -> str:
87
  print(f"Agent received question: {question[:200]}", flush=True)
88
 
89
  try:
90
 
91
- answer = self.answer_question(question)
92
  answer = str(answer).strip()
93
  print(f"Agent answer: {answer[:200]}", flush=True)
94
  return answer
 
83
  builder.add_edge("final_process", END)
84
  self.react_graph = builder.compile()
85
 
86
+ def __call__(self, question: str, task_id: str | None = None) -> str:
87
  print(f"Agent received question: {question[:200]}", flush=True)
88
 
89
  try:
90
 
91
+ answer = self.answer_question(question, task_id)
92
  answer = str(answer).strip()
93
  print(f"Agent answer: {answer[:200]}", flush=True)
94
  return answer