# 1. edit() - Replace entire source with new content
function = codebase.get_function("process_data")
function.edit("""
def process_data(input_data: dict) -> dict:
return transform(input_data)
""")
# 2. Replace - Substitute text while preserving context
class_def = codebase.get_class("UserModel")
class_def.replace("user_id", "account_id") # Updates all occurrences
# 3. Remove - Safely delete code with proper cleanup
unused_import = file.get_import("from utils import deprecated_func")
unused_import.remove() # Handles formatting, commas, etc
# 4. Insert - Add code before or after an element
function.insert_before("# Process user input") # Adds comment before function
function.insert_after("""
def validate_data(data: dict) -> bool:
return all(required in data for required in REQUIRED_FIELDS)
""") # Adds new function after