mcp-use 0.0.3__tar.gz → 0.0.4__tar.gz

mcp_use-0.0.4/.github/workflows/publish.yml

@@ -0,0 +1,31 @@
+name: Publish to PyPI
+
+on:
+  release:
+    types: [created]
+
+# Required for PyPI trusted publishing
+permissions:
+  id-token: write
+  contents: read
+
+jobs:
+  deploy:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.11"
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install build twine wheel
+      - name: Build package
+        run: |
+          python -m build
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          password: ${{ secrets.PYPI_API_TOKEN }}

mcp_use-0.0.4/.github/workflows/tests.yml

@@ -0,0 +1,31 @@
+name: Python Tests
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.11", "3.12"]
+
+    steps:
+      - uses: actions/checkout@v3
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v4
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install .[dev,anthropic,openai]
+      - name: Lint with ruff
+        run: |
+          ruff check .
+      - name: Test with pytest
+        run: |
+          pytest

mcp_use-0.0.4/.gitignore

@@ -0,0 +1,124 @@
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# PyInstaller
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# pipenv
+Pipfile.lock
+
+# poetry
+poetry.lock
+
+# Environment variables
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# VS Code
+.vscode/
+*.code-workspace
+
+# PyCharm
+.idea/
+*.iml
+
+# macOS
+.DS_Store

mcp_use-0.0.4/.pre-commit-config.yaml

@@ -0,0 +1,24 @@
+repos:
+-   repo: https://github.com/astral-sh/ruff-pre-commit
+    # Ruff version.
+    rev: v0.3.2
+    hooks:
+    -   id: ruff
+        args: [--fix, --exit-non-zero-on-fix, --config=pyproject.toml]
+        types: [python]
+    -   id: ruff-format
+        args: [--config=pyproject.toml]
+        types: [python]
+
+-   repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.5.0
+    hooks:
+    -   id: trailing-whitespace
+    -   id: end-of-file-fixer
+    -   id: check-yaml
+    -   id: check-added-large-files
+    -   id: debug-statements
+
+# Define configuration for the Python checks
+default_language_version:
+    python: python3.11

{mcp_use-0.0.3 → mcp_use-0.0.4}/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: mcp_use
-Version: 0.0.3
-Summary: Model-Agnostic MCP Library for LLMs
-Home-page: https://github.com/pietrozullo/mcp_use
-Author: Pietro Zullo
-Author-email: pietro.zullo@gmail.com
+Version: 0.0.4
+Summary: MCP Library for LLMs
+Author-email: Pietro Zullo <pietro.zullo@gmail.com>
+License: MIT
+License-File: LICENSE
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: MIT License
@@ -14,40 +14,28 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.11
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: mcp
-Requires-Dist: langchain>=0.1.0
-Requires-Dist: langchain-community>=0.0.10
-Requires-Dist: websockets>=12.0
 Requires-Dist: aiohttp>=3.9.0
-Requires-Dist: pydantic>=2.0.0
-Requires-Dist: typing-extensions>=4.8.0
 Requires-Dist: jsonschema-pydantic>=0.1.0
+Requires-Dist: langchain-community>=0.0.10
+Requires-Dist: langchain>=0.1.0
+Requires-Dist: mcp
+Requires-Dist: pydantic>=2.0.0
 Requires-Dist: python-dotenv>=1.0.0
-Provides-Extra: dev
-Requires-Dist: pytest>=7.4.0; extra == "dev"
-Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
-Requires-Dist: pytest-cov>=4.1.0; extra == "dev"
-Requires-Dist: black>=23.9.0; extra == "dev"
-Requires-Dist: isort>=5.12.0; extra == "dev"
-Requires-Dist: mypy>=1.5.0; extra == "dev"
-Requires-Dist: ruff>=0.1.0; extra == "dev"
+Requires-Dist: typing-extensions>=4.8.0
+Requires-Dist: websockets>=12.0
 Provides-Extra: anthropic
-Requires-Dist: anthropic>=0.15.0; extra == "anthropic"
+Requires-Dist: anthropic>=0.15.0; extra == 'anthropic'
+Provides-Extra: dev
+Requires-Dist: black>=23.9.0; extra == 'dev'
+Requires-Dist: isort>=5.12.0; extra == 'dev'
+Requires-Dist: mypy>=1.5.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.21.0; extra == 'dev'
+Requires-Dist: pytest-cov>=4.1.0; extra == 'dev'
+Requires-Dist: pytest>=7.4.0; extra == 'dev'
+Requires-Dist: ruff>=0.1.0; extra == 'dev'
 Provides-Extra: openai
-Requires-Dist: openai>=1.10.0; extra == "openai"
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: license-file
-Dynamic: provides-extra
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
+Requires-Dist: openai>=1.10.0; extra == 'openai'
+Description-Content-Type: text/markdown
 
 <picture>
   <source media="(prefers-color-scheme: dark)" srcset="./static/mcp-use-dark.png">
@@ -55,7 +43,7 @@ Dynamic: summary
   <img alt="Shows a black MCP-Use Logo in light color mode and a white one in dark color mode." src="./static/mcp-use.png"  width="full">
 </picture>
 
-<h1 align="center">Model-Agnostic MCP Library for LLMs 🤖</h1>
+<h1 align="center">Use MCPs directly from python 🤖</h1>
 
 [![GitHub stars](https://img.shields.io/github/stars/pietrozullo/mcp-use?style=social)](https://github.com/pietrozullo/mcp-use/stargazers)
 
@@ -356,7 +344,7 @@ If you use MCP-Use in your research or project, please cite:
 ```bibtex
 @software{mcp_use2024,
   author = {Zullo, Pietro},
-  title = {MCP-Use: Model-Agnostic MCP Library for LLMs},
+  title = {MCP-Use: MCP Library for Python},
   year = {2024},
   publisher = {GitHub},
   url = {https://github.com/pietrozullo/mcp-use}

{mcp_use-0.0.3 → mcp_use-0.0.4}/README.md

@@ -4,7 +4,7 @@
   <img alt="Shows a black MCP-Use Logo in light color mode and a white one in dark color mode." src="./static/mcp-use.png"  width="full">
 </picture>
 
-<h1 align="center">Model-Agnostic MCP Library for LLMs 🤖</h1>
+<h1 align="center">Use MCPs directly from python 🤖</h1>
 
 [![GitHub stars](https://img.shields.io/github/stars/pietrozullo/mcp-use?style=social)](https://github.com/pietrozullo/mcp-use/stargazers)
 
@@ -305,7 +305,7 @@ If you use MCP-Use in your research or project, please cite:
 ```bibtex
 @software{mcp_use2024,
   author = {Zullo, Pietro},
-  title = {MCP-Use: Model-Agnostic MCP Library for LLMs},
+  title = {MCP-Use: MCP Library for Python},
   year = {2024},
   publisher = {GitHub},
   url = {https://github.com/pietrozullo/mcp-use}

mcp_use-0.0.4/examples/airbnb_mcp.json

@@ -0,0 +1,8 @@
+{
+  "mcpServers": {
+      "airbnb": {
+          "command": "npx",
+          "args": ["-y", "@openbnb/mcp-server-airbnb"]
+      }
+  }
+}

mcp_use-0.0.4/examples/airbnb_use.py

@@ -0,0 +1,49 @@
+"""
+Example demonstrating how to use mcp_use with Airbnb.
+
+This example shows how to connect an LLM to Airbnb through MCP tools
+to perform tasks like searching for accommodations.
+"""
+
+import asyncio
+import os
+
+from dotenv import load_dotenv
+from langchain_anthropic import ChatAnthropic
+
+from mcp_use import MCPAgent, MCPClient
+
+
+async def run_airbnb_example():
+    """Run an example using Airbnb MCP server."""
+    # Load environment variables
+    load_dotenv()
+
+    # Create MCPClient with Airbnb configuration
+    client = MCPClient.from_config_file(os.path.join(os.path.dirname(__file__), "airbnb_mcp.json"))
+    # Create LLM - you can choose between different models
+    llm = ChatAnthropic(model="claude-3-5-sonnet-20240620")
+    # Alternative models:
+    # llm = init_chat_model(model="llama-3.1-8b-instant", model_provider="groq")
+    # llm = ChatOpenAI(model="gpt-4o")
+
+    # Create agent with the client
+    agent = MCPAgent(llm=llm, client=client, max_steps=30)
+
+    try:
+        # Run a query to search for accommodations
+        result = await agent.run(
+            "Find me a nice place to stay in Barcelona for 2 adults "
+            "for a week in August. I prefer places with a pool and "
+            "good reviews. Show me the top 3 options.",
+            max_steps=30,
+        )
+        print(f"\nResult: {result}")
+    finally:
+        # Ensure we clean up resources properly
+        if client.sessions:
+            await client.close_all_sessions()
+
+
+if __name__ == "__main__":
+    asyncio.run(run_airbnb_example())

mcp_use-0.0.4/examples/blender_use.py

@@ -0,0 +1,51 @@
+"""
+Blender MCP example for mcp_use.
+
+This example demonstrates how to use the mcp_use library with MCPClient
+to connect an LLM to Blender through MCP tools via WebSocket.
+The example assumes you have installed the Blender MCP addon from:
+https://github.com/gd3kr/BlenderGPT/tree/main/mcp_addon
+
+Make sure the addon is enabled in Blender preferences and the WebSocket
+server is running before executing this script.
+"""
+
+import asyncio
+
+from dotenv import load_dotenv
+from langchain_anthropic import ChatAnthropic
+
+from mcp_use import MCPAgent, MCPClient
+
+
+async def run_blender_example():
+    """Run the Blender MCP example."""
+    # Load environment variables
+    load_dotenv()
+
+    # Create MCPClient with Blender MCP configuration
+    config = {"mcpServers": {"blender": {"command": "uvx", "args": ["blender-mcp"]}}}
+    client = MCPClient.from_dict(config)
+
+    # Create LLM
+    llm = ChatAnthropic(model="claude-3-5-sonnet-20240620")
+
+    # Create agent with the client
+    agent = MCPAgent(llm=llm, client=client, max_steps=30)
+
+    try:
+        # Run the query
+        result = await agent.run(
+            "Create an inflatable cube with soft material and a plane as ground.",
+            max_steps=30,
+        )
+        print(f"\nResult: {result}")
+    finally:
+        # Ensure we clean up resources properly
+        if client.sessions:
+            await client.close_all_sessions()
+
+
+if __name__ == "__main__":
+    # Run the Blender example
+    asyncio.run(run_blender_example())

mcp_use-0.0.4/examples/browser_mcp.json

@@ -0,0 +1,11 @@
+{
+  "mcpServers": {
+    "playwright": {
+      "command": "npx",
+      "args": ["@playwright/mcp@latest"],
+      "env": {
+        "DISPLAY": ":1"
+      }
+    }
+  }
+}

mcp_use-0.0.4/examples/browser_use.py

@@ -0,0 +1,44 @@
+"""
+Basic usage example for mcp_use.
+
+This example demonstrates how to use the mcp_use library with MCPClient
+to connect any LLM to MCP tools through a unified interface.
+"""
+
+import asyncio
+import os
+
+from dotenv import load_dotenv
+from langchain_openai import ChatOpenAI
+
+from mcp_use import MCPAgent, MCPClient
+
+
+async def main():
+    """Run the example using a configuration file."""
+    # Load environment variables
+    load_dotenv()
+
+    # Create MCPClient from config file
+    client = MCPClient.from_config_file(os.path.join(os.path.dirname(__file__), "browser_mcp.json"))
+
+    # Create LLM
+    llm = ChatOpenAI(model="gpt-4o")
+    # llm = init_chat_model(model="llama-3.1-8b-instant", model_provider="groq")
+    # llm = ChatAnthropic(model="claude-3-")
+    # llm = ChatGroq(model="llama3-8b-8192")
+
+    # Create agent with the client
+    agent = MCPAgent(llm=llm, client=client, max_steps=30)
+
+    # Run the query
+    result = await agent.run(
+        "Find the best restaurant in San Francisco USING GOOGLE SEARCH",
+        max_steps=30,
+    )
+    print(f"\nResult: {result}")
+
+
+if __name__ == "__main__":
+    # Run the appropriate example
+    asyncio.run(main())

mcp_use-0.0.4/examples/chat_examle.py

@@ -0,0 +1,78 @@
+"""
+Simple chat example using MCPAgent with built-in conversation memory.
+
+This example demonstrates how to use the MCPAgent with its built-in
+conversation history capabilities for better contextual interactions.
+"""
+
+import asyncio
+
+from dotenv import load_dotenv
+from langchain_openai import ChatOpenAI
+
+from mcp_use import MCPAgent, MCPClient
+
+
+async def run_memory_chat():
+    """Run a chat using MCPAgent's built-in conversation memory."""
+    # Load environment variables for API keys
+    load_dotenv()
+
+    # Config file path - change this to your config file
+    config_file = "examples/browser_mcp.json"
+
+    print("Initializing chat...")
+
+    # Create MCP client and agent with memory enabled
+    client = MCPClient.from_config_file(config_file)
+    llm = ChatOpenAI(model="gpt-4o-mini")
+
+    # Create agent with memory_enabled=True
+    agent = MCPAgent(
+        llm=llm,
+        client=client,
+        max_steps=15,
+        memory_enabled=True,  # Enable built-in conversation memory
+    )
+
+    print("\n===== Interactive MCP Chat =====")
+    print("Type 'exit' or 'quit' to end the conversation")
+    print("Type 'clear' to clear conversation history")
+    print("==================================\n")
+
+    try:
+        # Main chat loop
+        while True:
+            # Get user input
+            user_input = input("\nYou: ")
+
+            # Check for exit command
+            if user_input.lower() in ["exit", "quit"]:
+                print("Ending conversation...")
+                break
+
+            # Check for clear history command
+            if user_input.lower() == "clear":
+                agent.clear_conversation_history()
+                print("Conversation history cleared.")
+                continue
+
+            # Get response from agent
+            print("\nAssistant: ", end="", flush=True)
+
+            try:
+                # Run the agent with the user input (memory handling is automatic)
+                response = await agent.run(user_input)
+                print(response)
+
+            except Exception as e:
+                print(f"\nError: {e}")
+
+    finally:
+        # Clean up
+        if client and client.sessions:
+            await client.close_all_sessions()
+
+
+if __name__ == "__main__":
+    asyncio.run(run_memory_chat())

{mcp_use-0.0.3 → mcp_use-0.0.4}/mcp_use/__init__.py

@@ -1,5 +1,5 @@
 """
-mcp_use - A model-agnostic MCP (Multi-Channel Platform) library for LLMs.
+mcp_use - An MCP library for LLMs.
 
 This library provides a unified interface for connecting different LLMs
 to MCP tools through existing LangChain adapters.

{mcp_use-0.0.3 → mcp_use-0.0.4}/mcp_use/agents/langchain_agent.py

@@ -1,8 +1,8 @@
 """
-LangChain agent implementation for MCP tools.
+LangChain agent implementation for MCP tools with customizable system message.
 
 This module provides a LangChain agent implementation that can use MCP tools
-through a unified interface.
+through a unified interface, with support for customizable system messages.
 """
 
 from typing import Any, NoReturn
@@ -73,8 +73,15 @@ class LangChainAgent:
     through a unified interface.
     """
 
+    # Default system message if none is provided
+    DEFAULT_SYSTEM_MESSAGE = "You are a helpful AI assistant that can use tools to help users."
+
     def __init__(
-        self, connector: BaseConnector, llm: BaseLanguageModel, max_steps: int = 5
+        self,
+        connector: BaseConnector,
+        llm: BaseLanguageModel,
+        max_steps: int = 5,
+        system_message: str | None = None,
     ) -> None:
         """Initialize a new LangChain agent.
 
@@ -82,13 +89,28 @@ class LangChainAgent:
             connector: The MCP connector to use.
             llm: The LangChain LLM to use.
             max_steps: The maximum number of steps to take.
+            system_message: Optional custom system message to use.
         """
         self.connector = connector
         self.llm = llm
         self.max_steps = max_steps
+        self.system_message = system_message or self.DEFAULT_SYSTEM_MESSAGE
         self.tools: list[BaseTool] = []
         self.agent: AgentExecutor | None = None
 
+    def set_system_message(self, message: str) -> None:
+        """Set a new system message and recreate the agent.
+
+        Args:
+            message: The new system message.
+        """
+        self.system_message = message
+
+        # Recreate the agent with the new system message if it exists
+        if self.agent and self.tools:
+            self.agent = self._create_agent()
+            logger.info("Agent recreated with new system message")
+
     async def initialize(self) -> None:
         """Initialize the agent and its tools."""
         self.tools = await self._create_langchain_tools()
@@ -184,7 +206,7 @@ class LangChainAgent:
         return langchain_tools
 
     def _create_agent(self) -> AgentExecutor:
-        """Create the LangChain agent.
+        """Create the LangChain agent with the configured system message.
 
         Returns:
             An initialized AgentExecutor.
@@ -193,7 +215,7 @@ class LangChainAgent:
             [
                 (
                     "system",
-                    "You are a helpful AI assistant that can use tools to help users.",
+                    self.system_message,
                 ),
                 MessagesPlaceholder(variable_name="chat_history"),
                 ("human", "{input}"),
@@ -234,7 +256,25 @@ class LangChainAgent:
         if chat_history is None:
             chat_history = []
 
+        # Add a hint to use tools for queries about current information
+        enhanced_query = query
+        if any(
+            keyword in query.lower()
+            for keyword in [
+                "weather",
+                "current",
+                "today",
+                "now",
+                "latest",
+                "news",
+                "price",
+                "stock",
+            ]
+        ):
+            # Just log this, don't modify the query
+            logger.info("Query involves current information that may benefit from tool use")
+
         # Invoke with all required variables
-        result = await self.agent.ainvoke({"input": query, "chat_history": chat_history})
+        result = await self.agent.ainvoke({"input": enhanced_query, "chat_history": chat_history})
 
         return result["output"]