| .. | ||
| docker | ||
| samples | ||
| custom_agent.py | ||
| custom_agent_agui.py | ||
| README.md | ||
haiku.rag Examples
This directory contains example scripts demonstrating various features of haiku.rag.
Docker Example
Directory: docker/
Complete Docker setup for running haiku.rag with all services:
- File monitoring for automatic document indexing
- MCP server for AI assistant integration
See docker/README.md for setup instructions.
Custom Agent
Script: custom_agent.py
Composes search, qa, and document toolsets into a pydantic-ai Agent using AgentDeps and prepare_context. Shows how to run queries and inspect accumulated state (citations, QA history).
uv run python examples/custom_agent.py /path/to/db.lancedb
Custom Agent with AG-UI Streaming
Script: custom_agent_agui.py
A Starlette app that serves an AG-UI streaming endpoint using composed toolsets, AgentDeps, and ToolContextCache for multi-session support.
DB_PATH=/path/to/db.lancedb uv run uvicorn examples.custom_agent_agui:app --reload --port 8000