File tree Expand file tree Collapse file tree
agents/matmaster_agent/flow_agents/plan_make_agent Expand file tree Collapse file tree Original file line number Diff line number Diff line change 99 DisallowTransferAndContentLimitSchemaAgent ,
1010)
1111from agents .matmaster_agent .logger import PrefixFilter
12+ from agents .matmaster_agent .state import MULTI_PLANS
1213from agents .matmaster_agent .utils .event_utils import update_state_event
1314
1415logger = logging .getLogger (__name__ )
1920class PlanMakeAgent (DisallowTransferAndContentLimitSchemaAgent ):
2021 @override
2122 async def _run_events (self , ctx : InvocationContext ) -> AsyncGenerator [Event , None ]:
22- async for event in super ()._run_events (ctx ):
23- yield event
23+ for _ in range (2 ):
24+ async for event in super ()._run_events (ctx ):
25+ yield event
2426
25- logger .info (
26- f'{ ctx .session .id } multi_plans = { ctx .session .state ["multi_plans" ]} '
27- )
27+ if ctx .session .state .get (MULTI_PLANS ):
28+ logger .info (
29+ f'{ ctx .session .id } multi_plans = { ctx .session .state [MULTI_PLANS ]} '
30+ )
31+ break
32+ else :
33+ logger .error (f'{ ctx .session .id } Multi Plans Generate Error, Retry' )
34+
35+ if not ctx .session .state .get (MULTI_PLANS ):
36+ raise RuntimeError (
37+ f'{ ctx .session .id } After Retry, Multi Plans Generate Still Error!!'
38+ )
2839
2940 # 计算 feasibility
3041 update_multi_plans = ctx .session .state ['multi_plans' ]
You can’t perform that action at this time.
0 commit comments