mcp-use 1.0.2__py3-none-any.whl → 1.1.4__py3-none-any.whl

mcp_use/agents/langchain_agent.py

@@ -82,6 +82,7 @@ class LangChainAgent:
         llm: BaseLanguageModel,
         max_steps: int = 5,
         system_message: str | None = None,
+        disallowed_tools: list[str] | None = None,
     ) -> None:
         """Initialize a new LangChain agent.
 
@@ -90,11 +91,13 @@ class LangChainAgent:
             llm: The LangChain LLM to use.
             max_steps: The maximum number of steps to take.
             system_message: Optional custom system message to use.
+            disallowed_tools: List of tool names that should not be available to the agent.
         """
         self.connectors = connectors
         self.llm = llm
         self.max_steps = max_steps
         self.system_message = system_message or self.DEFAULT_SYSTEM_MESSAGE
+        self.disallowed_tools = disallowed_tools or []
         self.tools: list[BaseTool] = []
         self.agent: AgentExecutor | None = None
 
@@ -137,17 +140,16 @@ class LangChainAgent:
         """Create LangChain tools from MCP tools.
 
         Returns:
-            A list of LangChain tools created from MCP tools.
+            A list of LangChain tools that wrap MCP tools.
         """
-        langchain_tools: list[BaseTool] = []
-
+        tools = []
         for connector in self.connectors:
-            tools = connector.tools
-            local_connector = connector
+            local_connector = connector  # Capture for closure
+            for tool in connector.tools:
+                # Skip disallowed tools
+                if tool.name in self.disallowed_tools:
+                    continue
 
-            # Wrap MCP tools into LangChain tools
-            for tool in tools:
-                # Define adapter class to convert MCP tool to LangChain format
                 class McpToLangChainAdapter(BaseTool):
                     name: str = tool.name or "NO NAME"
                     description: str = tool.description or ""
@@ -202,11 +204,11 @@ class LangChainAgent:
                                 return f"Error executing MCP tool: {str(e)}"
                             raise
 
-                langchain_tools.append(McpToLangChainAdapter())
+                tools.append(McpToLangChainAdapter())
 
         # Log available tools for debugging
-        logger.info(f"Available tools: {[tool.name for tool in langchain_tools]}")
-        return langchain_tools
+        logger.info(f"Available tools: {[tool.name for tool in tools]}")
+        return tools
 
     def _create_agent(self) -> AgentExecutor:
         """Create the LangChain agent with the configured system message.

mcp_use/agents/mcpagent.py

@@ -39,6 +39,7 @@ class MCPAgent:
         system_prompt: str | None = None,
         system_prompt_template: str | None = None,
         additional_instructions: str | None = None,
+        disallowed_tools: list[str] | None = None,
     ):
         """Initialize a new MCPAgent instance.
 
@@ -53,6 +54,7 @@ class MCPAgent:
             system_prompt: Complete system prompt to use (overrides template if provided).
             system_prompt_template: Template for system prompt with {tool_descriptions} placeholder.
             additional_instructions: Extra instructions to append to the system prompt.
+            disallowed_tools: List of tool names that should not be available to the agent.
         """
         self.llm = llm
         self.client = client
@@ -63,6 +65,7 @@ class MCPAgent:
         self.memory_enabled = memory_enabled
         self._initialized = False
         self._conversation_history: list[BaseMessage] = []
+        self.disallowed_tools = disallowed_tools or []
 
         # System prompt configuration
         self.system_prompt = system_prompt
@@ -102,6 +105,7 @@ class MCPAgent:
             llm=self.llm,
             max_steps=self.max_steps,
             system_message=(self._system_message.content if self._system_message else None),
+            disallowed_tools=self.disallowed_tools,
         )
 
         # Initialize the agent
@@ -125,6 +129,10 @@ class MCPAgent:
             tools = connector.tools
             # Generate tool descriptions
             for tool in tools:
+                # Skip disallowed tools
+                if tool.name in self.disallowed_tools:
+                    continue
+
                 # Escape curly braces in the description by doubling them
                 # (sometimes e.g. blender mcp they are used in the description)
                 description = (
@@ -188,6 +196,33 @@ class MCPAgent:
         if self._agent:
             self._agent.set_system_message(message)
 
+    def set_disallowed_tools(self, disallowed_tools: list[str]) -> None:
+        """Set the list of tools that should not be available to the agent.
+
+        This will take effect the next time the agent is initialized.
+
+        Args:
+            disallowed_tools: List of tool names that should not be available.
+        """
+        self.disallowed_tools = disallowed_tools
+
+        # If the agent is already initialized, we need to reinitialize it
+        # to apply the changes to the available tools
+        if self._initialized:
+            logger.info(
+                "Agent already initialized. Changes will take effect on next initialization."
+            )
+            # We don't automatically reinitialize here as it could be disruptive
+            # to ongoing operations. The user can call initialize() explicitly if needed.
+
+    def get_disallowed_tools(self) -> list[str]:
+        """Get the list of tools that are not available to the agent.
+
+        Returns:
+            List of tool names that are not available.
+        """
+        return self.disallowed_tools
+
     async def run(
         self,
         query: str,

{mcp_use-1.0.2.dist-info → mcp_use-1.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-use
-Version: 1.0.2
+Version: 1.1.4
 Summary: MCP Library for LLMs
 Author-email: Pietro Zullo <pietro.zullo@gmail.com>
 License: MIT
@@ -56,6 +56,19 @@ Description-Content-Type: text/markdown
 
 💡 Let developers easily connect any LLM to tools like web browsing, file operations, and more.
 
+# Features
+
+## ✨ Key Features
+
+| Feature | Description |
+|---------|-------------|
+| 🔄 **Ease of use** | Create your first MCP capable agent you need only 6 lines of code |
+| 🤖 **LLM Flexibility** | Works with any langchain supported LLM that supports tool calling (OpenAI, Anthropic, Groq, LLama etc.) |
+| 🌐 **HTTP Support** | Direct connection to MCP servers running on specific HTTP ports |
+| 🧩 **Multi-Server Support** | Use multiple MCP servers simultaneously in a single agent |
+| 🛡️ **Tool Restrictions** | Restrict potentially dangerous tools like file system or network access |
+
+
 # Quick start
 
 With pip:
@@ -72,7 +85,30 @@ cd mcp-use
 pip install -e .
 ```
 
-Spin up your agent:
+### Installing LangChain Providers
+
+mcp_use works with various LLM providers through LangChain. You'll need to install the appropriate LangChain provider package for your chosen LLM. For example:
+
+```bash
+# For OpenAI
+pip install langchain-openai
+
+# For Anthropic
+pip install langchain-anthropic
+
+# For other providers, check the [LangChain chat models documentation](https://python.langchain.com/docs/integrations/chat/)
+```
+
+and add your API keys for the provider you want to use to your `.env` file.
+
+```bash
+OPENAI_API_KEY=
+ANTHROPIC_API_KEY=
+```
+
+> **Important**: Only models with tool calling capabilities can be used with mcp_use. Make sure your chosen model supports function calling or tool use.
+
+### Spin up your agent:
 
 ```python
 import asyncio
@@ -85,8 +121,21 @@ async def main():
     # Load environment variables
     load_dotenv()
 
-    # Create MCPClient from config file
-    client = MCPClient.from_config_file("browser_mcp.json")
+    # Create configuration dictionary
+    config = {
+      "mcpServers": {
+        "playwright": {
+          "command": "npx",
+          "args": ["@playwright/mcp@latest"],
+          "env": {
+            "DISPLAY": ":1"
+          }
+        }
+      }
+    }
+
+    # Create MCPClient from configuration dictionary
+    client = MCPClient.from_dict(config)
 
     # Create LLM
     llm = ChatOpenAI(model="gpt-4o")
@@ -96,7 +145,7 @@ async def main():
 
     # Run the query
     result = await agent.run(
-        "Find the best restaurant in San Francisco USING GOOGLE SEARCH",
+        "Find the best restaurant in San Francisco",
     )
     print(f"\nResult: {result}")
 
@@ -104,6 +153,14 @@ if __name__ == "__main__":
     asyncio.run(main())
 ```
 
+You can also add the servers configuration from a config file like this:
+
+```python
+client = MCPClient.from_config_file(
+        os.path.join("browser_mcp.json")
+    )
+```
+
 Example configuration file (`browser_mcp.json`):
 
 ```json
@@ -120,15 +177,10 @@ Example configuration file (`browser_mcp.json`):
 }
 ```
 
-Add your API keys for the provider you want to use to your `.env` file.
-
-```bash
-OPENAI_API_KEY=
-ANTHROPIC_API_KEY=
-```
-
 For other settings, models, and more, check out the documentation.
 
+# Features
+
 # Example Use Cases
 
 ## Web Browsing with Playwright
@@ -286,6 +338,55 @@ if __name__ == "__main__":
     asyncio.run(main())
 ```
 
+## HTTP Connection Example
+
+MCP-Use now supports HTTP connections, allowing you to connect to MCP servers running on specific HTTP ports. This feature is particularly useful for integrating with web-based MCP servers.
+
+Here's an example of how to use the HTTP connection feature:
+
+```python
+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()
+
+    config = {
+        "mcpServers": {
+            "http": {
+                "url": "http://localhost:8931/sse"
+            }
+        }
+    }
+
+    # Create MCPClient from config file
+    client = MCPClient.from_dict(config)
+
+    # Create LLM
+    llm = ChatOpenAI(model="gpt-4o")
+
+    # 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())
+```
+
+This example demonstrates how to connect to an MCP server running on a specific HTTP port. Make sure to start your MCP server before running this example.
+
 # Multi-Server Support
 
 MCP-Use supports working with multiple MCP servers simultaneously, allowing you to combine tools from different servers in a single agent. This is useful for complex tasks that require multiple capabilities, such as web browsing combined with file operations or 3D modeling.
@@ -346,25 +447,58 @@ if __name__ == "__main__":
     asyncio.run(main())
 ```
 
-## Roadmap
+# Tool Access Control
+
+MCP-Use allows you to restrict which tools are available to the agent, providing better security and control over agent capabilities:
+
+```python
+import asyncio
+from mcp_use import MCPAgent, MCPClient
+from langchain_openai import ChatOpenAI
+
+async def main():
+    # Create client
+    client = MCPClient.from_config_file("config.json")
+
+    # Create agent with restricted tools
+    agent = MCPAgent(
+        llm=ChatOpenAI(model="gpt-4"),
+        client=client,
+        disallowed_tools=["file_system", "network"]  # Restrict potentially dangerous tools
+    )
+
+    # Run a query with restricted tool access
+    result = await agent.run(
+        "Find the best restaurant in San Francisco"
+    )
+    print(result)
+
+    # Clean up
+    await client.close_all_sessions()
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+# Roadmap
 
 <ul>
 <li>[x] Multiple Servers at once </li>
-<li>[ ] Test remote connectors (http, ws)</li>
+<li>[x] Test remote connectors (http, ws)</li>
 <li>[ ] ... </li>
 </ul>
 
-## Contributing
+# Contributing
 
 We love contributions! Feel free to open issues for bugs or feature requests.
 
-## Requirements
+# Requirements
 
 - Python 3.11+
 - MCP implementation (like Playwright MCP)
 - LangChain and appropriate model libraries (OpenAI, Anthropic, etc.)
 
-## Citation
+# Citation
 
 If you use MCP-Use in your research or project, please cite:
 
@@ -378,6 +512,6 @@ If you use MCP-Use in your research or project, please cite:
 }
 ```
 
-## License
+# License
 
 MIT

{mcp_use-1.0.2.dist-info → mcp_use-1.1.4.dist-info}/RECORD

@@ -5,8 +5,8 @@ mcp_use/logging.py,sha256=2-hSB7ZWcHEx_OFHNg8GIbSGCZx3MW4mZGGWxi2Ew3E,2690
 mcp_use/session.py,sha256=Z4EZTUnQUX0QyGMzkJIrMRTX4SDk6qQUoBld408LIJE,3449
 mcp_use/agents/__init__.py,sha256=ukchMTqCOID6ikvLmJ-6sldWTVFIzztGQo4BX6QeQr8,312
 mcp_use/agents/base.py,sha256=bfuldi_89AbSbNc8KeTiCArRT9V62CNxHOWYkLHWjyA,1605
-mcp_use/agents/langchain_agent.py,sha256=q6zIb9J9fc15HRGDjPAhmPdM_8UOqQToy8ESeyry1kc,10035
-mcp_use/agents/mcpagent.py,sha256=lTRutdT1QIMiTbMSKfSbqlqNq_Y6uDPfkjAzJAKb6H0,12727
+mcp_use/agents/langchain_agent.py,sha256=5fml081T3meLkZxA8o29eLuMmITyr5EfkKMPGy9UgHA,10165
+mcp_use/agents/mcpagent.py,sha256=YF-ApIGM2lM8mOdeopr-hPlMRuNRNMQpJpg4gm9u6Ns,14183
 mcp_use/agents/prompts/default.py,sha256=tnwt9vOiVBhdpu-lIHhwEJo3rvE6EobPfUgS9JURBzg,941
 mcp_use/connectors/__init__.py,sha256=jnd-7pPPJMb0UNJ6aD9lInj5Tlamc8lA_mFyG8RWJpo,385
 mcp_use/connectors/base.py,sha256=TCLVNJdt6qrflmphgXOZhD6xPKQQegbGqe5REmcLYg0,4813
@@ -18,7 +18,7 @@ mcp_use/task_managers/base.py,sha256=ksNdxTwq8N-zqymxVoKGnWXq9iqkLYC61uB91o6Mh-4
 mcp_use/task_managers/sse.py,sha256=WysmjwqRI3meXMZY_F4y9tSBMvSiUZfTJQfitM5l6jQ,2529
 mcp_use/task_managers/stdio.py,sha256=DEISpXv4mo3d5a-WT8lkWbrXJwUh7QW0nMT_IM3fHGg,2269
 mcp_use/task_managers/websocket.py,sha256=ZbCqdGgzCRtsXzRGFws-f2OzH8cPAkN4sJNDwEpRmCc,1915
-mcp_use-1.0.2.dist-info/METADATA,sha256=4Tqt1IT7KtUS5HVF9VIw0dt6Sgf2lBslMlv5ztkAg8k,10203
-mcp_use-1.0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_use-1.0.2.dist-info/licenses/LICENSE,sha256=7Pw7dbwJSBw8zH-WE03JnR5uXvitRtaGTP9QWPcexcs,1068
-mcp_use-1.0.2.dist-info/RECORD,,
+mcp_use-1.1.4.dist-info/METADATA,sha256=vE5PNvtxt7MIOI5EL1DRA73xzNoGjEe2Xa7WfBdV9rU,14015
+mcp_use-1.1.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_use-1.1.4.dist-info/licenses/LICENSE,sha256=7Pw7dbwJSBw8zH-WE03JnR5uXvitRtaGTP9QWPcexcs,1068
+mcp_use-1.1.4.dist-info/RECORD,,