From a61ebebaa0d873997f6ac3ebb33d22a7655c6d97 Mon Sep 17 00:00:00 2001
From: ochafik <ochafik@google.com>
Date: Wed, 10 Apr 2024 01:22:09 +0100
Subject: [PATCH] agent: hint at math import in python tool

---
 examples/agent/tools/unsafe_python_tools.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/agent/tools/unsafe_python_tools.py b/examples/agent/tools/unsafe_python_tools.py
index 4a8a103c5..0a3a2cb84 100644
--- a/examples/agent/tools/unsafe_python_tools.py
+++ b/examples/agent/tools/unsafe_python_tools.py
@@ -6,7 +6,7 @@ from typing import Dict, Union
 def execute_python(source: str) -> Union[Dict, str]:
     """
         Evaluate a Python program and return the globals it declared.
-        Can be used to compute mathematical expressions.
+        Can be used to compute mathematical expressions (e.g. after importing math module).
 
         Args:
             source: contain valid, executable and pure Python code. Should also import any required Python packages.