vanna 0.7.8__py3-none-any.whl → 2.0.0__py3-none-any.whl

vanna/core/__init__.py

@@ -0,0 +1,193 @@
+"""
+Core components of the Vanna Agents framework.
+
+This package contains the fundamental abstractions and implementations
+that form the foundation of the agent framework.
+"""
+
+# Core domains - re-export from new structure
+from .tool import T, Tool, ToolCall, ToolContext, ToolResult, ToolSchema
+from .llm import LlmMessage, LlmRequest, LlmResponse, LlmService, LlmStreamChunk
+from .storage import Conversation, ConversationStore, Message
+from .user import User, UserService
+from .agent import Agent, AgentConfig
+from .system_prompt import DefaultSystemPromptBuilder, SystemPromptBuilder
+from .lifecycle import LifecycleHook
+from .middleware import LlmMiddleware
+from .workflow import WorkflowHandler, WorkflowResult, DefaultWorkflowHandler
+from .recovery import ErrorRecoveryStrategy, RecoveryAction, RecoveryActionType
+from .enricher import ToolContextEnricher
+from .enhancer import LlmContextEnhancer, DefaultLlmContextEnhancer
+from .filter import ConversationFilter
+from .observability import ObservabilityProvider, Span, Metric
+from .audit import (
+    AuditLogger,
+    AuditEvent,
+    AuditEventType,
+    ToolAccessCheckEvent,
+    ToolInvocationEvent,
+    ToolResultEvent,
+    UiFeatureAccessCheckEvent,
+    AiResponseEvent,
+)
+
+# UI Components
+from .components import UiComponent
+from .rich_component import RichComponent
+from ..components import (
+    SimpleComponent,
+    SimpleComponentType,
+    SimpleImageComponent,
+    SimpleLinkComponent,
+    SimpleTextComponent,
+    ArtifactComponent,
+    BadgeComponent,
+    CardComponent,
+    DataFrameComponent,
+    IconTextComponent,
+    LogViewerComponent,
+    NotificationComponent,
+    ProgressBarComponent,
+    ProgressDisplayComponent,
+    RichTextComponent,
+    StatusCardComponent,
+    TaskListComponent,
+)
+
+# Exceptions
+from .errors import (
+    AgentError,
+    ConversationNotFoundError,
+    LlmServiceError,
+    PermissionError,
+    ToolExecutionError,
+    ToolNotFoundError,
+    ValidationError,
+)
+
+# Core implementations
+from .registry import ToolRegistry
+
+# Evaluation framework
+from .evaluation import (
+    Evaluator,
+    TestCase,
+    ExpectedOutcome,
+    AgentResult,
+    EvaluationResult,
+    TestCaseResult,
+    AgentVariant,
+    EvaluationRunner,
+    TrajectoryEvaluator,
+    OutputEvaluator,
+    LLMAsJudgeEvaluator,
+    EfficiencyEvaluator,
+    EvaluationReport,
+    ComparisonReport,
+    EvaluationDataset,
+)
+
+# Rebuild models to resolve forward references after all imports
+from .tool.models import ToolContext, ToolResult
+from .components import UiComponent  # Import UiComponent to ensure it's available
+
+ToolContext.model_rebuild()
+ToolResult.model_rebuild()
+
+__all__ = [
+    # Models
+    "User",
+    "Message",
+    "Conversation",
+    "ToolCall",
+    "ToolResult",
+    "ToolContext",
+    "ToolSchema",
+    "LlmMessage",
+    "LlmRequest",
+    "LlmResponse",
+    "LlmStreamChunk",
+    "RecoveryAction",
+    "RecoveryActionType",
+    "Span",
+    "Metric",
+    # Interfaces
+    "Tool",
+    "Agent",
+    "LlmService",
+    "ConversationStore",
+    "UserService",
+    "SystemPromptBuilder",
+    "LifecycleHook",
+    "LlmMiddleware",
+    "WorkflowHandler",
+    "DefaultWorkflowHandler",
+    "WorkflowResult",
+    "ErrorRecoveryStrategy",
+    "ToolContextEnricher",
+    "LlmContextEnhancer",
+    "DefaultLlmContextEnhancer",
+    "ConversationFilter",
+    "ObservabilityProvider",
+    "AuditLogger",
+    "T",
+    # Audit
+    "AuditEvent",
+    "AuditEventType",
+    "ToolAccessCheckEvent",
+    "ToolInvocationEvent",
+    "ToolResultEvent",
+    "UiFeatureAccessCheckEvent",
+    "AiResponseEvent",
+    # UI Components
+    "UiComponent",
+    # Simple Components
+    "SimpleComponent",
+    "SimpleComponentType",
+    "SimpleTextComponent",
+    "SimpleImageComponent",
+    "SimpleLinkComponent",
+    # Rich Components
+    "RichComponent",
+    "ArtifactComponent",
+    "BadgeComponent",
+    "CardComponent",
+    "DataFrameComponent",
+    "IconTextComponent",
+    "LogViewerComponent",
+    "NotificationComponent",
+    "ProgressBarComponent",
+    "ProgressDisplayComponent",
+    "RichTextComponent",
+    "StatusCardComponent",
+    "TaskListComponent",
+    # Core implementations
+    "ToolRegistry",
+    "Agent",
+    "AgentConfig",
+    "DefaultSystemPromptBuilder",
+    # Evaluation
+    "Evaluator",
+    "TestCase",
+    "ExpectedOutcome",
+    "AgentResult",
+    "EvaluationResult",
+    "TestCaseResult",
+    "AgentVariant",
+    "EvaluationRunner",
+    "TrajectoryEvaluator",
+    "OutputEvaluator",
+    "LLMAsJudgeEvaluator",
+    "EfficiencyEvaluator",
+    "EvaluationReport",
+    "ComparisonReport",
+    "EvaluationDataset",
+    # Exceptions
+    "AgentError",
+    "ToolExecutionError",
+    "ToolNotFoundError",
+    "PermissionError",
+    "ConversationNotFoundError",
+    "LlmServiceError",
+    "ValidationError",
+]

vanna/core/_compat.py

@@ -0,0 +1,19 @@
+"""
+Compatibility shims for different Python versions.
+
+This module provides compatibility utilities for features that vary across
+Python versions.
+"""
+
+try:
+    from enum import StrEnum  # Py 3.11+
+except ImportError:  # Py < 3.11
+    from enum import Enum
+
+    class StrEnum(str, Enum):  # type: ignore[no-redef]
+        """Minimal backport of StrEnum for Python < 3.11."""
+
+        pass
+
+
+__all__ = ["StrEnum"]

vanna/core/agent/__init__.py

@@ -0,0 +1,10 @@
+"""
+Agent module.
+
+This module contains the core Agent implementation and configuration.
+"""
+
+from .agent import Agent
+from .config import AgentConfig
+
+__all__ = ["Agent", "AgentConfig"]