@aymeric-roucher, I just realized that my proposed solution introduces a side effect: since static tools are decorated to enhance their safety, any output result that includes a reference to a static tool will actually use the decorated version instead.

For example:

code = "type_int = int"
result, _ = evaluate_python_code(code, {"int": int}, state={})
result is not int  # result is the decorated version of int

Do you think this could be problematic? Or would you consider this an acceptable edge case?

@albertvillanova I don't see an issue, don't think it is a problem yet. Could you double check that it doesn't disturb current tool use (as in, run some tests with our current tools) ? If that's good, let's merge and we can fix possible issues later on, if they arise.