From 6be88c6b0b61f8d3e93c0b3f6f802917b093af56 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 24 Mar 2024 19:38:32 -0300 Subject: [PATCH] Update render_tool_description method in XMLAgent.py --- src/backend/langflow/components/agents/XMLAgent.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/components/agents/XMLAgent.py b/src/backend/langflow/components/agents/XMLAgent.py index 47a3cc09a..60916fbfd 100644 --- a/src/backend/langflow/components/agents/XMLAgent.py +++ b/src/backend/langflow/components/agents/XMLAgent.py @@ -77,7 +77,14 @@ class XMLAgentComponent(CustomComponent): raise ValueError("Prompt must contain 'input' key.") def render_tool_description(tools): - return "\n".join([tool_template.format(name=tool.name, description=tool.description) for tool in tools]) + return "\n".join( + [ + tool_template.format( + name=tool.name, description=tool.description, args_schema=tool.args_schema, args=tool.args + ) + for tool in tools + ] + ) prompt_template = PromptTemplate.from_template(prompt) input_variables = prompt_template.input_variables