Agent for cleaning up named entities in YouTube video transcripts.

demo-enrichment-agent.py

...

class DemoEnrichmentAgent:
    ...

    async def start(self, transcript_text: str):
        print("Invoking agent")
        resp = await self.graph.ainvoke({ "transcript_text": transcript_text })
        print(f"Agent response: {resp}")

if __name__ == "__main__":
    import asyncio
    from dotenv import load_dotenv

    load_dotenv()

    enrichment_agent = DemoEnrichmentAgent()
    vid_transcript = "Hollywood director Kristoffer Nolen has announced his next big project, rumored to be another complex sci-fi thriller. Fans of Nolen’s earlier work are eager to see if this film will rival the success of ‘Inceptshun’ or ‘Dunkrik.’"
    asyncio.run(enrichment_agent.start(vid_transcript))