Upload agent.py

src/agent.py

@@ -46,7 +46,7 @@ class Agent:
         )
         self.graph = self.__build_graph()
 
-    def __call__(self, question: str) -> str:
+    def __call__(self, question: str) -> tuple[str, str]:
         """
         Reply to a question using the agent and return the agents full reply
         with reasoning included.
@@ -64,7 +64,8 @@ class Agent:
                 "callbacks": self.chat_model.callbacks
             }
         )
-        return final_state["messages"][-1].content
+        reply = json.loads(final_state["messages"][-1].content)
+        return reply["reasoning"], reply["answer"]
 
     def __build_graph(self):
         """