PyPI - atlas-chat - Versions diffs - 0.1.0__py3-none-any.whl - Mend

atlas-chat 0.1.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (250) hide show

atlas/__init__.py +40 -0
atlas/application/__init__.py +7 -0
atlas/application/chat/__init__.py +7 -0
atlas/application/chat/agent/__init__.py +10 -0
atlas/application/chat/agent/act_loop.py +179 -0
atlas/application/chat/agent/factory.py +142 -0
atlas/application/chat/agent/protocols.py +46 -0
atlas/application/chat/agent/react_loop.py +338 -0
atlas/application/chat/agent/think_act_loop.py +171 -0
atlas/application/chat/approval_manager.py +151 -0
atlas/application/chat/elicitation_manager.py +191 -0
atlas/application/chat/events/__init__.py +1 -0
atlas/application/chat/events/agent_event_relay.py +112 -0
atlas/application/chat/modes/__init__.py +1 -0
atlas/application/chat/modes/agent.py +125 -0
atlas/application/chat/modes/plain.py +74 -0
atlas/application/chat/modes/rag.py +81 -0
atlas/application/chat/modes/tools.py +179 -0
atlas/application/chat/orchestrator.py +213 -0
atlas/application/chat/policies/__init__.py +1 -0
atlas/application/chat/policies/tool_authorization.py +99 -0
atlas/application/chat/preprocessors/__init__.py +1 -0
atlas/application/chat/preprocessors/message_builder.py +92 -0
atlas/application/chat/preprocessors/prompt_override_service.py +104 -0
atlas/application/chat/service.py +454 -0
atlas/application/chat/utilities/__init__.py +6 -0
atlas/application/chat/utilities/error_handler.py +367 -0
atlas/application/chat/utilities/event_notifier.py +546 -0
atlas/application/chat/utilities/file_processor.py +613 -0
atlas/application/chat/utilities/tool_executor.py +789 -0
atlas/atlas_chat_cli.py +347 -0
atlas/atlas_client.py +238 -0
atlas/core/__init__.py +0 -0
atlas/core/auth.py +205 -0
atlas/core/authorization_manager.py +27 -0
atlas/core/capabilities.py +123 -0
atlas/core/compliance.py +215 -0
atlas/core/domain_whitelist.py +147 -0
atlas/core/domain_whitelist_middleware.py +82 -0
atlas/core/http_client.py +28 -0
atlas/core/log_sanitizer.py +102 -0
atlas/core/metrics_logger.py +59 -0
atlas/core/middleware.py +131 -0
atlas/core/otel_config.py +242 -0
atlas/core/prompt_risk.py +200 -0
atlas/core/rate_limit.py +0 -0
atlas/core/rate_limit_middleware.py +64 -0
atlas/core/security_headers_middleware.py +51 -0
atlas/domain/__init__.py +37 -0
atlas/domain/chat/__init__.py +1 -0
atlas/domain/chat/dtos.py +85 -0
atlas/domain/errors.py +96 -0
atlas/domain/messages/__init__.py +12 -0
atlas/domain/messages/models.py +160 -0
atlas/domain/rag_mcp_service.py +664 -0
atlas/domain/sessions/__init__.py +7 -0
atlas/domain/sessions/models.py +36 -0
atlas/domain/unified_rag_service.py +371 -0
atlas/infrastructure/__init__.py +10 -0
atlas/infrastructure/app_factory.py +135 -0
atlas/infrastructure/events/__init__.py +1 -0
atlas/infrastructure/events/cli_event_publisher.py +140 -0
atlas/infrastructure/events/websocket_publisher.py +140 -0
atlas/infrastructure/sessions/in_memory_repository.py +56 -0
atlas/infrastructure/transport/__init__.py +7 -0
atlas/infrastructure/transport/websocket_connection_adapter.py +33 -0
atlas/init_cli.py +226 -0
atlas/interfaces/__init__.py +15 -0
atlas/interfaces/events.py +134 -0
atlas/interfaces/llm.py +54 -0
atlas/interfaces/rag.py +40 -0
atlas/interfaces/sessions.py +75 -0
atlas/interfaces/tools.py +57 -0
atlas/interfaces/transport.py +24 -0
atlas/main.py +564 -0
atlas/mcp/api_key_demo/README.md +76 -0
atlas/mcp/api_key_demo/main.py +172 -0
atlas/mcp/api_key_demo/run.sh +56 -0
atlas/mcp/basictable/main.py +147 -0
atlas/mcp/calculator/main.py +149 -0
atlas/mcp/code-executor/execution_engine.py +98 -0
atlas/mcp/code-executor/execution_environment.py +95 -0
atlas/mcp/code-executor/main.py +528 -0
atlas/mcp/code-executor/result_processing.py +276 -0
atlas/mcp/code-executor/script_generation.py +195 -0
atlas/mcp/code-executor/security_checker.py +140 -0
atlas/mcp/corporate_cars/main.py +437 -0
atlas/mcp/csv_reporter/main.py +545 -0
atlas/mcp/duckduckgo/main.py +182 -0
atlas/mcp/elicitation_demo/README.md +171 -0
atlas/mcp/elicitation_demo/main.py +262 -0
atlas/mcp/env-demo/README.md +158 -0
atlas/mcp/env-demo/main.py +199 -0
atlas/mcp/file_size_test/main.py +284 -0
atlas/mcp/filesystem/main.py +348 -0
atlas/mcp/image_demo/main.py +113 -0
atlas/mcp/image_demo/requirements.txt +4 -0
atlas/mcp/logging_demo/README.md +72 -0
atlas/mcp/logging_demo/main.py +103 -0
atlas/mcp/many_tools_demo/main.py +50 -0
atlas/mcp/order_database/__init__.py +0 -0
atlas/mcp/order_database/main.py +369 -0
atlas/mcp/order_database/signal_data.csv +1001 -0
atlas/mcp/pdfbasic/main.py +394 -0
atlas/mcp/pptx_generator/main.py +760 -0
atlas/mcp/pptx_generator/requirements.txt +13 -0
atlas/mcp/pptx_generator/run_test.sh +1 -0
atlas/mcp/pptx_generator/test_pptx_generator_security.py +169 -0
atlas/mcp/progress_demo/main.py +167 -0
atlas/mcp/progress_updates_demo/QUICKSTART.md +273 -0
atlas/mcp/progress_updates_demo/README.md +120 -0
atlas/mcp/progress_updates_demo/main.py +497 -0
atlas/mcp/prompts/main.py +222 -0
atlas/mcp/public_demo/main.py +189 -0
atlas/mcp/sampling_demo/README.md +169 -0
atlas/mcp/sampling_demo/main.py +234 -0
atlas/mcp/thinking/main.py +77 -0
atlas/mcp/tool_planner/main.py +240 -0
atlas/mcp/ui-demo/badmesh.png +0 -0
atlas/mcp/ui-demo/main.py +383 -0
atlas/mcp/ui-demo/templates/button_demo.html +32 -0
atlas/mcp/ui-demo/templates/data_visualization.html +32 -0
atlas/mcp/ui-demo/templates/form_demo.html +28 -0
atlas/mcp/username-override-demo/README.md +320 -0
atlas/mcp/username-override-demo/main.py +308 -0
atlas/modules/__init__.py +0 -0
atlas/modules/config/__init__.py +34 -0
atlas/modules/config/cli.py +231 -0
atlas/modules/config/config_manager.py +1096 -0
atlas/modules/file_storage/__init__.py +22 -0
atlas/modules/file_storage/cli.py +330 -0
atlas/modules/file_storage/content_extractor.py +290 -0
atlas/modules/file_storage/manager.py +295 -0
atlas/modules/file_storage/mock_s3_client.py +402 -0
atlas/modules/file_storage/s3_client.py +417 -0
atlas/modules/llm/__init__.py +19 -0
atlas/modules/llm/caller.py +287 -0
atlas/modules/llm/litellm_caller.py +675 -0
atlas/modules/llm/models.py +19 -0
atlas/modules/mcp_tools/__init__.py +17 -0
atlas/modules/mcp_tools/client.py +2123 -0
atlas/modules/mcp_tools/token_storage.py +556 -0
atlas/modules/prompts/prompt_provider.py +130 -0
atlas/modules/rag/__init__.py +24 -0
atlas/modules/rag/atlas_rag_client.py +336 -0
atlas/modules/rag/client.py +129 -0
atlas/routes/admin_routes.py +865 -0
atlas/routes/config_routes.py +484 -0
atlas/routes/feedback_routes.py +361 -0
atlas/routes/files_routes.py +274 -0
atlas/routes/health_routes.py +40 -0
atlas/routes/mcp_auth_routes.py +223 -0
atlas/server_cli.py +164 -0
atlas/tests/conftest.py +20 -0
atlas/tests/integration/test_mcp_auth_integration.py +152 -0
atlas/tests/manual_test_sampling.py +87 -0
atlas/tests/modules/mcp_tools/test_client_auth.py +226 -0
atlas/tests/modules/mcp_tools/test_client_env.py +191 -0
atlas/tests/test_admin_mcp_server_management_routes.py +141 -0
atlas/tests/test_agent_roa.py +135 -0
atlas/tests/test_app_factory_smoke.py +47 -0
atlas/tests/test_approval_manager.py +439 -0
atlas/tests/test_atlas_client.py +188 -0
atlas/tests/test_atlas_rag_client.py +447 -0
atlas/tests/test_atlas_rag_integration.py +224 -0
atlas/tests/test_attach_file_flow.py +287 -0
atlas/tests/test_auth_utils.py +165 -0
atlas/tests/test_backend_public_url.py +185 -0
atlas/tests/test_banner_logging.py +287 -0
atlas/tests/test_capability_tokens_and_injection.py +203 -0
atlas/tests/test_compliance_level.py +54 -0
atlas/tests/test_compliance_manager.py +253 -0
atlas/tests/test_config_manager.py +617 -0
atlas/tests/test_config_manager_paths.py +12 -0
atlas/tests/test_core_auth.py +18 -0
atlas/tests/test_core_utils.py +190 -0
atlas/tests/test_docker_env_sync.py +202 -0
atlas/tests/test_domain_errors.py +329 -0
atlas/tests/test_domain_whitelist.py +359 -0
atlas/tests/test_elicitation_manager.py +408 -0
atlas/tests/test_elicitation_routing.py +296 -0
atlas/tests/test_env_demo_server.py +88 -0
atlas/tests/test_error_classification.py +113 -0
atlas/tests/test_error_flow_integration.py +116 -0
atlas/tests/test_feedback_routes.py +333 -0
atlas/tests/test_file_content_extraction.py +1134 -0
atlas/tests/test_file_extraction_routes.py +158 -0
atlas/tests/test_file_library.py +107 -0
atlas/tests/test_file_manager_unit.py +18 -0
atlas/tests/test_health_route.py +49 -0
atlas/tests/test_http_client_stub.py +8 -0
atlas/tests/test_imports_smoke.py +30 -0
atlas/tests/test_interfaces_llm_response.py +9 -0
atlas/tests/test_issue_access_denied_fix.py +136 -0
atlas/tests/test_llm_env_expansion.py +836 -0
atlas/tests/test_log_level_sensitive_data.py +285 -0
atlas/tests/test_mcp_auth_routes.py +341 -0
atlas/tests/test_mcp_client_auth.py +331 -0
atlas/tests/test_mcp_data_injection.py +270 -0
atlas/tests/test_mcp_get_authorized_servers.py +95 -0
atlas/tests/test_mcp_hot_reload.py +512 -0
atlas/tests/test_mcp_image_content.py +424 -0
atlas/tests/test_mcp_logging.py +172 -0
atlas/tests/test_mcp_progress_updates.py +313 -0
atlas/tests/test_mcp_prompt_override_system_prompt.py +102 -0
atlas/tests/test_mcp_prompts_server.py +39 -0
atlas/tests/test_mcp_tool_result_parsing.py +296 -0
atlas/tests/test_metrics_logger.py +56 -0
atlas/tests/test_middleware_auth.py +379 -0
atlas/tests/test_prompt_risk_and_acl.py +141 -0
atlas/tests/test_rag_mcp_aggregator.py +204 -0
atlas/tests/test_rag_mcp_service.py +224 -0
atlas/tests/test_rate_limit_middleware.py +45 -0
atlas/tests/test_routes_config_smoke.py +60 -0
atlas/tests/test_routes_files_download_token.py +41 -0
atlas/tests/test_routes_files_health.py +18 -0
atlas/tests/test_runtime_imports.py +53 -0
atlas/tests/test_sampling_integration.py +482 -0
atlas/tests/test_security_admin_routes.py +61 -0
atlas/tests/test_security_capability_tokens.py +65 -0
atlas/tests/test_security_file_stats_scope.py +21 -0
atlas/tests/test_security_header_injection.py +191 -0
atlas/tests/test_security_headers_and_filename.py +63 -0
atlas/tests/test_shared_session_repository.py +101 -0
atlas/tests/test_system_prompt_loading.py +181 -0
atlas/tests/test_token_storage.py +505 -0
atlas/tests/test_tool_approval_config.py +93 -0
atlas/tests/test_tool_approval_utils.py +356 -0
atlas/tests/test_tool_authorization_group_filtering.py +223 -0
atlas/tests/test_tool_details_in_config.py +108 -0
atlas/tests/test_tool_planner.py +300 -0
atlas/tests/test_unified_rag_service.py +398 -0
atlas/tests/test_username_override_in_approval.py +258 -0
atlas/tests/test_websocket_auth_header.py +168 -0
atlas/version.py +6 -0
atlas_chat-0.1.0.data/data/.env.example +253 -0
atlas_chat-0.1.0.data/data/config/defaults/compliance-levels.json +44 -0
atlas_chat-0.1.0.data/data/config/defaults/domain-whitelist.json +123 -0
atlas_chat-0.1.0.data/data/config/defaults/file-extractors.json +74 -0
atlas_chat-0.1.0.data/data/config/defaults/help-config.json +198 -0
atlas_chat-0.1.0.data/data/config/defaults/llmconfig-buggy.yml +11 -0
atlas_chat-0.1.0.data/data/config/defaults/llmconfig.yml +19 -0
atlas_chat-0.1.0.data/data/config/defaults/mcp.json +138 -0
atlas_chat-0.1.0.data/data/config/defaults/rag-sources.json +17 -0
atlas_chat-0.1.0.data/data/config/defaults/splash-config.json +16 -0
atlas_chat-0.1.0.dist-info/METADATA +236 -0
atlas_chat-0.1.0.dist-info/RECORD +250 -0
atlas_chat-0.1.0.dist-info/WHEEL +5 -0
atlas_chat-0.1.0.dist-info/entry_points.txt +4 -0
atlas_chat-0.1.0.dist-info/top_level.txt +1 -0

atlas/mcp/progress_updates_demo/README.md ADDED Viewed

@@ -0,0 +1,120 @@
+# Progress Updates Demo MCP Server
+This MCP server demonstrates the enhanced progress reporting capabilities that allow MCP servers to send viewable updates to the frontend during tool execution.
+## Features
+This demo shows three types of enhanced progress updates:
+1. **Canvas Updates**: Display HTML visualizations in the canvas panel during execution
+2. **System Messages**: Send rich messages that appear in chat history
+3. **Progress Artifacts**: Share file artifacts progressively as they're generated
+## Tools
+### `task_with_canvas_updates`
+Demonstrates sending HTML progress visualizations to the canvas during execution.
+**Parameters:**
+- `task_name` (str): Name of the task (default: "demo")
+- `steps` (int): Number of steps to process (default: 5)
+- `interval_seconds` (int): Delay between steps (default: 2)
+### `task_with_system_messages`
+Demonstrates sending rich system messages to chat history during execution.
+**Parameters:**
+- `task_name` (str): Name of the analysis task (default: "analysis")
+- `stages` (int): Number of stages to process (default: 4)
+- `interval_seconds` (int): Delay between stages (default: 2)
+### `task_with_artifacts`
+Demonstrates sending file artifacts progressively during execution.
+**Parameters:**
+- `task_name` (str): Name of the processing task (default: "data_processing")
+- `files_to_generate` (int): Number of intermediate files (default: 3)
+- `interval_seconds` (int): Delay between file generation (default: 2)
+## How It Works
+MCP servers can send structured progress updates by encoding JSON data in the progress message field with the prefix `"MCP_UPDATE:"`.
+### Supported Update Types
+#### 1. Canvas Update
+```python
+update_payload = {
+    "type": "canvas_update",
+    "content": "<html>...</html>",  # HTML content to display
+    "progress_message": "Processing..."  # Optional progress text
+}
+await ctx.report_progress(
+    progress=step,
+    total=total_steps,
+    message=f"MCP_UPDATE:{json.dumps(update_payload)}"
+)
+```
+#### 2. System Message
+```python
+update_payload = {
+    "type": "system_message",
+    "message": "Data validation complete!",
+    "subtype": "success",  # or "info", "warning", "error"
+    "progress_message": "Validating data..."
+}
+await ctx.report_progress(
+    progress=step,
+    total=total_steps,
+    message=f"MCP_UPDATE:{json.dumps(update_payload)}"
+)
+```
+#### 3. Artifacts
+```python
+update_payload = {
+    "type": "artifacts",
+    "artifacts": [
+        {
+            "name": "result.html",
+            "b64": base64_encoded_content,
+            "mime": "text/html",
+            "size": content_size,
+            "description": "Intermediate result",
+            "viewer": "html"
+        }
+    ],
+    "display": {
+        "open_canvas": True,
+        "primary_file": "result.html",
+        "mode": "replace"
+    },
+    "progress_message": "Generated result..."
+}
+await ctx.report_progress(
+    progress=step,
+    total=total_steps,
+    message=f"MCP_UPDATE:{json.dumps(update_payload)}"
+)
+```
+## Usage
+Try these example prompts:
+```
+Show me a task with canvas updates
+Run task_with_system_messages
+Generate artifacts progressively
+```
+## Benefits
+- **Better UX**: Users see real-time progress with visual feedback
+- **Reduced perceived latency**: Long-running tasks feel more responsive
+- **More informative**: Rich context about what's happening at each stage
+- **Flexible**: Can display any HTML content, images, or file artifacts

atlas/mcp/progress_updates_demo/main.py ADDED Viewed

@@ -0,0 +1,497 @@
+#!/usr/bin/env python3
+"""
+Progress Updates Demo MCP Server using FastMCP.
+This server demonstrates the enhanced progress reporting capabilities
+that allow MCP servers to send viewable updates to the frontend during
+tool execution, including:
+- Canvas updates (plots, HTML, images)
+- Rich system messages
+- Progress artifacts
+To use these features from an MCP server, send special formatted messages
+via ctx.report_progress() with the message field containing:
+  "MCP_UPDATE:{json_payload}"
+Supported update types:
+- canvas_update: Display HTML/images in canvas during execution
+- system_message: Add rich messages to chat history
+- artifacts: Send file artifacts during execution
+"""
+from __future__ import annotations
+import asyncio
+import base64
+import json
+from typing import Any, Dict
+from fastmcp import Context, FastMCP
+# Initialize the MCP server
+mcp = FastMCP("Progress Updates Demo")
+def create_progress_html(step: int, total: int, message: str) -> str:
+    """Create an HTML progress visualization."""
+    percentage = int((step / total) * 100)
+    return f"""
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <style>
+            body {{
+                font-family: Arial, sans-serif;
+                padding: 20px;
+                background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                color: white;
+            }}
+            .progress-container {{
+                background: rgba(255, 255, 255, 0.2);
+                border-radius: 10px;
+                padding: 20px;
+                backdrop-filter: blur(10px);
+            }}
+            .progress-bar {{
+                width: 100%;
+                height: 30px;
+                background: rgba(255, 255, 255, 0.3);
+                border-radius: 15px;
+                overflow: hidden;
+                margin: 10px 0;
+            }}
+            .progress-fill {{
+                height: 100%;
+                background: linear-gradient(90deg, #4CAF50, #8BC34A);
+                width: {percentage}%;
+                transition: width 0.3s ease;
+                display: flex;
+                align-items: center;
+                justify-content: center;
+                color: white;
+                font-weight: bold;
+            }}
+            h1 {{
+                margin: 0 0 10px 0;
+            }}
+            .step-info {{
+                font-size: 18px;
+                margin: 10px 0;
+            }}
+        </style>
+    </head>
+    <body>
+        <div class="progress-container">
+            <h1>Task Progress</h1>
+            <div class="step-info">Step {step} of {total}</div>
+            <div class="progress-bar">
+                <div class="progress-fill">{percentage}%</div>
+            </div>
+            <p>{message}</p>
+        </div>
+    </body>
+    </html>
+    """
+def create_chart_html(data: Dict[str, int]) -> str:
+    """Create a simple bar chart HTML."""
+    max_value = max(data.values()) if data else 1
+    bars = ""
+    for label, value in data.items():
+        percentage = int((value / max_value) * 100)
+        bars += f"""
+        <div class="bar-container">
+            <div class="bar-label">{label}</div>
+            <div class="bar-wrapper">
+                <div class="bar-fill" style="width: {percentage}%">
+                    <span class="bar-value">{value}</span>
+                </div>
+            </div>
+        </div>
+        """
+    return f"""
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <style>
+            body {{
+                font-family: Arial, sans-serif;
+                padding: 20px;
+                background: linear-gradient(135deg, #f093fb 0%, #f5576c 100%);
+            }}
+            .chart-container {{
+                background: white;
+                border-radius: 10px;
+                padding: 20px;
+                box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+            }}
+            h2 {{
+                color: #333;
+                margin-top: 0;
+            }}
+            .bar-container {{
+                margin: 15px 0;
+            }}
+            .bar-label {{
+                font-weight: bold;
+                margin-bottom: 5px;
+                color: #555;
+            }}
+            .bar-wrapper {{
+                background: #e0e0e0;
+                border-radius: 5px;
+                overflow: hidden;
+                height: 30px;
+            }}
+            .bar-fill {{
+                background: linear-gradient(90deg, #667eea, #764ba2);
+                height: 100%;
+                display: flex;
+                align-items: center;
+                justify-content: flex-end;
+                padding-right: 10px;
+                transition: width 0.5s ease;
+            }}
+            .bar-value {{
+                color: white;
+                font-weight: bold;
+            }}
+        </style>
+    </head>
+    <body>
+        <div class="chart-container">
+            <h2>Processing Results</h2>
+            {bars}
+        </div>
+    </body>
+    </html>
+    """
+@mcp.tool
+async def task_with_canvas_updates(
+    task_name: str = "demo",
+    steps: int = 5,
+    interval_seconds: int = 2,
+    ctx: Context | None = None,
+) -> Dict[str, Any]:
+    """
+    Execute a long-running task with visual progress updates in the canvas.
+    This tool demonstrates how MCP servers can send canvas updates during
+    execution, allowing users to see real-time visual progress indicators.
+    Args:
+        task_name: Name of the task to execute
+        steps: Number of steps to process (default: 5)
+        interval_seconds: Delay between steps (default: 2)
+        ctx: MCP context for progress reporting
+    Returns:
+        Task completion summary with final results
+    """
+    total = max(1, int(steps))
+    interval = max(1, int(interval_seconds))
+    # Initial progress
+    if ctx:
+        await ctx.report_progress(
+            progress=0,
+            total=total,
+            message=f"Starting {task_name}..."
+        )
+    # Process each step and send visual updates as artifacts
+    for step in range(1, total + 1):
+        await asyncio.sleep(interval)
+        # Create progress visualization HTML
+        html_content = create_progress_html(step, total, f"Processing {task_name}: Step {step}")
+        # Send progress HTML as an artifact so it uses the HTML viewer
+        if ctx:
+            artifact_html = base64.b64encode(html_content.encode("utf-8")).decode("utf-8")
+            update_payload = {
+                "type": "artifacts",
+                "artifacts": [
+                    {
+                        "name": f"progress_step_{step}.html",
+                        "b64": artifact_html,
+                        "mime": "text/html",
+                        "size": len(html_content),
+                        "description": f"Progress for {task_name} step {step}/{total}",
+                        "viewer": "html",
+                    }
+                ],
+                "display": {
+                    "open_canvas": True,
+                    "primary_file": f"progress_step_{step}.html",
+                    "mode": "replace",
+                },
+                "progress_message": f"{task_name}: Step {step}/{total}",
+            }
+            await ctx.report_progress(
+                progress=step,
+                total=total,
+                message=f"MCP_UPDATE:{json.dumps(update_payload)}",
+            )
+    # Final result with chart
+    result_data = {
+        "Items Processed": total * 10,
+        "Errors Found": 2,
+        "Warnings": 5,
+        "Success Rate": 95
+    }
+    chart_html = create_chart_html(result_data)
+    return {
+        "results": {
+            "task": task_name,
+            "status": "completed",
+            "steps_completed": total,
+            "summary": result_data
+        },
+        "artifacts": [
+            {
+                "name": "final_results.html",
+                "b64": base64.b64encode(chart_html.encode('utf-8')).decode('utf-8'),
+                "mime": "text/html",
+                "size": len(chart_html),
+                "description": "Final processing results chart",
+                "viewer": "html"
+            }
+        ],
+        "display": {
+            "open_canvas": True,
+            "primary_file": "final_results.html",
+            "mode": "replace",
+            "viewer_hint": "html"
+        }
+    }
+@mcp.tool
+async def task_with_system_messages(
+    task_name: str = "analysis",
+    stages: int = 4,
+    interval_seconds: int = 2,
+    ctx: Context | None = None,
+) -> Dict[str, Any]:
+    """
+    Execute a task with rich system messages displayed in chat history.
+    This tool demonstrates how MCP servers can send rich system messages
+    that appear as new items in the chat history during tool execution.
+    Args:
+        task_name: Name of the analysis task
+        stages: Number of stages to process (default: 4)
+        interval_seconds: Delay between stages (default: 2)
+        ctx: MCP context for progress reporting
+    Returns:
+        Analysis completion summary
+    """
+    stage_names = [
+        "Data Collection",
+        "Data Validation",
+        "Analysis",
+        "Report Generation"
+    ][:stages]
+    total = len(stage_names)
+    # Initial progress
+    if ctx:
+        await ctx.report_progress(
+            progress=0,
+            total=total,
+            message=f"Starting {task_name}..."
+        )
+    # Process each stage and send system messages
+    for i, stage in enumerate(stage_names, 1):
+        await asyncio.sleep(interval_seconds)
+        # Send system message
+        if ctx:
+            update_payload = {
+                "type": "system_message",
+                "message": f"**{stage}** - Stage {i}/{total} completed successfully",
+                "subtype": "success",
+                "progress_message": f"{task_name}: {stage}"
+            }
+            await ctx.report_progress(
+                progress=i,
+                total=total,
+                message=f"MCP_UPDATE:{json.dumps(update_payload)}"
+            )
+    return {
+        "results": {
+            "task": task_name,
+            "status": "completed",
+            "stages_completed": total,
+            "completion_message": f"All {total} stages completed successfully"
+        }
+    }
+@mcp.tool
+async def task_with_artifacts(
+    task_name: str = "data_processing",
+    files_to_generate: int = 3,
+    interval_seconds: int = 2,
+    ctx: Context | None = None,
+) -> Dict[str, Any]:
+    """
+    Execute a task that generates and displays artifacts progressively.
+    This tool demonstrates how MCP servers can send file artifacts during
+    execution, allowing users to see intermediate results as they're generated.
+    Args:
+        task_name: Name of the processing task
+        files_to_generate: Number of intermediate files to create (default: 3)
+        interval_seconds: Delay between file generation (default: 2)
+        ctx: MCP context for progress reporting
+    Returns:
+        Processing completion summary
+    """
+    total = max(1, int(files_to_generate))
+    interval = max(1, int(interval_seconds))
+    # Initial progress
+    if ctx:
+        await ctx.report_progress(
+            progress=0,
+            total=total,
+            message=f"Starting {task_name}..."
+        )
+    # Generate intermediate files
+    for file_num in range(1, total + 1):
+        await asyncio.sleep(interval)
+        # Create intermediate result HTML
+        intermediate_html = f"""
+        <!DOCTYPE html>
+        <html>
+        <head>
+            <style>
+                body {{
+                    font-family: Arial, sans-serif;
+                    padding: 20px;
+                    background: linear-gradient(135deg, #89f7fe 0%, #66a6ff 100%);
+                }}
+                .result {{
+                    background: white;
+                    border-radius: 10px;
+                    padding: 20px;
+                    box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+                }}
+                h2 {{ color: #333; margin-top: 0; }}
+                .data {{ color: #666; font-size: 18px; }}
+            </style>
+        </head>
+        <body>
+            <div class="result">
+                <h2>Intermediate Result {file_num}</h2>
+                <p class="data">Generated at step {file_num} of {total}</p>
+                <p class="data">Processing status: In Progress</p>
+            </div>
+        </body>
+        </html>
+        """
+        # Send artifact via structured progress message
+        if ctx:
+            artifact_data = {
+                "type": "artifacts",
+                "artifacts": [
+                    {
+                        "name": f"intermediate_result_{file_num}.html",
+                        "b64": base64.b64encode(intermediate_html.encode('utf-8')).decode('utf-8'),
+                        "mime": "text/html",
+                        "size": len(intermediate_html),
+                        "description": f"Intermediate result {file_num}",
+                        "viewer": "html"
+                    }
+                ],
+                "display": {
+                    "open_canvas": True,
+                    "primary_file": f"intermediate_result_{file_num}.html",
+                    "mode": "replace"
+                },
+                "progress_message": f"Generated file {file_num}/{total}"
+            }
+            await ctx.report_progress(
+                progress=file_num,
+                total=total,
+                message=f"MCP_UPDATE:{json.dumps(artifact_data)}"
+            )
+    # Final result
+    final_html = """
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <style>
+            body {
+                font-family: Arial, sans-serif;
+                padding: 20px;
+                background: linear-gradient(135deg, #a8edea 0%, #fed6e3 100%);
+            }
+            .final-result {
+                background: white;
+                border-radius: 10px;
+                padding: 30px;
+                box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+                text-align: center;
+            }
+            h1 { color: #4CAF50; }
+            .success-icon { font-size: 64px; }
+        </style>
+    </head>
+    <body>
+        <div class="final-result">
+            <div class="success-icon">✓</div>
+            <h1>Processing Complete!</h1>
+            <p>All files have been generated successfully.</p>
+        </div>
+    </body>
+    </html>
+    """
+    return {
+        "results": {
+            "task": task_name,
+            "status": "completed",
+            "files_generated": total
+        },
+        "artifacts": [
+            {
+                "name": "final_result.html",
+                "b64": base64.b64encode(final_html.encode('utf-8')).decode('utf-8'),
+                "mime": "text/html",
+                "size": len(final_html),
+                "description": "Final processing result",
+                "viewer": "html"
+            }
+        ],
+        "display": {
+            "open_canvas": True,
+            "primary_file": "final_result.html",
+            "mode": "replace",
+            "viewer_hint": "html"
+        }
+    }
+if __name__ == "__main__":
+    mcp.run()