feat: add customizable context parameter description for tools

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mcpcat"
-version = "0.1.8"
+version = "0.1.9"
 description = "Analytics Tool for MCP Servers - provides insights into MCP tool usage patterns"
 authors = [
     { name = "MCPCat", email = "support@mcpcat.io" },

src/mcpcat/modules/compatibility.py

@@ -30,35 +30,39 @@ def call_tool(self, name: str, arguments: dict) -> Any:
 
 def is_community_fastmcp_server(server: Any) -> bool:
     """Check if the server is a community FastMCP instance.
-    
+
     Community FastMCP comes from the 'fastmcp' package.
+    Supports FastMCP subclasses like FastMCPOpenAPI, FastMCPProxy, etc.
     """
     # Check by class name and module
     class_name = server.__class__.__name__
     module_name = server.__class__.__module__
-
-    # Community FastMCP has class name 'FastMCP' and module starts with 'fastmcp'
+
+    # Community FastMCP has class name containing 'FastMCP' and module starts with 'fastmcp'
+    # This supports FastMCPOpenAPI, FastMCPProxy, and other subclasses
     return (
-        class_name == "FastMCP" and 
+        "FastMCP" in class_name and
         module_name.startswith("fastmcp") and
-        hasattr(server, "_mcp_server") and 
+        hasattr(server, "_mcp_server") and
         hasattr(server, "_tool_manager")
     )
 
 def is_official_fastmcp_server(server: Any) -> bool:
     """Check if the server is an official FastMCP instance.
-    
+
     Official FastMCP comes from the 'mcp.server.fastmcp' module.
+    Supports FastMCP subclasses like FastMCPOpenAPI, FastMCPProxy, etc.
     """
     # Check by class name and module
     class_name = server.__class__.__name__
     module_name = server.__class__.__module__
-
-    # Official FastMCP has class name 'FastMCP' and module 'mcp.server.fastmcp'
+
+    # Official FastMCP has class name containing 'FastMCP' and module 'mcp.server.fastmcp'
+    # This supports FastMCPOpenAPI, FastMCPProxy, and other subclasses
     return (
-        class_name == "FastMCP" and 
+        "FastMCP" in class_name and
         module_name.startswith("mcp.server.fastmcp") and
-        hasattr(server, "_mcp_server") and 
+        hasattr(server, "_mcp_server") and
         hasattr(server, "_tool_manager")
     )
 

src/mcpcat/modules/constants.py

@@ -3,3 +3,4 @@
 SESSION_ID_PREFIX = "ses"
 EVENT_ID_PREFIX = "evt"
 MCPCAT_API_URL = "https://api.mcpcat.io"  # Default API URL for MCPCat events
+DEFAULT_CONTEXT_DESCRIPTION = "Describe why you are calling this tool and how it fits into your overall task"

src/mcpcat/modules/context_parameters.py

@@ -3,7 +3,9 @@
 from typing import Any
 
 
-def add_context_parameter_to_tools(tools: list[dict[str, Any]]) -> list[dict[str, Any]]:
+def add_context_parameter_to_tools(
+    tools: list[dict[str, Any]], custom_context_description: str
+) -> list[dict[str, Any]]:
     """Add context parameter to tool schemas."""
     modified_tools = []
 
@@ -13,15 +15,17 @@ def add_context_parameter_to_tools(tools: list[dict[str, Any]]) -> list[dict[str
 
         if "inputSchema" in modified_tool:
             modified_tool["inputSchema"] = add_context_parameter_to_schema(
-                modified_tool["inputSchema"]
+                modified_tool["inputSchema"], custom_context_description
             )
 
         modified_tools.append(modified_tool)
 
     return modified_tools
 
 
-def add_context_parameter_to_schema(schema: dict[str, Any]) -> dict[str, Any]:
+def add_context_parameter_to_schema(
+    schema: dict[str, Any], custom_context_description: str
+) -> dict[str, Any]:
     """Add context parameter to a JSON schema."""
     # Create a copy to avoid modifying original
     modified_schema = schema.copy()
@@ -36,7 +40,7 @@ def add_context_parameter_to_schema(schema: dict[str, Any]) -> dict[str, Any]:
     # Add context parameter
     modified_schema["properties"]["context"] = {
         "type": "string",
-        "description": "Describe why you are calling this tool and how it fits into your overall task",
+        "description": custom_context_description,
     }
 
     # Add to required fields

src/mcpcat/modules/overrides/community/tool_manager.py

@@ -79,6 +79,7 @@ async def get_more_tools(context: str | None = "") -> str:
 def patch_existing_tools(server: FastMCP) -> None:
     """Modify existing tools to include the context parameter."""
     try:
+        data = get_server_tracking_data(server._mcp_server)
         tool_manager = server._tool_manager
         if not hasattr(tool_manager, "_tools"):
             write_to_log("No _tools dictionary found on tool manager")
@@ -102,7 +103,7 @@ def patch_existing_tools(server: FastMCP) -> None:
             # Always overwrite the context property with MCPCat's version
             tool.parameters["properties"]["context"] = {
                 "type": "string",
-                "description": "Describe why you are calling this tool and how it fits into your overall task"
+                "description": data.options.custom_context_description,
             }
 
             # Add to required array
@@ -168,7 +169,7 @@ def patched_add_tool(tool: Any) -> Any:
                         # Always overwrite the context property with MCPCat's version
                         tool.parameters["properties"]["context"] = {
                             "type": "string",
-                            "description": "Describe why you are calling this tool and how it fits into your overall task"
+                            "description": data.options.custom_context_description
                         }
 
                         # Add to required array

src/mcpcat/modules/overrides/mcp_server.py

@@ -120,7 +120,7 @@ async def wrapped_list_tools_handler(request: ListToolsRequest) -> ServerResult:
 
                         tool.inputSchema["properties"]["context"] = {
                             "type": "string",
-                            "description": "Describe why you are calling this tool and how it fits into your overall task",
+                            "description": data.options.custom_context_description,
                         }
 
                         # Add context to required array if it exists

src/mcpcat/modules/overrides/official/monkey_patch.py

@@ -444,7 +444,7 @@ async def get_more_tools_fn(context: str) -> Any:
 
                                     tool.parameters["properties"]["context"] = {
                                         "type": "string",
-                                        "description": "Describe why you are calling this tool and how it fits into your overall task",
+                                        "description": current_data.options.custom_context_description,
                                     }
 
                                     # Add to required array

src/mcpcat/types.py

@@ -8,6 +8,8 @@
 from mcpcat_api import PublishEventRequest
 from pydantic import BaseModel
 
+from mcpcat.modules.constants import DEFAULT_CONTEXT_DESCRIPTION
+
 # Type alias for identify function
 IdentifyFunction = Callable[[dict[str, Any], Any], Optional["UserIdentity"]]
 # Type alias for redaction function
@@ -119,6 +121,7 @@ class MCPCatOptions:
     enable_report_missing: bool = True
     enable_tracing: bool = True
     enable_tool_call_context: bool = True
+    custom_context_description: str = DEFAULT_CONTEXT_DESCRIPTION
     identify: IdentifyFunction | None = None
     redact_sensitive_information: RedactionFunction | None = None
     exporters: dict[str, ExporterConfig] | None = None