Skip to content

Commit 816f2e8

Browse files
authored
Merge branch 'main' into feat/function-tools-enum-support
2 parents 35c8949 + c6dd444 commit 816f2e8

4 files changed

Lines changed: 5 additions & 7 deletions

File tree

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ dependencies = [
3838
"google-cloud-spanner>=3.56.0, <4.0.0", # For Spanner database
3939
"google-cloud-speech>=2.30.0, <3.0.0", # For Audio Transcription
4040
"google-cloud-storage>=2.18.0, <3.0.0", # For GCS Artifact service
41-
"google-genai>=1.21.1, <=1.40.0, !=1.37.0, !=1.38.0, !=1.39.0", # Google GenAI SDK
41+
"google-genai>=1.41.0, <2.0.0", # Google GenAI SDK
4242
"graphviz>=0.20.2, <1.0.0", # Graphviz for graph rendering
4343
"mcp>=1.8.0, <2.0.0;python_version>='3.10'", # For MCP Toolset
4444
"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.

src/google/adk/tools/computer_use/computer_use_toolset.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -206,9 +206,7 @@ async def process_llm_request(
206206
types.Environment.ENVIRONMENT_BROWSER,
207207
)
208208
llm_request.config.tools.append(
209-
types.Tool(
210-
computer_use=types.ToolComputerUse(environment=environment)
211-
)
209+
types.Tool(computer_use=types.ComputerUse(environment=environment))
212210
)
213211
logger.debug(
214212
"Added computer use tool with environment: %s",

tests/unittests/models/test_google_llm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def llm_request_with_computer_use():
128128
system_instruction="You are a helpful assistant",
129129
tools=[
130130
types.Tool(
131-
computer_use=types.ToolComputerUse(
131+
computer_use=types.ComputerUse(
132132
environment=types.Environment.ENVIRONMENT_BROWSER
133133
)
134134
)
@@ -1488,7 +1488,7 @@ async def test_computer_use_removes_system_instruction():
14881488
system_instruction="You are a helpful assistant",
14891489
tools=[
14901490
types.Tool(
1491-
computer_use=types.ToolComputerUse(
1491+
computer_use=types.ComputerUse(
14921492
environment=types.Environment.ENVIRONMENT_BROWSER
14931493
)
14941494
)

tests/unittests/tools/computer_use/test_computer_use_toolset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -370,7 +370,7 @@ async def test_process_llm_request_with_existing_computer_use(
370370
config=types.GenerateContentConfig(
371371
tools=[
372372
types.Tool(
373-
computer_use=types.ToolComputerUse(
373+
computer_use=types.ComputerUse(
374374
environment=types.Environment.ENVIRONMENT_BROWSER
375375
)
376376
)

0 commit comments

Comments
 (0)