refactor: hide non-public field and cleanup for tools

PiperOrigin-RevId: 759400805
This commit is contained in:
Xiang (Sean) Zhou 2025-05-15 19:23:17 -07:00 committed by Copybara-Service
parent e62e15e720
commit 770ecd8bb3
2 changed files with 3 additions and 5 deletions

View File

@ -14,7 +14,6 @@
from typing import Any
from . import _automatic_function_calling_util
from .langchain_tool import LangchainTool
@ -35,12 +34,12 @@ class ToolboxTool:
def __init__(self, url: str):
from toolbox_langchain import ToolboxClient
self.toolbox_client = ToolboxClient(url)
self._toolbox_client = ToolboxClient(url)
def get_tool(self, tool_name: str) -> LangchainTool:
tool = self.toolbox_client.load_tool(tool_name)
tool = self._toolbox_client.load_tool(tool_name)
return LangchainTool(tool)
def get_toolset(self, toolset_name: str) -> list[LangchainTool]:
tools = self.toolbox_client.load_toolset(toolset_name)
tools = self._toolbox_client.load_toolset(toolset_name)
return [LangchainTool(tool) for tool in tools]

View File

@ -15,7 +15,6 @@
from .tool_context import ToolContext
# TODO: make this internal, since user doesn't need to use this tool directly.
def transfer_to_agent(agent_name: str, tool_context: ToolContext):
"""Transfer the question to another agent."""
tool_context.actions.transfer_to_agent = agent_name