From 968e09f0b50d17f7c591baa977666b991a1e59b7 Mon Sep 17 00:00:00 2001 From: Taylor Mullen Date: Thu, 15 May 2025 23:51:53 -0700 Subject: fix: Ensure filename is available for diff rendering in write-file This commit resolves a bug where the `write-file` operation could fail to render content due to a missing filename. The fix involves: - Ensuring `fileName` is consistently passed to `DiffRenderer.tsx` through `ToolConfirmationMessage.tsx`, `ToolMessage.tsx`, and `useGeminiStream.ts`. - Modifying `edit.ts` and `write-file.ts` to include `fileName` in the `FileDiff` object. - Expanding the `FileDiff` interface in `tools.ts` to include `fileName`. Additionally, this commit enhances the diff rendering by: - Adding syntax highlighting based on file extension in `DiffRenderer.tsx`. - Adding more language mappings to `getLanguageFromExtension` in `DiffRenderer.tsx`. - Added lots of tests for all the above. Fixes https://b.corp.google.com/issues/418125982 --- packages/server/src/tools/write-file.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'packages/server/src/tools/write-file.ts') diff --git a/packages/server/src/tools/write-file.ts b/packages/server/src/tools/write-file.ts index 44794a2d..21178b5b 100644 --- a/packages/server/src/tools/write-file.ts +++ b/packages/server/src/tools/write-file.ts @@ -202,7 +202,7 @@ export class WriteFileTool extends BaseTool { ? `Successfully created and wrote to new file: ${params.file_path}` : `Successfully overwrote file: ${params.file_path}`; - const displayResult: FileDiff = { fileDiff }; + const displayResult: FileDiff = { fileDiff, fileName }; return { llmContent: llmSuccessMessage, -- cgit v1.2.3