Function replace_min_max_functions

src/chains/llm_math_chain.py:116–132

(expression)

def replace_min_max_functions(expression):
    # Parse the expression into an AST
    parsed_expression = ast.parse(expression, mode="eval")

    # Transform the AST
    transformer = MaxTransformer()
    transformed_ast = transformer.visit(parsed_expression)

    # Fix the missing locations in the AST
    transformed_ast = ast.fix_missing_locations(transformed_ast)

    # Compile the transformed AST
    compiled_code = compile(transformed_ast, "<string>", "eval")

    # Evaluate the compiled code
    result = eval(compiled_code)
    return str(result)


class LLMMathChain(Chain):