Add authorized_imports in importFrom errors (#506)

huggingface · Feb 9, 2025 · 01c5106 · 01c5106
1 parent 881ce13
commit 01c5106
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/smolagents/local_python_executor.py b/src/smolagents/local_python_executor.py
@@ -1067,7 +1067,9 @@ def check_module_authorized(module_name):
                     else:
                         raise InterpreterError(f"Module {expression.module} has no attribute {alias.name}")
         else:
-            raise InterpreterError(f"Import from {expression.module} is not allowed.")
+            raise InterpreterError(
+                f"Import from {expression.module} is not allowed. Authorized imports are: {str(authorized_imports)}"
+            )
         return None