PradeepBodhi commited on
Commit
fbfef16
·
verified ·
1 Parent(s): cf2a43c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -24,14 +24,14 @@ class BasicAgent:
24
  self.graph = build_graph()
25
 
26
  def __call__(self, question: str) -> str:
27
- try:
28
- messages = [HumanMessage(content=question)]
29
- messages = self.graph.invoke({"messages": messages})
30
- answer = messages['messages'][-1].content
31
- return answer[14:] if answer.startswith("FINAL ANSWER:") else answer
32
- except Exception as e:
33
- print(f"Agent Error Details: {str(e)}")
34
- return f"AGENT ERROR: {str(e)}"
35
 
36
 
37
  def run_and_submit_all( profile: gr.OAuthProfile | None):
 
24
  self.graph = build_graph()
25
 
26
  def __call__(self, question: str) -> str:
27
+ try:
28
+ messages = [HumanMessage(content=question)]
29
+ messages = self.graph.invoke({"messages": messages})
30
+ answer = messages['messages'][-1].content
31
+ return answer[14:] if answer.startswith("FINAL ANSWER:") else answer
32
+ except Exception as e:
33
+ print(f"Agent Error Details: {str(e)}")
34
+ return f"AGENT ERROR: {str(e)}"
35
 
36
 
37
  def run_and_submit_all( profile: gr.OAuthProfile | None):