Agent for cleaning up named entities in YouTube video transcripts.

demo-enrichment-agent.py

...
from typing import Annotated
import operator

...

class WorkerState(TypedDict):
    named_entity: NamedEntity
    verified_entities: Annotated[list, operator.add]

class DemoEnrichmentAgent:
    ...

    async def get_verified_entity_worker(self, state: WorkerState):
        entity = state["named_entity"]
        search_res = await self.research_entity(entity)
        verified_entity = await self.get_canonical_name(entity, search_res)
        return { "verified_entities": [verified_entity] }

    ...