mcp-use 1.3.12__py3-none-any.whl → 1.4.0__py3-none-any.whl

mcp_use/agents/prompts/system_prompt_builder.py

@@ -1,4 +1,4 @@
-from langchain.schema import SystemMessage
+from langchain_core.messages import SystemMessage
 from langchain_core.tools import BaseTool
 
 

mcp_use/agents/remote.py

@@ -9,10 +9,10 @@ from typing import Any, TypeVar
 from uuid import UUID
 
 import httpx
-from langchain.schema import BaseMessage
+from langchain_core.messages import BaseMessage
 from pydantic import BaseModel
 
-from ..logging import logger
+from mcp_use.logging import logger
 
 T = TypeVar("T", bound=BaseModel)
 
@@ -282,6 +282,7 @@ class RemoteAgent:
                 if event.startswith("0:"):  # Text event
                     try:
                         text_data = json.loads(event[2:])  # Remove "0:" prefix
+                        # Normal text accumulation
                         if final_result is None:
                             final_result = ""
                         final_result += text_data
@@ -308,6 +309,18 @@ class RemoteAgent:
                     except json.JSONDecodeError as e:
                         raise RuntimeError("Agent execution failed with unknown error") from e
 
+                elif event.startswith("f:"):  # Structured final event
+                    try:
+                        structured_data = json.loads(event[2:])  # Remove "f:" prefix
+                        logger.info(f"📋 [{self.chat_id}] Received structured final event")
+
+                        # Replace accumulated text with structured output
+                        final_result = structured_data
+                        logger.info(f"📋 [{self.chat_id}] Replaced accumulated text with structured output")
+                    except json.JSONDecodeError:
+                        logger.warning(f"Failed to parse structured final event: {event[:100]}")
+                        continue
+
             # Log completion of stream consumption
             logger.info(f"Stream consumption complete. Finished: {finished}, Steps taken: {steps_taken}")
 

mcp_use/auth/__init__.py

@@ -1,6 +1,21 @@
-"""Authentication support for MCP clients."""
+# mcp_use/auth/__init__.py
+import warnings
 
-from .bearer import BearerAuth
-from .oauth import OAuth
+from typing_extensions import deprecated
 
-__all__ = ["BearerAuth", "OAuth"]
+from mcp_use.client.auth import BearerAuth as _BearerAuth
+from mcp_use.client.auth import OAuth as _OAuth
+
+warnings.warn(
+    "mcp_use.auth is deprecated. Use mcp_use.client.auth. This import will be removed in version 1.4.0",
+    DeprecationWarning,
+    stacklevel=2,
+)
+
+
+@deprecated("Use mcp_use.client.auth.BearerAuth")
+class BearerAuth(_BearerAuth): ...
+
+
+@deprecated("Use mcp_use.client.auth.OAuth")
+class OAuth(_OAuth): ...

mcp_use/auth/bearer.py

@@ -1,17 +1,16 @@
-"""Bearer token authentication support."""
+# mcp_use/auth/bearer.py
+import warnings
 
-from collections.abc import Generator
+from typing_extensions import deprecated
 
-import httpx
-from pydantic import BaseModel, SecretStr
+from mcp_use.client.auth.bearer import BearerAuth as _BearerAuth
 
+warnings.warn(
+    "mcp_use.auth.bearer is deprecated. Use mcp_use.client.auth.bearer. This import will be removed in version 1.4.0",
+    DeprecationWarning,
+    stacklevel=2,
+)
 
-class BearerAuth(httpx.Auth, BaseModel):
-    """Bearer token authentication for HTTP requests."""
 
-    token: SecretStr
-
-    def auth_flow(self, request: httpx.Request) -> Generator[httpx.Request, httpx.Response, None]:
-        """Apply bearer token authentication to the request."""
-        request.headers["Authorization"] = f"Bearer {self.token.get_secret_value()}"
-        yield request
+@deprecated("Use mcp_use.client.auth.bearer.BearerAuth")
+class BearerAuth(_BearerAuth): ...