Skip to content

Commit b07002e

Browse files
MarkDaoustcopybara-github
MarkDaoust
authored and
copybara-github
committed
feat: Add service tier for interactions.
PiperOrigin-RevId: 890642503
1 parent ad38e3a commit b07002e

File tree

3 files changed

+43
-0
lines changed

3 files changed

+43
-0
lines changed

‎google/genai/_interactions/resources/interactions.py‎

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ def create(
7777
response_format: object | Omit = omit,
7878
response_mime_type: str | Omit = omit,
7979
response_modalities: List[Literal["text", "image", "audio"]] | Omit = omit,
80+
service_tier: Literal["flex", "standard", "priority"] | Omit = omit,
8081
store: bool | Omit = omit,
8182
stream: Literal[False] | Omit = omit,
8283
system_instruction: str | Omit = omit,
@@ -109,6 +110,8 @@ def create(
109110
110111
response_modalities: The requested modalities of the response (TEXT, IMAGE, AUDIO).
111112
113+
service_tier: The service tier for the interaction.
114+
112115
store: Input only. Whether to store the response and request for later retrieval.
113116
114117
stream: Input only. Whether the interaction will be streamed.
@@ -141,6 +144,7 @@ def create(
141144
response_format: object | Omit = omit,
142145
response_mime_type: str | Omit = omit,
143146
response_modalities: List[Literal["text", "image", "audio"]] | Omit = omit,
147+
service_tier: Literal["flex", "standard", "priority"] | Omit = omit,
144148
store: bool | Omit = omit,
145149
system_instruction: str | Omit = omit,
146150
tools: Iterable[ToolParam] | Omit = omit,
@@ -174,6 +178,8 @@ def create(
174178
175179
response_modalities: The requested modalities of the response (TEXT, IMAGE, AUDIO).
176180
181+
service_tier: The service tier for the interaction.
182+
177183
store: Input only. Whether to store the response and request for later retrieval.
178184
179185
system_instruction: System instruction for the interaction.
@@ -203,6 +209,7 @@ def create(
203209
response_format: object | Omit = omit,
204210
response_mime_type: str | Omit = omit,
205211
response_modalities: List[Literal["text", "image", "audio"]] | Omit = omit,
212+
service_tier: Literal["flex", "standard", "priority"] | Omit = omit,
206213
store: bool | Omit = omit,
207214
stream: Literal[False] | Omit = omit,
208215
system_instruction: str | Omit = omit,
@@ -235,6 +242,8 @@ def create(
235242
236243
response_modalities: The requested modalities of the response (TEXT, IMAGE, AUDIO).
237244
245+
service_tier: The service tier for the interaction.
246+
238247
store: Input only. Whether to store the response and request for later retrieval.
239248
240249
stream: Input only. Whether the interaction will be streamed.
@@ -267,6 +276,7 @@ def create(
267276
response_format: object | Omit = omit,
268277
response_mime_type: str | Omit = omit,
269278
response_modalities: List[Literal["text", "image", "audio"]] | Omit = omit,
279+
service_tier: Literal["flex", "standard", "priority"] | Omit = omit,
270280
store: bool | Omit = omit,
271281
system_instruction: str | Omit = omit,
272282
tools: Iterable[ToolParam] | Omit = omit,
@@ -300,6 +310,8 @@ def create(
300310
301311
response_modalities: The requested modalities of the response (TEXT, IMAGE, AUDIO).
302312
313+
service_tier: The service tier for the interaction.
314+
303315
store: Input only. Whether to store the response and request for later retrieval.
304316
305317
system_instruction: System instruction for the interaction.
@@ -330,6 +342,7 @@ def create(
330342
response_format: object | Omit = omit,
331343
response_mime_type: str | Omit = omit,
332344
response_modalities: List[Literal["text", "image", "audio"]] | Omit = omit,
345+
service_tier: Literal["flex", "standard", "priority"] | Omit = omit,
333346
store: bool | Omit = omit,
334347
system_instruction: str | Omit = omit,
335348
tools: Iterable[ToolParam] | Omit = omit,
@@ -363,6 +376,8 @@ def create(
363376
364377
response_modalities: The requested modalities of the response (TEXT, IMAGE, AUDIO).
365378
379+
service_tier: The service tier for the interaction.
380+
366381
store: Input only. Whether to store the response and request for later retrieval.
367382
368383
system_instruction: System instruction for the interaction.
@@ -392,6 +407,7 @@ def create(
392407
response_format: object | Omit = omit,
393408
response_mime_type: str | Omit = omit,
394409
response_modalities: List[Literal["text", "image", "audio"]] | Omit = omit,
410+
service_tier: Literal["flex", "standard", "priority"] | Omit = omit,
395411
store: bool | Omit = omit,
396412
stream: Literal[False] | Literal[True] | Omit = omit,
397413
system_instruction: str | Omit = omit,
@@ -425,6 +441,7 @@ def create(
425441
"response_format": response_format,
426442
"response_mime_type": response_mime_type,
427443
"response_modalities": response_modalities,
444+
"service_tier": service_tier,
428445
"store": store,
429446
"stream": stream,
430447
"system_instruction": system_instruction,
@@ -705,6 +722,7 @@ async def create(
705722
response_format: object | Omit = omit,
706723
response_mime_type: str | Omit = omit,
707724
response_modalities: List[Literal["text", "image", "audio"]] | Omit = omit,
725+
service_tier: Literal["flex", "standard", "priority"] | Omit = omit,
708726
store: bool | Omit = omit,
709727
stream: Literal[False] | Omit = omit,
710728
system_instruction: str | Omit = omit,
@@ -737,6 +755,8 @@ async def create(
737755
738756
response_modalities: The requested modalities of the response (TEXT, IMAGE, AUDIO).
739757
758+
service_tier: The service tier for the interaction.
759+
740760
store: Input only. Whether to store the response and request for later retrieval.
741761
742762
stream: Input only. Whether the interaction will be streamed.
@@ -769,6 +789,7 @@ async def create(
769789
response_format: object | Omit = omit,
770790
response_mime_type: str | Omit = omit,
771791
response_modalities: List[Literal["text", "image", "audio"]] | Omit = omit,
792+
service_tier: Literal["flex", "standard", "priority"] | Omit = omit,
772793
store: bool | Omit = omit,
773794
system_instruction: str | Omit = omit,
774795
tools: Iterable[ToolParam] | Omit = omit,
@@ -802,6 +823,8 @@ async def create(
802823
803824
response_modalities: The requested modalities of the response (TEXT, IMAGE, AUDIO).
804825
826+
service_tier: The service tier for the interaction.
827+
805828
store: Input only. Whether to store the response and request for later retrieval.
806829
807830
system_instruction: System instruction for the interaction.
@@ -831,6 +854,7 @@ async def create(
831854
response_format: object | Omit = omit,
832855
response_mime_type: str | Omit = omit,
833856
response_modalities: List[Literal["text", "image", "audio"]] | Omit = omit,
857+
service_tier: Literal["flex", "standard", "priority"] | Omit = omit,
834858
store: bool | Omit = omit,
835859
stream: Literal[False] | Omit = omit,
836860
system_instruction: str | Omit = omit,
@@ -863,6 +887,8 @@ async def create(
863887
864888
response_modalities: The requested modalities of the response (TEXT, IMAGE, AUDIO).
865889
890+
service_tier: The service tier for the interaction.
891+
866892
store: Input only. Whether to store the response and request for later retrieval.
867893
868894
stream: Input only. Whether the interaction will be streamed.
@@ -895,6 +921,7 @@ async def create(
895921
response_format: object | Omit = omit,
896922
response_mime_type: str | Omit = omit,
897923
response_modalities: List[Literal["text", "image", "audio"]] | Omit = omit,
924+
service_tier: Literal["flex", "standard", "priority"] | Omit = omit,
898925
store: bool | Omit = omit,
899926
system_instruction: str | Omit = omit,
900927
tools: Iterable[ToolParam] | Omit = omit,
@@ -928,6 +955,8 @@ async def create(
928955
929956
response_modalities: The requested modalities of the response (TEXT, IMAGE, AUDIO).
930957
958+
service_tier: The service tier for the interaction.
959+
931960
store: Input only. Whether to store the response and request for later retrieval.
932961
933962
system_instruction: System instruction for the interaction.
@@ -958,6 +987,7 @@ async def create(
958987
response_format: object | Omit = omit,
959988
response_mime_type: str | Omit = omit,
960989
response_modalities: List[Literal["text", "image", "audio"]] | Omit = omit,
990+
service_tier: Literal["flex", "standard", "priority"] | Omit = omit,
961991
store: bool | Omit = omit,
962992
system_instruction: str | Omit = omit,
963993
tools: Iterable[ToolParam] | Omit = omit,
@@ -991,6 +1021,8 @@ async def create(
9911021
9921022
response_modalities: The requested modalities of the response (TEXT, IMAGE, AUDIO).
9931023
1024+
service_tier: The service tier for the interaction.
1025+
9941026
store: Input only. Whether to store the response and request for later retrieval.
9951027
9961028
system_instruction: System instruction for the interaction.
@@ -1020,6 +1052,7 @@ async def create(
10201052
response_format: object | Omit = omit,
10211053
response_mime_type: str | Omit = omit,
10221054
response_modalities: List[Literal["text", "image", "audio"]] | Omit = omit,
1055+
service_tier: Literal["flex", "standard", "priority"] | Omit = omit,
10231056
store: bool | Omit = omit,
10241057
stream: Literal[False] | Literal[True] | Omit = omit,
10251058
system_instruction: str | Omit = omit,
@@ -1053,6 +1086,7 @@ async def create(
10531086
"response_format": response_format,
10541087
"response_mime_type": response_mime_type,
10551088
"response_modalities": response_modalities,
1089+
"service_tier": service_tier,
10561090
"store": store,
10571091
"stream": stream,
10581092
"system_instruction": system_instruction,

‎google/genai/_interactions/types/interaction.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,6 +138,9 @@ class Interaction(BaseModel):
138138
role: Optional[str] = None
139139
"""Output only. The role of the interaction."""
140140

141+
service_tier: Optional[Literal["flex", "standard", "priority"]] = None
142+
"""The service tier for the interaction."""
143+
141144
system_instruction: Optional[str] = None
142145
"""System instruction for the interaction."""
143146

‎google/genai/_interactions/types/interaction_create_params.py‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,9 @@ class BaseCreateModelInteractionParams(TypedDict, total=False):
9090
response_modalities: List[Literal["text", "image", "audio"]]
9191
"""The requested modalities of the response (TEXT, IMAGE, AUDIO)."""
9292

93+
service_tier: Literal["flex", "standard", "priority"]
94+
"""The service tier for the interaction."""
95+
9396
store: bool
9497
"""Input only. Whether to store the response and request for later retrieval."""
9598

@@ -157,6 +160,9 @@ class BaseCreateAgentInteractionParams(TypedDict, total=False):
157160
response_modalities: List[Literal["text", "image", "audio"]]
158161
"""The requested modalities of the response (TEXT, IMAGE, AUDIO)."""
159162

163+
service_tier: Literal["flex", "standard", "priority"]
164+
"""The service tier for the interaction."""
165+
160166
store: bool
161167
"""Input only. Whether to store the response and request for later retrieval."""
162168

0 commit comments

Comments
 (0)