mirror of
https://github.com/OpenBMB/ChatDev.git
synced 2024-09-16 04:37:11 +03:00
21 lines
658 B
Python
21 lines
658 B
Python
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))
|