csefrassia commited on
Commit
538f24c
·
1 Parent(s): cbd96fd

fix: prompting and answers

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -29,9 +29,11 @@ class BasicAgent:
29
  messages = [HumanMessage(content=question)]
30
  messages = self.graph.invoke({"messages": messages})
31
  answer = messages['messages'][-1].content
32
- # full_answer = messages['messages'][-1].content
 
 
33
  # print(f"Agent response: {full_answer}")
34
- return answer[14:]
35
  # return full_answer
36
 
37
  def run_and_submit_all( profile: gr.OAuthProfile | None):
 
29
  messages = [HumanMessage(content=question)]
30
  messages = self.graph.invoke({"messages": messages})
31
  answer = messages['messages'][-1].content
32
+ full_answer = messages['messages'][-1].content
33
+ if "Final answer:" in answer:
34
+ answer = full_answer.split("Final answer:")[-1].strip()
35
  # print(f"Agent response: {full_answer}")
36
+ return answer
37
  # return full_answer
38
 
39
  def run_and_submit_all( profile: gr.OAuthProfile | None):