fix: Adapt to new computer use tool name in genai sdk 1.41.0

1.40.0 has some bug that caused some UT tests failures

PiperOrigin-RevId: 815098429
This commit is contained in:
Xiang (Sean) Zhou
2025-10-04 08:53:13 -07:00
committed by Copybara-Service
parent d1efc8461e
commit c6dd444fc9
4 changed files with 5 additions and 7 deletions
+2 -2
View File
@@ -128,7 +128,7 @@ def llm_request_with_computer_use():
system_instruction="You are a helpful assistant",
tools=[
types.Tool(
computer_use=types.ToolComputerUse(
computer_use=types.ComputerUse(
environment=types.Environment.ENVIRONMENT_BROWSER
)
)
@@ -1488,7 +1488,7 @@ async def test_computer_use_removes_system_instruction():
system_instruction="You are a helpful assistant",
tools=[
types.Tool(
computer_use=types.ToolComputerUse(
computer_use=types.ComputerUse(
environment=types.Environment.ENVIRONMENT_BROWSER
)
)
@@ -370,7 +370,7 @@ class TestComputerUseToolset:
config=types.GenerateContentConfig(
tools=[
types.Tool(
computer_use=types.ToolComputerUse(
computer_use=types.ComputerUse(
environment=types.Environment.ENVIRONMENT_BROWSER
)
)