mcp-use 1.2.8__py3-none-any.whl → 1.2.10__py3-none-any.whl

mcp_use/agents/mcpagent.py

@@ -6,6 +6,7 @@ to provide a simple interface for using MCP tools with different LLMs.
 """
 
 import logging
+from collections.abc import AsyncIterator
 
 from langchain.agents import AgentExecutor, create_tool_calling_agent
 from langchain.globals import set_debug
@@ -304,6 +305,86 @@ class MCPAgent:
         """
         return self.disallowed_tools
 
+    async def _generate_response_chunks_async(
+        self,
+        query: str,
+        max_steps: int | None = None,
+        manage_connector: bool = True,
+        external_history: list[BaseMessage] | None = None,
+    ) -> AsyncIterator[str]:
+        """Internal async generator yielding response chunks.
+
+        The implementation purposefully keeps the logic compact:
+        1. Ensure the agent is initialised (optionally handling connector
+           lifecycle).
+        2. Forward the *same* inputs we use for ``run`` to LangChain's
+           ``AgentExecutor.astream``.
+        3. Diff the growing ``output`` field coming from LangChain and yield
+           only the new part so the caller receives *incremental* chunks.
+        4. Persist conversation history when memory is enabled.
+        """
+
+        # 1. Initialise on-demand ------------------------------------------------
+        initialised_here = False
+        if (manage_connector and not self._initialized) or (
+            not self._initialized and self.auto_initialize
+        ):
+            await self.initialize()
+            initialised_here = True
+
+        if not self._agent_executor:
+            raise RuntimeError("MCP agent failed to initialise – call initialise() first?")
+
+        # 2. Build inputs --------------------------------------------------------
+        effective_max_steps = max_steps or self.max_steps
+        self._agent_executor.max_iterations = effective_max_steps
+
+        if self.memory_enabled:
+            self.add_to_history(HumanMessage(content=query))
+
+        history_to_use = (
+            external_history if external_history is not None else self._conversation_history
+        )
+        langchain_history: list[BaseMessage] = [
+            m for m in history_to_use if isinstance(m, HumanMessage | AIMessage)
+        ]
+        inputs = {"input": query, "chat_history": langchain_history}
+
+        # 3. Stream & diff -------------------------------------------------------
+        accumulated = ""
+        async for event in self._agent_executor.astream(inputs):
+            yield event
+
+        # 4. Persist assistant message ------------------------------------------
+        if self.memory_enabled and accumulated:
+            self.add_to_history(AIMessage(content=accumulated))
+
+        # 5. House-keeping -------------------------------------------------------
+        if initialised_here and manage_connector:
+            await self.close()
+
+    async def astream(
+        self,
+        query: str,
+        max_steps: int | None = None,
+        manage_connector: bool = True,
+        external_history: list[BaseMessage] | None = None,
+    ) -> AsyncIterator[str]:
+        """Asynchronous streaming interface.
+
+        Example::
+
+            async for chunk in agent.astream("hello"):
+                print(chunk, end="|", flush=True)
+        """
+        async for chunk in self._generate_response_chunks_async(
+            query=query,
+            max_steps=max_steps,
+            manage_connector=manage_connector,
+            external_history=external_history,
+        ):
+            yield chunk
+
     async def run(
         self,
         query: str,

mcp_use/connectors/base.py

@@ -131,6 +131,26 @@ class BaseConnector(ABC):
         resource = await self.client.read_resource(uri)
         return resource.content, resource.mimeType
 
+    async def list_prompts(self) -> list[dict[str, Any]]:
+        """List all available prompts from the MCP implementation."""
+        if not self.client:
+            raise RuntimeError("MCP client is not connected")
+
+        logger.debug("Listing prompts")
+        prompts = await self.client.list_prompts()
+        return prompts
+
+    async def get_prompt(
+        self, name: str, arguments: dict[str, Any] | None = None
+    ) -> tuple[bytes, str]:
+        """Get a prompt by name."""
+        if not self.client:
+            raise RuntimeError("MCP client is not connected")
+
+        logger.debug(f"Getting prompt: {name}")
+        prompt = await self.client.get_prompt(name, arguments)
+        return prompt
+
     async def request(self, method: str, params: dict[str, Any] | None = None) -> Any:
         """Send a raw request to the MCP implementation."""
         if not self.client:

mcp_use/managers/tools/search_tools.py

@@ -3,7 +3,6 @@ import time
 from typing import ClassVar
 
 import numpy as np
-from fastembed import TextEmbedding
 from langchain_core.tools import BaseTool
 from pydantic import BaseModel, Field
 
@@ -114,11 +113,22 @@ class ToolSearchEngine:
         if self.model is not None:
             return True
 
+        try:
+            from fastembed import TextEmbedding  # optional dependency install with [search]
+        except ImportError:
+            logger.error(
+                "The 'fastembed' library is not installed. "
+                "To use the search functionality, please install it by running: "
+                "pip install mcp-use[search]"
+            )
+            return False
+
         try:
             self.model = TextEmbedding(model_name="BAAI/bge-small-en-v1.5")
             self.embedding_function = lambda texts: list(self.model.embed(texts))
             return True
-        except Exception:
+        except Exception as e:
+            logger.error(f"Failed to load the embedding model: {e}")
             return False
 
     async def start_indexing(self) -> None:

{mcp_use-1.2.8.dist-info → mcp_use-1.2.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-use
-Version: 1.2.8
+Version: 1.2.10
 Summary: MCP Library for LLMs
 Author-email: Pietro Zullo <pietro.zullo@gmail.com>
 License: MIT
@@ -50,6 +50,7 @@ Description-Content-Type: text/markdown
 [![PyPI Version](https://img.shields.io/pypi/v/mcp_use.svg)](https://pypi.org/project/mcp_use/)
 [![Python Versions](https://img.shields.io/pypi/pyversions/mcp_use.svg)](https://pypi.org/project/mcp_use/)
 [![Documentation](https://img.shields.io/badge/docs-mcp--use.io-blue)](https://docs.mcp-use.io)
+[![Website](https://img.shields.io/badge/website-mcp--use.io-blue)](https://mcp-use.io)
 [![License](https://img.shields.io/github/license/pietrozullo/mcp-use)](https://github.com/pietrozullo/mcp-use/blob/main/LICENSE)
 [![Code style: Ruff](https://img.shields.io/badge/code%20style-ruff-000000.svg)](https://github.com/astral-sh/ruff)
 [![GitHub stars](https://img.shields.io/github/stars/pietrozullo/mcp-use?style=social)](https://github.com/pietrozullo/mcp-use/stargazers)
@@ -67,7 +68,8 @@ Description-Content-Type: text/markdown
 |---------|-------------|
 | 🔄 [**Ease of use**](#quick-start) | Create your first MCP capable agent you need only 6 lines of code |
 | 🤖 [**LLM Flexibility**](#installing-langchain-providers) | Works with any langchain supported LLM that supports tool calling (OpenAI, Anthropic, Groq, LLama etc.) |
-| 🌐 [**HTTP Support**](#http-connection-example) | Direct connection to MCP servers running on specific HTTP ports |
+| 🌐 [**Code Builder**](https://mcp-use.io/builder) | Explore MCP capabilities and generate starter code with the interactive [code builder](https://mcp-use.io/builder). |
+| 🔗 [**HTTP Support**](#http-connection-example) | Direct connection to MCP servers running on specific HTTP ports |
 | ⚙️ [**Dynamic Server Selection**](#dynamic-server-selection-server-manager) | Agents can dynamically choose the most appropriate MCP server for a given task from the available pool |
 | 🧩 [**Multi-Server Support**](#multi-server-support) | Use multiple MCP servers simultaneously in a single agent |
 | 🛡️ [**Tool Restrictions**](#tool-access-control) | Restrict potentially dangerous tools like file system or network access |
@@ -184,6 +186,43 @@ Example configuration file (`browser_mcp.json`):
 
 For other settings, models, and more, check out the documentation.
 
+## Streaming Agent Output
+
+MCP-Use supports asynchronous streaming of agent output using the `astream` method on `MCPAgent`. This allows you to receive incremental results, tool actions, and intermediate steps as they are generated by the agent, enabling real-time feedback and progress reporting.
+
+### How to use
+
+Call `agent.astream(query)` and iterate over the results asynchronously:
+
+```python
+async for chunk in agent.astream("Find the best restaurant in San Francisco"):
+    print(chunk["messages"], end="", flush=True)
+```
+
+Each chunk is a dictionary containing keys such as `actions`, `steps`, `messages`, and (on the last chunk) `output`. This enables you to build responsive UIs or log agent progress in real time.
+
+#### Example: Streaming in Practice
+
+```python
+import asyncio
+import os
+from dotenv import load_dotenv
+from langchain_openai import ChatOpenAI
+from mcp_use import MCPAgent, MCPClient
+
+async def main():
+    load_dotenv()
+    client = MCPClient.from_config_file("browser_mcp.json")
+    llm = ChatOpenAI(model="gpt-4o")
+    agent = MCPAgent(llm=llm, client=client, max_steps=30)
+    async for chunk in agent.astream("Look for job at nvidia for machine learning engineer."):
+        print(chunk["messages"], end="", flush=True)
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+This streaming interface is ideal for applications that require real-time updates, such as chatbots, dashboards, or interactive notebooks.
 
 # Example Use Cases
 
@@ -344,7 +383,7 @@ if __name__ == "__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.
+MCP-Use 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:
 

{mcp_use-1.2.8.dist-info → mcp_use-1.2.10.dist-info}/RECORD

@@ -8,11 +8,11 @@ mcp_use/adapters/base.py,sha256=ixLHXp8FWdyZPx7Kh6s-4jEVs3qT4DWrApSLXfqzNws,6141
 mcp_use/adapters/langchain_adapter.py,sha256=s8IHPPtqqXMmWQfeBqwESs3SZA6_ECSiGRwdTOIWki0,6417
 mcp_use/agents/__init__.py,sha256=N3eVYP2PxqNO2KcQv5fY8UMUX2W3eLTNkkzuFIJ1DUA,261
 mcp_use/agents/base.py,sha256=bfuldi_89AbSbNc8KeTiCArRT9V62CNxHOWYkLHWjyA,1605
-mcp_use/agents/mcpagent.py,sha256=khzBFeW-mJxA4YeCA6v0jD7czu9MwOjtIXD0_WzNkQE,23795
+mcp_use/agents/mcpagent.py,sha256=1U2HHOk-P-W4O81wdmEXaVWcGU2VQ9xyGaWykK1VCVQ,26954
 mcp_use/agents/prompts/system_prompt_builder.py,sha256=GH5Pvl49IBpKpZA9YTI83xMsdYSkRN_hw4LFHkKtxbg,4122
 mcp_use/agents/prompts/templates.py,sha256=AZKrGWuI516C-PmyOPvxDBibNdqJtN24sOHTGR06bi4,1933
 mcp_use/connectors/__init__.py,sha256=jnd-7pPPJMb0UNJ6aD9lInj5Tlamc8lA_mFyG8RWJpo,385
-mcp_use/connectors/base.py,sha256=5TcXB-I5zrwPtedB6dShceNucsK3wHBeGC2yDVq8X48,4885
+mcp_use/connectors/base.py,sha256=5V8XHh3K-_zMBG22c6cxHynLyZps7M8FTIMt29OJa8o,5599
 mcp_use/connectors/http.py,sha256=2ZG5JxcK1WZ4jkTfTir6bEQLMxXBTPHyi0s42RHGeFs,2837
 mcp_use/connectors/stdio.py,sha256=MTzsqmVVihACUKngE-g5BignK3jAFds2CFv3aSzbJfs,2608
 mcp_use/connectors/websocket.py,sha256=LeU53YI3zjbwKq5GzFRziqA_z9Dn5qACiNyxWDrn2ns,9540
@@ -24,14 +24,14 @@ mcp_use/managers/tools/connect_server.py,sha256=MGYQCl11q-w6gSIYuT44dDk7ILV3Oh7k
 mcp_use/managers/tools/disconnect_server.py,sha256=4487QlLbXAh9JyfGioc6DMWd0n_dkaa8RLMvsoNZv3E,1602
 mcp_use/managers/tools/get_active_server.py,sha256=LRcHbKZopMl1PiO4D4JS4s0fwtrvtMtvb4kpnoAE8fQ,1015
 mcp_use/managers/tools/list_servers_tool.py,sha256=OPDSMNe-VuAhlUyhDnR4CiuZFpoMhnhWpAablwO5S0k,1897
-mcp_use/managers/tools/search_tools.py,sha256=0BfFesCBJsdCAG2tPCM-c49tmBJLwLQoR_U-sj-rp2s,11628
+mcp_use/managers/tools/search_tools.py,sha256=sT2fe66IyOeASTGjdTsjyzSpqkIGKLVXBF8wXUtWXd4,12055
 mcp_use/managers/tools/use_tool.py,sha256=r7k7uMYzrk353qw7M5h1utu_IR2G85uMZkrNcg2RyZA,6824
 mcp_use/task_managers/__init__.py,sha256=4dgW5N61iiPLpwjU2rrn_uqrL8mmDJFDaF9Lukzk65A,486
 mcp_use/task_managers/base.py,sha256=ksNdxTwq8N-zqymxVoKGnWXq9iqkLYC61uB91o6Mh-4,4888
 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.2.8.dist-info/METADATA,sha256=ld-cA_dlzxM-D5VkJMLqgr3WeyGzHVRx_pQ_sKQxRL8,18493
-mcp_use-1.2.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_use-1.2.8.dist-info/licenses/LICENSE,sha256=7Pw7dbwJSBw8zH-WE03JnR5uXvitRtaGTP9QWPcexcs,1068
-mcp_use-1.2.8.dist-info/RECORD,,
+mcp_use-1.2.10.dist-info/METADATA,sha256=WnpB7b1j-Py3BtKajWprmb6PLf89JxtWdCsTuquAuis,20199
+mcp_use-1.2.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_use-1.2.10.dist-info/licenses/LICENSE,sha256=7Pw7dbwJSBw8zH-WE03JnR5uXvitRtaGTP9QWPcexcs,1068
+mcp_use-1.2.10.dist-info/RECORD,,