Add model version to GenerateContent Output

ai-logic/firebase-ai/CHANGELOG.md

@@ -1,5 +1,6 @@
 # Unreleased
 
+- [feature] Added `modelVersion` property to `GenerateContentResponse` (#8227)
 - [feature] Added the `retrievalConfig` argument to `TemplateToolConfig` (#8107)
 - [fixed] Fixed citation indices to be native UTF-16 instead of UTF-8. (#8056)
 

ai-logic/firebase-ai/api.txt

@@ -679,6 +679,7 @@ package com.google.firebase.ai.type {
     method public java.util.List<com.google.firebase.ai.type.FunctionCallPart> getFunctionCalls();
     method public com.google.firebase.ai.InferenceSource getInferenceSource();
     method public java.util.List<com.google.firebase.ai.type.InlineDataPart> getInlineDataParts();
+    method public String? getModelVersion();
     method public com.google.firebase.ai.type.PromptFeedback? getPromptFeedback();
     method public String? getText();
     method public String? getThoughtSummary();
@@ -687,6 +688,7 @@ package com.google.firebase.ai.type {
     property public final java.util.List<com.google.firebase.ai.type.FunctionCallPart> functionCalls;
     property public final com.google.firebase.ai.InferenceSource inferenceSource;
     property public final java.util.List<com.google.firebase.ai.type.InlineDataPart> inlineDataParts;
+    property public final String? modelVersion;
     property public final com.google.firebase.ai.type.PromptFeedback? promptFeedback;
     property public final String? text;
     property public final String? thoughtSummary;

ai-logic/firebase-ai/src/main/kotlin/com/google/firebase/ai/type/GenerateContentResponse.kt

@@ -17,6 +17,7 @@
 package com.google.firebase.ai.type
 
 import com.google.firebase.ai.InferenceSource
+import kotlinx.serialization.SerialName
 import kotlinx.serialization.Serializable
 
 /**
@@ -35,6 +36,7 @@ internal constructor(
   public val inferenceSource: InferenceSource,
   public val promptFeedback: PromptFeedback?,
   public val usageMetadata: UsageMetadata?,
+  public val modelVersion: String? = null,
 ) {
 
   // To maintain backwards compatibility with possible outside usage, we maintain a public
@@ -47,7 +49,7 @@ internal constructor(
     candidates: List<Candidate>,
     promptFeedback: PromptFeedback?,
     usageMetadata: UsageMetadata?,
-  ) : this(candidates, InferenceSource.IN_CLOUD, promptFeedback, usageMetadata)
+  ) : this(candidates, InferenceSource.IN_CLOUD, promptFeedback, usageMetadata, null)
 
   /**
    * Convenience field representing all the text parts in the response as a single string.
@@ -118,13 +120,15 @@ internal constructor(
     val candidates: List<Candidate.Internal>? = null,
     val promptFeedback: PromptFeedback.Internal? = null,
     val usageMetadata: UsageMetadata.Internal? = null,
+    @SerialName("model_version") val modelVersion: String? = null,
   ) : Response {
     internal fun toPublic(): GenerateContentResponse {
       return GenerateContentResponse(
         candidates?.map { it.toPublic() }.orEmpty(),
         InferenceSource.IN_CLOUD,
         promptFeedback?.toPublic(),
-        usageMetadata?.toPublic()
+        usageMetadata?.toPublic(),
+        modelVersion
       )
     }
   }

ai-logic/firebase-ai/src/test/java/com/google/firebase/ai/SerializationTests.kt

@@ -122,6 +122,9 @@ internal class SerializationTests {
             },
             "usageMetadata": {
                 "${'$'}ref": "UsageMetadata"
+            },
+            "model_version": {
+                "type": "string"
             }
         }
      }