jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: Include Missing Field in Tool Status (#2118)

Browse source 
Modified the status to include args_schema when the tool has it.

You can use this translation in your commit message.
This commit is contained in:
YAMON.IO 2024-06-10 10:01:27 +09:00 • committed by GitHub
commit b40bb95909
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/base/tools/base.py
View file

@ -9,7 +9,7 @@ def build_status_from_tool(tool: Tool):
tool (Tool): The tool object to build the status for. tool (Tool): The tool object to build the status for.
Returns: Returns:
str: The status string representation of the tool, including its name, description, and arguments (if any). str: The status string representation of the tool, including its name, description, arguments (if any), and args_schema (if any).
""" """
description_repr = repr(tool.description).strip("'") description_repr = repr(tool.description).strip("'")
args_str = "\n".join( args_str = "\n".join(
@ -19,5 +19,8 @@ def build_status_from_tool(tool: Tool):
if "description" in arg_data if "description" in arg_data
] ]
) )
# Include args_schema information
args_schema_str = repr(tool.args_schema) if tool.args_schema else "None"
status = f"Name: {tool.name}\nDescription: {description_repr}" status = f"Name: {tool.name}\nDescription: {description_repr}"
status += f"\nArgs Schema: {args_schema_str}"
return status + (f"\nArguments:\n{args_str}" if args_str else "") return status + (f"\nArguments:\n{args_str}" if args_str else "")