mcp-use 1.0.2__py3-none-any.whl → 1.0.3__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.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-use
-Version: 1.0.2
+Version: 1.0.3
 Summary: MCP Library for LLMs
 Author-email: Pietro Zullo <pietro.zullo@gmail.com>
 License: MIT
@@ -346,25 +346,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 +411,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.0.3.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.0.3.dist-info/METADATA,sha256=3HbO5h9azPZyPz6HmK0rWZEB7pdxh4U3jCSEQKtGdtc,11043
+mcp_use-1.0.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_use-1.0.3.dist-info/licenses/LICENSE,sha256=7Pw7dbwJSBw8zH-WE03JnR5uXvitRtaGTP9QWPcexcs,1068
+mcp_use-1.0.3.dist-info/RECORD,,