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

vanna/servers/base/models.py

@@ -0,0 +1,111 @@
+"""
+Request and response models for server endpoints.
+"""
+
+import time
+import uuid
+from typing import Any, Dict, List, Optional, Union
+
+from pydantic import BaseModel, Field
+
+from ...components import UiComponent, RichComponent
+from ...core.component_manager import ComponentUpdate
+from ...core.user.request_context import RequestContext
+
+
+class ChatRequest(BaseModel):
+    """Request model for chat endpoints."""
+
+    message: str = Field(description="User message")
+    conversation_id: Optional[str] = Field(default=None, description="Conversation ID")
+    request_id: Optional[str] = Field(
+        default=None, description="Request ID for tracing"
+    )
+    request_context: RequestContext = Field(
+        default_factory=RequestContext,
+        description="Request context for user resolution",
+    )
+    metadata: Dict[str, Any] = Field(
+        default_factory=dict, description="Additional metadata"
+    )
+
+
+class ChatStreamChunk(BaseModel):
+    """Single chunk in a streaming chat response."""
+
+    rich: Dict[str, Any] = Field(description="Rich component data for advanced UIs")
+    simple: Optional[Dict[str, Any]] = Field(
+        default=None, description="Simple component data for basic UIs"
+    )
+
+    # Stream metadata
+    conversation_id: str = Field(description="Conversation ID")
+    request_id: str = Field(description="Request ID")
+    timestamp: float = Field(default_factory=time.time, description="Timestamp")
+
+    @classmethod
+    def from_component(
+        cls,
+        component: Union[UiComponent, RichComponent],
+        conversation_id: str,
+        request_id: str,
+    ) -> "ChatStreamChunk":
+        """Create chunk from UI component or rich component."""
+
+        if isinstance(component, UiComponent):
+            # Full UiComponent with both rich and simple
+            rich_data = component.rich_component.serialize_for_frontend()
+            simple_data = None
+            if component.simple_component:
+                simple_data = component.simple_component.serialize_for_frontend()
+
+            return cls(
+                rich=rich_data,
+                simple=simple_data,
+                conversation_id=conversation_id,
+                request_id=request_id,
+            )
+
+        # Rich component only (no simple fallback)
+        rich_data = component.serialize_for_frontend()
+        return cls(
+            rich=rich_data,
+            simple=None,
+            conversation_id=conversation_id,
+            request_id=request_id,
+        )
+
+    @classmethod
+    def from_component_update(
+        cls, update: ComponentUpdate, conversation_id: str, request_id: str
+    ) -> "ChatStreamChunk":
+        """Create chunk from component update."""
+        update_payload = update.serialize_for_frontend()
+        return cls(
+            rich=update_payload,
+            simple=None,  # Component updates don't have simple representations
+            conversation_id=conversation_id,
+            request_id=request_id,
+        )
+
+
+class ChatResponse(BaseModel):
+    """Complete chat response for polling endpoints."""
+
+    chunks: List[ChatStreamChunk] = Field(description="Response chunks")
+    conversation_id: str = Field(description="Conversation ID")
+    request_id: str = Field(description="Request ID")
+    total_chunks: int = Field(description="Total number of chunks")
+
+    @classmethod
+    def from_chunks(cls, chunks: List[ChatStreamChunk]) -> "ChatResponse":
+        """Create response from chunks."""
+        if not chunks:
+            return cls(chunks=[], conversation_id="", request_id="", total_chunks=0)
+
+        return cls(
+            chunks=chunks,
+            conversation_id=chunks[0].conversation_id,
+            request_id=chunks[0].request_id,
+            total_chunks=len(chunks),
+        )

vanna/servers/base/rich_chat_handler.py

@@ -0,0 +1,141 @@
+# """
+# Rich component-aware chat handling logic.
+# """
+
+# import uuid
+# from typing import AsyncGenerator, Callable, List, Optional, Union
+
+# from ...core import Agent, User
+# from ...core.rich_components import RichComponent
+# from ...core.component_manager import ComponentManager, ComponentUpdate
+# from .models import ChatRequest, ChatResponse, ChatStreamChunk
+
+
+# class RichChatHandler:
+#     """Rich component-aware chat handling logic."""
+
+#     def __init__(
+#         self,
+#         agent: Agent,
+#         default_user_factory: Optional[Callable[[Optional[str]], User]] = None,
+#     ):
+#         """Initialize rich chat handler.
+
+#         Args:
+#             agent: The agent to handle chat requests
+#             default_user_factory: Function to create default user from user_id
+#         """
+#         self.agent = agent
+#         self.default_user_factory = default_user_factory or self._create_default_user
+#         self.component_managers: dict[str, ComponentManager] = {}  # Per conversation
+
+#     async def handle_stream(
+#         self, request: ChatRequest
+#     ) -> AsyncGenerator[ChatStreamChunk, None]:
+#         """Stream chat responses with rich component support.
+
+#         Args:
+#             request: Chat request
+
+#         Yields:
+#             Chat stream chunks including rich component updates
+#         """
+#         user = self._resolve_user(request.user_id)
+#         conversation_id = request.conversation_id or self._generate_conversation_id()
+#         request_id = request.request_id or str(uuid.uuid4())
+
+#         # Get or create component manager for this conversation
+#         if conversation_id not in self.component_managers:
+#             self.component_managers[conversation_id] = ComponentManager()
+
+#         component_manager = self.component_managers[conversation_id]
+
+#         async for component in self.agent.send_message(
+#             conversation_id=conversation_id,
+#             user=user,
+#             message=request.message,
+#             request_id=request_id,
+#         ):
+#             if isinstance(component, RichComponent):
+#                 # Handle rich component through manager
+#                 update = component_manager.emit(component)
+#                 yield ChatStreamChunk.from_component_update(update, conversation_id, request_id)
+#             else:
+#                 # Handle legacy components
+#                 yield ChatStreamChunk.from_component(component, conversation_id, request_id)
+
+#     async def handle_poll(self, request: ChatRequest) -> ChatResponse:
+#         """Handle polling request with rich component support.
+
+#         Args:
+#             request: Chat request
+
+#         Returns:
+#             Complete chat response with all components
+#         """
+#         chunks: List[ChatStreamChunk] = []
+
+#         async for chunk in self.handle_stream(request):
+#             chunks.append(chunk)
+
+#         return ChatResponse.from_chunks(chunks)
+
+#     def get_component_manager(self, conversation_id: str) -> Optional[ComponentManager]:
+#         """Get the component manager for a conversation."""
+#         return self.component_managers.get(conversation_id)
+
+#     def get_component(self, conversation_id: str, component_id: str) -> Optional[RichComponent]:
+#         """Get a specific component from a conversation."""
+#         manager = self.get_component_manager(conversation_id)
+#         return manager.get_component(component_id) if manager else None
+
+#     def get_all_components(self, conversation_id: str) -> List[RichComponent]:
+#         """Get all components in a conversation."""
+#         manager = self.get_component_manager(conversation_id)
+#         return manager.get_all_components() if manager else []
+
+#     def update_component(
+#         self,
+#         conversation_id: str,
+#         component_id: str,
+#         **updates
+#     ) -> Optional[ComponentUpdate]:
+#         """Update a component in a conversation."""
+#         manager = self.get_component_manager(conversation_id)
+#         return manager.update_component(component_id, **updates) if manager else None
+
+#     def remove_component(
+#         self,
+#         conversation_id: str,
+#         component_id: str
+#     ) -> Optional[ComponentUpdate]:
+#         """Remove a component from a conversation."""
+#         manager = self.get_component_manager(conversation_id)
+#         return manager.remove_component(component_id) if manager else None
+
+#     def clear_conversation_components(self, conversation_id: str):
+#         """Clear all components for a conversation."""
+#         if conversation_id in self.component_managers:
+#             del self.component_managers[conversation_id]
+
+#     def _resolve_user(self, user_id: Optional[str]) -> User:
+#         """Resolve user from ID or create default."""
+#         if user_id:
+#             # In a real implementation, you'd fetch from a user store
+#             return User(id=user_id, username=f"user_{user_id}", email="", permissions=[])
+
+#         return self.default_user_factory(user_id)
+
+#     def _create_default_user(self, user_id: Optional[str]) -> User:
+#         """Create a default user."""
+#         user_id = user_id or "anonymous"
+#         return User(
+#             id=user_id,
+#             username=f"user_{user_id}",
+#             email="",
+#             permissions=[]
+#         )
+
+#     def _generate_conversation_id(self) -> str:
+#         """Generate a new conversation ID."""
+#         return str(uuid.uuid4())

vanna/servers/base/templates.py

@@ -0,0 +1,331 @@
+"""
+HTML templates for Vanna Agents servers.
+"""
+
+from typing import Optional
+
+
+def get_vanna_component_script(
+    dev_mode: bool = False,
+    static_path: str = "/static",
+    cdn_url: str = "https://img.vanna.ai/vanna-components.js",
+) -> str:
+    """Get the script tag for loading Vanna web components.
+
+    Args:
+        dev_mode: If True, load from local static files
+        static_path: Path to static assets in dev mode
+        cdn_url: CDN URL for production
+
+    Returns:
+        HTML script tag for loading components
+    """
+    if dev_mode:
+        return (
+            f'<script type="module" src="{static_path}/vanna-components.js"></script>'
+        )
+    else:
+        return f'<script type="module" src="{cdn_url}"></script>'
+
+
+def get_index_html(
+    dev_mode: bool = False,
+    static_path: str = "/static",
+    cdn_url: str = "https://img.vanna.ai/vanna-components.js",
+    api_base_url: str = "",
+) -> str:
+    """Generate index HTML with configurable component loading.
+
+    Args:
+        dev_mode: If True, load components from local static files
+        static_path: Path to static assets in dev mode
+        cdn_url: CDN URL for production components
+        api_base_url: Base URL for API endpoints
+
+    Returns:
+        Complete HTML page as string
+    """
+    component_script = get_vanna_component_script(dev_mode, static_path, cdn_url)
+
+    return f"""<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Vanna Agents Chat</title>
+    <link rel="preconnect" href="https://fonts.googleapis.com">
+    <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
+    <link href="https://fonts.googleapis.com/css2?family=Roboto+Slab:wght@400;500;600;700&family=Space+Grotesk:wght@300;400;500;600;700&family=Space+Mono:wght@400;700&display=swap" rel="stylesheet">
+    <script src="https://cdn.tailwindcss.com"></script>
+    <script>
+        tailwind.config = {{
+            theme: {{
+                extend: {{
+                    colors: {{
+                        'vanna-navy': '#023d60',
+                        'vanna-cream': '#e7e1cf',
+                        'vanna-teal': '#15a8a8',
+                        'vanna-orange': '#fe5d26',
+                        'vanna-magenta': '#bf1363',
+                    }},
+                    fontFamily: {{
+                        'sans': ['Space Grotesk', 'ui-sans-serif', 'system-ui'],
+                        'serif': ['Roboto Slab', 'ui-serif', 'Georgia'],
+                        'mono': ['Space Mono', 'ui-monospace', 'monospace'],
+                    }}
+                }}
+            }}
+        }}
+    </script>
+    <style>
+        body {{
+            background: linear-gradient(to bottom, #e7e1cf, #ffffff, #e7e1cf);
+            min-height: 100vh;
+            position: relative;
+            overflow-x: hidden;
+        }}
+
+        /* Background decorations matching landing page */
+        body::before {{
+            content: '';
+            position: fixed;
+            inset: 0;
+            pointer-events: none;
+            z-index: 0;
+            /* Radial gradients with brand colors */
+            background:
+                radial-gradient(circle at top left, rgba(21, 168, 168, 0.12), transparent 60%),
+                radial-gradient(circle at bottom right, rgba(254, 93, 38, 0.08), transparent 65%);
+        }}
+
+        body::after {{
+            content: '';
+            position: fixed;
+            inset: 0;
+            pointer-events: none;
+            z-index: 0;
+            /* Dot pattern with retro computing aesthetic */
+            background-image: radial-gradient(circle at 2px 2px, rgba(2, 61, 96, 0.3) 1px, transparent 0);
+            background-size: 32px 32px;
+            /* Grid overlay */
+            background-image:
+                radial-gradient(circle at 2px 2px, rgba(2, 61, 96, 0.3) 1px, transparent 0),
+                linear-gradient(rgba(2, 61, 96, 0.1) 1px, transparent 1px),
+                linear-gradient(90deg, rgba(2, 61, 96, 0.1) 1px, transparent 1px);
+            background-size: 32px 32px, 100px 100px, 100px 100px;
+        }}
+
+        /* Ensure content is above background */
+        body > * {{
+            position: relative;
+            z-index: 1;
+        }}
+
+        vanna-chat {{
+            width: 100%;
+            height: 100%;
+            display: block;
+        }}
+    </style>
+    {component_script}
+</head>
+<body>
+    <div class="max-w-6xl mx-auto p-5">
+        <!-- Header -->
+        <div class="text-center mb-8">
+            <h1 class="text-4xl font-bold text-vanna-navy mb-2 font-serif">Vanna Agents</h1>
+            <p class="text-lg font-mono font-bold text-vanna-teal mb-4">DATA-FIRST AGENTS</p>
+            <p class="text-slate-600 mb-4">Interactive AI Assistant powered by Vanna Agents Framework</p>
+            <a href="javascript:window.location='view-source:'+window.location.href" class="inline-flex items-center gap-2 px-4 py-2 bg-vanna-teal text-white text-sm font-medium rounded-lg hover:bg-vanna-navy transition">
+                <svg class="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+                    <path stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="M10 20l4-16m4 4l4 4-4 4M6 16l-4-4 4-4"/>
+                </svg>
+                View Page Source
+            </a>
+        </div>
+
+        {('    <div class="bg-vanna-orange/10 border border-vanna-orange/30 rounded-lg p-3 mb-5 text-vanna-orange text-sm font-medium">📦 Development Mode: Loading components from local assets</div>' if dev_mode else "")}
+
+        <!-- Login Form -->
+        <div id="loginContainer" class="max-w-md mx-auto mb-10 bg-white p-8 rounded-xl shadow-lg border border-vanna-teal/30">
+            <div class="text-center mb-6">
+                <h2 class="text-2xl font-semibold text-vanna-navy mb-2 font-serif">Login to Continue</h2>
+                <p class="text-sm text-slate-600">Select your email to access the chat</p>
+            </div>
+
+            <div class="mb-5">
+                <label for="emailInput" class="block mb-2 text-sm font-medium text-vanna-navy">Email Address</label>
+                <select
+                    id="emailInput"
+                    class="w-full px-4 py-3 text-sm border border-vanna-teal/30 rounded-lg focus:outline-none focus:ring-2 focus:ring-vanna-teal focus:border-transparent bg-white"
+                >
+                    <option value="">Select an email...</option>
+                    <option value="admin@example.com">admin@example.com</option>
+                    <option value="user@example.com">user@example.com</option>
+                </select>
+            </div>
+
+            <button id="loginButton" class="w-full px-4 py-3 bg-vanna-teal text-white text-sm font-medium rounded-lg hover:bg-vanna-navy focus:outline-none focus:ring-2 focus:ring-vanna-teal focus:ring-offset-2 transition disabled:bg-gray-400 disabled:cursor-not-allowed">
+                Continue
+            </button>
+
+            <div class="mt-5 p-3 bg-vanna-teal/10 border-l-4 border-vanna-teal rounded text-xs text-vanna-navy leading-relaxed">
+                <strong>Demo Mode:</strong> This is a frontend-only authentication demo.
+                Your email will be stored as a cookie and automatically sent with all API requests.
+            </div>
+        </div>
+
+        <!-- Logged In Status (hidden by default) -->
+        <div id="loggedInStatus" class="hidden text-center p-4 bg-vanna-teal/10 border border-vanna-teal/30 rounded-lg mb-5">
+            Logged in as <span id="loggedInEmail" class="font-semibold text-vanna-navy"></span>
+            <br>
+            <button id="logoutButton" class="mt-2 px-3 py-1.5 bg-vanna-navy text-white text-xs rounded hover:bg-vanna-teal transition">
+                Logout
+            </button>
+        </div>
+
+        <!-- Chat Container (hidden by default) -->
+        <div id="chatSections" class="hidden">
+            <div class="bg-white rounded-xl shadow-lg h-[600px] overflow-hidden border border-vanna-teal/30">
+                <vanna-chat
+                    api-base="{api_base_url}"
+                    sse-endpoint="{api_base_url}/api/vanna/v2/chat_sse"
+                    ws-endpoint="{api_base_url}/api/vanna/v2/chat_websocket"
+                    poll-endpoint="{api_base_url}/api/vanna/v2/chat_poll">
+                </vanna-chat>
+            </div>
+
+            <div class="mt-8 p-5 bg-white rounded-lg shadow border border-vanna-teal/30">
+                <h3 class="text-lg font-semibold text-vanna-navy mb-3 font-serif">API Endpoints</h3>
+                <ul class="space-y-2">
+                    <li class="p-2 bg-vanna-cream/50 rounded font-mono text-sm">
+                        <span class="font-bold text-vanna-teal mr-2">POST</span>{api_base_url}/api/vanna/v2/chat_sse - Server-Sent Events streaming
+                    </li>
+                    <li class="p-2 bg-vanna-cream/50 rounded font-mono text-sm">
+                        <span class="font-bold text-vanna-teal mr-2">WS</span>{api_base_url}/api/vanna/v2/chat_websocket - WebSocket real-time chat
+                    </li>
+                    <li class="p-2 bg-vanna-cream/50 rounded font-mono text-sm">
+                        <span class="font-bold text-vanna-teal mr-2">POST</span>{api_base_url}/api/vanna/v2/chat_poll - Request/response polling
+                    </li>
+                    <li class="p-2 bg-vanna-cream/50 rounded font-mono text-sm">
+                        <span class="font-bold text-vanna-teal mr-2">GET</span>{api_base_url}/health - Health check
+                    </li>
+                </ul>
+            </div>
+        </div>
+    </div>
+
+    <script>
+        // Cookie helpers
+        const getCookie = (name) => {{
+            const value = `; ${{document.cookie}}`;
+            const parts = value.split(`; ${{name}}=`);
+            return parts.length === 2 ? parts.pop().split(';').shift() : null;
+        }};
+
+        const setCookie = (name, value) => {{
+            const expires = new Date(Date.now() + 365 * 864e5).toUTCString();
+            document.cookie = `${{name}}=${{value}}; expires=${{expires}}; path=/; SameSite=Lax`;
+        }};
+
+        const deleteCookie = (name) => {{
+            document.cookie = `${{name}}=; expires=Thu, 01 Jan 1970 00:00:00 UTC; path=/;`;
+        }};
+
+        // Login/Logout
+        document.addEventListener('DOMContentLoaded', () => {{
+            const email = getCookie('vanna_email');
+
+            // Check if already logged in
+            if (email) {{
+                loginContainer.classList.add('hidden');
+                loggedInStatus.classList.remove('hidden');
+                chatSections.classList.remove('hidden');
+                loggedInEmail.textContent = email;
+            }}
+
+            // Login button
+            loginButton.addEventListener('click', () => {{
+                const email = emailInput.value.trim();
+                if (!email) {{
+                    alert('Please select an email address');
+                    return;
+                }}
+                setCookie('vanna_email', email);
+                loginContainer.classList.add('hidden');
+                loggedInStatus.classList.remove('hidden');
+                chatSections.classList.remove('hidden');
+                loggedInEmail.textContent = email;
+            }});
+
+            // Logout button
+            logoutButton.addEventListener('click', () => {{
+                deleteCookie('vanna_email');
+                loginContainer.classList.remove('hidden');
+                loggedInStatus.classList.add('hidden');
+                chatSections.classList.add('hidden');
+                emailInput.value = '';
+            }});
+
+            // Enter key
+            emailInput.addEventListener('keypress', (e) => {{
+                if (e.key === 'Enter') loginButton.click();
+            }});
+        }});
+    </script>
+
+    <script>
+        // Artifact demo event listener
+        document.addEventListener('DOMContentLoaded', () => {{
+            const vannaChat = document.querySelector('vanna-chat');
+
+            if (vannaChat) {{
+                // Add artifact event listener to demonstrate external rendering
+                vannaChat.addEventListener('artifact-opened', (event) => {{
+                    const {{ artifactId, type, title, trigger }} = event.detail;
+
+                    console.log('🎨 Artifact Event:', {{ artifactId, type, title, trigger }});
+
+                    // For demo: open all artifacts externally
+                    setTimeout(() => {{
+                        const newWindow = window.open('', '_blank', 'width=900,height=700');
+                        if (newWindow) {{
+                            newWindow.document.write(event.detail.getStandaloneHTML());
+                            newWindow.document.close();
+                            newWindow.document.title = title || 'Vanna Artifact';
+                            console.log(`📱 Opened ${{title}} in new window`);
+                        }}
+                    }}, 100);
+
+                    // Prevent default in-chat rendering
+                    event.detail.preventDefault();
+                    console.log('✋ Showing placeholder in chat instead of full artifact');
+                }});
+
+                console.log('🎯 Artifact demo mode: All artifacts will open externally');
+            }}
+        }});
+
+        // Fallback if web component doesn't load
+        if (!customElements.get('vanna-chat')) {{
+            setTimeout(() => {{
+                if (!customElements.get('vanna-chat')) {{
+                    document.querySelector('vanna-chat').innerHTML = `
+                        <div class="p-10 text-center text-gray-600">
+                            <h3 class="text-xl font-semibold mb-2">Vanna Chat Component</h3>
+                            <p class="mb-2">Web component failed to load. Please check your connection.</p>
+                            <p class="text-sm text-gray-400">
+                                {("Loading from: local static assets" if dev_mode else f"Loading from: {cdn_url}")}
+                            </p>
+                        </div>
+                    `;
+                }}
+            }}, 2000);
+        }}
+    </script>
+</body>
+</html>"""
+
+
+# Backward compatibility - default production HTML
+INDEX_HTML = get_index_html()

vanna/servers/cli/__init__.py

@@ -0,0 +1,7 @@
+"""
+CLI components for Vanna Agents servers.
+"""
+
+from .server_runner import ExampleAgentLoader
+
+__all__ = ["ExampleAgentLoader"]