Skip to content

Commit 96d644c

Browse files
BenjaminKazemicopybara-github
authored andcommitted
feat: Add minimal and medium thinking levels.
PiperOrigin-RevId: 845014005
1 parent 356c320 commit 96d644c

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

google/genai/types.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -267,14 +267,18 @@ class PhishBlockThreshold(_common.CaseInSensitiveEnum):
267267

268268

269269
class ThinkingLevel(_common.CaseInSensitiveEnum):
270-
"""The level of thoughts tokens that the model should generate."""
270+
"""The number of thoughts tokens that the model should generate."""
271271

272272
THINKING_LEVEL_UNSPECIFIED = 'THINKING_LEVEL_UNSPECIFIED'
273-
"""Default value."""
273+
"""Unspecified thinking level."""
274274
LOW = 'LOW'
275275
"""Low thinking level."""
276+
MEDIUM = 'MEDIUM'
277+
"""Medium thinking level."""
276278
HIGH = 'HIGH'
277279
"""High thinking level."""
280+
MINIMAL = 'MINIMAL'
281+
"""MINIMAL thinking level."""
278282

279283

280284
class HarmCategory(_common.CaseInSensitiveEnum):
@@ -4611,7 +4615,7 @@ class ThinkingConfig(_common.BaseModel):
46114615
)
46124616
thinking_level: Optional[ThinkingLevel] = Field(
46134617
default=None,
4614-
description="""Optional. The level of thoughts tokens that the model should generate.""",
4618+
description="""Optional. The number of thoughts tokens that the model should generate.""",
46154619
)
46164620

46174621

@@ -4627,7 +4631,7 @@ class ThinkingConfigDict(TypedDict, total=False):
46274631
"""
46284632

46294633
thinking_level: Optional[ThinkingLevel]
4630-
"""Optional. The level of thoughts tokens that the model should generate."""
4634+
"""Optional. The number of thoughts tokens that the model should generate."""
46314635

46324636

46334637
ThinkingConfigOrDict = Union[ThinkingConfig, ThinkingConfigDict]

0 commit comments

Comments
 (0)