diff --git a/libs/langchain/langchain/agents/openai_functions_multi_agent/base.py b/libs/langchain/langchain/agents/openai_functions_multi_agent/base.py index 026171b25e..480f102d90 100644 --- a/libs/langchain/langchain/agents/openai_functions_multi_agent/base.py +++ b/libs/langchain/langchain/agents/openai_functions_multi_agent/base.py @@ -58,7 +58,12 @@ def _parse_ai_message(message: BaseMessage) -> Union[List[AgentAction], AgentFin final_tools: List[AgentAction] = [] for tool_schema in tools: - _tool_input = tool_schema["action"] + if "action" in tool_schema: + _tool_input = tool_schema["action"] + else: + # drop action_name from schema + _tool_input = tool_schema.copy() + del _tool_input["action_name"] function_name = tool_schema["action_name"] # HACK HACK HACK: