misc testing

Author: hallvictoria

azure/functions/decorators/function_app.py

@@ -43,7 +43,7 @@
     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, MCPToolContext, _TYPE_MAPPING, _extract_type_and_description, _get_user_function
 from .retry_policy import RetryPolicy
 from .function_name import FunctionName
 from .warmup import WarmUpTrigger
@@ -473,7 +473,8 @@ def mcp_tool(self) -> Callable[[Callable], Callable]:
         - Extracts parameters and types for tool properties
         - Handles MCPToolContext injection
         """
-        def decorator(target_func: Callable) -> Callable:
+        def decorator(user_func: Callable) -> Callable:
+            target_func = _get_user_function(user_func)
             sig = inspect.signature(target_func)
             tool_name = target_func.__name__
             description = (target_func.__doc__ or "").strip().split("\n")[0]

azure/functions/decorators/mcp.py

@@ -1,10 +1,12 @@
 from typing import Optional
 from typing import Any, Dict, Tuple, get_args, get_origin, Annotated
+import logging
 
 from azure.functions.decorators.constants import (
     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 = {
@@ -46,3 +48,28 @@ 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