diff --git a/libs/langchain/langchain/agents/format_scratchpad/tools.py b/libs/langchain/langchain/agents/format_scratchpad/tools.py index 4fbf16d8a34..82cff2a7bd1 100644 --- a/libs/langchain/langchain/agents/format_scratchpad/tools.py +++ b/libs/langchain/langchain/agents/format_scratchpad/tools.py @@ -14,12 +14,12 @@ from langchain.agents.output_parsers.tools import ToolAgentAction def _create_tool_message( agent_action: ToolAgentAction, observation: str ) -> ToolMessage: - """Convert agent action and observation into a function message. + """Convert agent action and observation into a tool message. Args: agent_action: the tool invocation request from the agent observation: the result of the tool invocation Returns: - FunctionMessage that corresponds to the original tool invocation + ToolMessage that corresponds to the original tool invocation """ if not isinstance(observation, str): try: @@ -38,7 +38,7 @@ def _create_tool_message( def format_to_tool_messages( intermediate_steps: Sequence[Tuple[AgentAction, str]], ) -> List[BaseMessage]: - """Convert (AgentAction, tool output) tuples into FunctionMessages. + """Convert (AgentAction, tool output) tuples into ToolMessages. Args: intermediate_steps: Steps the LLM has taken to date, along with observations