Wikipedia Reader (Async)

import asyncio

from agno.agent import Agent
from agno.knowledge.knowledge import Knowledge
from agno.knowledge.reader.arxiv_reader import ArxivReader
from agno.knowledge.reader.wikipedia_reader import WikipediaReader
from agno.vectordb.pgvector import PgVector

# Create Knowledge Instance
knowledge = Knowledge(
    name="Multi-Source Knowledge Base",
    description="Knowledge base combining Wikipedia and ArXiv content",
    vector_db=PgVector(
        table_name="multi_vectors", 
        db_url="postgresql+psycopg://ai:ai@localhost:5532/ai"
    ),
)

async def main():
    # Add topics from Wikipedia
    await knowledge.add_content_async(
        metadata={"source": "wikipedia", "type": "encyclopedia"},
        topics=["Manchester United", "Machine Learning"],
        reader=WikipediaReader(),
    )

    # Add topics from ArXiv
    await knowledge.add_content_async(
        metadata={"source": "arxiv", "type": "academic"},
        topics=["Carbon Dioxide", "Neural Networks"],
        reader=ArxivReader(),
    )

    # Create an agent with the knowledge
    agent = Agent(
        knowledge=knowledge,
        search_knowledge=True,
    )

    # Query the knowledge base
    await agent.aprint_response(
        "What can you tell me about Machine Learning from both general and academic sources?",
        markdown=True
    )

if __name__ == "__main__":
    asyncio.run(main())