1 files changed, 26 insertions, 0 deletions
diff --git a/packages/core/src/tools/write-file.ts b/packages/core/src/tools/write-file.ts
index b9e07034..b19b00ac 100644
--- a/packages/core/src/tools/write-file.ts
+++ b/packages/core/src/tools/write-file.ts
@@ -26,6 +26,11 @@ import {
 import { GeminiClient } from '../core/client.js';
 import { DEFAULT_DIFF_OPTIONS } from './diffOptions.js';
 import { ModifiableTool, ModifyContext } from './modifiable-tool.js';
+import { getSpecificMimeType } from '../utils/fileUtils.js';
+import {
+  recordFileOperationMetric,
+  FileOperation,
+} from '../telemetry/metrics.js';
 
 /**
  * Parameters for the WriteFile tool
@@ -271,6 +276,27 @@ export class WriteFileTool
 
       const displayResult: FileDiff = { fileDiff, fileName };
 
+      const lines = fileContent.split('\n').length;
+      const mimetype = getSpecificMimeType(params.file_path);
+      const extension = path.extname(params.file_path); // Get extension
+      if (isNewFile) {
+        recordFileOperationMetric(
+          this.config,
+          FileOperation.CREATE,
+          lines,
+          mimetype,
+          extension,
+        );
+      } else {
+        recordFileOperationMetric(
+          this.config,
+          FileOperation.UPDATE,
+          lines,
+          mimetype,
+          extension,
+        );
+      }
+
       return {
         llmContent: llmSuccessMessage,
         returnDisplay: displayResult,