create MCPToolContext class

Author: hallvictoria

azure/functions/__init__.py

@@ -10,7 +10,7 @@
                          DecoratorApi, DataType, AuthLevel,
                          Cardinality, AccessRights, HttpMethod,
                          AsgiFunctionApp, WsgiFunctionApp,
-                         ExternalHttpFunctionApp, BlobSource, MCPToolContext)
+                         ExternalHttpFunctionApp, BlobSource)
 from ._durable_functions import OrchestrationContext, EntityContext
 from .decorators.function_app import (FunctionRegister, TriggerApi,
                                       BindingApi, SettingsApi)
@@ -19,6 +19,7 @@
 from ._http_wsgi import WsgiMiddleware
 from ._http_asgi import AsgiMiddleware
 from .kafka import KafkaEvent, KafkaConverter, KafkaTriggerConverter
+from .mcp import MCPToolContext
 from .meta import get_binding_registry
 from ._queue import QueueMessage
 from ._servicebus import ServiceBusMessage
@@ -32,6 +33,7 @@
 from . import eventhub  # NoQA
 from . import http  # NoQA
 from . import kafka # NoQA
+from . import mcp  # NoQA
 from . import queue  # NoQA
 from . import servicebus  # NoQA
 from . import timer  # NoQA

azure/functions/decorators/__init__.py

@@ -4,7 +4,7 @@
 from .function_app import FunctionApp, Function, DecoratorApi, DataType, \
     AuthLevel, Blueprint, ExternalHttpFunctionApp, AsgiFunctionApp, \
     WsgiFunctionApp, FunctionRegister, TriggerApi, BindingApi, \
-    SettingsApi, BlobSource, MCPToolContext
+    SettingsApi, BlobSource
 from .http import HttpMethod
 
 __all__ = [
@@ -24,6 +24,5 @@
     'Cardinality',
     'AccessRights',
     'HttpMethod',
-    'BlobSource',
-    'MCPToolContext'
+    'BlobSource'
 ]

azure/functions/decorators/function_app.py

@@ -43,10 +43,11 @@
     AssistantQueryInput, AssistantPostInput, InputType, EmbeddingsInput, \
     semantic_search_system_prompt, \
     SemanticSearchInput, EmbeddingsStoreOutput
-from .mcp import MCPToolTrigger, MCPToolContext, _TYPE_MAPPING, _extract_type_and_description
+from .mcp import MCPToolTrigger, _TYPE_MAPPING, _extract_type_and_description
 from .retry_policy import RetryPolicy
 from .function_name import FunctionName
 from .warmup import WarmUpTrigger
+from ..mcp import MCPToolContext
 from .._http_asgi import AsgiMiddleware
 from .._http_wsgi import WsgiMiddleware, Context
 from azure.functions.decorators.mysql import MySqlInput, MySqlOutput, \
@@ -502,7 +503,7 @@ def decorator(fb: FunctionBuilder) -> FunctionBuilder:
                     "description": param_desc,
                 })
 
-            tool_properties_json = json.dumps(tool_properties)\
+            tool_properties_json = json.dumps(tool_properties)
 
             bound_params = [
                 inspect.Parameter(name, inspect.Parameter.POSITIONAL_OR_KEYWORD)

azure/functions/decorators/mcp.py

@@ -6,7 +6,6 @@
     MCP_TOOL_TRIGGER
 )
 from azure.functions.decorators.core import Trigger, DataType
-from azure.functions.decorators.function_app import FunctionBuilder
 
 # Mapping Python types to MCP property types
 _TYPE_MAPPING = {
@@ -34,10 +33,6 @@ def __init__(self,
         self.tool_properties = tool_properties
         super().__init__(name=name, data_type=data_type)
 
-# MCP-specific context object
-class MCPToolContext(Dict[str, Any]):
-    """Injected context object for MCP tool triggers."""
-    pass
 
 # Helper to extract actual type and description from Annotated types
 def _extract_type_and_description(param_name: str, type_hint: Any) -> Tuple[Any, str]:
@@ -48,28 +43,3 @@ def _extract_type_and_description(param_name: str, type_hint: Any) -> Tuple[Any,
         param_description = next((a for a in args[1:] if isinstance(a, str)), f"The {param_name} parameter.")
         return actual_type, param_description
     return type_hint, f"The {param_name} parameter."
-
-def _get_user_function(target_func):
-    """
-    Unwraps decorated or builder-wrapped functions to find the original
-    user-defined function (the one starting with 'def' or 'async def').
-    """
-    logging.info("HELLO FROM THE SDK")
-    # Case 1: It's a FunctionBuilder object
-    if isinstance(target_func, FunctionBuilder):
-        # Access the internal user function
-        try:
-            return target_func._function.get_user_function()
-        except AttributeError:
-            pass
-
-    # Case 2: It's already the user-defined function
-    if callable(target_func) and hasattr(target_func, "__name__"):
-        return target_func
-
-    # Case 3: It might be a partially wrapped callable
-    if hasattr(target_func, "__wrapped__"):
-        return _get_user_function(target_func.__wrapped__)
-
-    # Default fallback
-    return target_func

azure/functions/mcp.py

@@ -3,12 +3,18 @@
 from . import meta
 
 
+# MCP-specific context object
+class MCPToolContext(typing.Dict[str, typing.Any]):
+    """Injected context object for MCP tool triggers."""
+    pass
+
+
 class MCPToolTriggerConverter(meta.InConverter, binding='mcpToolTrigger',
                               trigger=True):
 
     @classmethod
     def check_input_type_annotation(cls, pytype: type) -> bool:
-        return issubclass(pytype, (str, dict, bytes))
+        return issubclass(pytype, (str, dict, bytes, MCPToolContext))
 
     @classmethod
     def has_implicit_output(cls) -> bool: