Agent for cleaning up named entities in YouTube video transcripts.

demo-enrichment-agent.py

...

class AgentState(TypedDict):
    ...
    replacement_loop_idx: Annotated[int, operator.add]

...

class DemoEnrichmentAgent:
    ...

    def continue_replacement_router(self, state: AgentState):
        replacement_loop_idx = state.get("replacement_loop_idx", 0)
        if replacement_loop_idx < len(state.get("verified_entities", [])):
            print(f"Replacement ongoing for entity at index {replacement_loop_idx}")
            return "continue"

        print(f"Replacement complete for all entities")
        return "end"

    ...