Add VenvUv support and alwaysUseUv setting

Co-authored-by: eleanorjboyd <26030610+eleanorjboyd@users.noreply.github.com>

Author: Copilot

package.json

@@ -127,6 +127,12 @@
                     "items": {
                         "type": "string"
                     }
+                },
+                "python-envs.alwaysUseUv": {
+                    "type": "boolean",
+                    "description": "%python-envs.alwaysUseUv.description%",
+                    "default": true,
+                    "scope": "machine"
                 }
             }
         },

package.nls.json

@@ -39,5 +39,6 @@
     "python-envs.terminal.deactivate.title": "Deactivate Environment in Current Terminal",
     "python-envs.uninstallPackage.title": "Uninstall Package",
     "python-envs.revealProjectInExplorer.title": "Reveal Project in Explorer",
-    "python-envs.runPetInTerminal.title": "Run Python Environment Tool (PET) in Terminal..."
+    "python-envs.runPetInTerminal.title": "Run Python Environment Tool (PET) in Terminal...",
+    "python-envs.alwaysUseUv.description": "When set to true, uv will be used to manage all virtual environments if available. When set to false, uv will only manage virtual environments explicitly created by uv."
 }

src/managers/builtin/helpers.ts

@@ -3,6 +3,8 @@ import { CancellationError, CancellationToken, LogOutputChannel } from 'vscode';
 import { createDeferred } from '../../common/utils/deferred';
 import { sendTelemetryEvent } from '../../common/telemetry/sender';
 import { EventNames } from '../../common/telemetry/constants';
+import { getConfiguration } from '../../common/workspace.apis';
+import { NativePythonEnvironmentKind } from '../common/nativePythonFinder';
 
 const available = createDeferred<boolean>();
 export async function isUvInstalled(log?: LogOutputChannel): Promise<boolean> {
@@ -24,6 +26,35 @@ export async function isUvInstalled(log?: LogOutputChannel): Promise<boolean> {
     return available.promise;
 }
 
+/**
+ * Determines if uv should be used for managing a virtual environment.
+ * @param envKind - The kind of environment (Venv, VenvUv, etc.)
+ * @param log - Optional log output channel
+ * @returns True if uv should be used, false otherwise
+ */
+export async function shouldUseUv(
+    envKind?: NativePythonEnvironmentKind,
+    log?: LogOutputChannel,
+): Promise<boolean> {
+    // If the environment is explicitly a VenvUv type, always use uv
+    if (envKind === NativePythonEnvironmentKind.venvUv) {
+        return await isUvInstalled(log);
+    }
+
+    // Check the alwaysUseUv setting
+    const config = getConfiguration('python-envs');
+    const alwaysUseUv = config.get<boolean>('alwaysUseUv', true);
+
+    // If alwaysUseUv is true and uv is installed, use it
+    if (alwaysUseUv) {
+        return await isUvInstalled(log);
+    }
+
+    // Otherwise, only use uv for VenvUv environments (already handled above)
+    return false;
+}
+
+
 export async function runUV(
     args: string[],
     cwd?: string,

src/managers/builtin/utils.ts

@@ -18,7 +18,7 @@ import {
     NativePythonFinder,
 } from '../common/nativePythonFinder';
 import { shortVersion, sortEnvironments } from '../common/utils';
-import { isUvInstalled, runPython, runUV } from './helpers';
+import { shouldUseUv, runPython, runUV } from './helpers';
 import { parsePipList, PipPackage } from './pipListUtils';
 
 function asPackageQuickPickItem(name: string, version?: string): QuickPickItem {
@@ -139,7 +139,7 @@ export async function refreshPythons(
 }
 
 async function refreshPipPackagesRaw(environment: PythonEnvironment, log?: LogOutputChannel): Promise<string> {
-    const useUv = await isUvInstalled();
+    const useUv = await shouldUseUv(undefined, log);
     if (useUv) {
         return await runUV(['pip', 'list', '--python', environment.execInfo.run.executable], undefined, log);
     }
@@ -194,7 +194,7 @@ export async function managePackages(
         throw new Error('Python 2.* is not supported (deprecated)');
     }
 
-    const useUv = await isUvInstalled();
+    const useUv = await shouldUseUv(undefined, manager.log);
     const uninstallArgs = ['pip', 'uninstall'];
     if (options.uninstall && options.uninstall.length > 0) {
         if (useUv) {

src/managers/builtin/venvUtils.ts

@@ -25,7 +25,7 @@ import {
     NativePythonFinder,
 } from '../common/nativePythonFinder';
 import { getShellActivationCommands, shortVersion, sortEnvironments } from '../common/utils';
-import { isUvInstalled, runPython, runUV } from './helpers';
+import { shouldUseUv, runPython, runUV } from './helpers';
 import { getProjectInstallable, PipPackages } from './pipUtils';
 import { resolveSystemPythonEnvironmentPath } from './utils';
 import { createStepBasedVenvFlow } from './venvStepBasedFlow';
@@ -176,7 +176,7 @@ export async function findVirtualEnvironments(
     const envs = data
         .filter((e) => isNativeEnvInfo(e))
         .map((e) => e as NativeEnvInfo)
-        .filter((e) => e.kind === NativePythonEnvironmentKind.venv);
+        .filter((e) => e.kind === NativePythonEnvironmentKind.venv || e.kind === NativePythonEnvironmentKind.venvUv);
 
     for (const e of envs) {
         if (!(e.prefix && e.executable && e.version)) {
@@ -290,7 +290,7 @@ export async function createWithProgress(
         async () => {
             const result: CreateEnvironmentResult = {};
             try {
-                const useUv = await isUvInstalled(log);
+                const useUv = await shouldUseUv(undefined, log);
                 // env creation
                 if (basePython.execInfo?.run.executable) {
                     if (useUv) {
@@ -459,7 +459,7 @@ export async function resolveVenvPythonEnvironmentPath(
 ): Promise<PythonEnvironment | undefined> {
     const resolved = await nativeFinder.resolve(fsPath);
 
-    if (resolved.kind === NativePythonEnvironmentKind.venv) {
+    if (resolved.kind === NativePythonEnvironmentKind.venv || resolved.kind === NativePythonEnvironmentKind.venvUv) {
         const envInfo = await getPythonInfo(resolved);
         return api.createPythonEnvironmentItem(envInfo, manager);
     }

src/managers/common/nativePythonFinder.ts

@@ -69,6 +69,7 @@ export enum NativePythonEnvironmentKind {
     linuxGlobal = 'LinuxGlobal',
     macXCode = 'MacXCode',
     venv = 'Venv',
+    venvUv = 'VenvUv',
     virtualEnv = 'VirtualEnv',
     virtualEnvWrapper = 'VirtualEnvWrapper',
     windowsStore = 'WindowsStore',