Agent for cleaning up named entities in YouTube video transcripts.

demo-enrichment-agent.py

...
from langgraph.types import Send

...

class DemoEnrichmentAgent:
    ...

    async def spawn_workers(self, state: AgentState):
        extracted_entities = state["extracted_entities"]

        # spawn a worker to verify each extracted entity
        return [
            Send("get_verified_entity_worker", {"named_entity": ne})
            for ne in extracted_entities.named_entities
        ]

    ...