Merge pull request LykosAI#1206 from ionite34/fix-pkg-resources-for-real

Fix the missing pkg_resources error for real this time

(cherry picked from commit d9a25ac26919aa0ee00d0fe4f2e6f43ce9b86d82)

Author: mohnjiles

StabilityMatrix.Core/Models/Packages/A3WebUI.cs

@@ -226,7 +226,7 @@ torchIndex is TorchIndex.Cuda
         // 1. Configure the entire install process declaratively.
         var config = new PipInstallConfig
         {
-            RequirementsFilePaths = ["requirements_versions.txt", "setuptools<82"],
+            RequirementsFilePaths = ["requirements_versions.txt"],
             TorchVersion = torchIndex == TorchIndex.Mps ? "==2.3.1" : (isBlackwell ? "" : "==2.1.2"),
             TorchvisionVersion = torchIndex == TorchIndex.Mps ? "==0.18.1" : (isBlackwell ? "" : "==0.16.2"),
             XformersVersion = isBlackwell ? " " : "==0.0.23.post1",
@@ -235,7 +235,6 @@ torchIndex is TorchIndex.Cuda
             ExtraPipArgs =
             [
                 "https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip",
-                "setuptools<82",
             ],
         };
 

StabilityMatrix.Core/Models/Packages/BaseGitPackage.cs

@@ -235,6 +235,14 @@ await PyInstallationManager.GetInstallationAsync(pythonVersion.Value).ConfigureA
             await venvRunner.Setup(true, onConsoleOutput).ConfigureAwait(false);
         }
 
+        // Constrain setuptools<82 in uv's isolated build environments.
+        // setuptools 82+ removed pkg_resources, breaking source builds that import it.
+        var buildConstraintsPath = Path.Combine(installedPackagePath, venvName, "uv-build-constraints.txt");
+        await File.WriteAllTextAsync(buildConstraintsPath, "setuptools<82\n").ConfigureAwait(false);
+        venvRunner.UpdateEnvironmentVariables(env =>
+            env.SetItem("UV_BUILD_CONSTRAINT", buildConstraintsPath)
+        );
+
         // ensure pip is installed
         await venvRunner.PipInstall("pip", onConsoleOutput).ConfigureAwait(false);
 

StabilityMatrix.Core/Models/Packages/SDWebForge.cs

@@ -185,11 +185,9 @@ public override async Task InstallPackage(
             torchIndex is TorchIndex.Cuda
             && (SettingsManager.Settings.PreferredGpu?.IsBlackwellGpu() ?? HardwareHelper.HasBlackwellGpu());
 
-        var isAmd = torchIndex is TorchIndex.Rocm;
-
         var config = new PipInstallConfig
         {
-            PrePipInstallArgs = ["joblib", "setuptools<82"],
+            PrePipInstallArgs = ["joblib"],
             RequirementsFilePaths = requirementsPaths,
             TorchVersion = "",
             TorchvisionVersion = "",
@@ -199,7 +197,7 @@ torchIndex is TorchIndex.Cuda
             [
                 "https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip",
             ],
-            PostInstallPipArgs = ["numpy==1.26.4", "setuptools<82"],
+            PostInstallPipArgs = ["numpy==1.26.4"],
         };
 
         await StandardPipInstallProcessAsync(