Skip to content

Commit 48a0df6

Browse files
committed
Updatges
1 parent 31836c0 commit 48a0df6

File tree

2 files changed

+5
-14
lines changed

2 files changed

+5
-14
lines changed

src/client/chat/configurePythonEnvTool.ts

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,6 @@ import {
2626
import { resolveFilePath } from './utils';
2727
import { ITerminalHelper } from '../common/terminal/types';
2828
import { IRecommendedEnvironmentService } from '../interpreter/configuration/types';
29-
import { IDiscoveryAPI } from '../pythonEnvironments/base/locator';
3029
import { CreateVirtualEnvTool } from './createVirtualEnvTool';
3130
import { SelectPythonEnvTool } from './selectEnvTool';
3231

@@ -36,9 +35,9 @@ export class ConfigurePythonEnvTool implements LanguageModelTool<IResourceRefere
3635
private readonly recommendedEnvService: IRecommendedEnvironmentService;
3736
public static readonly toolName = 'configure_python_environment';
3837
constructor(
39-
private readonly discoveryApi: IDiscoveryAPI,
4038
private readonly api: PythonExtension['environments'],
4139
private readonly serviceContainer: IServiceContainer,
40+
private readonly createVirtualEnvTool: CreateVirtualEnvTool,
4241
) {
4342
this.terminalExecutionService = this.serviceContainer.get<TerminalCodeExecutionProvider>(
4443
ICodeExecutionService,
@@ -78,11 +77,7 @@ export class ConfigurePythonEnvTool implements LanguageModelTool<IResourceRefere
7877

7978
let reason: 'cancelled' | undefined;
8079
if (
81-
// eslint-disable-next-line @typescript-eslint/no-use-before-define
82-
await new CreateVirtualEnvTool(this.discoveryApi, this.api, this.serviceContainer).canCreateNewVirtualEnv(
83-
resolveFilePath(options.input.resourcePath),
84-
token,
85-
)
80+
await this.createVirtualEnvTool.canCreateNewVirtualEnv(resolveFilePath(options.input.resourcePath), token)
8681
) {
8782
reason = 'cancelled';
8883
try {

src/client/chat/index.ts

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -44,19 +44,15 @@ export function registerTools(
4444
new InstallPackagesTool(environmentsApi, serviceContainer, discoverApi),
4545
),
4646
);
47-
ourTools.add(
48-
lm.registerTool(
49-
CreateVirtualEnvTool.toolName,
50-
new CreateVirtualEnvTool(discoverApi, environmentsApi, serviceContainer),
51-
),
52-
);
47+
const createVirtualEnvTool = new CreateVirtualEnvTool(discoverApi, environmentsApi, serviceContainer);
48+
ourTools.add(lm.registerTool(CreateVirtualEnvTool.toolName, createVirtualEnvTool));
5349
ourTools.add(
5450
lm.registerTool(SelectPythonEnvTool.toolName, new SelectPythonEnvTool(environmentsApi, serviceContainer)),
5551
);
5652
ourTools.add(
5753
lm.registerTool(
5854
ConfigurePythonEnvTool.toolName,
59-
new ConfigurePythonEnvTool(discoverApi, environmentsApi, serviceContainer),
55+
new ConfigurePythonEnvTool(environmentsApi, serviceContainer, createVirtualEnvTool),
6056
),
6157
);
6258
ourTools.add(

0 commit comments

Comments
 (0)