mirror of
https://github.com/EvolutionAPI/adk-python.git
synced 2025-07-13 15:14:50 -06:00
feat: add Toolbox Toolset
PiperOrigin-RevId: 759741922
This commit is contained in:
parent
015c667917
commit
939b45a1c8
@ -29,6 +29,7 @@ from .long_running_tool import LongRunningFunctionTool
|
||||
from .preload_memory_tool import preload_memory_tool as preload_memory
|
||||
from .tool_context import ToolContext
|
||||
from .transfer_to_agent_tool import transfer_to_agent
|
||||
from .toolbox_toolset import ToolboxToolset
|
||||
|
||||
|
||||
__all__ = [
|
||||
@ -47,5 +48,6 @@ __all__ = [
|
||||
'LongRunningFunctionTool',
|
||||
'preload_memory',
|
||||
'ToolContext',
|
||||
'ToolboxToolset',
|
||||
'transfer_to_agent',
|
||||
]
|
||||
|
77
src/google/adk/tools/toolbox_toolset.py
Normal file
77
src/google/adk/tools/toolbox_toolset.py
Normal file
@ -0,0 +1,77 @@
|
||||
# Copyright 2025 Google LLC
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from typing import List, Optional, override
|
||||
|
||||
import toolbox_core as toolbox
|
||||
|
||||
from ..agents.readonly_context import ReadonlyContext
|
||||
from .base_tool import BaseTool
|
||||
from .base_toolset import BaseToolset
|
||||
from .function_tool import FunctionTool
|
||||
|
||||
|
||||
class ToolboxToolset(BaseToolset):
|
||||
"""A class that provides access to toolbox toolsets.
|
||||
|
||||
Example:
|
||||
```python
|
||||
toolbox_toolset = ToolboxToolset("http://127.0.0.1:5000",
|
||||
toolset_name="my-toolset")
|
||||
)
|
||||
```
|
||||
"""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
server_url: str,
|
||||
toolset_name: Optional[str] = None,
|
||||
tool_names: Optional[List[str]] = None,
|
||||
):
|
||||
"""Args:
|
||||
|
||||
server_url: The URL of the toolbox server.
|
||||
toolset_name: The name of the toolbox toolset to load.
|
||||
tool_names: The names of the tools to load.
|
||||
The resulting ToolboxToolset will contain both tools loaded by tool_names
|
||||
and toolset_name.
|
||||
"""
|
||||
if not tool_names and not toolset_name:
|
||||
raise ValueError("tool_names and toolset_name cannot both be None")
|
||||
super().__init__()
|
||||
self._server_url = server_url
|
||||
self._toolbox_client = toolbox.ToolboxSyncClient(server_url)
|
||||
self._toolset_name = toolset_name
|
||||
self._tool_names = tool_names
|
||||
|
||||
@override
|
||||
async def get_tools(
|
||||
self, readonly_context: Optional[ReadonlyContext] = None
|
||||
) -> list[BaseTool]:
|
||||
tools = []
|
||||
if self._toolset_name:
|
||||
tools.extend([
|
||||
FunctionTool(tool)
|
||||
for tool in self._toolbox_client.load_toolset(self._toolset_name)
|
||||
])
|
||||
if self._tool_names:
|
||||
tools.extend([
|
||||
FunctionTool(self._toolbox_client.load_tool(tool_name))
|
||||
for tool_name in self._tool_names
|
||||
])
|
||||
return tools
|
||||
|
||||
@override
|
||||
async def close(self):
|
||||
self._toolbox_client.close()
|
Loading…
Reference in New Issue
Block a user