class WorkflowAssignedAgents:
    def __init__(self, workflow_actions=None):
        if workflow_actions is None:
            workflow_actions = {}
        self.actions = workflow_actions

    def get(self, key):
        assigned_entities = {
            "agentIds": [],
            "teamIds": []
        }
        for action in self.actions:
            if action.get('type', '') == key:
                agent_ids = [aid for aid in action['agentIds'] if aid]
                assigned_entities['agentIds'] = agent_ids
                team_ids = [tid for tid in action['teamIds'] if tid]
                assigned_entities['teamIds'] = team_ids

        return assigned_entities