File size: 447 Bytes
d64fd55
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from core.pipeline.step import PipelineStep

class PlanStep(PipelineStep):
    name = "plan_agent"

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

    async def run(self, context):
        plan_result = await self.agent.run(
            context.summary,
            language=context.language,
            profile=context.runner_profile,
            goal=context.active_goal
        )
        context.plan = plan_result.get("plan", "")