Function prepare_execution_globals

optillm/z3_solver.py:18–62

Prepare globals dictionary for Z3/SymPy code execution. WARNING: This is NOT a security sandbox. The name "execution_globals" reflects that this simply provides the execution environment for solver code, not a security boundary. The code is executed via exec() with access to z3, sy

()

    pass

def prepare_execution_globals():
    """
    Prepare globals dictionary for Z3/SymPy code execution.

    WARNING: This is NOT a security sandbox. The name "execution_globals" reflects
    that this simply provides the execution environment for solver code, not a
    security boundary. The code is executed via exec() with access to z3, sympy,
    and math libraries. Only execute trusted code.
    """
    execution_globals = {
        'print': print,
        '__builtins__': {
            'True': True,
            'False': False,
            'None': None,
            'abs': abs,
            'float': float,
            'int': int,
            'len': len,
            'max': max,
            'min': min,
            'round': round,
            'sum': sum,
            'complex': complex,
        }
    }
    
    # Add common math functions
    execution_globals.update({
        'log': math.log,
        'log2': math.log2,
        'sqrt': math.sqrt,
        'exp': math.exp,
        'sin': math.sin,
        'cos': math.cos,
        'tan': math.tan,
        'pi': math.pi,
        'e': math.e,
    })

    # Add complex number support
    execution_globals['I'] = complex(0, 1)
    execution_globals['Complex'] = complex

    return execution_globals

def execute_code_in_process(code: str):
    import z3