encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

feat: Support custom tool_name_prefix in auto-generated GoogleApiToolset

Browse Source 
PiperOrigin-RevId: 795508179
This commit is contained in:
Xiang (Sean) Zhou
2025-08-15 09:44:16 -07:00
committed by Copybara-Service
parent 1328e6ef62
commit a2832d5ac7
4 changed files with 161 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unittests/tools/google_api_tool/test_google_api_toolset.py
+26
View File
@@ -425,3 +425,29 @@ class TestGoogleApiToolset:
tool_set.set_tool_filter(new_filter_predicate)
assert tool_set.tool_filter == new_filter_predicate
@mock.patch(
"google.adk.tools.google_api_tool.google_api_toolset.OpenAPIToolset"
)
@mock.patch(
"google.adk.tools.google_api_tool.google_api_toolset.GoogleApiToOpenApiConverter"
)
def test_init_with_tool_name_prefix(
self,
mock_converter_class,
mock_openapi_toolset_class,
mock_converter_instance,
mock_openapi_toolset_instance,
):
"""Test GoogleApiToolset initialization with tool_name_prefix."""
mock_converter_class.return_value = mock_converter_instance
mock_openapi_toolset_class.return_value = mock_openapi_toolset_instance
tool_name_prefix = "test_prefix"
tool_set = GoogleApiToolset(
api_name=TEST_API_NAME,
api_version=TEST_API_VERSION,
tool_name_prefix=tool_name_prefix,
)
assert tool_set.tool_name_prefix == tool_name_prefix