improve code comment

BoundaryML · Dec 30, 2024 · 9dc5135 · 9dc5135
1 parent 6bd1eb4
commit 9dc5135
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/engine/baml-runtime/src/internal/llm_client/primitive/google/googleai_client.rs b/engine/baml-runtime/src/internal/llm_client/primitive/google/googleai_client.rs
@@ -455,9 +455,14 @@ impl ToProviderMessage for GoogleAIClient {
     }
 }
 
-/// The Google Gemini 2 model returns `candidates` with 2 parts each.
+/// The Google Gemini 2 model has an experimental feature
+/// called Flash Thinking Mode, which is turned on in a particular
+/// named model: gemini-2.0-flash-thinking-exp-1219
+/// 
+/// When run in this mode, Gemini returns `candidates` with 2 parts each.
 /// Part 0 is the chain of thought, part 1 is the actual output.
 /// Other Gemini models put the output data in part 0.
+/// 
 /// TODO: Explicitly represent Flash Thinking Mode response and
 /// do more thorough checking for the content part.
 /// For examples of how to introspect the response more safely, see: