fix: convert tool functions to async and await ctx.info calls

Author: msanatan

MCPForUnity/UnityMcpServer~/src/tools/manage_asset.py

@@ -32,7 +32,7 @@ async def manage_asset(
     page_size: Annotated[int, "Page size for pagination"] | None = None,
     page_number: Annotated[int, "Page number for pagination"] | None = None
 ) -> dict[str, Any]:
-    ctx.info(f"Processing manage_asset: {action}")
+    await ctx.info(f"Processing manage_asset: {action}")
     # Ensure properties is a dict if None
     if properties is None:
         properties = {}

MCPForUnity/UnityMcpServer~/src/tools/manage_editor.py

@@ -9,7 +9,7 @@
 @mcp_for_unity_tool(
     description="Controls and queries the Unity editor's state and settings"
 )
-def manage_editor(
+async def manage_editor(
     ctx: Context,
     action: Annotated[Literal["telemetry_status", "telemetry_ping", "play", "pause", "stop", "get_state", "get_project_root", "get_windows",
                               "get_active_tool", "get_selection", "get_prefab_stage", "set_active_tool", "add_tag", "remove_tag", "get_tags", "add_layer", "remove_layer", "get_layers"], "Get and update the Unity Editor state."],
@@ -22,7 +22,7 @@ def manage_editor(
     layer_name: Annotated[str,
                           "Layer name when adding and removing layers"] | None = None,
 ) -> dict[str, Any]:
-    ctx.info(f"Processing manage_editor: {action}")
+    await ctx.info(f"Processing manage_editor: {action}")
     try:
         # Diagnostics: quick telemetry checks
         if action == "telemetry_status":

MCPForUnity/UnityMcpServer~/src/tools/manage_gameobject.py

@@ -8,7 +8,7 @@
 @mcp_for_unity_tool(
     description="Manage GameObjects. Note: for 'get_components', the `data` field contains a dictionary of component names and their serialized properties. For 'get_component', specify 'component_name' to retrieve only that component's serialized data."
 )
-def manage_gameobject(
+async def manage_gameobject(
     ctx: Context,
     action: Annotated[Literal["create", "modify", "delete", "find", "add_component", "remove_component", "set_component_property", "get_components", "get_component"], "Perform CRUD operations on GameObjects and components."],
     target: Annotated[str,
@@ -64,7 +64,7 @@ def manage_gameobject(
     includeNonPublicSerialized: Annotated[bool,
                                           "Controls whether serialization of private [SerializeField] fields is included"] | None = None,
 ) -> dict[str, Any]:
-    ctx.info(f"Processing manage_gameobject: {action}")
+    await ctx.info(f"Processing manage_gameobject: {action}")
     try:
         # Validate parameter usage to prevent silent failures
         if action == "find":

MCPForUnity/UnityMcpServer~/src/tools/manage_menu_item.py

@@ -22,7 +22,7 @@ async def manage_menu_item(
     refresh: Annotated[bool,
                        "Optional flag to force refresh of the menu cache when listing"] | None = None,
 ) -> dict[str, Any]:
-    ctx.info(f"Processing manage_menu_item: {action}")
+    await ctx.info(f"Processing manage_menu_item: {action}")
     # Prepare parameters for the C# handler
     params_dict: dict[str, Any] = {
         "action": action,

MCPForUnity/UnityMcpServer~/src/tools/manage_prefabs.py

@@ -8,7 +8,7 @@
 @mcp_for_unity_tool(
     description="Bridge for prefab management commands (stage control and creation)."
 )
-def manage_prefabs(
+async def manage_prefabs(
     ctx: Context,
     action: Annotated[Literal[
         "open_stage",
@@ -29,7 +29,7 @@ def manage_prefabs(
     search_inactive: Annotated[bool,
                                "Include inactive objects when resolving the target name"] | None = None,
 ) -> dict[str, Any]:
-    ctx.info(f"Processing manage_prefabs: {action}")
+    await ctx.info(f"Processing manage_prefabs: {action}")
     try:
         params: dict[str, Any] = {"action": action}
 

MCPForUnity/UnityMcpServer~/src/tools/manage_scene.py

@@ -6,7 +6,7 @@
 
 
 @mcp_for_unity_tool(description="Manage Unity scenes")
-def manage_scene(
+async def manage_scene(
     ctx: Context,
     action: Annotated[Literal["create", "load", "save", "get_hierarchy", "get_active", "get_build_settings"], "Perform CRUD operations on Unity scenes."],
     name: Annotated[str,
@@ -16,7 +16,7 @@ def manage_scene(
     build_index: Annotated[int,
                            "Build index for load/build settings actions"] | None = None,
 ) -> dict[str, Any]:
-    ctx.info(f"Processing manage_scene: {action}")
+    await ctx.info(f"Processing manage_scene: {action}")
     try:
         # Coerce numeric inputs defensively
         def _coerce_int(value, default=None):

MCPForUnity/UnityMcpServer~/src/tools/manage_script.py

@@ -75,7 +75,7 @@ def _split_uri(uri: str) -> tuple[str, str]:
         - Lines, columns are 1-indexed
         - Tabs count as 1 column"""
 ))
-def apply_text_edits(
+async def apply_text_edits(
     ctx: Context,
     uri: Annotated[str, "URI of the script to edit under Assets/ directory, unity://path/Assets/... or file://... or Assets/..."],
     edits: Annotated[list[dict[str, Any]], "List of edits to apply to the script, i.e. a list of {startLine,startCol,endLine,endCol,newText} (1-indexed!)"],
@@ -86,7 +86,7 @@ def apply_text_edits(
     options: Annotated[dict[str, Any],
                        "Optional options, used to pass additional options to the script editor"] | None = None,
 ) -> dict[str, Any]:
-    ctx.info(f"Processing apply_text_edits: {uri}")
+    await ctx.info(f"Processing apply_text_edits: {uri}")
     name, directory = _split_uri(uri)
 
     # Normalize common aliases/misuses for resilience:
@@ -360,7 +360,7 @@ def create_script(
     script_type: Annotated[str, "Script type (e.g., 'C#')"] | None = None,
     namespace: Annotated[str, "Namespace for the script"] | None = None,
 ) -> dict[str, Any]:
-    ctx.info(f"Processing create_script: {path}")
+    await ctx.info(f"Processing create_script: {path}")
     name = os.path.splitext(os.path.basename(path))[0]
     directory = os.path.dirname(path)
     # Local validation to avoid round-trips on obviously bad input
@@ -396,7 +396,7 @@ def delete_script(
     uri: Annotated[str, "URI of the script to delete under Assets/ directory, unity://path/Assets/... or file://... or Assets/..."]
 ) -> dict[str, Any]:
     """Delete a C# script by URI."""
-    ctx.info(f"Processing delete_script: {uri}")
+    await ctx.info(f"Processing delete_script: {uri}")
     name, directory = _split_uri(uri)
     if not directory or directory.split("/")[0].lower() != "assets":
         return {"success": False, "code": "path_outside_assets", "message": "URI must resolve under 'Assets/'."}
@@ -414,7 +414,7 @@ def validate_script(
     include_diagnostics: Annotated[bool,
                                    "Include full diagnostics and summary"] = False
 ) -> dict[str, Any]:
-    ctx.info(f"Processing validate_script: {uri}")
+    await ctx.info(f"Processing validate_script: {uri}")
     name, directory = _split_uri(uri)
     if not directory or directory.split("/")[0].lower() != "assets":
         return {"success": False, "code": "path_outside_assets", "message": "URI must resolve under 'Assets/'."}
@@ -451,7 +451,7 @@ def manage_script(
                            "Type hint (e.g., 'MonoBehaviour')"] | None = None,
     namespace: Annotated[str, "Namespace for the script"] | None = None,
 ) -> dict[str, Any]:
-    ctx.info(f"Processing manage_script: {action}")
+    await ctx.info(f"Processing manage_script: {action}")
     try:
         # Prepare parameters for Unity
         params = {
@@ -509,7 +509,7 @@ def manage_script(
         - guards: header/using guard enabled flag"""
 ))
 def manage_script_capabilities(ctx: Context) -> dict[str, Any]:
-    ctx.info("Processing manage_script_capabilities")
+    await ctx.info("Processing manage_script_capabilities")
     try:
         # Keep in sync with server/Editor ManageScript implementation
         ops = [
@@ -537,7 +537,7 @@ def get_sha(
     ctx: Context,
     uri: Annotated[str, "URI of the script to edit under Assets/ directory, unity://path/Assets/... or file://... or Assets/..."]
 ) -> dict[str, Any]:
-    ctx.info(f"Processing get_sha: {uri}")
+    await ctx.info(f"Processing get_sha: {uri}")
     try:
         name, directory = _split_uri(uri)
         params = {"action": "get_sha", "name": name, "path": directory}

MCPForUnity/UnityMcpServer~/src/tools/manage_shader.py

@@ -9,15 +9,15 @@
 @mcp_for_unity_tool(
     description="Manages shader scripts in Unity (create, read, update, delete)."
 )
-def manage_shader(
+async def manage_shader(
     ctx: Context,
     action: Annotated[Literal['create', 'read', 'update', 'delete'], "Perform CRUD operations on shader scripts."],
     name: Annotated[str, "Shader name (no .cs extension)"],
     path: Annotated[str, "Asset path (default: \"Assets/\")"],
     contents: Annotated[str,
                         "Shader code for 'create'/'update'"] | None = None,
 ) -> dict[str, Any]:
-    ctx.info(f"Processing manage_shader: {action}")
+    await ctx.info(f"Processing manage_shader: {action}")
     try:
         # Prepare parameters for Unity
         params = {

MCPForUnity/UnityMcpServer~/src/tools/read_console.py

@@ -25,7 +25,7 @@ def read_console(
     include_stacktrace: Annotated[bool,
                                   "Include stack traces in output"] | None = None
 ) -> dict[str, Any]:
-    ctx.info(f"Processing read_console: {action}")
+    await ctx.info(f"Processing read_console: {action}")
     # Set defaults if values are None
     action = action if action is not None else 'get'
     types = types if types is not None else ['error', 'warning', 'log']

MCPForUnity/UnityMcpServer~/src/tools/resource_tools.py

@@ -142,7 +142,7 @@ async def list_resources(
     limit: Annotated[int, "Page limit"] = 200,
     project_root: Annotated[str, "Project path"] | None = None,
 ) -> dict[str, Any]:
-    ctx.info(f"Processing list_resources: {pattern}")
+    await ctx.info(f"Processing list_resources: {pattern}")
     try:
         project = _resolve_project_root(project_root)
         base = (project / under).resolve()
@@ -202,7 +202,7 @@ async def read_resource(
                             "The project root directory"] | None = None,
     request: Annotated[str, "The request ID"] | None = None,
 ) -> dict[str, Any]:
-    ctx.info(f"Processing read_resource: {uri}")
+    await ctx.info(f"Processing read_resource: {uri}")
     try:
         # Serve the canonical spec directly when requested (allow bare or with scheme)
         if uri in ("unity://spec/script-edits", "spec/script-edits", "script-edits"):
@@ -357,7 +357,7 @@ async def find_in_file(
     max_results: Annotated[int,
                            "Cap results to avoid huge payloads"] = 200,
 ) -> dict[str, Any]:
-    ctx.info(f"Processing find_in_file: {uri}")
+    await ctx.info(f"Processing find_in_file: {uri}")
     try:
         project = _resolve_project_root(project_root)
         p = _resolve_safe_path_from_uri(uri, project)