From 5ccf46b5a0412a41d73b0b6397f9a0744adf75f0 Mon Sep 17 00:00:00 2001
From: Sandy Tao <sandytao520@icloud.com>
Date: Wed, 3 Sep 2025 10:56:51 -0700
Subject: [PATCH] Fix(core): Log exact model version from API response (#7666)

---
 packages/core/src/core/loggingContentGenerator.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/packages/core/src/core/loggingContentGenerator.ts b/packages/core/src/core/loggingContentGenerator.ts
index 58565625a3..82e6dd1104 100644
--- a/packages/core/src/core/loggingContentGenerator.ts
+++ b/packages/core/src/core/loggingContentGenerator.ts
@@ -60,6 +60,7 @@ export class LoggingContentGenerator implements ContentGenerator {
 
   private _logApiResponse(
     durationMs: number,
+    model: string,
     prompt_id: string,
     usageMetadata?: GenerateContentResponseUsageMetadata,
     responseText?: string,
@@ -67,7 +68,7 @@ export class LoggingContentGenerator implements ContentGenerator {
     logApiResponse(
       this.config,
       new ApiResponseEvent(
-        this.config.getModel(),
+        model,
         durationMs,
         prompt_id,
         this.config.getContentGeneratorConfig()?.authType,
@@ -112,6 +113,7 @@ export class LoggingContentGenerator implements ContentGenerator {
       const durationMs = Date.now() - startTime;
       this._logApiResponse(
         durationMs,
+        response.modelVersion || '',
         userPromptId,
         response.usageMetadata,
         JSON.stringify(response),
@@ -171,6 +173,7 @@ export class LoggingContentGenerator implements ContentGenerator {
       this._logApiResponse(
         durationMs,
         userPromptId,
+        responses[0]?.modelVersion || '',
         lastUsageMetadata,
         JSON.stringify(responses),
       );