Fix

Author: Scriptwonder

MCPForUnity/Editor/Resources/Editor/ToolStates.cs

@@ -49,7 +49,7 @@ public static object HandleCommand(JObject @params)
                         ["is_built_in"] = tool.IsBuiltIn,
                         ["structured_output"] = tool.StructuredOutput,
                         ["requires_polling"] = tool.RequiresPolling,
-                        ["poll_action"] = tool.PollAction,
+                        ["poll_action"] = tool.PollAction ?? "status",
                         ["max_poll_seconds"] = tool.MaxPollSeconds,
                         ["parameters"] = paramsArray
                     });

MCPForUnity/Editor/Services/Transport/Transports/WebSocketTransportClient.cs

@@ -540,7 +540,7 @@ private async Task SendRegisterToolsAsync(CancellationToken token)
                     ["description"] = tool.Description,
                     ["structured_output"] = tool.StructuredOutput,
                     ["requires_polling"] = tool.RequiresPolling,
-                    ["poll_action"] = tool.PollAction,
+                    ["poll_action"] = tool.PollAction ?? "status",
                     ["max_poll_seconds"] = tool.MaxPollSeconds,
                     ["group"] = string.IsNullOrWhiteSpace(tool.Group) ? "core" : tool.Group
                 };

Server/src/services/tools/__init__.py

@@ -193,7 +193,7 @@ async def sync_tool_visibility_from_unity(
                                 name=p.get("name", ""),
                                 description=p.get("description", ""),
                                 type=p.get("type", "string"),
-                                required=p.get("required", False),
+                                required=p.get("required", True),
                                 default_value=p.get("default_value"),
                             )
                             for p in td.get("parameters", [])
@@ -204,7 +204,7 @@ async def sync_tool_visibility_from_unity(
                                 description=td.get("description", ""),
                                 structured_output=td.get("structured_output", True),
                                 requires_polling=td.get("requires_polling", False),
-                                poll_action=td.get("poll_action", "status"),
+                                poll_action=td.get("poll_action") or "status",
                                 max_poll_seconds=td.get("max_poll_seconds", 0),
                                 parameters=params,
                             )

Server/src/transport/legacy/unity_connection.py

@@ -945,19 +945,23 @@ async def async_send_command_with_retry(
         # After a successful command, check if the connection was freshly
         # established (reconnection after domain reload).  If so, re-sync
         # tool visibility and custom tool registration from Unity.
-        # Skip if *this* call is the sync itself to avoid recursion.
-        if command_type != "get_tool_states":
-            try:
-                pool = get_unity_connection_pool()
-                conn = pool.get_connection(instance_id)
-                if getattr(conn, "_needs_tool_resync", False):
-                    conn._needs_tool_resync = False
+        # Always clear the flag, but only schedule the background resync
+        # when this call is not itself get_tool_states (to avoid recursion).
+        try:
+            pool = get_unity_connection_pool()
+            conn = pool.get_connection(instance_id)
+            if getattr(conn, "_needs_tool_resync", False):
+                conn._needs_tool_resync = False
+                if command_type != "get_tool_states":
                     logger.info(
                         "Detected reconnection to Unity; scheduling tool re-sync"
                     )
                     asyncio.ensure_future(_resync_tools_after_reconnect(instance_id))
-            except Exception:
-                pass  # Best-effort; don't fail the actual command
+        except Exception as exc:
+            logger.debug(
+                "Failed to schedule post-reconnection tool re-sync: %s",
+                exc,
+            )
 
         return result
     except Exception as e:

Server/tests/integration/test_stdio_custom_tool_sync.py

@@ -240,8 +240,8 @@ async def test_no_resync_for_get_tool_states():
         from transport.legacy.unity_connection import async_send_command_with_retry
         await async_send_command_with_retry("get_tool_states", {})
 
-    # Flag should NOT be cleared — get_tool_states is excluded
-    assert mock_conn._needs_tool_resync is True
+    # Flag should be cleared, but no re-sync task should be scheduled
+    assert mock_conn._needs_tool_resync is False
     mock_resync.assert_not_awaited()