mikethor007 commited on
Commit
e198ae4
·
verified ·
1 Parent(s): b24a24f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -136,13 +136,15 @@ class BasicAgent:
136
  for event in events:
137
  listMessages.extend(event["messages"])
138
 
 
 
139
  answer = listMessages[-1].content
140
 
141
  #answer = supervisor.invoke(messages)
142
- print(f"Answer: {answer}...")
143
 
144
- final_answer = re.search(r"FINAL ANSWER:\s*(.*)", answer)
145
- print(f"Agent returning answer: {final_answer}")
146
  return final_answer
147
 
148
  def run_and_submit_all( profile: gr.OAuthProfile | None):
 
136
  for event in events:
137
  listMessages.extend(event["messages"])
138
 
139
+ print(f"messages: {listMessages}/n")
140
+
141
  answer = listMessages[-1].content
142
 
143
  #answer = supervisor.invoke(messages)
144
+ print(f"Answer: {answer}/n")
145
 
146
+ final_answer = answer.removeprefix("FINAL ANSWER: ") #re.search(r"FINAL ANSWER:\s*(.*)", answer)
147
+ print(f"Agent returning answer: {final_answer}/n")
148
  return final_answer
149
 
150
  def run_and_submit_all( profile: gr.OAuthProfile | None):