Final_Assignment_Template

Sleeping

notsakeeb commited on May 15

Commit

c14c18f

verified ·

1 Parent(s): a17efc4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -69,7 +69,7 @@ class BasicAgent:
         messages = [HumanMessage(content=question)]
         messages = self.graph.invoke({"messages": messages})
         answer = messages['messages'][-1].content
-        m = re.search(r"FINAL ANSWER:\s*(.*)", answer, flags=re.DOTALL)
         if m:
             result = m.group(1)
         else:

         messages = [HumanMessage(content=question)]
         messages = self.graph.invoke({"messages": messages})
         answer = messages['messages'][-1].content
+        m = re.search(r"delim_\s*(.*)", answer, flags=re.DOTALL)
         if m:
             result = m.group(1)
         else: