diff --git a/genai/text_generation/model_optimizer_textgen_with_txt.py b/genai/text_generation/model_optimizer_textgen_with_txt.py
index c6ae8651f05..983b3ece091 100644
--- a/genai/text_generation/model_optimizer_textgen_with_txt.py
+++ b/genai/text_generation/model_optimizer_textgen_with_txt.py
@@ -16,12 +16,22 @@
 def generate_content() -> str:
     # [START googlegenaisdk_model_optimizer_textgen_with_txt]
     from google import genai
-    from google.genai.types import HttpOptions
+    from google.genai.types import (
+        FeatureSelectionPreference,
+        GenerateContentConfig,
+        HttpOptions,
+        ModelSelectionConfig
+    )
 
-    client = genai.Client(http_options=HttpOptions(api_version="v1"))
+    client = genai.Client(http_options=HttpOptions(api_version="v1beta1"))
     response = client.models.generate_content(
         model="model-optimizer-exp-04-09",
         contents="How does AI work?",
+        config=GenerateContentConfig(
+            model_selection_config=ModelSelectionConfig(
+                feature_selection_preference=FeatureSelectionPreference.BALANCED  # Options: PRIORITIZE_QUALITY, BALANCED, PRIORITIZE_COST
+            ),
+        ),
     )
     print(response.text)
     # Example response:
diff --git a/genai/text_generation/requirements.txt b/genai/text_generation/requirements.txt
index 73d0828cb4e..0705f93a9f3 100644
--- a/genai/text_generation/requirements.txt
+++ b/genai/text_generation/requirements.txt
@@ -1 +1 @@
-google-genai==1.7.0
+google-genai==1.12.1