from core.pipeline.step import PipelineStep

class FeatureEngineeringStep(PipelineStep):
    name = "feature_engineering"

    def __init__(self, agent):
        self.agent = agent

    async def run(self, context):
        features_result = self.agent.run(context.runs)
        context.runs = features_result.get("detailed_features", [])