update to new package api

Author: eleanorjboyd

src/features/copilotTools.ts

@@ -9,6 +9,7 @@ import {
     Uri,
 } from 'vscode';
 import {
+    PackageManagementOptions,
     PythonEnvironment,
     PythonEnvironmentExecutionInfo,
     PythonPackageGetterApi,
@@ -205,7 +206,10 @@ export class InstallPackageTool implements LanguageModelTool<IInstallPackageInpu
             }
 
             // Install the packages
-            await this.api.installPackages(environment, parameters.packageList);
+            const pkgManagementOptions: PackageManagementOptions = {
+                install: parameters.packageList,
+            };
+            await this.api.managePackages(environment, pkgManagementOptions);
             const resultMessage = `Successfully installed ${packagePlurality}: ${parameters.packageList.join(', ')}`;
 
             deferredReturn.resolve({

src/test/copilotTools.unit.test.ts

@@ -35,9 +35,6 @@ suite('InstallPackageTool Tests', () => {
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         mockEnvironment.setup((x: any) => x.then).returns(() => undefined);
 
-        // refresh will always return a resolved promise
-        mockApi.setup((x) => x.refreshPackages(typeMoq.It.isAny())).returns(() => Promise.resolve());
-
         // Create an instance of InstallPackageTool with the mock functions
         installPackageTool = new InstallPackageTool(mockApi.object);
     });
@@ -114,8 +111,6 @@ suite('InstallPackageTool Tests', () => {
                 return Promise.resolve(mockEnvironment.object);
             });
 
-        mockApi.setup((x) => x.refreshPackages(typeMoq.It.isAny())).returns(() => Promise.resolve());
-
         const options = { input: testFile, toolInvocationToken: undefined };
         const tokenSource = new vscode.CancellationTokenSource();
         const token = tokenSource.token;
@@ -145,9 +140,8 @@ suite('InstallPackageTool Tests', () => {
                 return Promise.resolve(mockEnvironment.object);
             });
 
-        mockApi.setup((x) => x.refreshPackages(typeMoq.It.isAny())).returns(() => Promise.resolve());
         mockApi
-            .setup((x) => x.installPackages(typeMoq.It.isAny(), typeMoq.It.isAny()))
+            .setup((x) => x.managePackages(typeMoq.It.isAny(), typeMoq.It.isAny()))
             .returns(() => {
                 const deferred = createDeferred<void>();
                 deferred.resolve();
@@ -177,9 +171,8 @@ suite('InstallPackageTool Tests', () => {
                 return Promise.resolve(mockEnvironment.object);
             });
 
-        mockApi.setup((x) => x.refreshPackages(typeMoq.It.isAny())).returns(() => Promise.resolve());
         mockApi
-            .setup((x) => x.installPackages(typeMoq.It.isAny(), typeMoq.It.isAny()))
+            .setup((x) => x.managePackages(typeMoq.It.isAny(), typeMoq.It.isAny()))
             .returns(() => {
                 const deferred = createDeferred<void>();
                 deferred.reject(new Error('Installation failed'));
@@ -228,9 +221,8 @@ suite('InstallPackageTool Tests', () => {
             .returns(async () => {
                 return Promise.resolve(mockEnvironment.object);
             });
-        mockApi.setup((x) => x.refreshPackages(typeMoq.It.isAny())).returns(() => Promise.resolve());
         mockApi
-            .setup((x) => x.installPackages(typeMoq.It.isAny(), typeMoq.It.isAny()))
+            .setup((x) => x.managePackages(typeMoq.It.isAny(), typeMoq.It.isAny()))
             .returns(() => {
                 const deferred = createDeferred<void>();
                 deferred.resolve();