From f69682f31b745ba0a9a09fd9b11aee1723755cec Mon Sep 17 00:00:00 2001
From: Pavan Kumar <66913595+ascender1729@users.noreply.github.com>
Date: Thu, 17 Apr 2025 18:08:47 +0000
Subject: [PATCH] chore: add detailed logging to MCP server for end-to-end data
 flow visibility
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- Instrumented `mock_math_server.py` to log incoming agent requests and tool responses
- Ensured logs clarify the MCP protocol: agent → MCP server → tool → response → agent
- Supports debugging and compliance with Model Context Protocol (MCP) flow
---
 examples/mcp_example/mock_math_server.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/examples/mcp_example/mock_math_server.py b/examples/mcp_example/mock_math_server.py
index d2b40b71..e49368a2 100644
--- a/examples/mcp_example/mock_math_server.py
+++ b/examples/mcp_example/mock_math_server.py
@@ -10,10 +10,15 @@ mcp = FastMCP("Math-Mock-Server")
 def add(a: int, b: int) -> int:
     """Add two numbers together"""
     try:
+        print(f"MCP Server: Processing addition request for {a} + {b}")
         time.sleep(0.1)  # Simulate processing time
-        return a + b
+        result = a + b
+        print(f"MCP Server: Returning result {result}")
+        return result
     except Exception as e:
-        return {"error": f"Error adding numbers: {str(e)}"}
+        error_msg = f"Error adding numbers: {str(e)}"
+        print(f"MCP Server: {error_msg}")
+        return {"error": error_msg}
 
 @mcp.tool()
 def multiply(a: int, b: int) -> int: