from praisonaiagents import Agent, Knowledge

# Create knowledge base from files
knowledge = Knowledge(
    sources=["docs/", "data.pdf", "faq.txt"]
)

# Create agent with knowledge
agent = Agent(
    name="Support Agent",
    instructions="Answer questions using the knowledge base.",
    knowledge=knowledge
)

result = agent.start("What is your refund policy?")

knowledge = Knowledge(
    sources=[
        "documents/",      # Directory
        "manual.pdf",      # PDF
        "faq.md",          # Markdown
        "data.csv"         # CSV
    ]
)

knowledge = Knowledge(
    sources=[
        "https://docs.example.com/api",
        "https://example.com/faq"
    ]
)

knowledge = Knowledge(
    sources=[],
    texts=[
        "Company policy: All refunds within 30 days.",
        "Support hours: 9 AM - 5 PM EST"
    ]
)

knowledge = Knowledge(
    sources=["docs/"],
    chunk_size=500,           # Characters per chunk
    chunk_overlap=50,         # Overlap between chunks
    embedding_model="text-embedding-3-small",
    vector_store="chroma",    # or "faiss", "qdrant"
    top_k=5                   # Results to retrieve
)