vanna 2.0.0rc1__py3-none-any.whl → 2.0.2__py3-none-any.whl

vanna-2.0.0rc1.dist-info/METADATA

@@ -1,868 +0,0 @@
-Metadata-Version: 2.4
-Name: vanna
-Version: 2.0.0rc1
-Summary: Generate SQL queries from natural language
-Author-email: Zain Hoda <zain@vanna.ai>
-Requires-Python: >=3.9
-Description-Content-Type: text/markdown
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-License-File: LICENSE
-Requires-Dist: pydantic>=2.0.0
-Requires-Dist: click>=8.0.0
-Requires-Dist: pandas
-Requires-Dist: httpx>=0.28.0
-Requires-Dist: PyYAML
-Requires-Dist: plotly
-Requires-Dist: tabulate
-Requires-Dist: sqlparse
-Requires-Dist: sqlalchemy
-Requires-Dist: requests
-Requires-Dist: psycopg2-binary ; extra == "all"
-Requires-Dist: db-dtypes ; extra == "all"
-Requires-Dist: PyMySQL ; extra == "all"
-Requires-Dist: google-cloud-bigquery ; extra == "all"
-Requires-Dist: snowflake-connector-python ; extra == "all"
-Requires-Dist: duckdb ; extra == "all"
-Requires-Dist: openai ; extra == "all"
-Requires-Dist: qianfan ; extra == "all"
-Requires-Dist: mistralai>=1.0.0 ; extra == "all"
-Requires-Dist: chromadb>=1.1.0 ; extra == "all"
-Requires-Dist: anthropic ; extra == "all"
-Requires-Dist: zhipuai ; extra == "all"
-Requires-Dist: marqo ; extra == "all"
-Requires-Dist: google-generativeai ; extra == "all"
-Requires-Dist: google-cloud-aiplatform ; extra == "all"
-Requires-Dist: qdrant-client ; extra == "all"
-Requires-Dist: fastembed ; extra == "all"
-Requires-Dist: ollama ; extra == "all"
-Requires-Dist: httpx ; extra == "all"
-Requires-Dist: opensearch-py ; extra == "all"
-Requires-Dist: opensearch-dsl ; extra == "all"
-Requires-Dist: transformers ; extra == "all"
-Requires-Dist: pinecone ; extra == "all"
-Requires-Dist: pymilvus[model] ; extra == "all"
-Requires-Dist: weaviate-client ; extra == "all"
-Requires-Dist: azure-search-documents ; extra == "all"
-Requires-Dist: azure-identity ; extra == "all"
-Requires-Dist: azure-common ; extra == "all"
-Requires-Dist: faiss-cpu ; extra == "all"
-Requires-Dist: boto ; extra == "all"
-Requires-Dist: boto3 ; extra == "all"
-Requires-Dist: botocore ; extra == "all"
-Requires-Dist: langchain_core ; extra == "all"
-Requires-Dist: langchain_postgres ; extra == "all"
-Requires-Dist: langchain-community ; extra == "all"
-Requires-Dist: langchain-huggingface ; extra == "all"
-Requires-Dist: xinference-client ; extra == "all"
-Requires-Dist: anthropic ; extra == "anthropic"
-Requires-Dist: openai ; extra == "azureopenai"
-Requires-Dist: azure-identity ; extra == "azureopenai"
-Requires-Dist: azure-search-documents ; extra == "azuresearch"
-Requires-Dist: azure-identity ; extra == "azuresearch"
-Requires-Dist: azure-common ; extra == "azuresearch"
-Requires-Dist: fastembed ; extra == "azuresearch"
-Requires-Dist: boto3 ; extra == "bedrock"
-Requires-Dist: botocore ; extra == "bedrock"
-Requires-Dist: google-cloud-bigquery ; extra == "bigquery"
-Requires-Dist: chromadb>=1.1.0 ; extra == "chromadb"
-Requires-Dist: clickhouse_connect ; extra == "clickhouse"
-Requires-Dist: pytest>=7.0.0 ; extra == "dev"
-Requires-Dist: pytest-asyncio>=0.21.0 ; extra == "dev"
-Requires-Dist: pytest-mock>=3.10.0 ; extra == "dev"
-Requires-Dist: pytest-cov>=4.0.0 ; extra == "dev"
-Requires-Dist: tox>=4.0.0 ; extra == "dev"
-Requires-Dist: mypy ; extra == "dev"
-Requires-Dist: ruff ; extra == "dev"
-Requires-Dist: pandas-stubs ; extra == "dev"
-Requires-Dist: plotly-stubs ; extra == "dev"
-Requires-Dist: types-PyYAML ; extra == "dev"
-Requires-Dist: types-requests ; extra == "dev"
-Requires-Dist: types-tabulate ; extra == "dev"
-Requires-Dist: duckdb ; extra == "duckdb"
-Requires-Dist: faiss-cpu ; extra == "faiss-cpu"
-Requires-Dist: faiss-gpu ; extra == "faiss-gpu"
-Requires-Dist: fastapi>=0.68.0 ; extra == "fastapi"
-Requires-Dist: uvicorn>=0.15.0 ; extra == "fastapi"
-Requires-Dist: flask>=2.0.0 ; extra == "flask"
-Requires-Dist: flask-cors>=4.0.0 ; extra == "flask"
-Requires-Dist: google-genai ; extra == "gemini"
-Requires-Dist: google-generativeai ; extra == "google"
-Requires-Dist: google-cloud-aiplatform ; extra == "google"
-Requires-Dist: transformers ; extra == "hf"
-Requires-Dist: pyhive ; extra == "hive"
-Requires-Dist: thrift ; extra == "hive"
-Requires-Dist: marqo ; extra == "marqo"
-Requires-Dist: pymilvus[model] ; extra == "milvus"
-Requires-Dist: mistralai>=1.0.0 ; extra == "mistralai"
-Requires-Dist: pyodbc ; extra == "mssql"
-Requires-Dist: PyMySQL ; extra == "mysql"
-Requires-Dist: ollama ; extra == "ollama"
-Requires-Dist: httpx ; extra == "ollama"
-Requires-Dist: openai ; extra == "openai"
-Requires-Dist: opensearch-py ; extra == "opensearch"
-Requires-Dist: opensearch-dsl ; extra == "opensearch"
-Requires-Dist: langchain-community ; extra == "opensearch"
-Requires-Dist: langchain-huggingface ; extra == "opensearch"
-Requires-Dist: oracledb ; extra == "oracle"
-Requires-Dist: chromadb<1.0.0 ; extra == "oracle"
-Requires-Dist: langchain-postgres>=0.0.12 ; extra == "pgvector"
-Requires-Dist: pinecone ; extra == "pinecone"
-Requires-Dist: fastembed ; extra == "pinecone"
-Requires-Dist: psycopg2-binary ; extra == "postgres"
-Requires-Dist: db-dtypes ; extra == "postgres"
-Requires-Dist: pyhive ; extra == "presto"
-Requires-Dist: thrift ; extra == "presto"
-Requires-Dist: qdrant-client ; extra == "qdrant"
-Requires-Dist: fastembed ; extra == "qdrant"
-Requires-Dist: qianfan ; extra == "qianfan"
-Requires-Dist: vanna[flask, fastapi] ; extra == "servers"
-Requires-Dist: snowflake-connector-python ; extra == "snowflake"
-Requires-Dist: pytest>=7.0.0 ; extra == "test"
-Requires-Dist: pytest-asyncio>=0.21.0 ; extra == "test"
-Requires-Dist: pytest-mock>=3.10.0 ; extra == "test"
-Requires-Dist: pytest-cov>=4.0.0 ; extra == "test"
-Requires-Dist: tox>=4.0.0 ; extra == "test"
-Requires-Dist: vllm ; extra == "vllm"
-Requires-Dist: weaviate-client ; extra == "weaviate"
-Requires-Dist: xinference-client ; extra == "xinference-client"
-Requires-Dist: zhipuai ; extra == "zhipuai"
-Project-URL: Bug Tracker, https://github.com/vanna-ai/vanna/issues
-Project-URL: Homepage, https://github.com/vanna-ai/vanna
-Provides-Extra: all
-Provides-Extra: anthropic
-Provides-Extra: azureopenai
-Provides-Extra: azuresearch
-Provides-Extra: bedrock
-Provides-Extra: bigquery
-Provides-Extra: chromadb
-Provides-Extra: clickhouse
-Provides-Extra: dev
-Provides-Extra: duckdb
-Provides-Extra: faiss-cpu
-Provides-Extra: faiss-gpu
-Provides-Extra: fastapi
-Provides-Extra: flask
-Provides-Extra: gemini
-Provides-Extra: google
-Provides-Extra: hf
-Provides-Extra: hive
-Provides-Extra: marqo
-Provides-Extra: milvus
-Provides-Extra: mistralai
-Provides-Extra: mssql
-Provides-Extra: mysql
-Provides-Extra: ollama
-Provides-Extra: openai
-Provides-Extra: opensearch
-Provides-Extra: oracle
-Provides-Extra: pgvector
-Provides-Extra: pinecone
-Provides-Extra: postgres
-Provides-Extra: presto
-Provides-Extra: qdrant
-Provides-Extra: qianfan
-Provides-Extra: servers
-Provides-Extra: snowflake
-Provides-Extra: test
-Provides-Extra: vllm
-Provides-Extra: weaviate
-Provides-Extra: xinference-client
-Provides-Extra: zhipuai
-
-# Vanna 2.0+: Web-First, User-Aware Agent Framework
-
-> [!WARNING]
-> This version of Vanna is actively under development and may contain breaking changes until it is officially released to PyPI.
->
-> To install while in development, use:
-> ```bash
-> pip install --force-reinstall --no-cache-dir 'vanna[flask,anthropic] @ git+https://github.com/vanna-ai/vanna.git@v2'
-> ```
-
-> [!IMPORTANT]
-> If you're upgrading from an older version of Vanna, use the [Migration Guide](MIGRATION_GUIDE.md).
-
-> **Turn natural language into data insights — with enterprise-grade security baked in**
-
-Vanna is purpose-built for **data analytics** with **user awareness** as a first-class concern. Drop in our web component, connect your existing auth, and start querying data securely.
-
-[![Python](https://img.shields.io/badge/python-3.8+-blue.svg)](https://python.org)
-[![License](https://img.shields.io/badge/license-MIT-green.svg)](LICENSE)
-[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
-
-![Vanna2 Diagram](img/architecture.png)
-
----
-
-## Why Vanna?
-
-```mermaid
-graph LR
-    A[👤 User asks:<br/>'Show me Q4 sales'] --> B[🔐 Identity flows through<br/>every layer]
-    B --> C[🧰 Tools execute securely<br/>with permissions]
-    C --> D[📊 Rich UI streams back<br/>tables, charts, code]
-
-    style B fill:#FFD93D,stroke:#333,stroke-width:3px
-```
-
-### What Makes Vanna Different
-
-**Built for Production Data Analytics**
-- **Pre-built web component + backend** — No need to build your own chat UI
-- **User-aware at every layer** — Identity and permissions flow through the entire system
-- **Rich streaming responses** — Tables, charts, SQL code blocks, not just text
-- **Works with your existing auth** — Cookies, JWTs, session tokens
-- **Enterprise security built-in** — Row-level security, audit logs, rate limiting
-
-
-
----
-
-## Quick Start
-
-### Installation
-
-```bash
-pip install vanna[anthropic]  # or [openai]
-```
-
-### Basic Example
-
-```python
-from vanna import Agent, AgentConfig
-from vanna.servers.fastapi import VannaFastAPIServer
-from vanna.core.registry import ToolRegistry
-from vanna.core.user import UserResolver, User, RequestContext
-from vanna.integrations.anthropic import AnthropicLlmService
-from vanna.tools import RunSqlTool
-from vanna.integrations.sqlite import SqliteRunner
-
-# 1. Define how to resolve users from requests
-class SimpleUserResolver(UserResolver):
-    async def resolve_user(self, request_context: RequestContext) -> User:
-        # In production, validate cookies/JWTs here
-        user_id = request_context.get_cookie('user_id') or 'demo_user'
-        return User(id=user_id, group_memberships=['read_sales'])
-
-# 2. Set up LLM and tools
-llm = AnthropicLlmService(model="claude-sonnet-4-5")
-tools = ToolRegistry()
-tools.register(RunSqlTool(sql_runner=SqliteRunner(database_path="./data.db")))
-
-# 3. Create agent
-agent = Agent(
-    llm_service=llm,
-    tool_registry=tools,
-    user_resolver=SimpleUserResolver()
-)
-
-# 4. Create and run server
-server = VannaFastAPIServer(agent)
-app = server.create_app()
-
-# Run with: uvicorn main:app --host 0.0.0.0 --port 8000 --reload
-# Visit http://localhost:8000 to see the web UI
-```
-
----
-
-![Vanna2 Diagram](img/vanna2.svg)
-
----
-
-## What Makes Vanna Unique
-
-### 1. User-Aware by Design
-
-Every layer of the system knows **who the user is** and **what they can access**.
-
-```mermaid
-sequenceDiagram
-    participant U as 👤 User
-    participant W as 🌐 Web Component
-    participant S as 🐍 Flask/FastAPI
-    participant R as 🪪 User Resolver
-    participant A as 🤖 Agent
-    participant T as 🧰 Tools
-
-    U->>W: "Show Q4 sales"
-    W->>S: POST /api/vanna/v2/chat_sse
-    S->>R: Extract user identity
-    R->>A: User(id=alice, group_memberships=[read_sales])
-    A->>A: Generate personalized system prompt
-    A->>T: Execute SQL tool (user-aware)
-    T->>T: Apply row-level security
-    T->>A: Return filtered results
-    A->>W: Stream: Table → Chart → Summary
-    W->>U: Display results
-```
-
-**Not just authentication — authorization at every step:**
-
-- System prompt customized per user
-- Tools check permissions before execution
-- SQL queries filtered by row-level security
-- Audit logs per user
-- Rate limiting per user
-
-### 2. Drop-in Web Component
-
-```html
-<!-- Works with any existing app -->
-<vanna-chat
-  api-endpoint="/api/vanna/v2/chat_sse"
-  initial-message="What can I help you with?"
-  theme="dark">
-</vanna-chat>
-```
-
-**Features:**
-- Uses your existing cookies/JWTs (no new login system)
-- Renders streaming tables, charts, SQL code blocks
-- Responsive and customizable
-- Framework-agnostic (works with React, Vue, plain HTML)
-
-### 3. Purpose-Built for Data Analytics
-
-**Out-of-the-box tools:**
-- SQL generation and execution (with user permissions)
-- Data visualization with Plotly
-- File system operations (for coding agents)
-- Python code execution (sandboxed)
-
-```python
-from vanna.tools import RunSqlTool, VisualizeDataTool
-from vanna.integrations.sqlite import SqliteRunner
-from vanna.integrations.local import LocalFileSystem
-
-file_system = LocalFileSystem("./data")
-
-tools.register(RunSqlTool(
-    sql_runner=SqliteRunner(database_path="./data.db"),
-    file_system=file_system
-))
-tools.register(VisualizeDataTool(file_system=file_system))
-```
-
-### 4. Enterprise-Ready
-
-| Feature | Description |
-|---------|-------------|
-| **Row-Level Security** | SQL tools respect database permissions |
-| **Audit Logs** | Every query and tool call logged per user |
-| **Rate Limiting** | Per-user token/request limits via lifecycle hooks |
-| **Observability** | Built-in tracing and debugging hooks |
-| **Conversation Management** | Persistent conversation storage |
-| **Content Filtering** | Extensible filtering system |
-
----
-
-## Architecture
-
-```mermaid
-graph TB
-    subgraph Frontend["🌐 Frontend"]
-        WC[Web Component<br/>&lt;vanna-chat&gt;]
-    end
-
-    subgraph Backend["🐍 Python Server (Flask/FastAPI)"]
-        direction TB
-        SSE[SSE/WebSocket Handler]
-    end
-
-    subgraph Agent["🤖 User-Aware Agent"]
-        direction TB
-        UR[🪪 User Resolver<br/>YOUR auth system]
-        SP[⚙️ System Prompt<br/>Per-user customization]
-        LLM[🧠 LLM<br/>Claude/GPT/Gemini]
-        Tools[🧰 Tools<br/>SQL, Charts, Memory]
-        Comp[📄 UI Components<br/>Tables, Charts, Code]
-
-        UR --> SP
-        SP --> LLM
-        UR --> Tools
-        Tools <--> LLM
-        Tools --> Comp
-        LLM --> Comp
-    end
-
-    WC -->|User question<br/>+ cookies/JWT| SSE
-    SSE -->|Request context| UR
-    Comp -->|Streaming components| SSE
-    SSE -->|Progressive updates| WC
-
-    style UR fill:#95E1D3
-    style Tools fill:#FFD93D
-    style Comp fill:#C7F1FF
-```
-
-### Core Concepts
-
-**1. User Resolver** — You define this!
-
-```python
-from vanna.core.user import UserResolver, User, RequestContext
-
-class MyUserResolver(UserResolver):
-    async def resolve_user(self, request_context: RequestContext) -> User:
-        # Extract from your existing auth system
-        token = request_context.get_header('Authorization')
-        user_data = self.decode_jwt(token)  # Your logic
-
-        return User(
-            id=user_data['id'],
-            email=user_data['email'],
-            group_memberships=user_data['groups'],  # Key!
-            metadata={'role': user_data['role']}
-        )
-```
-
-**2. User-Aware Tools** — Check permissions automatically
-
-```python
-from vanna.core.tool import Tool, ToolContext, ToolResult
-from pydantic import BaseModel, Field
-from typing import Type
-
-class QueryArgs(BaseModel):
-    query: str = Field(description="SQL query to execute")
-
-class CustomSQLTool(Tool[QueryArgs]):
-    @property
-    def name(self) -> str:
-        return "query_database"
-
-    @property
-    def description(self) -> str:
-        return "Execute a SQL query against the database"
-
-    @property
-    def access_groups(self) -> list[str]:
-        return ["read_sales"]  # Only users in this group can use this tool
-
-    def get_args_schema(self) -> Type[QueryArgs]:
-        return QueryArgs
-
-    async def execute(self, context: ToolContext, args: QueryArgs) -> ToolResult:
-        user = context.user  # Automatically injected
-
-        # Apply row-level security
-        filtered_query = self.add_user_filters(args.query, user)
-        results = await self.db.execute(filtered_query)
-
-        return ToolResult(
-            success=True,
-            result_for_llm=str(results)
-        )
-```
-
-**3. Streaming UI Components**
-
-```python
-async for component in agent.send_message(request_context=ctx, message="Show sales"):
-    # Rich component: structured data (tables, charts, status cards)
-    print(type(component.rich_component).__name__)
-
-    # Simple component: plain text fallback
-    print(component.simple_component.text)
-```
-
-Output:
-```
-StatusBarUpdateComponent    # "Processing..."
-TaskTrackerUpdateComponent  # "Load conversation context"
-RichTextComponent          # "Let me query the sales data..."
-StatusCardComponent        # "Executing run_sql"
-DataFrameComponent         # Tabular results
-RichTextComponent          # "Here are your top customers..."
-```
-
----
-
-## Add to Existing FastAPI App
-
-If you already have a FastAPI application, you can add Vanna as additional routes:
-
-```python
-from fastapi import FastAPI
-from vanna import Agent, AgentConfig
-from vanna.servers.base import ChatHandler
-from vanna.servers.fastapi.routes import register_chat_routes
-from vanna.core.registry import ToolRegistry
-from vanna.core.user import UserResolver, User, RequestContext
-from vanna.integrations.anthropic import AnthropicLlmService
-from vanna.tools import RunSqlTool
-from vanna.integrations.sqlite import SqliteRunner
-
-# Your existing FastAPI app
-app = FastAPI()
-
-# Your existing routes
-@app.get('/api/users')
-async def get_users():
-    return {'users': [...]}
-
-@app.get('/api/products')
-async def get_products():
-    return {'products': [...]}
-
-# Add Vanna agent
-class CookieUserResolver(UserResolver):
-    async def resolve_user(self, request_context: RequestContext) -> User:
-        user_id = request_context.get_cookie('user_id') or 'anonymous'
-        role = request_context.get_cookie('role') or 'guest'
-
-        groups = []
-        if role == 'admin':
-            groups = ['read_sales', 'read_confidential', 'admin']
-        elif role == 'analyst':
-            groups = ['read_sales']
-
-        return User(id=user_id, group_memberships=groups)
-
-# Set up agent
-llm = AnthropicLlmService(model="claude-sonnet-4-5")
-tools = ToolRegistry()
-tools.register(RunSqlTool(sql_runner=SqliteRunner(database_path="./data.db")))
-
-agent = Agent(
-    llm_service=llm,
-    tool_registry=tools,
-    user_resolver=CookieUserResolver()
-)
-
-# Add Vanna routes to your existing app
-chat_handler = ChatHandler(agent)
-register_chat_routes(app, chat_handler)
-
-# Run with: uvicorn main:app --host 0.0.0.0 --port 8000 --reload
-```
-
-This adds these endpoints to your existing FastAPI app:
-- `GET /` - Vanna web component UI (you may want to change this)
-- `POST /api/vanna/v2/chat_sse` - Server-Sent Events streaming
-- `POST /api/vanna/v2/chat_poll` - Polling endpoint
-- `GET /health` - Health check
-
-To customize the routes or serve the UI at a different path, see the [server configuration docs](https://docs.vanna.ai).
-
----
-
-## Custom Tools
-
-Create custom tools by extending the `Tool` base class:
-
-```python
-from vanna.core.tool import Tool, ToolContext, ToolResult
-from vanna.components import UiComponent, NotificationComponent, SimpleTextComponent, ComponentType
-from pydantic import BaseModel, Field
-from typing import Type
-
-class EmailArgs(BaseModel):
-    recipient: str = Field(description="Email recipient")
-    subject: str = Field(description="Email subject")
-    body: str = Field(description="Email body")
-
-class EmailTool(Tool[EmailArgs]):
-    @property
-    def name(self) -> str:
-        return "send_email"
-
-    @property
-    def description(self) -> str:
-        return "Send an email to a user"
-
-    @property
-    def access_groups(self) -> list[str]:
-        return ["send_email"]  # Only users in this group can use this tool
-
-    def get_args_schema(self) -> Type[EmailArgs]:
-        return EmailArgs
-
-    async def execute(self, context: ToolContext, args: EmailArgs) -> ToolResult:
-        user = context.user
-
-        # Check domain restrictions
-        if not args.recipient.endswith('@company.com'):
-            error_msg = "Can only send to company email addresses"
-            return ToolResult(
-                success=False,
-                result_for_llm=error_msg,
-                error=error_msg,
-                ui_component=UiComponent(
-                    rich_component=NotificationComponent(
-                        type=ComponentType.NOTIFICATION,
-                        level="error",
-                        message=error_msg
-                    ),
-                    simple_component=SimpleTextComponent(text=error_msg)
-                )
-            )
-
-        # Send email (your logic)
-        await self.email_service.send(
-            from_email=user.email,
-            to=args.recipient,
-            subject=args.subject,
-            body=args.body
-        )
-
-        success_msg = f"Email sent to {args.recipient}"
-        return ToolResult(
-            success=True,
-            result_for_llm=success_msg,
-            ui_component=UiComponent(
-                rich_component=NotificationComponent(
-                    type=ComponentType.NOTIFICATION,
-                    level="success",
-                    message=success_msg
-                ),
-                simple_component=SimpleTextComponent(text=success_msg)
-            )
-        )
-
-# Register tool
-tools.register(EmailTool())
-```
-
----
-
-## Configuration
-
-### Agent Configuration
-
-```python
-from vanna import AgentConfig
-
-config = AgentConfig(
-    max_tool_iterations=10,           # Max tool calls per message
-    stream_responses=True,             # Enable streaming
-    temperature=0.7,                   # LLM temperature
-    include_thinking_indicators=True,  # Show "Thinking..." states
-    auto_save_conversations=True,      # Auto-persist conversations
-    max_tokens=None                    # Maximum response tokens
-)
-
-agent = Agent(
-    llm_service=llm,
-    tool_registry=tools,
-    user_resolver=user_resolver,
-    config=config
-)
-```
-
-### Environment Variables
-
-```bash
-# Anthropic
-export ANTHROPIC_API_KEY="sk-ant-..."
-export ANTHROPIC_MODEL="claude-sonnet-4-5"
-
-# OpenAI
-export OPENAI_API_KEY="sk-..."
-export OPENAI_MODEL="gpt-5"
-
-# Database
-export DATABASE_URL="postgresql://localhost/mydb"
-```
-
----
-
-## Advanced Features
-
-### 1. Conversation Storage
-
-```python
-from vanna.integrations.local import MemoryConversationStore
-
-# Use in-memory storage (default)
-store = MemoryConversationStore()
-
-agent = Agent(
-    llm_service=llm,
-    tool_registry=tools,
-    user_resolver=user_resolver,
-    conversation_store=store
-)
-
-# List user's conversations
-alice = User(id="alice")
-conversations = await store.list_conversations(user=alice)
-
-# Get conversation history
-conversation = await store.get_conversation(
-    conversation_id="conv_123",
-    user=alice
-)
-```
-
-### 2. Lifecycle Hooks
-
-```python
-from vanna.core.lifecycle import LifecycleHook
-
-class QuotaCheckHook(LifecycleHook):
-    async def before_message(self, user: User, message: str) -> str:
-        # Check if user has quota remaining
-        if not await self.check_quota(user.id):
-            raise Exception("Quota exceeded")
-        return message
-
-    async def after_tool(self, result: ToolResult) -> ToolResult:
-        # Log tool execution
-        await self.log_tool_execution(result)
-        return result
-
-agent = Agent(
-    llm_service=llm,
-    tool_registry=tools,
-    user_resolver=user_resolver,
-    lifecycle_hooks=[QuotaCheckHook()]
-)
-```
-
-### 3. LLM Middlewares
-
-```python
-from vanna.core.middleware import LlmMiddleware
-from vanna.core.llm import LlmRequest, LlmResponse
-
-class CachingMiddleware(LlmMiddleware):
-    async def before_llm_request(self, request: LlmRequest) -> LlmRequest:
-        # Check cache before sending to LLM
-        cached = await self.cache.get(request)
-        if cached:
-            return cached
-        return request
-
-    async def after_llm_response(
-        self,
-        request: LlmRequest,
-        response: LlmResponse
-    ) -> LlmResponse:
-        # Cache the response
-        await self.cache.set(request, response)
-        return response
-
-agent = Agent(
-    llm_service=llm,
-    tool_registry=tools,
-    user_resolver=user_resolver,
-    llm_middlewares=[CachingMiddleware()]
-)
-```
-
-### 4. Observability
-
-```python
-from vanna.core.observability import ObservabilityProvider
-
-class LoggingProvider(ObservabilityProvider):
-    async def create_span(self, name: str, attributes: dict):
-        print(f"Starting: {name}")
-        return Span(name, attributes)
-
-    async def record_metric(self, name: str, value: float, unit: str, tags: dict):
-        print(f"Metric: {name} = {value}{unit}")
-
-agent = Agent(
-    llm_service=llm,
-    tool_registry=tools,
-    user_resolver=user_resolver,
-    observability_provider=LoggingProvider()
-)
-```
-
-### 5. Context Enrichers
-
-```python
-from vanna.core.enricher import ToolContextEnricher
-from vanna.core.tool import ToolContext
-
-class UserMetadataEnricher(ToolContextEnricher):
-    async def enrich_context(self, context: ToolContext) -> ToolContext:
-        # Add additional user metadata from database
-        user_metadata = await self.db.get_user_metadata(context.user.id)
-        context.user.metadata.update(user_metadata)
-        return context
-
-agent = Agent(
-    llm_service=llm,
-    tool_registry=tools,
-    user_resolver=user_resolver,
-    context_enrichers=[UserMetadataEnricher()]
-)
-```
-
-### 6. LLM Context Enhancers
-
-Enhance LLM system prompts and messages with additional context (e.g., from memory, RAG, documentation):
-
-```python
-from vanna.core.enhancer import LlmContextEnhancer, DefaultLlmContextEnhancer
-from vanna.core.llm import LlmMessage
-from vanna.core.user import User
-
-class CustomLlmContextEnhancer(LlmContextEnhancer):
-    async def enhance_system_prompt(
-        self,
-        system_prompt: str,
-        user_message: str,
-        user: User
-    ) -> str:
-        # Add relevant context to system prompt based on user message
-        relevant_docs = await self.search_documentation(user_message)
-        return system_prompt + f"\n\nRelevant documentation:\n{relevant_docs}"
-
-    async def enhance_user_messages(
-        self,
-        messages: list[LlmMessage],
-        user: User
-    ) -> list[LlmMessage]:
-        # Optionally modify user messages
-        return messages
-
-# Use default implementation (uses AgentMemory for RAG)
-agent = Agent(
-    llm_service=llm,
-    tool_registry=tools,
-    user_resolver=user_resolver,
-    agent_memory=agent_memory,
-    llm_context_enhancer=DefaultLlmContextEnhancer(agent_memory)  # Default if not provided
-)
-
-# Or use custom implementation
-agent = Agent(
-    llm_service=llm,
-    tool_registry=tools,
-    user_resolver=user_resolver,
-    llm_context_enhancer=CustomLlmContextEnhancer()
-)
-```
-
-**Key difference:**
-- **Context Enrichers** (ToolContextEnricher): Enrich tool execution context
-- **LLM Context Enhancers** (LlmContextEnhancer): Enhance LLM prompts and messages
-
----
-
-## When to Use Vanna
-
-**Vanna is ideal for:**
-- Building data analytics applications with natural language interfaces
-- Applications requiring user-aware permissions throughout
-- Teams that want a pre-built web component + backend integration
-- Enterprise environments with strict security requirements
-- Use cases needing rich streaming responses (tables, charts, SQL)
-- Integrating with existing authentication systems
-
----
-
-## Documentation
-
-- **Migration Guide**: [Migrating from Vanna 1.x to 2.0+](MIGRATION_GUIDE.md)
-- **GitHub Discussions**: [GitHub Discussions](https://github.com/vanna-ai/vanna/discussions)
-- **Email**: support@vanna.ai
-