Apply suggestions from code review

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

Author: edvilme

scripts/compare_package_versions.py

@@ -63,7 +63,7 @@ def main() -> int:
     if extension_version != api_version:
         print(
             f"::error::Version mismatch: package.json is {extension_version} but "
-            f"api/package.json is {api_version}. Update api/package.json to match."
+            f"api/package.json is {api_version}. Update package.json and/or api/package.json so both versions match."
         )
         return 1
 

src/api.ts

@@ -1421,15 +1421,13 @@ export namespace PythonEnvironments {
     /**
      * Returns the API exposed by the Python Environments extension in VS Code.
      */
-    export async function api(): Promise<PythonEnvironmentApi> {
-        const extension = extensions.getExtension(EXTENSION_ID);
-        if (extension === undefined) {
-            throw new Error(`Python Environments extension is not installed or is disabled`);
-        }
-        if (!extension.isActive) {
-            await extension.activate();
-        }
-        const pythonEnvsApi: PythonEnvironmentApi = extension.exports;
-        return pythonEnvsApi;
+export async function api(): Promise<PythonEnvironmentApi> {
+    const extension = extensions.getExtension<PythonEnvironmentApi>(EXTENSION_ID);
+    if (extension === undefined) {
+        throw new Error(`Python Environments extension (${EXTENSION_ID}) is not installed or is disabled`);
     }
+    if (!extension.isActive) {
+        await extension.activate();
+    }
+    return extension.exports;
 }