Agent for cleaning up named entities in YouTube video transcripts.

demo-enrichment-agent.py

...
import os
from tavily import AsyncTavilyClient

...

class DemoEnrichmentAgent:
    ...
    tavily_client: AsyncTavilyClient

    def __init__(self):
        ...
        self.tavily_client = AsyncTavilyClient(api_key=os.getenv("TAVILY_API_KEY"))

    async def research_entity(self, entity: NamedEntity):
        print(f"Performing web search to verify entity: {entity.entity_name}")
        search_response = await self.tavily_client.search(
            f"{entity.entity_name} {entity.entity_context}", max_results=3
        )
        search_results = search_response.get("results", [])
        print(f"Found {len(search_results)} results for entity: {entity.entity_name}")
        return search_results

    ...