diff --git a/python/helpers/log.py b/python/helpers/log.py index 808e8fa4c..acfddd6a4 100644 --- a/python/helpers/log.py +++ b/python/helpers/log.py @@ -28,6 +28,7 @@ Type = Literal[ "progress", "response", "tool", + "mcp", "input", "user", "util", diff --git a/python/helpers/mcp_handler.py b/python/helpers/mcp_handler.py index 5936f4864..482724057 100644 --- a/python/helpers/mcp_handler.py +++ b/python/helpers/mcp_handler.py @@ -23,6 +23,7 @@ from datetime import timedelta import json from python.helpers import errors from python.helpers import settings +from python.helpers.log import LogItem import httpx @@ -100,6 +101,14 @@ def initialize_mcp(mcp_servers_config: str): class MCPTool(Tool): """MCP Tool wrapper""" + def get_log_object(self) -> LogItem: + return self.agent.context.log.log( + type="mcp", + heading=f"icon://extension {self.agent.agent_name}: Using MCP tool '{self.name}'", + content="", + kvps={"tool_name": self.name, **self.args}, + ) + async def execute(self, **kwargs: Any): error = "" try: