Agent for cleaning up named entities in YouTube video transcripts.

demo-enrichment-agent.py

...

class DemoEnrichmentAgent:
    ...

    async def extractor_node(self, state: AgentState):
        sys_prompt = SystemMessage(
            content=(
                'You are a master at data extraction. '
                'You will receive the transcript of a YouTube video. '
                'Extract all entities named in the video and '
                'provide some context for each one. '
                'ONLY output a JSON array of items each with 2 keys: '
                '`entity_name` and `entity_context`. For example, '
                '`[{ "entity_name": "Jamaican", "entity_context": "vacation spot" }]`. '
                'The context will be used for a web search so it should be '
                'brief and without fluff.'
            )
        )
        ...