Agent for cleaning up named entities in YouTube video transcripts.

demo-enrichment-agent.py

...

class DemoEnrichmentAgent:
    ...

    def __init__(self):
        ...
        graph_builder.add_node(
            "get_verified_entity_worker", self.get_verified_entity_worker
        )

        # set edges
        graph_builder.add_edge(START, "extractor")
        graph_builder.add_conditional_edges(
            "extractor", self.spawn_workers, ["get_verified_entity_worker"]
        )
        graph_builder.add_edge("get_verified_entity_worker", END)

        # compile the graph
        self.graph = graph_builder.compile()

    ...