aline-ai 0.2.6__py3-none-any.whl → 0.3.0__py3-none-any.whl

realign/mcp_server.py

@@ -1,581 +1,305 @@
 #!/usr/bin/env python3
-"""Aline MCP Server - AI Agent Chat Session Tracker via Model Context Protocol."""
+"""Aline MCP Server - Query shared conversations via Model Context Protocol."""
 
 import asyncio
+import hashlib
 import json
-import subprocess
 import sys
-from pathlib import Path
 from typing import Any, Optional
+from urllib.parse import urlparse
+
+try:
+    import httpx
+    HTTPX_AVAILABLE = True
+except ImportError:
+    HTTPX_AVAILABLE = False
 
 from mcp.server import Server
 from mcp.server.stdio import stdio_server
 from mcp.types import Tool, TextContent
 
-from .config import ReAlignConfig
-from .commands import init, search, show, commit
-from .mcp_watcher import start_watcher
-
 from . import __version__
 
 # Initialize MCP server
 app = Server("aline")
 
 
-@app.list_tools()
-async def list_tools() -> list[Tool]:
-    """List available Aline tools."""
-    return [
-        Tool(
-            name="aline_init",
-            description=(
-                "Initialize Aline in a directory to enable AI conversation tracking. "
-                "This command automatically sets up the workspace and configures session tracking. "
-                "Use this to start tracking AI conversations in your project."
-            ),
-            inputSchema={
-                "type": "object",
-                "properties": {
-                    "repo_path": {
-                        "type": "string",
-                        "description": "Path to directory to initialize (default: current directory)",
-                    },
-                },
-            },
-        ),
-        Tool(
-            name="aline_search",
-            description=(
-                "Search through AI agent chat sessions. "
-                "Finds sessions with matching keywords in session content. "
-                "Returns session information and optionally full content. "
-                "Can search in sessions only or include metadata."
-            ),
-            inputSchema={
-                "type": "object",
-                "properties": {
-                    "keyword": {
-                        "type": "string",
-                        "description": "Keyword to search for in sessions",
-                    },
-                    "repo_path": {
-                        "type": "string",
-                        "description": "Path to workspace (default: current directory)",
-                    },
-                    "show_session": {
-                        "type": "boolean",
-                        "description": "Include full session content in results",
-                        "default": False,
-                    },
-                    "max_results": {
-                        "type": "integer",
-                        "description": "Maximum number of results to return",
-                        "default": 20,
-                    },
-                    "session_only": {
-                        "type": "boolean",
-                        "description": "Search only in session files. Useful for finding topics discussed in historical conversations.",
-                        "default": False,
-                    },
-                    "commits_only": {
-                        "type": "boolean",
-                        "description": "Search only in metadata, not session files.",
-                        "default": False,
-                    },
-                },
-                "required": ["keyword"],
-            },
-        ),
-        Tool(
-            name="aline_show",
-            description=(
-                "Display an AI agent chat session. "
-                "Supports navigation through the session with line ranges. "
-                "Use aline_search first to find matching lines, then use this tool to view context around those lines."
-            ),
-            inputSchema={
-                "type": "object",
-                "properties": {
-                    "commit_hash": {
-                        "type": "string",
-                        "description": "Session identifier (full or short)",
-                    },
-                    "repo_path": {
-                        "type": "string",
-                        "description": "Path to workspace (default: current directory)",
-                    },
-                    "session_path": {
-                        "type": "string",
-                        "description": "Direct path to session file (alternative to commit_hash)",
-                    },
-                    "format": {
-                        "type": "string",
-                        "enum": ["pretty", "json", "raw"],
-                        "description": "Output format",
-                        "default": "pretty",
-                    },
-                    "from_line": {
-                        "type": "integer",
-                        "description": "Start from line number (inclusive, 1-indexed)",
-                    },
-                    "to_line": {
-                        "type": "integer",
-                        "description": "End at line number (inclusive, 1-indexed)",
-                    },
-                    "around_line": {
-                        "type": "integer",
-                        "description": "Show lines around this line number (use with context)",
-                    },
-                    "context": {
-                        "type": "integer",
-                        "description": "Number of lines before/after when using around_line (default: 5)",
-                        "default": 5,
-                    },
-                    "first": {
-                        "type": "integer",
-                        "description": "Show only first N lines",
-                    },
-                    "last": {
-                        "type": "integer",
-                        "description": "Show only last N lines",
-                    },
-                },
-            },
-        ),
-        Tool(
-            name="aline_get_latest_session",
-            description=(
-                "Get the path to the most recent AI agent chat session. "
-                "Automatically detects Claude Code sessions or uses configured history path. "
-                "Useful for retrieving the current conversation context."
-            ),
-            inputSchema={
-                "type": "object",
-                "properties": {
-                    "repo_path": {
-                        "type": "string",
-                        "description": "Path to workspace (for Claude auto-detection)",
-                    },
-                },
-            },
-        ),
-        Tool(
-            name="aline_version",
-            description=(
-                f"Get the current version of Aline ({__version__}). "
-                "Aline is a tool that tracks and preserves AI agent chat sessions."
-            ),
-            inputSchema={
-                "type": "object",
-                "properties": {},
-            },
-        ),
-        Tool(
-            name="aline_review",
-            description=(
-                "Review unpushed commits before pushing. "
-                "Shows commit messages, LLM summaries, session files modified, and line counts. "
-                "Helps identify what content will be made public when you push."
-            ),
-            inputSchema={
-                "type": "object",
-                "properties": {
-                    "repo_path": {
-                        "type": "string",
-                        "description": "Path to repository root (default: current directory)",
-                    },
-                    "verbose": {
-                        "type": "boolean",
-                        "description": "Show detailed information",
-                        "default": False,
-                    },
-                    "detect_secrets": {
-                        "type": "boolean",
-                        "description": "Run sensitive content detection",
-                        "default": False,
-                    },
-                },
-            },
-        ),
-        Tool(
-            name="aline_hide",
-            description=(
-                "Hide (redact) specific unpushed commits by rewriting git history. "
-                "This redacts commit messages and session content. "
-                "WARNING: This rewrites git history. Use with caution."
-            ),
-            inputSchema={
-                "type": "object",
-                "properties": {
-                    "indices": {
-                        "type": "string",
-                        "description": (
-                            "Commit indices to hide (e.g., '1', '1,3,5', '2-4', or '--all'). "
-                            "Use aline_review first to see the commit indices."
-                        ),
-                    },
-                    "repo_path": {
-                        "type": "string",
-                        "description": "Path to repository root (default: current directory)",
-                    },
-                    "force": {
-                        "type": "boolean",
-                        "description": "Skip confirmation prompt",
-                        "default": False,
-                    },
-                },
-                "required": ["indices"],
-            },
-        ),
-    ]
+def extract_share_id(share_url: str) -> str:
+    """
+    Extract share ID from share URL.
 
+    Examples:
+        https://realign-server.vercel.app/share/abc123 -> abc123
+        https://example.com/share/xyz789/chat -> xyz789
+    """
+    # Remove trailing slash
+    url = share_url.rstrip('/')
 
-@app.call_tool()
-async def call_tool(name: str, arguments: Any) -> list[TextContent]:
-    """Handle tool execution."""
+    # Extract path components
+    parsed = urlparse(url)
+    path_parts = [p for p in parsed.path.split('/') if p]
 
-    try:
-        # Execute the tool
-        if name == "aline_init":
-            result = await handle_init(arguments)
-        elif name == "aline_search":
-            result = await handle_search(arguments)
-        elif name == "aline_show":
-            result = await handle_show(arguments)
-        elif name == "aline_get_latest_session":
-            result = await handle_get_latest_session(arguments)
-        elif name == "aline_version":
-            result = await handle_version(arguments)
-        elif name == "aline_review":
-            result = await handle_review(arguments)
-        elif name == "aline_hide":
-            result = await handle_hide(arguments)
-        else:
-            return [TextContent(type="text", text=f"Unknown tool: {name}")]
+    # Find 'share' in path and get next component
+    if 'share' in path_parts:
+        share_idx = path_parts.index('share')
+        if share_idx + 1 < len(path_parts):
+            return path_parts[share_idx + 1]
 
-        return result
+    raise ValueError(f"Could not extract share ID from URL: {share_url}")
 
-    except Exception as e:
-        return [TextContent(type="text", text=f"Error: {str(e)}")]
-
-
-async def handle_init(args: dict) -> list[TextContent]:
-    """Handle aline_init tool."""
-    from .commands.init import init_repository
-
-    repo_path = args.get("repo_path", ".")
-
-    # Call Python function directly instead of subprocess
-    # This avoids PATH issues completely
-    result = init_repository(
-        repo_path=repo_path,
-        auto_init_git=True,
-        skip_commit=False,
-    )
-
-    # Format output with detailed parameters
-    if result["success"]:
-        output_lines = [
-            "✅ Aline initialized successfully\n",
-            "Parameters:",
-            f"  • Workspace Path: {result.get('repo_path', 'N/A')}",
-            f"  • Workspace Root: {result.get('repo_root', 'N/A')}",
-            f"  • Aline Directory: {result.get('realign_dir', 'N/A')}",
-            f"  • Config Path: {result.get('config_path', 'N/A')}",
-            f"  • History Directory: {result.get('history_dir', 'N/A')}",
-            f"  • Initialized: {result.get('git_initialized', False)}",
-            f"  • Hooks Created: {', '.join(result.get('hooks_created', [])) or 'None'}",
-            f"  • Auto-committed: {result.get('committed', False)}",
-            "",
-            "Aline is now tracking your AI conversations automatically.",
-            "Use aline_search to find past discussions.",
-        ]
-        return [TextContent(type="text", text="\n".join(output_lines))]
-    else:
-        error_lines = [
-            "❌ Failed to initialize Aline\n",
-            f"Error: {result['message']}",
-        ]
-        if result.get("errors"):
-            error_lines.append("\nDetails:")
-            for error in result["errors"]:
-                error_lines.append(f"  • {error}")
-        return [TextContent(type="text", text="\n".join(error_lines))]
-
-
-async def handle_search(args: dict) -> list[TextContent]:
-    """Handle aline_search tool."""
-    keyword = args["keyword"]
-    repo_path = args.get("repo_path", ".")
-    show_session = args.get("show_session", False)
-    max_results = args.get("max_results", 20)
-    session_only = args.get("session_only", False)
-    commits_only = args.get("commits_only", False)
-
-    # Build command
-    cmd = ["aline", "search", keyword, "--max", str(max_results)]
-    if show_session:
-        cmd.append("--show-session")
-    if session_only:
-        cmd.append("--session-only")
-    if commits_only:
-        cmd.append("--commits-only")
-
-    # Run command
-    result = subprocess.run(
-        cmd,
-        cwd=repo_path,
-        capture_output=True,
-        text=True,
-    )
-
-    output = result.stdout
-    if result.returncode == 0:
-        return [TextContent(type="text", text=output or "No results found")]
-    else:
-        return [TextContent(
-            type="text",
-            text=f"Error searching: {result.stderr}"
-        )]
 
+def extract_base_url(share_url: str) -> str:
+    """
+    Extract base URL from share URL.
 
-async def handle_show(args: dict) -> list[TextContent]:
-    """Handle aline_show tool."""
-    commit_hash = args.get("commit_hash")
-    repo_path = args.get("repo_path", ".")
-    session_path = args.get("session_path")
-    format_type = args.get("format", "pretty")
-    from_line = args.get("from_line")
-    to_line = args.get("to_line")
-    around_line = args.get("around_line")
-    context = args.get("context", 5)
-    first = args.get("first")
-    last = args.get("last")
-
-    # Build command
-    cmd = ["aline", "show"]
-    if session_path:
-        cmd.extend(["--session", session_path])
-    elif commit_hash:
-        cmd.append(commit_hash)
-    else:
-        return [TextContent(
-            type="text",
-            text="Error: Must provide either commit_hash or session_path"
-        )]
+    Examples:
+        https://realign-server.vercel.app/share/abc123 -> https://realign-server.vercel.app
+    """
+    parsed = urlparse(share_url)
+    return f"{parsed.scheme}://{parsed.netloc}"
 
-    cmd.extend(["--format", format_type])
-
-    # Add range parameters
-    if from_line is not None:
-        cmd.extend(["--from", str(from_line)])
-    if to_line is not None:
-        cmd.extend(["--to", str(to_line)])
-    if around_line is not None:
-        cmd.extend(["--around", str(around_line)])
-        if context != 5:  # Only add if not default
-            cmd.extend(["--context", str(context)])
-    if first is not None:
-        cmd.extend(["--first", str(first)])
-    if last is not None:
-        cmd.extend(["--last", str(last)])
-
-    # Run command
-    result = subprocess.run(
-        cmd,
-        cwd=repo_path,
-        capture_output=True,
-        text=True,
-    )
-
-    if result.returncode == 0:
-        return [TextContent(type="text", text=result.stdout)]
-    else:
-        return [TextContent(
-            type="text",
-            text=f"Error showing session: {result.stderr}"
-        )]
 
+async def authenticate_share(share_url: str, password: Optional[str] = None) -> tuple[str, str, str]:
+    """
+    Authenticate with share and get session token.
+
+    Args:
+        share_url: The share URL
+        password: Optional password for encrypted shares
+
+    Returns:
+        tuple of (base_url, share_id, session_token)
+
+    Raises:
+        ValueError: If authentication fails
+        httpx.HTTPError: If network request fails
+    """
+    if not HTTPX_AVAILABLE:
+        raise RuntimeError("httpx package is required. Install with: pip install httpx")
+
+    share_id = extract_share_id(share_url)
+    base_url = extract_base_url(share_url)
 
-async def handle_get_latest_session(args: dict) -> list[TextContent]:
-    """Handle aline_get_latest_session tool."""
-    from .claude_detector import find_claude_sessions_dir
+    async with httpx.AsyncClient(timeout=30.0) as client:
+        # Step 1: Check if password is required
+        info_resp = await client.get(f"{base_url}/api/share/{share_id}/info")
+        info_resp.raise_for_status()
+        info = info_resp.json()
 
-    repo_path = args.get("repo_path", ".")
-    config = ReAlignConfig.load()
+        requires_password = info.get("requires_password", False)
 
-    # Try Claude auto-detection first
-    if config.auto_detect_claude:
-        claude_dir = find_claude_sessions_dir(Path(repo_path))
-        if claude_dir:
-            # Find latest session in Claude directory
-            session_files = sorted(
-                claude_dir.glob("*.jsonl"),
-                key=lambda p: p.stat().st_mtime,
-                reverse=True,
+        # Step 2: Authenticate
+        if requires_password:
+            if not password:
+                raise ValueError("This share requires a password, but none was provided")
+
+            # Hash password (SHA-256)
+            password_hash = hashlib.sha256(password.encode()).hexdigest()
+
+            auth_resp = await client.post(
+                f"{base_url}/api/share/{share_id}/auth",
+                json={"password_hash": password_hash}
+            )
+            auth_resp.raise_for_status()
+            data = auth_resp.json()
+        else:
+            # No password needed - create session directly
+            session_resp = await client.post(
+                f"{base_url}/api/share/{share_id}/session"
             )
-            if session_files:
-                return [TextContent(
-                    type="text",
-                    text=f"Latest Claude Code session: {session_files[0]}"
-                )]
+            session_resp.raise_for_status()
+            data = session_resp.json()
+
+        session_token = data.get("session_token")
+        if not session_token:
+            raise ValueError("Failed to obtain session token")
+
+        return base_url, share_id, session_token
+
+
+async def ask_conversation(
+    base_url: str,
+    share_id: str,
+    session_token: str,
+    question: str
+) -> str:
+    """
+    Send a question to the remote AI agent and receive the answer.
+
+    Args:
+        base_url: Base URL of the share service
+        share_id: Share identifier
+        session_token: Session token from authentication
+        question: The question to ask
+
+    Returns:
+        The answer from the remote AI agent
+
+    Raises:
+        httpx.HTTPError: If network request fails
+    """
+    if not HTTPX_AVAILABLE:
+        raise RuntimeError("httpx package is required. Install with: pip install httpx")
+
+    async with httpx.AsyncClient(timeout=120.0) as client:
+        # Send question to chat API with proper UIMessage format
+        # UIMessage requires 'parts' instead of 'content'
+        resp = await client.post(
+            f"{base_url}/api/chat/{share_id}",
+            headers={"x-session-token": session_token},
+            json={"messages": [{"role": "user", "parts": [{"type": "text", "text": question}]}]}
+        )
+        resp.raise_for_status()
+
+        # Parse streaming response from Vercel AI SDK
+        # The AI SDK returns a UIMessageStreamResponse with multiple data chunks
+        # We only need to extract the final text content
+        text_chunks = []
+
+        async for chunk in resp.aiter_text():
+            # Split into lines and process each
+            for line in chunk.split('\n'):
+                line = line.strip()
+                if not line:
+                    continue
+
+                # Remove "data: " prefix if present
+                if line.startswith('data: '):
+                    line = line[6:]
+
+                # Try to parse as JSON
+                try:
+                    data = json.loads(line)
+
+                    # Vercel AI SDK sends different types of chunks
+                    # We're looking for text deltas (type: "text-delta")
+                    if isinstance(data, dict):
+                        # Extract text from text-delta chunks
+                        # The field name is 'delta' not 'textDelta'
+                        if data.get('type') == 'text-delta':
+                            delta = data.get('delta', '')
+                            if delta:
+                                text_chunks.append(delta)
+
+                except json.JSONDecodeError:
+                    # Not JSON, skip
+                    continue
+
+        # Combine all text chunks to get the final answer
+        answer = ''.join(text_chunks)
+
+        # Apply reasonable length limit to prevent overwhelming the MCP client
+        # If answer is too long, truncate and add notice
+        MAX_RESPONSE_LENGTH = 50000  # ~50KB of text
+        if len(answer) > MAX_RESPONSE_LENGTH:
+            answer = answer[:MAX_RESPONSE_LENGTH] + "\n\n[Response truncated due to length. Please ask more specific questions to get complete answers.]"
+
+        return answer if answer else "No response received from the agent."
+
+
+async def handle_ask_tool(
+    share_url: str,
+    question: str,
+    password: Optional[str] = None
+) -> list[TextContent]:
+    """
+    Handle the ask_shared_conversation tool.
+
+    Args:
+        share_url: URL of the shared conversation
+        question: Question to ask
+        password: Optional password for encrypted shares
+
+    Returns:
+        List of TextContent with the answer or error
+    """
+    try:
+        # Authenticate and get session token
+        base_url, share_id, token = await authenticate_share(share_url, password)
 
-    # Fallback to configured history path
-    history_path = config.expanded_local_history_path
+        # Ask the question
+        answer = await ask_conversation(base_url, share_id, token, question)
 
-    if not history_path.exists():
+        # Return the answer
         return [TextContent(
             type="text",
-            text=f"No sessions found. History path does not exist: {history_path}"
+            text=answer
         )]
 
-    # Find latest session file
-    session_files = sorted(
-        history_path.glob("*.jsonl"),
-        key=lambda p: p.stat().st_mtime,
-        reverse=True,
-    )
-
-    if session_files:
+    except ValueError as e:
         return [TextContent(
             type="text",
-            text=f"Latest session: {session_files[0]}"
+            text=f"Authentication error: {str(e)}"
         )]
-    else:
+    except Exception as e:
         return [TextContent(
             type="text",
-            text=f"No session files found in {history_path}"
+            text=f"Error querying shared conversation: {str(e)}"
         )]
 
 
-async def handle_version(args: dict) -> list[TextContent]:
-    """Handle aline_version tool."""
-    return [TextContent(
-        type="text",
-        text=f"Aline version {__version__}"
-    )]
-
-
-async def handle_review(args: dict) -> list[TextContent]:
-    """Handle aline_review tool."""
-    from .commands.review import review_command
-    from io import StringIO
-    import sys
-
-    repo_path = args.get("repo_path")
-    verbose = args.get("verbose", False)
-    detect_secrets = args.get("detect_secrets", False)
-
-    # Convert repo_path to Path if provided
-    repo_root = Path(repo_path) if repo_path else None
-
-    # Capture stdout
-    old_stdout = sys.stdout
-    sys.stdout = captured_output = StringIO()
-
-    try:
-        # Call the review command
-        exit_code = review_command(
-            repo_root=repo_root,
-            verbose=verbose,
-            detect_secrets=detect_secrets
-        )
-
-        # Get the output
-        output = captured_output.getvalue()
-
-        if exit_code == 0:
-            return [TextContent(type="text", text=output or "No unpushed commits found.")]
-        else:
-            return [TextContent(type="text", text=f"Error: Review failed\n{output}")]
-    finally:
-        sys.stdout = old_stdout
-
-
-async def handle_hide(args: dict) -> list[TextContent]:
-    """Handle aline_hide tool."""
-    from .commands.hide import hide_command
-    from io import StringIO
-    import sys
-
-    indices = args["indices"]
-    repo_path = args.get("repo_path")
-    force = args.get("force", False)
-
-    # Convert repo_path to Path if provided
-    repo_root = Path(repo_path) if repo_path else None
+@app.list_tools()
+async def list_tools() -> list[Tool]:
+    """List available tools."""
+    return [
+        Tool(
+            name="ask_shared_conversation",
+            description=(
+                "Ask a question to a shared conversation. The remote AI agent will "
+                "search through the conversation history and provide an answer. "
+                "This enables agent-to-agent communication where your local agent "
+                "can query information from a remote conversation share."
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "share_url": {
+                        "type": "string",
+                        "description": (
+                            "The full URL of the shared conversation "
+                            "(e.g., https://realign-server.vercel.app/share/abc123xyz)"
+                        ),
+                    },
+                    "question": {
+                        "type": "string",
+                        "description": (
+                            "The question to ask about the conversation. "
+                            "Be specific to get better answers from the remote agent."
+                        ),
+                    },
+                    "password": {
+                        "type": "string",
+                        "description": (
+                            "Password for encrypted shares (optional). "
+                            "Leave empty for public shares."
+                        ),
+                    },
+                },
+                "required": ["share_url", "question"],
+            },
+        ),
+    ]
 
-    # Capture stdout
-    old_stdout = sys.stdout
-    sys.stdout = captured_output = StringIO()
 
-    try:
-        # Call the hide command
-        exit_code = hide_command(
-            indices=indices,
-            repo_root=repo_root,
-            force=force
+@app.call_tool()
+async def call_tool(name: str, arguments: Any) -> list[TextContent]:
+    """Handle tool execution."""
+    if name == "ask_shared_conversation":
+        return await handle_ask_tool(
+            share_url=arguments.get("share_url", ""),
+            question=arguments.get("question", ""),
+            password=arguments.get("password")
         )
-
-        # Get the output
-        output = captured_output.getvalue()
-
-        if exit_code == 0:
-            return [TextContent(type="text", text=output or "Hide operation completed.")]
-        else:
-            return [TextContent(type="text", text=f"Error: Hide failed\n{output}")]
-    finally:
-        sys.stdout = old_stdout
-
-
-def _server_log(msg: str):
-    """Log MCP server messages to both stderr and file."""
-    from datetime import datetime
-    timestamp = datetime.now().strftime("%H:%M:%S")
-    print(f"[MCP Server] {msg}", file=sys.stderr)
-
-    # Also write to the watcher log file for consistency
-    log_path = Path.home() / ".aline_watcher.log"
-    try:
-        with open(log_path, "a") as f:
-            f.write(f"[{timestamp}] [MCP Server] {msg}\n")
-    except Exception:
-        pass
-
-
-# Log immediately when module is imported
-_early_log_path = Path.home() / ".aline_mcp_startup.log"
-try:
-    with open(_early_log_path, "a") as f:
-        from datetime import datetime
-        f.write(f"\n{'='*60}\n")
-        f.write(f"[{datetime.now()}] MCP SERVER MODULE LOADED\n")
-        f.write(f"Python: {sys.executable}\n")
-        f.write(f"CWD: {Path.cwd()}\n")
-        f.write(f"HOME: {Path.home()}\n")
-        f.write(f"Version: {__version__}\n")
-        f.write(f"{'='*60}\n")
-except Exception as e:
-    print(f"[MCP Server] Early log failed: {e}", file=sys.stderr)
+    else:
+        return [TextContent(
+            type="text",
+            text=f"Unknown tool: {name}"
+        )]
 
 
 async def async_main():
-    """Run the MCP server (async)."""
-    _server_log("Starting Aline MCP server...")
-    _server_log(f"Current working directory: {Path.cwd()}")
-    _server_log(f"Home directory: {Path.home()}")
-
-    # Start the watcher (no repo_path needed - it will extract dynamically from sessions)
-    try:
-        await start_watcher()
-        print("[MCP Server] Watcher started (will auto-detect project paths from sessions)", file=sys.stderr)
-    except Exception as e:
-        print(f"[MCP Server] Warning: Could not start watcher: {e}", file=sys.stderr)
-        import traceback
-        traceback.print_exc(file=sys.stderr)
-
-    _server_log("MCP server ready")
-
+    """Main async entry point for the MCP server."""
+    # Start stdio server
     async with stdio_server() as (read_stream, write_stream):
         await app.run(
             read_stream,
@@ -585,28 +309,18 @@ async def async_main():
 
 
 def main():
-    """Entry point for the MCP server command."""
-    # Log entry point
-    try:
-        with open(_early_log_path, "a") as f:
-            from datetime import datetime
-            f.write(f"[{datetime.now()}] main() called, starting asyncio.run\n")
-    except Exception:
-        pass
+    """Entry point for the aline-mcp command."""
+    # Check if httpx is available
+    if not HTTPX_AVAILABLE:
+        print(
+            "Error: httpx package is required for aline-mcp.\n"
+            "Install with: pip install httpx",
+            file=sys.stderr
+        )
+        sys.exit(1)
 
-    try:
-        asyncio.run(async_main())
-    except Exception as e:
-        # Log any exceptions
-        try:
-            with open(_early_log_path, "a") as f:
-                from datetime import datetime
-                import traceback
-                f.write(f"[{datetime.now()}] EXCEPTION in main: {e}\n")
-                f.write(traceback.format_exc())
-        except Exception:
-            pass
-        raise
+    # Run the async main
+    asyncio.run(async_main())
 
 
 if __name__ == "__main__":