Skip to content

Commit aaac8d8

Browse files
hkt74copybara-github
hkt74
authored and
copybara-github
committed
feat: rename ComputerUse tool (early access)
PiperOrigin-RevId: 813396097
1 parent 538c755 commit aaac8d8

File tree

6 files changed

+157
-173
lines changed

6 files changed

+157
-173
lines changed

‎google/genai/_live_converters.py‎

Lines changed: 38 additions & 42 deletions
Original file line numberDiff line numberDiff line change
@@ -291,6 +291,42 @@ def _Blob_to_vertex(
291291
return to_object
292292

293293

294+
def _ComputerUse_to_mldev(
295+
from_object: Union[dict[str, Any], object],
296+
parent_object: Optional[dict[str, Any]] = None,
297+
) -> dict[str, Any]:
298+
to_object: dict[str, Any] = {}
299+
if getv(from_object, ['environment']) is not None:
300+
setv(to_object, ['environment'], getv(from_object, ['environment']))
301+
302+
if getv(from_object, ['excluded_predefined_functions']) is not None:
303+
setv(
304+
to_object,
305+
['excludedPredefinedFunctions'],
306+
getv(from_object, ['excluded_predefined_functions']),
307+
)
308+
309+
return to_object
310+
311+
312+
def _ComputerUse_to_vertex(
313+
from_object: Union[dict[str, Any], object],
314+
parent_object: Optional[dict[str, Any]] = None,
315+
) -> dict[str, Any]:
316+
to_object: dict[str, Any] = {}
317+
if getv(from_object, ['environment']) is not None:
318+
setv(to_object, ['environment'], getv(from_object, ['environment']))
319+
320+
if getv(from_object, ['excluded_predefined_functions']) is not None:
321+
setv(
322+
to_object,
323+
['excludedPredefinedFunctions'],
324+
getv(from_object, ['excluded_predefined_functions']),
325+
)
326+
327+
return to_object
328+
329+
294330
def _Content_from_mldev(
295331
from_object: Union[dict[str, Any], object],
296332
parent_object: Optional[dict[str, Any]] = None,
@@ -3189,42 +3225,6 @@ def _SpeechConfig_to_vertex(
31893225
return to_object
31903226

31913227

3192-
def _ToolComputerUse_to_mldev(
3193-
from_object: Union[dict[str, Any], object],
3194-
parent_object: Optional[dict[str, Any]] = None,
3195-
) -> dict[str, Any]:
3196-
to_object: dict[str, Any] = {}
3197-
if getv(from_object, ['environment']) is not None:
3198-
setv(to_object, ['environment'], getv(from_object, ['environment']))
3199-
3200-
if getv(from_object, ['excluded_predefined_functions']) is not None:
3201-
setv(
3202-
to_object,
3203-
['excludedPredefinedFunctions'],
3204-
getv(from_object, ['excluded_predefined_functions']),
3205-
)
3206-
3207-
return to_object
3208-
3209-
3210-
def _ToolComputerUse_to_vertex(
3211-
from_object: Union[dict[str, Any], object],
3212-
parent_object: Optional[dict[str, Any]] = None,
3213-
) -> dict[str, Any]:
3214-
to_object: dict[str, Any] = {}
3215-
if getv(from_object, ['environment']) is not None:
3216-
setv(to_object, ['environment'], getv(from_object, ['environment']))
3217-
3218-
if getv(from_object, ['excluded_predefined_functions']) is not None:
3219-
setv(
3220-
to_object,
3221-
['excludedPredefinedFunctions'],
3222-
getv(from_object, ['excluded_predefined_functions']),
3223-
)
3224-
3225-
return to_object
3226-
3227-
32283228
def _Tool_to_mldev(
32293229
from_object: Union[dict[str, Any], object],
32303230
parent_object: Optional[dict[str, Any]] = None,
@@ -3278,9 +3278,7 @@ def _Tool_to_mldev(
32783278
setv(
32793279
to_object,
32803280
['computerUse'],
3281-
_ToolComputerUse_to_mldev(
3282-
getv(from_object, ['computer_use']), to_object
3283-
),
3281+
_ComputerUse_to_mldev(getv(from_object, ['computer_use']), to_object),
32843282
)
32853283

32863284
if getv(from_object, ['code_execution']) is not None:
@@ -3352,9 +3350,7 @@ def _Tool_to_vertex(
33523350
setv(
33533351
to_object,
33543352
['computerUse'],
3355-
_ToolComputerUse_to_vertex(
3356-
getv(from_object, ['computer_use']), to_object
3357-
),
3353+
_ComputerUse_to_vertex(getv(from_object, ['computer_use']), to_object),
33583354
)
33593355

33603356
if getv(from_object, ['code_execution']) is not None:

‎google/genai/_tokens_converters.py‎

Lines changed: 19 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,24 @@ def _Blob_to_mldev(
106106
return to_object
107107

108108

109+
def _ComputerUse_to_mldev(
110+
from_object: Union[dict[str, Any], object],
111+
parent_object: Optional[dict[str, Any]] = None,
112+
) -> dict[str, Any]:
113+
to_object: dict[str, Any] = {}
114+
if getv(from_object, ['environment']) is not None:
115+
setv(to_object, ['environment'], getv(from_object, ['environment']))
116+
117+
if getv(from_object, ['excluded_predefined_functions']) is not None:
118+
setv(
119+
to_object,
120+
['excludedPredefinedFunctions'],
121+
getv(from_object, ['excluded_predefined_functions']),
122+
)
123+
124+
return to_object
125+
126+
109127
def _Content_to_mldev(
110128
from_object: Union[dict[str, Any], object],
111129
parent_object: Optional[dict[str, Any]] = None,
@@ -743,24 +761,6 @@ def _SpeechConfig_to_mldev(
743761
return to_object
744762

745763

746-
def _ToolComputerUse_to_mldev(
747-
from_object: Union[dict[str, Any], object],
748-
parent_object: Optional[dict[str, Any]] = None,
749-
) -> dict[str, Any]:
750-
to_object: dict[str, Any] = {}
751-
if getv(from_object, ['environment']) is not None:
752-
setv(to_object, ['environment'], getv(from_object, ['environment']))
753-
754-
if getv(from_object, ['excluded_predefined_functions']) is not None:
755-
setv(
756-
to_object,
757-
['excludedPredefinedFunctions'],
758-
getv(from_object, ['excluded_predefined_functions']),
759-
)
760-
761-
return to_object
762-
763-
764764
def _Tool_to_mldev(
765765
from_object: Union[dict[str, Any], object],
766766
parent_object: Optional[dict[str, Any]] = None,
@@ -814,9 +814,7 @@ def _Tool_to_mldev(
814814
setv(
815815
to_object,
816816
['computerUse'],
817-
_ToolComputerUse_to_mldev(
818-
getv(from_object, ['computer_use']), to_object
819-
),
817+
_ComputerUse_to_mldev(getv(from_object, ['computer_use']), to_object),
820818
)
821819

822820
if getv(from_object, ['code_execution']) is not None:

‎google/genai/batches.py‎

Lines changed: 19 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -467,6 +467,24 @@ def _CitationMetadata_from_mldev(
467467
return to_object
468468

469469

470+
def _ComputerUse_to_mldev(
471+
from_object: Union[dict[str, Any], object],
472+
parent_object: Optional[dict[str, Any]] = None,
473+
) -> dict[str, Any]:
474+
to_object: dict[str, Any] = {}
475+
if getv(from_object, ['environment']) is not None:
476+
setv(to_object, ['environment'], getv(from_object, ['environment']))
477+
478+
if getv(from_object, ['excluded_predefined_functions']) is not None:
479+
setv(
480+
to_object,
481+
['excludedPredefinedFunctions'],
482+
getv(from_object, ['excluded_predefined_functions']),
483+
)
484+
485+
return to_object
486+
487+
470488
def _ContentEmbedding_from_mldev(
471489
from_object: Union[dict[str, Any], object],
472490
parent_object: Optional[dict[str, Any]] = None,
@@ -1877,24 +1895,6 @@ def _ThinkingConfig_to_mldev(
18771895
return to_object
18781896

18791897

1880-
def _ToolComputerUse_to_mldev(
1881-
from_object: Union[dict[str, Any], object],
1882-
parent_object: Optional[dict[str, Any]] = None,
1883-
) -> dict[str, Any]:
1884-
to_object: dict[str, Any] = {}
1885-
if getv(from_object, ['environment']) is not None:
1886-
setv(to_object, ['environment'], getv(from_object, ['environment']))
1887-
1888-
if getv(from_object, ['excluded_predefined_functions']) is not None:
1889-
setv(
1890-
to_object,
1891-
['excludedPredefinedFunctions'],
1892-
getv(from_object, ['excluded_predefined_functions']),
1893-
)
1894-
1895-
return to_object
1896-
1897-
18981898
def _ToolConfig_to_mldev(
18991899
from_object: Union[dict[str, Any], object],
19001900
parent_object: Optional[dict[str, Any]] = None,
@@ -1974,9 +1974,7 @@ def _Tool_to_mldev(
19741974
setv(
19751975
to_object,
19761976
['computerUse'],
1977-
_ToolComputerUse_to_mldev(
1978-
getv(from_object, ['computer_use']), to_object
1979-
),
1977+
_ComputerUse_to_mldev(getv(from_object, ['computer_use']), to_object),
19801978
)
19811979

19821980
if getv(from_object, ['code_execution']) is not None:

‎google/genai/caches.py‎

Lines changed: 38 additions & 42 deletions
Original file line numberDiff line numberDiff line change
@@ -181,6 +181,42 @@ def _CachedContent_from_vertex(
181181
return to_object
182182

183183

184+
def _ComputerUse_to_mldev(
185+
from_object: Union[dict[str, Any], object],
186+
parent_object: Optional[dict[str, Any]] = None,
187+
) -> dict[str, Any]:
188+
to_object: dict[str, Any] = {}
189+
if getv(from_object, ['environment']) is not None:
190+
setv(to_object, ['environment'], getv(from_object, ['environment']))
191+
192+
if getv(from_object, ['excluded_predefined_functions']) is not None:
193+
setv(
194+
to_object,
195+
['excludedPredefinedFunctions'],
196+
getv(from_object, ['excluded_predefined_functions']),
197+
)
198+
199+
return to_object
200+
201+
202+
def _ComputerUse_to_vertex(
203+
from_object: Union[dict[str, Any], object],
204+
parent_object: Optional[dict[str, Any]] = None,
205+
) -> dict[str, Any]:
206+
to_object: dict[str, Any] = {}
207+
if getv(from_object, ['environment']) is not None:
208+
setv(to_object, ['environment'], getv(from_object, ['environment']))
209+
210+
if getv(from_object, ['excluded_predefined_functions']) is not None:
211+
setv(
212+
to_object,
213+
['excludedPredefinedFunctions'],
214+
getv(from_object, ['excluded_predefined_functions']),
215+
)
216+
217+
return to_object
218+
219+
184220
def _Content_to_mldev(
185221
from_object: Union[dict[str, Any], object],
186222
parent_object: Optional[dict[str, Any]] = None,
@@ -1135,42 +1171,6 @@ def _RetrievalConfig_to_vertex(
11351171
return to_object
11361172

11371173

1138-
def _ToolComputerUse_to_mldev(
1139-
from_object: Union[dict[str, Any], object],
1140-
parent_object: Optional[dict[str, Any]] = None,
1141-
) -> dict[str, Any]:
1142-
to_object: dict[str, Any] = {}
1143-
if getv(from_object, ['environment']) is not None:
1144-
setv(to_object, ['environment'], getv(from_object, ['environment']))
1145-
1146-
if getv(from_object, ['excluded_predefined_functions']) is not None:
1147-
setv(
1148-
to_object,
1149-
['excludedPredefinedFunctions'],
1150-
getv(from_object, ['excluded_predefined_functions']),
1151-
)
1152-
1153-
return to_object
1154-
1155-
1156-
def _ToolComputerUse_to_vertex(
1157-
from_object: Union[dict[str, Any], object],
1158-
parent_object: Optional[dict[str, Any]] = None,
1159-
) -> dict[str, Any]:
1160-
to_object: dict[str, Any] = {}
1161-
if getv(from_object, ['environment']) is not None:
1162-
setv(to_object, ['environment'], getv(from_object, ['environment']))
1163-
1164-
if getv(from_object, ['excluded_predefined_functions']) is not None:
1165-
setv(
1166-
to_object,
1167-
['excludedPredefinedFunctions'],
1168-
getv(from_object, ['excluded_predefined_functions']),
1169-
)
1170-
1171-
return to_object
1172-
1173-
11741174
def _ToolConfig_to_mldev(
11751175
from_object: Union[dict[str, Any], object],
11761176
parent_object: Optional[dict[str, Any]] = None,
@@ -1276,9 +1276,7 @@ def _Tool_to_mldev(
12761276
setv(
12771277
to_object,
12781278
['computerUse'],
1279-
_ToolComputerUse_to_mldev(
1280-
getv(from_object, ['computer_use']), to_object
1281-
),
1279+
_ComputerUse_to_mldev(getv(from_object, ['computer_use']), to_object),
12821280
)
12831281

12841282
if getv(from_object, ['code_execution']) is not None:
@@ -1350,9 +1348,7 @@ def _Tool_to_vertex(
13501348
setv(
13511349
to_object,
13521350
['computerUse'],
1353-
_ToolComputerUse_to_vertex(
1354-
getv(from_object, ['computer_use']), to_object
1355-
),
1351+
_ComputerUse_to_vertex(getv(from_object, ['computer_use']), to_object),
13561352
)
13571353

13581354
if getv(from_object, ['code_execution']) is not None:

0 commit comments

Comments
 (0)