Skip to content

Commit e1745d7

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
feat: Update AdkApp template to match ADK async interface. After this change, AdkApp will only work with ADK >= 1.0.0.
PiperOrigin-RevId: 763877804
1 parent 1b7d29c commit e1745d7

File tree

3 files changed

+227
-56
lines changed

3 files changed

+227
-56
lines changed

setup.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,8 @@
143143
]
144144

145145
adk_extra_require = [
146-
"google-adk >= 0.0.2, < 1.0.0",
146+
# 1.0.0 contains breaking changes, so we need to pin to 1.0.0.
147+
"google-adk >= 1.0.0, < 2.0.0",
147148
]
148149

149150
reasoning_engine_extra_require = [

tests/unit/vertex_adk/test_reasoning_engine_templates_adk.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def simple_span_processor_mock():
8787
def mock_adk_major_version():
8888
with mock.patch(
8989
"google.cloud.aiplatform.vertexai.preview.reasoning_engines.templates.adk.get_adk_major_version",
90-
return_value=0,
90+
return_value=1,
9191
):
9292
yield
9393

@@ -153,13 +153,13 @@ class TestAdkApp:
153153
def test_adk_major_version(self):
154154
with mock.patch(
155155
"google.cloud.aiplatform.vertexai.preview.reasoning_engines.templates.adk.get_adk_major_version",
156-
return_value=1,
156+
return_value=0,
157157
):
158158
with pytest.raises(
159159
ValueError,
160160
match=(
161-
"Unsupported google-adk major version: 1, please use"
162-
" google-adk<1.0.0 for AdkApp deployment."
161+
"Unsupported google-adk major version: 0, please use"
162+
" google-adk>=1.0.0 for AdkApp deployment."
163163
),
164164
):
165165
reasoning_engines.AdkApp(

0 commit comments

Comments
 (0)