Knowledge Bases Overview

from lyzr import Studio

studio = Studio(api_key="your-api-key")

# Create a knowledge base
kb = studio.create_knowledge_base(
    name="product_docs",
    vector_store="qdrant",
    embedding_model="text-embedding-3-large"
)

# Add documents
kb.add_pdf("manual.pdf")
kb.add_website("https://docs.example.com", max_pages=50)

# Create an agent with the knowledge base
agent = studio.create_agent(
    name="Support Bot",
    provider="gpt-4o",
    role="Customer support",
    goal="Answer questions using documentation",
    instructions="Use the knowledge base to answer questions accurately"
)

# Query with the knowledge base
response = agent.run(
    "How do I reset my password?",
    knowledge_bases=[kb]
)
print(response.response)

kb = studio.create_knowledge_base(
    name="my_kb",
    vector_store="qdrant"
)

kb.add_pdf("document.pdf")
kb.add_website("https://example.com")
kb.add_text("Custom content", source="custom")

results = kb.query("search term", top_k=5)
for result in results:
    print(f"{result.score:.2f}: {result.text}")

response = agent.run(
    "What is the return policy?",
    knowledge_bases=[kb]
)