src/agents/executor.py

from src.utils.shared import llm_model

def execute_step(step):
    prompt = f"""You are an expert Python developer.
Generate Python code for this step:

Step: {step}

Output only code, no explanations."""
    return llm_model(prompt)