vrr/qwen-code
2
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2026-05-02 13:40:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

Enable "modify" in write tool (#1044)

Browse source
This commit is contained in:
Leo 2025-06-14 19:20:04 +01:00 • committed by GitHub
parent 93909a2dd3
commit 2c6aae863a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 626 additions and 287 deletions
Download patch file Download diff file Expand all files Collapse all files

37
packages/core/src/tools/write-file.ts
View file

@ -25,6 +25,7 @@ import {
} from '../utils/editCorrector.js';
import { GeminiClient } from '../core/client.js';
import { DEFAULT_DIFF_OPTIONS } from './diffOptions.js';
import { ModifiableTool, ModifyContext } from './modifiable-tool.js';
/**
* Parameters for the WriteFile tool
@ -51,7 +52,10 @@ interface GetCorrectedFileContentResult {
/**
* Implementation of the WriteFile tool logic
*/
export class WriteFileTool extends BaseTool<WriteFileToolParams, ToolResult> {
export class WriteFileTool
extends BaseTool<WriteFileToolParams, ToolResult>
implements ModifiableTool<WriteFileToolParams>
{
static readonly Name: string = 'write_file';
private readonly client: GeminiClient;
@ -336,4 +340,35 @@ export class WriteFileTool extends BaseTool<WriteFileToolParams, ToolResult> {
}
return { originalContent, correctedContent, fileExists };
}
getModifyContext(
abortSignal: AbortSignal,
): ModifyContext<WriteFileToolParams> {
return {
getFilePath: (params: WriteFileToolParams) => params.file_path,
getCurrentContent: async (params: WriteFileToolParams) => {
const correctedContentResult = await this._getCorrectedFileContent(
params.file_path,
params.content,
abortSignal,
);
return correctedContentResult.originalContent;
},
getProposedContent: async (params: WriteFileToolParams) => {
const correctedContentResult = await this._getCorrectedFileContent(
params.file_path,
params.content,
abortSignal,
);
return correctedContentResult.correctedContent;
},
createUpdatedParams: (
modifiedProposedContent: string,
originalParams: WriteFileToolParams,
) => ({
...originalParams,
content: modifiedProposedContent,
}),
};
}
}