Agent for cleaning up named entities in YouTube video transcripts.

demo-enrichment-agent.py

...

class DemoEnrichmentAgent:
    ...

    def __init__(self):
        ...
        graph_builder.add_node("replace_entity", self.replace_entity_node)
        graph_builder.add_node("replacement_reviewer", self.replacement_reviewer_node)

        # set edges
        ...
        graph_builder.add_edge("get_verified_entity_worker", "replacement_reviewer")
        graph_builder.add_conditional_edges(
            "replacement_reviewer",
            self.continue_replacement_router,
            {"continue": "replace_entity", "end": END},
        )
        graph_builder.add_edge("replace_entity", "replacement_reviewer")

        ...

    async def replace_entity_node(self, state: AgentState):
        # 🤖 - I will replace the entities in the text
        pass

    async def replacement_reviewer_node(self, state: AgentState):
        # 🤖 - I will review the replacements for completeness
        pass

    def continue_replacement_router(self, state: AgentState):
        # 🔄 - I will decide if the loop should continue or stop
        pass

    ...