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
+1 -1
View File
@@ -38,7 +38,7 @@ dependencies = [
"google-cloud-spanner>=3.56.0, <4.0.0", # For Spanner database
"google-cloud-speech>=2.30.0, <3.0.0", # For Audio Transcription
"google-cloud-storage>=2.18.0, <3.0.0", # For GCS Artifact service
"google-genai>=1.21.1, <=1.40.0, !=1.37.0, !=1.38.0, !=1.39.0", # Google GenAI SDK
"google-genai>=1.41.0, <2.0.0", # Google GenAI SDK
"graphviz>=0.20.2, <1.0.0", # Graphviz for graph rendering
"mcp>=1.8.0, <2.0.0;python_version>='3.10'", # For MCP Toolset
"opentelemetry-api>=1.37.0, <=1.37.0", # OpenTelemetry - limit upper version for sdk and api to not risk breaking changes from unstable _logs package.
@@ -206,9 +206,7 @@ class ComputerUseToolset(BaseToolset):
types.Environment.ENVIRONMENT_BROWSER,
)
llm_request.config.tools.append(
types.Tool(
computer_use=types.ToolComputerUse(environment=environment)
)
types.Tool(computer_use=types.ComputerUse(environment=environment))
)
logger.debug(
"Added computer use tool with environment: %s",
+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
)
)