Upload agent.py

src/agent.py

@@ -89,7 +89,11 @@ class Agent:
                 "callbacks": self.chat_model.callbacks
             }
         )
-        reply = json.loads(final_state["messages"][-1].content)
+        content: str = final_state["messages"][-1].content
+        if content.startwith("```json"):
+            # If the reply starts with a code fence, remove it
+            content = content[7:-3]
+        reply = json.loads(content)
         return reply["reasoning"], reply["answer"]
 
     def __build_graph(self):