update for bug fix

Author: Scriptwonder

MCPForUnity/Editor/Services/PackageJobManager.cs

@@ -129,8 +129,7 @@ private static void TryRecoverJob(PackageJob job, long nowMs)
             {
                 string packageName = ExtractPackageName(job.Package);
                 var allPackages = PackageInfo.GetAllRegisteredPackages();
-                var info = allPackages.FirstOrDefault(p =>
-                    string.Equals(p.name, packageName, StringComparison.OrdinalIgnoreCase));
+                var info = FindPackageInfo(allPackages, packageName, job.Package);
 
                 if (job.Operation == "add")
                 {
@@ -177,6 +176,34 @@ private static void TryRecoverJob(PackageJob job, long nowMs)
             }
         }
 
+        /// <summary>
+        /// Find a PackageInfo by name, falling back to packageId or git/local source for non-standard identifiers.
+        /// </summary>
+        private static PackageInfo FindPackageInfo(PackageInfo[] allPackages, string packageName, string originalIdentifier)
+        {
+            // Direct name match (handles normal com.company.package identifiers)
+            var info = allPackages.FirstOrDefault(p =>
+                string.Equals(p.name, packageName, StringComparison.OrdinalIgnoreCase));
+            if (info != null)
+                return info;
+
+            // For git URLs / file: paths, packageName == originalIdentifier and won't match .name.
+            // Try matching by packageId or source (git/local).
+            bool isGitOrFile = originalIdentifier.StartsWith("http", StringComparison.OrdinalIgnoreCase)
+                               || originalIdentifier.StartsWith("git", StringComparison.OrdinalIgnoreCase)
+                               || originalIdentifier.StartsWith("file:", StringComparison.OrdinalIgnoreCase)
+                               || originalIdentifier.EndsWith(".git", StringComparison.OrdinalIgnoreCase);
+
+            if (!isGitOrFile)
+                return null;
+
+            return allPackages.FirstOrDefault(p =>
+                p.source == PackageSource.Git || p.source == PackageSource.Local
+                    ? p.packageId != null && p.packageId.Contains(originalIdentifier)
+                      || p.resolvedPath != null && p.resolvedPath.Contains(originalIdentifier)
+                    : false);
+        }
+
         internal static string ExtractPackageName(string packageIdentifier)
         {
             if (string.IsNullOrEmpty(packageIdentifier))

MCPForUnity/Editor/Tools/ManagePackages.cs

@@ -38,35 +38,42 @@ public static object HandleCommand(JObject @params)
 
             string action = actionResult.Value.ToLowerInvariant();
 
-            switch (action)
-            {
-                case "add_package":
-                    return AddPackage(p);
-                case "remove_package":
-                    return RemovePackage(p);
-                case "status":
-                    return GetStatus(p);
-                case "list_packages":
-                    return ListPackages(p);
-                case "search_packages":
-                    return SearchPackages(p);
-                case "get_package_info":
-                    return GetPackageInfo(p);
-                case "list_registries":
-                    return ListRegistries();
-                case "add_registry":
-                    return AddRegistry(p);
-                case "remove_registry":
-                    return RemoveRegistry(p);
-                case "embed_package":
-                    return EmbedPackage(p);
-                case "resolve_packages":
-                    return ResolvePackages();
-                case "ping":
-                    return Ping();
-                default:
-                    return new ErrorResponse(
-                        $"Unknown action: '{action}'. Supported actions: add_package, remove_package, status, list_packages, search_packages, get_package_info, list_registries, add_registry, remove_registry, embed_package, resolve_packages, ping.");
+            try
+            {
+                switch (action)
+                {
+                    case "add_package":
+                        return AddPackage(p);
+                    case "remove_package":
+                        return RemovePackage(p);
+                    case "status":
+                        return GetStatus(p);
+                    case "list_packages":
+                        return ListPackages(p);
+                    case "search_packages":
+                        return SearchPackages(p);
+                    case "get_package_info":
+                        return GetPackageInfo(p);
+                    case "list_registries":
+                        return ListRegistries();
+                    case "add_registry":
+                        return AddRegistry(p);
+                    case "remove_registry":
+                        return RemoveRegistry(p);
+                    case "embed_package":
+                        return EmbedPackage(p);
+                    case "resolve_packages":
+                        return ResolvePackages();
+                    case "ping":
+                        return Ping();
+                    default:
+                        return new ErrorResponse(
+                            $"Unknown action: '{action}'. Supported actions: add_package, remove_package, status, list_packages, search_packages, get_package_info, list_registries, add_registry, remove_registry, embed_package, resolve_packages, ping.");
+                }
+            }
+            catch (Exception ex)
+            {
+                return new ErrorResponse(ex.Message, new { stackTrace = ex.StackTrace });
             }
         }
 
@@ -391,7 +398,7 @@ private static object ListRegistries()
                 }).ToArray();
 
                 return new SuccessResponse(
-                    $"Found {result.Length} scoped registr(y/ies).",
+                    $"Found {result.Length} scoped {(result.Length == 1 ? "registry" : "registries")}.",
                     new { registries = result, count = result.Length }
                 );
             }

Server/src/cli/commands/packages.py

@@ -141,7 +141,11 @@ def embed_package(package: str):
     result = run_command("manage_packages", {"action": "embed_package", "package": package}, config)
     click.echo(format_output(result, config.format))
     if result.get("success"):
-        print_success(f"Package embedded: {package}")
+        job_id = (result.get("data") or {}).get("job_id")
+        if job_id:
+            print_info(f"Embedding started. Poll with: unity-mcp packages status {job_id}")
+        else:
+            print_success(f"Package embedded: {package}")
 
 
 @packages.command("resolve")

Server/src/services/tools/manage_packages.py

@@ -11,7 +11,7 @@
 ALL_ACTIONS = [
     "list_packages", "search_packages", "get_package_info", "ping", "status",
     "add_package", "remove_package", "embed_package", "resolve_packages",
-    "add_registry", "remove_registry",
+    "add_registry", "remove_registry", "list_registries",
 ]
 
 
@@ -35,11 +35,12 @@ async def _send_packages_command(
         "- search_packages: Search Unity registry by keyword\n"
         "- get_package_info: Get details about a specific installed package\n"
         "- ping: Check package manager availability\n"
-        "- status: Poll async job status (for add/remove/list/search operations)\n\n"
+        "- status: Poll async job status (job_id required for list/search; optional for add/remove/embed)\n\n"
         "INSTALL/REMOVE:\n"
         "- add_package: Install a package (name, name@version, git URL, or file: path)\n"
         "- remove_package: Remove a package (checks dependents; use force=true to override)\n\n"
         "REGISTRIES:\n"
+        "- list_registries: List all scoped registries\n"
         "- add_registry: Add a scoped registry (e.g., OpenUPM)\n"
         "- remove_registry: Remove a scoped registry\n\n"
         "UTILITY:\n"

unity-mcp-skill/references/tools-reference.md

@@ -1086,7 +1086,7 @@ Manage Unity packages: query, install, remove, embed, and configure registries.
 | `search_packages` | `query` | Search Unity registry by keyword (async, returns job_id) |
 | `get_package_info` | `package` | Get details about a specific installed package |
 | `ping` | — | Check package manager availability, Unity version, package count |
-| `status` | `job_id` (optional) | Poll async job status; omit job_id for latest job |
+| `status` | `job_id` (required for list/search; optional for add/remove/embed) | Poll async job status; omit job_id to poll latest add/remove/embed job |
 
 **Mutating Actions:**