Apply suggestions from code review

Co-authored-by: Sampath Kumar <[email protected]>

Author: holtskinner

genai/text_generation/test_text_generation_examples.py

@@ -33,7 +33,7 @@
 import textgen_with_txt
 import textgen_with_txt_img
 import textgen_with_txt_stream
-import textgen_with_txt_thinking
+import thinking_textgen_with_txt
 import textgen_with_video
 import textgen_with_youtube_video
 
@@ -85,7 +85,7 @@ def test_textgen_with_txt_img() -> None:
 
 
 def test_textgen_with_txt_thinking() -> None:
-    response = textgen_with_txt_thinking.generate_content()
+    response = thinking_textgen_with_txt.generate_content()
     assert response
 
 

genai/text_generation/textgen_with_txt_thinking.py

@@ -14,7 +14,7 @@
 
 
 def generate_content() -> str:
-    # [START googlegenaisdk_textgen_with_txt_thinking]
+    # [START googlegenaisdk_thinking_textgen_with_txt]
     from google import genai
     from google.genai.types import HttpOptions
 
@@ -58,7 +58,7 @@ def generate_content() -> str:
 
     #     Final Answer: The final answer is $\boxed{-2}$
 
-    # [END googlegenaisdk_textgen_with_txt_thinking]
+    # [END googlegenaisdk_thinking_textgen_with_txt]
     return response.text