Agent for cleaning up named entities in YouTube video transcripts.

demo-enrichment-agent.py

...
import json

...

class AgentState(TypedDict):
    transcript_text: str
    extracted_entities: NamedEntities

...

class DemoEnrichmentAgent:
    ...

    async def extractor_node(self, state: AgentState):
        ...

        return {"extracted_entities": resp_msg}

    async def start(self, transcript_text: str):
        ...

        # pretty print the agent's response JSON
        print("Agent response:\n", json.dumps(resp, indent=2, default=lambda o: o.__dict__))