Method _task_assign

SWF/src/base.py:495–510

(self, tasks, survived_agent_names, statistics)

class StepRandomEnv(CacheEnv):

    def _task_assign(self, tasks, survived_agent_names, statistics):
        env_record, env_table = self.get_records(survived_agent_names)
        team = env_table['Name']
        idx = random.randint(0, 100000) % len(team)
        idx = self.name2idx[team[idx]]
        if int(os.environ.get('SLEEP', 0)) > 0:
            time.sleep(int(os.environ['SLEEP']))
        
        return Response(
            role='assistant',
            raw=self.agents[idx].name,
            content=self.agents[idx].name,
            end=True,
            usage=0,
            cost=0,
        )


@dataclass