traia-iatp 0.1.29__py3-none-any.whl

traia_iatp/mcp/templates/gitignore.j2

@@ -0,0 +1,77 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# Virtual Environment
+venv/
+ENV/
+env/
+.venv/
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+*~
+
+# Environment variables
+.env
+.env.local
+.env.*.local
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Logs
+*.log
+logs/
+
+# Testing
+.pytest_cache/
+.coverage
+htmlcov/
+.tox/
+.hypothesis/
+
+# Package managers
+uv.lock
+poetry.lock
+pip-log.txt
+
+# Docker
+.dockerignore
+
+# Output files
+output_tests/
+*.txt
+*.json
+
+# Temporary files
+*.tmp
+*.temp
+*.bak
+
+# {{ api_name }} specific
+{% if additional_gitignore_entries -%}
+{{ additional_gitignore_entries }}
+{%- endif %} 

traia_iatp/mcp/templates/mcp_health_check.py.j2

@@ -0,0 +1,150 @@
+#!/usr/bin/env python
+"""
+{{ api_name }} MCP Server Health Check Script
+
+This script properly connects to the {{ api_name }} MCP server and checks its health by:
+1. Establishing a session
+2. Requesting server info
+3. Listing available tools
+"""
+
+import sys
+import json
+import requests
+import argparse
+from typing import Dict, Any
+import uuid
+
+def create_mcp_session(base_url: str) -> Dict[str, Any]:
+    """Create an MCP session and return session info"""
+    # Generate a session ID
+    session_id = str(uuid.uuid4())
+    
+    # MCP requires specific headers for streamable-http
+    headers = {
+        "Content-Type": "application/json",
+        "Accept": "application/json, text/event-stream",
+        "X-Session-ID": session_id
+    }
+    
+    return {"session_id": session_id, "headers": headers, "base_url": base_url}
+
+def send_mcp_request(session: Dict[str, Any], method: str, params: Dict = None) -> Dict[str, Any]:
+    """Send an MCP JSON-RPC request"""
+    request_data = {
+        "jsonrpc": "2.0",
+        "method": method,
+        "params": params or {},
+        "id": str(uuid.uuid4())
+    }
+    
+    try:
+        response = requests.post(
+            f"{session['base_url']}/mcp/",
+            json=request_data,
+            headers=session['headers'],
+            timeout=5
+        )
+        
+        # Handle both JSON and SSE responses
+        if response.headers.get('content-type', '').startswith('text/event-stream'):
+            # For SSE, we'd need to parse the event stream
+            return {"status": "ok", "message": "Server returned SSE stream"}
+        else:
+            return response.json()
+            
+    except requests.exceptions.RequestException as e:
+        return {"error": str(e)}
+
+def check_mcp_server_health(url: str) -> bool:
+    """Check if {{ api_name }} MCP server is healthy"""
+    print(f"🔍 Checking {{ api_name }} MCP server health at {url}")
+    
+    # Create session
+    session = create_mcp_session(url)
+    print(f"📝 Created session: {session['session_id']}")
+    
+    # Try to get server info
+    print("\n1️⃣ Testing server.info method...")
+    result = send_mcp_request(session, "server.info")
+    
+    if "error" in result and "session" not in str(result.get("error", "")):
+        print(f"❌ Server info failed: {result}")
+        return False
+    else:
+        print(f"✅ Server responded: {json.dumps(result, indent=2)[:200]}...")
+    
+    # Try to list tools
+    print("\n2️⃣ Testing tools/list method...")
+    result = send_mcp_request(session, "tools/list")
+    
+    if "error" in result and "session" not in str(result.get("error", "")):
+        print(f"❌ List tools failed: {result}")
+        return False
+    else:
+        print(f"✅ Tools list responded: {json.dumps(result, indent=2)[:200]}...")
+        
+        # Check if we have the expected tools
+        if "result" in result and "tools" in result["result"]:
+            tools = result["result"]["tools"]
+            tool_names = [tool.get("name", "") for tool in tools]
+            print(f"📋 Available tools: {', '.join(tool_names)}")
+            
+            # Check for expected tools
+            expected_tools = ["example_tool", "get_api_info"]
+            missing_tools = [tool for tool in expected_tools if tool not in tool_names]
+            
+            if missing_tools:
+                print(f"⚠️  Missing expected tools: {', '.join(missing_tools)}")
+            else:
+                print("✅ All expected {{ api_name }} tools are available!")
+    
+    # Alternative: Try connecting with CrewAI adapter
+    print("\n3️⃣ Testing CrewAI adapter connection...")
+    try:
+        from crewai_tools import MCPServerAdapter
+        
+        server_params = {
+            "url": f"{url}/mcp/",
+            "transport": "streamable-http"
+        }
+        
+        with MCPServerAdapter(server_params) as mcp_tools:
+            tools = list(mcp_tools)
+            print(f"✅ CrewAI connected successfully! Found {len(tools)} tools")
+            
+            # Print first few tools
+            for i, tool in enumerate(tools[:3]):
+                print(f"   - {tool.name}")
+                
+            if len(tools) > 3:
+                print(f"   ... and {len(tools) - 3} more")
+                
+            return True
+            
+    except Exception as e:
+        print(f"⚠️  CrewAI adapter test failed: {e}")
+        # This might fail but server could still be healthy
+        
+    return True
+
+def main():
+    parser = argparse.ArgumentParser(description="Check {{ api_name }} MCP Server Health")
+    parser.add_argument("--url", default="http://localhost:8000", 
+                       help="{{ api_name }} MCP server URL (default: http://localhost:8000)")
+    args = parser.parse_args()
+    
+    print(f"🚀 {{ api_name }} MCP Server Health Check")
+    print(f"📍 Server URL: {args.url}")
+    print(f"📰 Expected tools: example_tool, get_api_info")
+    print("="*50)
+    
+    if check_mcp_server_health(args.url):
+        print("\n✅ {{ api_name }} MCP Server is healthy and responding!")
+        return 0
+    else:
+        print("\n❌ {{ api_name }} MCP Server health check failed!")
+        return 1
+
+if __name__ == "__main__":
+    sys.exit(main()) 

traia_iatp/mcp/templates/pyproject.toml.j2

@@ -0,0 +1,32 @@
+[project]
+name = "{{ api_slug }}-mcp-server"
+version = "0.1.0"
+description = "MCP server for {{ api_name }} API with {% if auth_description %} authentication and {% endif %} HTTP 402 payment protocol support"
+requires-python = ">=3.12"
+dependencies = [
+    "anyio>=4.0.0",
+    "mcp>=1.1.2",
+    "python-dotenv>=1.1.1",
+    "requests>=2.32.5",
+    "starlette>=0.45.0",
+    "retry>=0.9.2",
+    "traia-iatp>=0.1.27",  # For d402 payment protocol support
+    "uvicorn>=0.37.0",
+    "web3>=6.15.0",  # For blockchain payment verification
+{% if sdk_package %}
+    "{{ sdk_package }}",
+{% endif %}
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.metadata]
+allow-direct-references = true
+
+[tool.hatch.build.targets.wheel]
+include = [
+    "server.py",
+    "mcp_health_check.py",
+] 

traia_iatp/mcp/templates/pyrightconfig.json.j2

@@ -0,0 +1,22 @@
+{
+  "include": [
+    "server.py",
+    "mcp_health_check.py"
+  ],
+  "exclude": [
+    "**/__pycache__",
+    ".venv",
+    "**/*.pyc"
+  ],
+  "extraPaths": [
+    "../../../../IATP/src",
+    "/Users/eitanlavi/workspace/traia/IATP/src"
+  ],
+  "pythonVersion": "3.12",
+  "typeCheckingMode": "basic",
+  "reportMissingImports": true,
+  "reportMissingTypeStubs": false,
+  "venvPath": ".",
+  "venv": ".venv"
+}
+

traia_iatp/mcp/templates/run_local_docker.sh.j2

@@ -0,0 +1,390 @@
+#!/bin/bash
+
+# Script to build and run the {{ api_name }} MCP Server locally in Docker
+
+set -e  # Exit on error
+
+# Color codes for output
+GREEN='\033[0;32m'
+BLUE='\033[0;34m'
+YELLOW='\033[1;33m'
+RED='\033[0;31m'
+NC='\033[0m' # No Color
+
+# Configuration (will be overridden by .env if present)
+IMAGE_NAME="{{ api_slug }}-mcp-server"
+CONTAINER_NAME="{{ api_slug }}-mcp-local"
+HOST_PORT=${PORT:-8000}
+CONTAINER_PORT=${PORT:-8000}
+
+echo -e "${BLUE}🚀 Building and running {{ api_name }} MCP Server...${NC}"
+echo
+
+# Setup .env file for local development
+# Step 1: Copy .env.example to .env (if .env doesn't exist)
+if [ ! -f .env ]; then
+    echo -e "${BLUE}📋 Setting up .env file for local development...${NC}"
+    
+    if [ -f .env.example ]; then
+        echo -e "${BLUE}   Copying .env.example to .env...${NC}"
+        cp .env.example .env
+    else
+        echo -e "${YELLOW}⚠️  .env.example not found, creating minimal .env template...${NC}"
+        # Create minimal .env template
+        cat > .env << EOF
+# {{ api_name }} MCP Server Configuration
+PORT=8000
+STAGE=MAINNET
+LOG_LEVEL=INFO
+
+{% if requires_auth %}
+# API Authentication (server's internal key for payment mode)
+{{ api_key_env_var }}=
+{% endif %}
+# D402 Payment Protocol Configuration
+SERVER_ADDRESS=
+MCP_OPERATOR_PRIVATE_KEY=
+MCP_OPERATOR_ADDRESS=
+D402_FACILITATOR_URL=https://facilitator.d402.net
+D402_FACILITATOR_API_KEY=
+DEFAULT_SETTLEMENT_TOKEN=0x036CbD53842c5426634e7929541eC2318f3dCF7e
+DEFAULT_SETTLEMENT_NETWORK=sepolia
+D402_TESTING_MODE=true
+EOF
+    fi
+fi
+
+# Step 2: Fix pyproject.toml to use local IATP path for account generation
+# This is needed because uv run python requires a valid pyproject.toml
+echo -e "${BLUE}🔧 Fixing pyproject.toml for local development...${NC}"
+
+# Find local IATP path
+IATP_PATH=""
+if [ -d "/Users/eitanlavi/workspace/traia/IATP" ]; then
+    IATP_PATH="/Users/eitanlavi/workspace/traia/IATP"
+elif [ -d "$(dirname $(pwd))/IATP" ]; then
+    IATP_PATH="$(dirname $(pwd))/IATP"
+elif [ -d "$HOME/workspace/traia/IATP" ]; then
+    IATP_PATH="$HOME/workspace/traia/IATP"
+elif [ -n "$LOCAL_IATP_PATH" ] && [ -d "$LOCAL_IATP_PATH" ]; then
+    IATP_PATH="$LOCAL_IATP_PATH"
+fi
+
+# Temporarily update pyproject.toml to use local IATP path (if /tmp/IATP is present)
+if grep -q "file:///tmp/IATP" pyproject.toml 2>/dev/null; then
+    if [ -n "$IATP_PATH" ] && [ -d "$IATP_PATH" ]; then
+        echo -e "${BLUE}   Updating pyproject.toml to use: $IATP_PATH${NC}"
+        if [[ "$OSTYPE" == "darwin"* ]]; then
+            sed -i '' "s|file:///tmp/IATP|file://$IATP_PATH|g" pyproject.toml
+        else
+            sed -i "s|file:///tmp/IATP|file://$IATP_PATH|g" pyproject.toml
+        fi
+    else
+        echo -e "${RED}❌ Local IATP not found. Cannot generate accounts.${NC}"
+        echo -e "${YELLOW}   Please set LOCAL_IATP_PATH environment variable or ensure IATP is at /Users/eitanlavi/workspace/traia/IATP${NC}"
+        exit 1
+    fi
+fi
+
+# Step 3: Generate Ethereum accounts for local development (workaround)
+# Always regenerate keys for local dev to ensure they're set
+echo -e "${BLUE}🔐 Generating Ethereum accounts for local testing (workaround for local dev)...${NC}"
+
+# Generate accounts using Python (web3 library)
+# Use uv run to ensure web3 is available from the project dependencies
+uv run python << 'PYTHON_SCRIPT'
+import sys
+import secrets
+
+try:
+    from web3 import Web3
+    
+    # Generate server address (EOA for local testing)
+    # This is a workaround - in production, SERVER_ADDRESS comes from IATP wallet contract
+    server_private_key = "0x" + secrets.token_hex(32)
+    server_account = Web3().eth.account.from_key(server_private_key)
+    server_address = server_account.address
+    
+    # Generate operator private key and address
+    # This is a workaround - in production, these come from IATP wallet contract
+    operator_private_key = "0x" + secrets.token_hex(32)
+    operator_account = Web3().eth.account.from_key(operator_private_key)
+    operator_address = operator_account.address
+    
+    # Write to temporary file
+    with open('.env.generated', 'w') as f:
+        f.write(f"SERVER_ADDRESS={server_address}\n")
+        f.write(f"MCP_OPERATOR_PRIVATE_KEY={operator_private_key}\n")
+        f.write(f"MCP_OPERATOR_ADDRESS={operator_address}\n")
+    
+    print(f"Generated SERVER_ADDRESS: {server_address}")
+    print(f"Generated MCP_OPERATOR_ADDRESS: {operator_address}")
+    sys.exit(0)
+except ImportError:
+    print("ERROR: web3 library not found. Install it with: pip install web3", file=sys.stderr)
+    sys.exit(1)
+except Exception as e:
+    print(f"ERROR: Failed to generate accounts: {e}", file=sys.stderr)
+    sys.exit(1)
+PYTHON_SCRIPT
+
+# Step 4: Fill in generated values into .env
+if [ $? -eq 0 ] && [ -f .env.generated ]; then
+    # Load generated values
+    source .env.generated
+    
+    # Update .env file with generated values (macOS-compatible sed)
+    # Use a temporary file approach for portability
+    if grep -q "^SERVER_ADDRESS=" .env; then
+        if [[ "$OSTYPE" == "darwin"* ]]; then
+            sed -i '' "s|^SERVER_ADDRESS=.*|SERVER_ADDRESS=$SERVER_ADDRESS|" .env
+        else
+            sed -i "s|^SERVER_ADDRESS=.*|SERVER_ADDRESS=$SERVER_ADDRESS|" .env
+        fi
+    else
+        echo "SERVER_ADDRESS=$SERVER_ADDRESS" >> .env
+    fi
+    
+    if grep -q "^MCP_OPERATOR_PRIVATE_KEY=" .env; then
+        if [[ "$OSTYPE" == "darwin"* ]]; then
+            sed -i '' "s|^MCP_OPERATOR_PRIVATE_KEY=.*|MCP_OPERATOR_PRIVATE_KEY=$MCP_OPERATOR_PRIVATE_KEY|" .env
+        else
+            sed -i "s|^MCP_OPERATOR_PRIVATE_KEY=.*|MCP_OPERATOR_PRIVATE_KEY=$MCP_OPERATOR_PRIVATE_KEY|" .env
+        fi
+    else
+        echo "MCP_OPERATOR_PRIVATE_KEY=$MCP_OPERATOR_PRIVATE_KEY" >> .env
+    fi
+    
+    if grep -q "^MCP_OPERATOR_ADDRESS=" .env; then
+        if [[ "$OSTYPE" == "darwin"* ]]; then
+            sed -i '' "s|^MCP_OPERATOR_ADDRESS=.*|MCP_OPERATOR_ADDRESS=$MCP_OPERATOR_ADDRESS|" .env
+        else
+            sed -i "s|^MCP_OPERATOR_ADDRESS=.*|MCP_OPERATOR_ADDRESS=$MCP_OPERATOR_ADDRESS|" .env
+        fi
+    else
+        echo "MCP_OPERATOR_ADDRESS=$MCP_OPERATOR_ADDRESS" >> .env
+    fi
+    
+    # Set D402_TESTING_MODE if not set
+    if ! grep -q "^D402_TESTING_MODE=" .env; then
+        echo "D402_TESTING_MODE=true" >> .env
+    else
+        if [[ "$OSTYPE" == "darwin"* ]]; then
+            sed -i '' "s|^D402_TESTING_MODE=.*|D402_TESTING_MODE=true|" .env
+        else
+            sed -i "s|^D402_TESTING_MODE=.*|D402_TESTING_MODE=true|" .env
+        fi
+    fi
+    
+    rm -f .env.generated
+    echo -e "${GREEN}✅ Generated Ethereum accounts and updated .env${NC}"
+else
+    echo -e "${RED}❌ Failed to generate accounts. Please install web3: pip install web3${NC}"
+    echo -e "${YELLOW}   Or set SERVER_ADDRESS, MCP_OPERATOR_PRIVATE_KEY, and MCP_OPERATOR_ADDRESS manually in .env${NC}"
+    exit 1
+fi
+
+# Step 5: Prompt for API key if required (only if not already set)
+{% if requires_auth %}
+if ! grep -q "^{{ api_key_env_var }}=[^[:space:]]" .env; then
+    if [ -t 0 ]; then
+        echo ""
+        echo -e "${YELLOW}📝 Please enter your {{ api_name }} API key:${NC}"
+        echo -e "${YELLOW}   (This is the server's internal API key used when clients pay via 402)${NC}"
+        read -p "{{ api_key_env_var }}: " API_KEY_VALUE
+        if [ -n "$API_KEY_VALUE" ]; then
+            if grep -q "^{{ api_key_env_var }}=" .env; then
+                if [[ "$OSTYPE" == "darwin"* ]]; then
+                    sed -i '' "s|^{{ api_key_env_var }}=.*|{{ api_key_env_var }}=$API_KEY_VALUE|" .env
+                else
+                    sed -i "s|^{{ api_key_env_var }}=.*|{{ api_key_env_var }}=$API_KEY_VALUE|" .env
+                fi
+            else
+                echo "{{ api_key_env_var }}=$API_KEY_VALUE" >> .env
+            fi
+            echo -e "${GREEN}✅ API key saved to .env${NC}"
+        else
+            echo -e "${YELLOW}⚠️  No API key provided. You can set it later in .env${NC}"
+        fi
+    else
+        echo -e "${YELLOW}⚠️  Non-interactive mode: Please set {{ api_key_env_var }} in .env manually${NC}"
+    fi
+else
+    echo -e "${GREEN}✅ {{ api_key_env_var }} already set in .env${NC}"
+fi
+{% endif %}
+
+echo ""
+echo -e "${GREEN}✅ .env file configured for local development!${NC}"
+echo -e "${BLUE}📋 Review .env file if needed before continuing...${NC}"
+if [ -t 0 ]; then
+    echo -e "${YELLOW}Press Enter to continue, or Ctrl+C to cancel...${NC}"
+    read
+fi
+
+# Load environment variables from .env file
+echo -e "${BLUE}📋 Loading environment variables from .env file...${NC}"
+set -a  # Export all variables
+source .env
+set +a  # Stop exporting
+
+# Update port variables after loading .env
+HOST_PORT=${PORT:-8000}
+CONTAINER_PORT=${PORT:-8000}
+
+# Check if Docker is installed
+if ! command -v docker &> /dev/null; then
+    echo -e "${RED}❌ Docker is not installed. Please install Docker first.${NC}"
+    exit 1
+fi
+
+# Stop and remove existing container if it exists
+if docker ps -a | grep -q $CONTAINER_NAME; then
+    echo -e "${YELLOW}🛑 Stopping existing container...${NC}"
+    docker stop $CONTAINER_NAME >/dev/null 2>&1 || true
+    docker rm $CONTAINER_NAME >/dev/null 2>&1 || true
+fi
+
+# Build the Docker image
+echo -e "${BLUE}🔨 Building Docker image...${NC}"
+
+# Check if pyproject.toml uses local IATP path (for local development)
+if grep -q "file://" pyproject.toml 2>/dev/null; then
+    # Extract IATP path from pyproject.toml (macOS-compatible, no Perl regex)
+    IATP_PATH_FROM_TOML=$(grep "file://" pyproject.toml | sed -E 's|.*file://([^"]+).*|\1|' | head -1)
+    
+    # If path is /tmp/IATP (Docker path), find the actual local IATP path
+    if [ "$IATP_PATH_FROM_TOML" = "/tmp/IATP" ] || [ ! -d "$IATP_PATH_FROM_TOML" ]; then
+        # Try to find local IATP path (common locations)
+        if [ -d "/Users/eitanlavi/workspace/traia/IATP" ]; then
+            IATP_PATH="/Users/eitanlavi/workspace/traia/IATP"
+        elif [ -d "$(dirname $(pwd))/IATP" ]; then
+            IATP_PATH="$(dirname $(pwd))/IATP"
+        elif [ -d "$HOME/workspace/traia/IATP" ]; then
+            IATP_PATH="$HOME/workspace/traia/IATP"
+        elif [ -n "$LOCAL_IATP_PATH" ] && [ -d "$LOCAL_IATP_PATH" ]; then
+            IATP_PATH="$LOCAL_IATP_PATH"
+        else
+            IATP_PATH=""
+        fi
+    else
+        IATP_PATH="$IATP_PATH_FROM_TOML"
+    fi
+    
+    if [ -n "$IATP_PATH" ] && [ -d "$IATP_PATH" ]; then
+        echo -e "${BLUE}📦 Using local IATP package: $IATP_PATH${NC}"
+        # Copy IATP into build context for Docker (exclude .venv to save space!)
+        mkdir -p .docker-iatp
+        
+        # Use rsync if available (much better), otherwise cp with find
+        if command -v rsync &> /dev/null; then
+            rsync -a --exclude='.venv' --exclude='__pycache__' --exclude='.git' --exclude='build' --exclude='dist' --exclude='*.egg-info' "$IATP_PATH/" .docker-iatp/IATP/
+            echo -e "${BLUE}   Copied IATP to build context (excluded .venv, ${GREEN}saved ~900MB${BLUE})${NC}"
+        else
+            # Fallback: use cp but warn about size
+            cp -r "$IATP_PATH" .docker-iatp/IATP
+            # Try to remove .venv after copy
+            rm -rf .docker-iatp/IATP/.venv .docker-iatp/IATP/__pycache__ .docker-iatp/IATP/.git 2>/dev/null
+            echo -e "${BLUE}   Copied IATP to build context (cleaned .venv after copy)${NC}"
+        fi
+        # Temporarily update pyproject.toml to use Docker path
+        if [[ "$OSTYPE" == "darwin"* ]]; then
+            sed -i.bak 's|file://.*IATP|file:///tmp/IATP|g' pyproject.toml
+        else
+            sed -i.bak 's|file://.*IATP|file:///tmp/IATP|g' pyproject.toml
+        fi
+        docker build --no-cache -t $IMAGE_NAME .
+        BUILD_EXIT_CODE=$?
+        # Restore original pyproject.toml
+        mv pyproject.toml.bak pyproject.toml
+        rm -rf .docker-iatp
+        if [ $BUILD_EXIT_CODE -ne 0 ]; then
+            exit $BUILD_EXIT_CODE
+        fi
+    else
+        echo -e "${YELLOW}⚠️  Local IATP path not found${NC}"
+        echo -e "${YELLOW}   Falling back to published version (may fail if not published)${NC}"
+        # If pyproject.toml has /tmp/IATP, replace with published version
+        if grep -q "file:///tmp/IATP" pyproject.toml 2>/dev/null; then
+            if [[ "$OSTYPE" == "darwin"* ]]; then
+                sed -i.bak 's|file:///tmp/IATP|traia-iatp>=0.1.27|g' pyproject.toml
+            else
+                sed -i.bak 's|file:///tmp/IATP|traia-iatp>=0.1.27|g' pyproject.toml
+            fi
+            docker build --no-cache -t $IMAGE_NAME .
+            BUILD_EXIT_CODE=$?
+            mv pyproject.toml.bak pyproject.toml
+            if [ $BUILD_EXIT_CODE -ne 0 ]; then
+                exit $BUILD_EXIT_CODE
+            fi
+        else
+            docker build --no-cache -t $IMAGE_NAME .
+        fi
+    fi
+else
+    # Using published version, normal build
+    docker build --no-cache -t $IMAGE_NAME .
+fi
+
+# Run the container
+echo -e "${BLUE}🏃 Starting container...${NC}"
+# Use --env-file to load all variables from .env file
+docker run -d \
+    --name $CONTAINER_NAME \
+    -p $HOST_PORT:$CONTAINER_PORT \
+    --env-file .env \
+    $IMAGE_NAME
+
+# Wait for the server to start
+echo -e "${YELLOW}⏳ Waiting for server to start...${NC}"
+sleep 3
+
+# Check if container is running
+if ! docker ps | grep -q $CONTAINER_NAME; then
+    echo -e "${RED}❌ Container failed to start. Checking logs:${NC}"
+    docker logs $CONTAINER_NAME
+    exit 1
+fi
+
+# Get container info
+CONTAINER_ID=$(docker ps -q -f name=$CONTAINER_NAME)
+CONTAINER_IP=$(docker inspect -f '{% raw %}{{range.NetworkSettings.Networks}}{{.IPAddress}}{{end}}{% endraw %}' $CONTAINER_ID)
+
+# Output connection information
+echo
+echo -e "${GREEN}✅ {{ api_name }} MCP Server is running!${NC}"
+echo
+echo -e "${BLUE}📍 Connection Information:${NC}"
+echo -e "   Local URL:        ${GREEN}http://localhost:${HOST_PORT}/mcp${NC}"
+echo -e "   Container IP:     ${GREEN}${CONTAINER_IP}:${CONTAINER_PORT}${NC}"
+echo -e "   Container Name:   ${GREEN}${CONTAINER_NAME}${NC}"
+echo -e "   Container ID:     ${GREEN}${CONTAINER_ID:0:12}${NC}"
+echo
+echo -e "${BLUE}📝 Useful commands:${NC}"
+echo -e "   View logs:        ${YELLOW}docker logs -f ${CONTAINER_NAME}${NC}"
+echo -e "   Stop server:      ${YELLOW}docker stop ${CONTAINER_NAME}${NC}"
+echo -e "   Remove container: ${YELLOW}docker rm ${CONTAINER_NAME}${NC}"
+echo -e "   Shell access:     ${YELLOW}docker exec -it ${CONTAINER_NAME} /bin/bash${NC}"
+echo
+echo -e "${BLUE}🔌 MCP Server Endpoint:${NC}"
+echo -e "   ${GREEN}http://localhost:${HOST_PORT}/mcp${NC}"
+echo
+
+# Check if the server is responding
+echo -e "${YELLOW}🔍 Checking server health...${NC}"
+if curl -s -o /dev/null -w "%{http_code}" "http://localhost:${HOST_PORT}/mcp" | grep -q "200\|404\|405"; then
+    echo -e "${GREEN}✅ Server is responding!${NC}"
+else
+    echo -e "${YELLOW}⚠️  Server may still be starting up. Check logs with: docker logs -f ${CONTAINER_NAME}${NC}"
+fi
+
+# Cleanup: Restore pyproject.toml to use /tmp/IATP (git-committed version)
+echo -e "${BLUE}🧹 Cleaning up...${NC}"
+if grep -q "file:///Users/eitanlavi/workspace/traia/IATP" pyproject.toml 2>/dev/null; then
+    if [[ "$OSTYPE" == "darwin"* ]]; then
+        sed -i '' "s|file:///Users/eitanlavi/workspace/traia/IATP|file:///tmp/IATP|g" pyproject.toml
+    else
+        sed -i "s|file:///Users/eitanlavi/workspace/traia/IATP|file:///tmp/IATP|g" pyproject.toml
+    fi
+    echo -e "${GREEN}✅ Restored pyproject.toml to git state${NC}"
+fi