feat: Implement async method to retrieve enabled tools on the main thread

Author: Jordonh18

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

@@ -8,9 +8,11 @@
 using System.Threading.Tasks;
 using MCPForUnity.Editor.Config;
 using MCPForUnity.Editor.Helpers;
+using MCPForUnity.Editor.Services;
 using MCPForUnity.Editor.Services.Transport;
 using Newtonsoft.Json;
 using Newtonsoft.Json.Linq;
+using UnityEditor;
 using UnityEngine;
 
 namespace MCPForUnity.Editor.Services.Transport.Transports
@@ -65,6 +67,26 @@ public WebSocketTransportClient(IToolDiscoveryService toolDiscoveryService = nul
         public string TransportName => TransportDisplayName;
         public TransportState State => _state;
 
+        private Task<List<ToolMetadata>> GetEnabledToolsOnMainThreadAsync()
+        {
+            var tcs = new TaskCompletionSource<List<ToolMetadata>>(TaskCreationOptions.RunContinuationsAsynchronously);
+
+            EditorApplication.delayCall += () =>
+            {
+                try
+                {
+                    var tools = _toolDiscoveryService?.GetEnabledTools() ?? new List<ToolMetadata>();
+                    tcs.TrySetResult(tools);
+                }
+                catch (Exception ex)
+                {
+                    tcs.TrySetException(ex);
+                }
+            };
+
+            return tcs.Task;
+        }
+
         public async Task<bool> StartAsync()
         {
             // Capture identity values on the main thread before any async context switching
@@ -421,7 +443,9 @@ private async Task SendRegisterToolsAsync(CancellationToken token)
         {
             if (_toolDiscoveryService == null) return;
 
-            var tools = _toolDiscoveryService.GetEnabledTools();
+            token.ThrowIfCancellationRequested();
+            var tools = await GetEnabledToolsOnMainThreadAsync().ConfigureAwait(false);
+            token.ThrowIfCancellationRequested();
             McpLog.Info($"[WebSocket] Preparing to register {tools.Count} tool(s) with the bridge.");
             var toolsArray = new JArray();