Support for "complex" in python executor

[joaopauloschuler]

The following code produced by DeepSeek fails:

def generate_mandelbrot(width, height, x_min, x_max, y_min, y_max, max_iter):                                      
    image = np.zeros((height, width))                                                                              
    for row in range(height):                                                                                      
        for col in range(width):                                                                                   
            x = x_min + (x_max - x_min) * col / width                                                              
            y = y_min + (y_max - y_min) * row / height                                                             
            c = complex(x, y)                                                                                      
            m = mandelbrot(c, max_iter)                                                                            
            color = 1 - m / max_iter                                                                               
            image[row, col] = color                                                                                
    return image 

[batrlatom]

what is the fail message? I think that in this line :

        m = mandelbrot(c, max_iter)                                                                            

mandelbrot is not defined. And if there is some import needs to be done, you can add it to the additional imports like this:

manager_agent = CodeAgent(
tools=[],
model=model,
additional_authorized_imports=["time", "numpy", "pandas"],
)

Also, this is not secure, but you can add specific function names in the local_python_executor here:

BASE_PYTHON_TOOLS = {
"print": custom_print,
"isinstance": isinstance,
"range": range,
"float": float,
"int": int,
"bool": bool,
"str": str,
"set": set,
"list": list,
"dict": dict,
"tuple": tuple,
"round": round,
"ceil": math.ceil,
"floor": math.floor,
"log": math.log,
"exp": math.exp,
"sin": math.sin,
"cos": math.cos,
"tan": math.tan,
"asin": math.asin,
"acos": math.acos,
"atan": math.atan,
"atan2": math.atan2,
"degrees": math.degrees,
"radians": math.radians,
"pow": math.pow,
"sqrt": math.sqrt,
"len": len,
"sum": sum,
"max": max,
"min": min,
"abs": abs,
"enumerate": enumerate,
"zip": zip,
"reversed": reversed,
"sorted": sorted,
"all": all,
"any": any,
"map": map,
"filter": filter,
"ord": ord,
"chr": chr,
"next": next,
"iter": iter,
"divmod": divmod,
"callable": callable,
"getattr": getattr,
"hasattr": hasattr,
"setattr": setattr,
"issubclass": issubclass,
"type": type,
}

I, for example added "open": open, for executor to open files natively

[joaopauloschuler]

The problem happens at: c = complex(x, y). Adding complex to BASE_PYTHON_TOOLS fixes the problem. This is the fix: #79 .

[aymeric-roucher]

Closed by merging #79! 😄