Update agent.py

agent.py

@@ -8,7 +8,7 @@ from langgraph.graph import StateGraph, START, END
 from typing import Literal
 
 from langchain_community.tools import BraveSearch  # web search
-from langchain.tools.python.tool import PythonAstREPLTool  # for logic/math problems
+from langchain.tools.python.tool import PythonREPLTool  # for logic/math problems
 
 from tools import (datetime_tools, transcribe_audio_tool)
 from prompt import system_prompt