mcp-use 1.3.9__py3-none-any.whl → 1.3.10__py3-none-any.whl

mcp_use/cli.py

@@ -0,0 +1,581 @@
+#!/usr/bin/env python
+"""
+MCP-Use CLI Tool - All-in-one CLI for creating and deploying MCP projects.
+"""
+
+import argparse
+import sys
+import threading
+import time
+from pathlib import Path
+
+from mcp_use import __version__
+
+# ============= SPINNER CLASS =============
+
+
+class Spinner:
+    """Simple loading spinner similar to UV's style."""
+
+    def __init__(self, message: str = "Loading"):
+        self.message = message
+        self.running = False
+        self.thread = None
+        self.frames = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
+        self.current = 0
+
+    def start(self):
+        """Start the spinner."""
+        self.running = True
+        self.thread = threading.Thread(target=self._spin)
+        self.thread.daemon = True
+        self.thread.start()
+
+    def _spin(self):
+        """Spin animation loop."""
+        while self.running:
+            frame = self.frames[self.current % len(self.frames)]
+            print(f"\r{frame} {self.message}...", end="", flush=True)
+            self.current += 1
+            time.sleep(0.1)
+
+    def stop(self, success_message=None):
+        """Stop the spinner and optionally print success message."""
+        self.running = False
+        if self.thread:
+            self.thread.join()
+        if success_message:
+            print(f"\r✓ {success_message}        ")
+        else:
+            print("\r" + " " * (len(self.message) + 10), end="\r")
+
+
+# ============= PROJECT CREATION FUNCTIONS =============
+
+
+def print_header():
+    """Print the CLI header."""
+    print("\n mcp-use create")
+    print("━" * 50)
+    print()
+
+
+def get_project_name() -> str:
+    """Get project name from user."""
+    while True:
+        name = input("📝 Project name: ").strip().replace("-", "_")
+        if not name:
+            print("   ⚠️  Project name cannot be empty")
+            continue
+        if " " in name:
+            print("   ⚠️  Project name cannot contain spaces")
+            continue
+        if Path(name).exists():
+            print(f"   ⚠️  Directory '{name}' already exists")
+            continue
+        return name
+
+
+def get_project_type() -> str:
+    """Get project type from user."""
+    print("\n📦 What would you like to create?")
+    print("   1) Server + Agent")
+    print("   2) Server only")
+    print("   3) Agent only")
+
+    while True:
+        choice = input("\n   Choice (1-3): ").strip()
+        if choice == "1":
+            return "server_agent"
+        elif choice == "2":
+            return "server"
+        elif choice == "3":
+            return "agent"
+        else:
+            print("   ⚠️  Please enter 1, 2, or 3")
+
+
+def create_server_structure(project_dir: Path, project_name: str):
+    """Create server file in nested project folder."""
+    # Create nested project folder
+    nested_dir = project_dir / project_name
+    nested_dir.mkdir(parents=True)
+
+    # Create server.py
+    server_content = f'''"""
+MCP Server for {project_name}
+"""
+
+from mcp.server import FastMCP
+
+# Create server instance
+server = FastMCP("{project_name}-server")
+
+
+# ============= TOOLS =============
+
+
+@server.tool()
+async def add_numbers(a: float, b: float) -> str:
+    """Add two numbers together."""
+    result = a + b
+    return f"{{result}}"
+
+
+# ============= RESOURCES =============
+
+
+@server.resource("config://app")
+async def get_app_config() -> str:
+    """Get the application configuration."""
+    return "App: {project_name}, Version: 0.1.0, Status: Active"
+
+
+# ============= PROMPTS =============
+
+
+@server.prompt()
+async def assistant_prompt() -> str:
+    """Generate a helpful assistant prompt."""
+    return "You are a helpful assistant for {project_name}. Be concise and friendly."
+
+
+# ============= MAIN =============
+
+if __name__ == "__main__":
+    server.run("stdio")
+'''
+    (nested_dir / "server.py").write_text(server_content)
+
+
+def create_agent_structure(project_dir: Path, project_name: str, project_type: str):
+    """Create agent file in nested project folder."""
+    # Create nested project folder if it doesn't exist
+    nested_dir = project_dir / project_name
+    if not nested_dir.exists():
+        nested_dir.mkdir(parents=True)
+
+    if project_type == "server_agent":
+        # For server_agent mode, embed config directly in agent.py
+        agent_content = f'''"""
+MCP Agent implementation for {project_name}
+"""
+
+from langchain_openai import ChatOpenAI
+
+from mcp_use import MCPAgent, MCPClient
+
+config = {{
+    "mcpServers": {{
+        "{project_name}": {{
+            "command": "python",
+            "args": ["{project_name}/server.py"],
+        }}
+    }}
+}}
+
+client = MCPClient(config=config)
+agent = MCPAgent(llm=ChatOpenAI(model="gpt-4o"), client=client, max_steps=10, memory_enabled=True)
+'''
+    else:
+        # For agent-only mode, use external JSON config file
+        agent_content = f'''"""
+MCP Agent implementation for {project_name}
+"""
+
+from langchain_openai import ChatOpenAI
+
+from mcp_use import MCPAgent, MCPClient
+
+client = MCPClient.from_config_file("{project_name}/mcp_servers.json")
+agent = MCPAgent(llm=ChatOpenAI(model="gpt-4o"), client=client, max_steps=10, memory_enabled=True)
+'''
+        # Create mcp_servers.json for agent-only mode
+        mcp_servers_json = (
+            '''{
+    "mcpServers": {
+        "'''
+            + project_name
+            + """": {
+            "command": "npx",
+            "args": ["-y", "@modelcontextprotocol/server-everything"]
+        }
+    }
+}"""
+        )
+        (nested_dir / "mcp_servers.json").write_text(mcp_servers_json)
+
+    (nested_dir / "agent.py").write_text(agent_content)
+
+
+def create_common_files(project_dir: Path, project_name: str, project_type: str):
+    """Create common project files."""
+
+    # pyproject.toml
+    pyproject = f"""[project]
+name = "{project_name}"
+version = "0.1.0"
+description = "An MCP project"
+requires-python = ">=3.10"
+dependencies = [
+    "mcp>=1.0.0",
+    "mcp-use>=0.1.0",
+    "langchain-openai>=0.1.0",
+    "python-dotenv>=1.0.0",
+]
+"""
+    (project_dir / "pyproject.toml").write_text(pyproject)
+
+    # requirements.txt
+    requirements = """mcp>=1.0.0
+mcp-use>=0.1.0
+langchain-openai>=0.1.0
+python-dotenv>=1.0.0
+"""
+    (project_dir / "requirements.txt").write_text(requirements)
+
+    # .gitignore
+    gitignore = """__pycache__/
+*.py[cod]
+.env
+venv/
+.venv/
+*.egg-info/
+dist/
+build/
+.pytest_cache/
+.coverage
+"""
+    (project_dir / ".gitignore").write_text(gitignore)
+
+    # README.md
+    readme = f"""# {project_name}
+
+An MCP project created with mcp-use.
+
+## Project Structure
+
+```
+{project_name}/
+"""
+
+    if project_type in ["server_agent", "server"]:
+        readme += f"""├── {project_name}/
+│   ├── server.py    # MCP server with all components
+"""
+
+    if project_type in ["server_agent", "agent"]:
+        if project_type == "agent":
+            readme += f"""├── {project_name}/
+│   ├── agent.py     # MCP agent implementation
+│   └── mcp_servers.json  # Server configuration
+"""
+        else:
+            readme += """│   └── agent.py     # MCP agent implementation
+"""
+
+    if project_type in ["server_agent", "agent"]:
+        readme += """├── run.py           # Simple example
+├── chat.py          # Interactive chat
+"""
+
+    readme += """├── pyproject.toml
+├── requirements.txt
+├── .gitignore
+└── README.md
+```
+
+## Setup
+
+1. Install dependencies:
+```bash
+   pip install -r requirements.txt
+```
+
+2. Configure environment:
+```bash
+   export OPENAI_API_KEY=your-api-key-here
+```
+"""
+
+    if project_type in ["server_agent", "server"]:
+        readme += f"""
+## Running the Server
+
+```bash
+python {project_name}/server.py
+```
+
+The server uses FastMCP and includes:
+- **Tools**: Simple tool functions (e.g., add_numbers)
+- **Resources**: Data resources (e.g., config)
+- **Prompts**: Prompt templates for the LLM
+"""
+
+    if project_type in ["server_agent", "agent"]:
+        readme += f"""
+## Using the Agent
+
+```python
+from {project_name}.agent import agent
+
+result = await agent.run("Your prompt")
+```
+"""
+
+    (project_dir / "README.md").write_text(readme)
+
+
+def create_example_files(project_dir: Path, project_name: str):
+    """Create example files."""
+
+    # run.py
+    run_content = f'''"""
+Example usage of {project_name}.
+"""
+
+import asyncio
+import os
+
+from dotenv import load_dotenv
+
+from {project_name}.agent import agent
+
+
+async def main():
+    load_dotenv()
+
+    if not os.getenv("OPENAI_API_KEY"):
+        print("Error: OPENAI_API_KEY not found")
+        return
+
+    result = await agent.run("What tools are available?")
+    print(f"Result: {{result}}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())
+'''
+    (project_dir / "run.py").write_text(run_content)
+
+    # chat.py
+    chat_content = f'''"""
+Interactive chat for {project_name}.
+"""
+
+import asyncio
+import os
+
+from dotenv import load_dotenv
+
+from {project_name}.agent import agent
+
+
+async def chat():
+    load_dotenv()
+
+    if not os.getenv("OPENAI_API_KEY"):
+        print("Error: OPENAI_API_KEY not found")
+        return
+
+    print("Chat started (type 'exit' to quit)")
+
+    while True:
+        user_input = input("\\nYou: ")
+        if user_input.lower() == "exit":
+            break
+
+        print("Assistant: ", end="", flush=True)
+        response = await agent.run(user_input)
+        print(response)
+
+
+if __name__ == "__main__":
+    asyncio.run(chat())
+'''
+    (project_dir / "chat.py").write_text(chat_content)
+
+
+def create_project(project_name: str, project_type: str) -> bool:
+    """Create the project structure."""
+    project_dir = Path.cwd() / project_name
+
+    try:
+        # Create main directory
+        spinner = Spinner("Creating project directory")
+        spinner.start()
+        time.sleep(0.5)  # Simulate work
+        project_dir.mkdir(parents=True)
+        spinner.stop("Created project directory")
+
+        # Create server if needed
+        if project_type in ["server_agent", "server"]:
+            spinner = Spinner("Creating server")
+            spinner.start()
+            time.sleep(0.3)
+            create_server_structure(project_dir, project_name)
+            spinner.stop("Created server structure")
+
+        # Create agent if needed
+        if project_type in ["server_agent", "agent"]:
+            spinner = Spinner("Creating agent")
+            spinner.start()
+            time.sleep(0.3)
+            create_agent_structure(project_dir, project_name, project_type)
+            spinner.stop("Created agent structure")
+
+        # Create common files
+        spinner = Spinner("Creating configuration files")
+        spinner.start()
+        time.sleep(0.3)
+        create_common_files(project_dir, project_name, project_type)
+        spinner.stop("Created configuration files")
+
+        # Create examples for server_agent and agent
+        if project_type in ["server_agent", "agent"]:
+            spinner = Spinner("Creating example files")
+            spinner.start()
+            time.sleep(0.3)
+            create_example_files(project_dir, project_name)
+            spinner.stop("Created example files")
+
+        return True
+
+    except Exception as e:
+        print(f"\n❌ Error: {str(e)}")
+        return False
+
+
+# ============= MAIN CLI FUNCTIONS =============
+
+
+def show_help():
+    """Show the main help message."""
+    help_text = """
+╔══════════════════════════════════════════════════════════════════╗
+║                       MCP-Use CLI Tool                           ║
+║                                                                  ║
+║  Create and deploy MCP servers and agents with ease              ║
+╚══════════════════════════════════════════════════════════════════╝
+
+Usage: uvx mcp-use <command> [options]
+
+Available Commands:
+  create     🚀 Create a new MCP project (server, agent, or both)
+             Interactive wizard to scaffold your MCP project
+
+  deploy     ☁️  Deploy your MCP project to the cloud
+             (Coming soon - Cloud deployment from CLI)
+
+Examples:
+  uvx mcp-use create           # Start interactive project creation
+  uvx mcp-use deploy           # Deploy to cloud (coming soon)
+  uvx mcp-use --help           # Show this help message
+  uvx mcp-use --version        # Show version information
+
+For more information, visit: https://mcp-use.com
+    """
+    print(help_text)
+
+
+def handle_create():
+    """Handle the create command."""
+    print_header()
+
+    # Get project configuration
+    project_name = get_project_name()
+    project_type = get_project_type()
+
+    print(f"\n⚙️  Creating {project_type.replace('_', ' + ')} project: {project_name}")
+    print()
+
+    # Create the project
+    if create_project(project_name, project_type):
+        print(f"\n✨ Successfully created '{project_name}'!")
+        print("\n📋 Next steps:")
+        print(f"   cd {project_name}")
+        print("   pip install -r requirements.txt")
+        print("   export OPENAI_API_KEY=your-api-key-here")
+
+        if project_type in ["server_agent", "server"]:
+            print("\n   # Test the server:")
+            print(f"   python {project_name}/server.py")
+
+        if project_type in ["server_agent", "agent"]:
+            print("\n   # Run examples:")
+            print("   python run.py    # Simple example")
+            print("   python chat.py   # Interactive chat")
+
+        print()
+    else:
+        sys.exit(1)
+
+
+def handle_deploy():
+    """Handle the deploy command (placeholder for future implementation)."""
+    print("\n" + "=" * 60)
+    print("🚀 MCP Cloud Deployment")
+    print("=" * 60)
+
+    print("\n📝 Please login to MCP Cloud to continue...")
+    print("   Visit: https://cloud.mcp-use.com/login")
+    print()
+
+    # Simulate login prompt
+    print("Enter your MCP Cloud credentials:")
+    email = input("Email: ")
+
+    if email:
+        print(f"\n✨ Welcome {email}!")
+        print()
+        print("ℹ️  Deployment from CLI is coming soon!")
+        print("   For now, please use the web interface at:")
+        print("   https://cloud.mcp-use.com/deploy")
+        print()
+        print("Stay tuned for updates! 🎉")
+    else:
+        print("\n❌ Login cancelled")
+
+    print("=" * 60)
+
+
+def main(args=None):
+    """Main entry point for the CLI."""
+    parser = argparse.ArgumentParser(
+        prog="mcp-use",
+        description="MCP-Use CLI Tool - Create and deploy MCP projects",
+        add_help=False,  # We'll handle help ourselves
+    )
+
+    # Add version argument
+    parser.add_argument(
+        "--version", "-v", action="version", version=f"mcp-use {__version__}", help="Show version information"
+    )
+
+    # Add help argument
+    parser.add_argument("--help", "-h", action="store_true", help="Show help message")
+
+    # Add subcommand as positional argument
+    parser.add_argument("command", nargs="?", choices=["create", "deploy"], help="Command to execute")
+
+    # Parse arguments
+    parsed_args = parser.parse_args(args)
+
+    # Handle help flag or no command
+    if parsed_args.help or not parsed_args.command:
+        show_help()
+        sys.exit(0)
+
+    # Handle commands
+    if parsed_args.command == "create":
+        handle_create()
+    elif parsed_args.command == "deploy":
+        handle_deploy()
+    else:
+        print(f"Unknown command: {parsed_args.command}")
+        show_help()
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()

mcp_use/logging.py

@@ -80,6 +80,9 @@ class Logger:
 
         root_logger.setLevel(level)
 
+        # Set propagate to True to ensure child loggers inherit settings
+        root_logger.propagate = True
+
         # Clear existing handlers
         for handler in root_logger.handlers[:]:
             root_logger.removeHandler(handler)
@@ -91,6 +94,7 @@ class Logger:
         if log_to_console:
             console_handler = logging.StreamHandler(sys.stdout)
             console_handler.setFormatter(formatter)
+            console_handler.setLevel(level)  # Ensure handler respects the level
             root_logger.addHandler(console_handler)
 
         # Add file handler if requested
@@ -102,6 +106,7 @@ class Logger:
 
             file_handler = logging.FileHandler(log_to_file)
             file_handler.setFormatter(formatter)
+            file_handler.setLevel(level)  # Ensure handler respects the level
             root_logger.addHandler(file_handler)
 
     @classmethod
@@ -114,24 +119,34 @@ class Logger:
         global MCP_USE_DEBUG
         MCP_USE_DEBUG = debug_level
 
-        # Update log level for existing loggers
+        # Determine the target level
         if debug_level == 2:
-            for logger in cls._loggers.values():
-                logger.setLevel(logging.DEBUG)
-                langchain_set_debug(True)
+            target_level = logging.DEBUG
+            langchain_set_debug(True)
         elif debug_level == 1:
-            for logger in cls._loggers.values():
-                logger.setLevel(logging.INFO)
-                langchain_set_debug(False)
+            target_level = logging.INFO
+            langchain_set_debug(False)
         else:
-            # Reset to default (WARNING)
-            for logger in cls._loggers.values():
-                logger.setLevel(logging.WARNING)
-                langchain_set_debug(False)
+            target_level = logging.WARNING
+            langchain_set_debug(False)
+
+        # Update log level for existing loggers in our registry
+        for logger in cls._loggers.values():
+            logger.setLevel(target_level)
+            # Also update handler levels
+            for handler in logger.handlers:
+                handler.setLevel(target_level)
+
+        # Also update all mcp_use child loggers that might exist
+        # This ensures loggers created with logging.getLogger() are also updated
+        base_logger = logging.getLogger("mcp_use")
+        base_logger.setLevel(target_level)
+        for handler in base_logger.handlers:
+            handler.setLevel(target_level)
 
 
 # Check environment variable for debug flag
-debug_env = os.environ.get("DEBUG", "").lower()
+debug_env = os.environ.get("MCP_USE_DEBUG", "").lower() or os.environ.get("DEBUG", "").lower()
 if debug_env == "2":
     MCP_USE_DEBUG = 2
 elif debug_env == "1":

mcp_use/managers/base.py

@@ -0,0 +1,36 @@
+from abc import ABC, abstractmethod
+
+from langchain_core.tools import BaseTool
+
+
+class BaseServerManager(ABC):
+    """Abstract base class for server managers.
+
+    This class defines the interface for server managers that can be used with MCPAgent.
+    Custom server managers should inherit from this class and implement the required methods.
+    """
+
+    @abstractmethod
+    async def initialize(self) -> None:
+        """Initialize the server manager."""
+        raise NotImplementedError
+
+    @property
+    @abstractmethod
+    def tools(self) -> list[BaseTool]:
+        """Get all server management tools and tools from the active server.
+
+        Returns:
+            list of LangChain tools for server management plus tools from active server
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def has_tool_changes(self, current_tool_names: set[str]) -> bool:
+        """Check if the available tools have changed.
+        Args:
+            current_tool_names: Set of currently known tool names
+        Returns:
+            True if tools have changed, False otherwise
+        """
+        raise NotImplementedError

mcp_use/managers/server_manager.py

@@ -4,10 +4,11 @@ from mcp_use.client import MCPClient
 from mcp_use.logging import logger
 
 from ..adapters.base import BaseAdapter
+from .base import BaseServerManager
 from .tools import ConnectServerTool, DisconnectServerTool, GetActiveServerTool, ListServersTool, SearchToolsTool
 
 
-class ServerManager:
+class ServerManager(BaseServerManager):
     """Manages MCP servers and provides tools for server selection and management.
 
     This class allows an agent to discover and select which MCP server to use,