Update agent.py

agent.py

@@ -185,7 +185,13 @@ def image_caption(dir: str) -> str:
 
 # 2. Coding
 from langchain_experimental.tools import PythonREPLTool
-python_tool = PythonREPLTool()
+@tool
+def run_python(code: str):
+    """ Run the given python code
+    Args:
+        code: the python code
+    """
+    return PythonREPLTool().run(code)
 
 @tool
 def multiply(a: float, b: float) -> float:
@@ -273,11 +279,12 @@ class BasicAgent:
                 - csv_read: read the content of the attached csv file by passing the TASK-ID as input
                 - mp3_listen: listen to the content of the attached mp3 file by passing the TASK-ID as input
                 - image_caption: understand the visual content of the attached image by passing the TASK-ID as input
-                - python_tool: run the python code
-                
+                - run_python: run the python code
+
                 If Task ID is included in the question, remember to call the relevant read tools [ie. read_file, excel_read, csv_read, mp3_listen, image_caption]
+                Note: python_tool is called when the question mentions the term "Python" or any math calculation.
         """
-        self.tools = [duckduck_websearch, serper_websearch, visit_webpage, wiki_search, text_splitter, youtube_transcript, read_file, excel_read, csv_read, mp3_listen, image_caption, python_tool]
+        self.tools = [duckduck_websearch, serper_websearch, visit_webpage, wiki_search, text_splitter, youtube_transcript, read_file, excel_read, csv_read, mp3_listen, image_caption, run_python]
         # self.model_with_tools = self.model.bind_tools(self.tools)
         # self.sys_msg = SystemMessage(content=self.sys_prompt)