githubmirror/ChatDev

chatdev/roster.py

20 строк658 B

thinkwee

Initial commit

04 сен 2023, 14:24

04 сен 2023, 14:24acb93cf

class Roster():
    def __init__(self) -> None:
        self.agents = list()

    def _recruit(self, agent_name: str):
        self.agents.append(agent_name)

    def _exist_employee(self, agent_name: str):
        names = self.agents + [agent_name]
        names = [name.lower().strip() for name in names]
        names = [name.replace(" ", "").replace("_", "") for name in names]
        agent_name = names[-1]
        if agent_name in names[:-1]:
            return True
        return False

    def _print_employees(self):
        names = self.agents
        names = [name.lower().strip() for name in names]
        print("Employees: {}".format(names))