def METHOD_NAME(self, agent_id):
    agent = self._agent_dict["server"].get(agent_id)
    if agent is not None:
        return agent
    agent = self._agent_dict["client"].get(agent_id)
    if agent is not None:
        return agent
    raise ValueError(f"{agent_id} not found in current agent list")