fix nodes advanced param

This commit is contained in:
Gabriel Almeida 2023-04-29 19:23:12 -03:00
commit 79a18b8c40
2 changed files with 10 additions and 0 deletions

View file

@ -141,9 +141,11 @@ class ToolCreator(LangChainTypeCreator):
for param in params: for param in params:
field = TOOL_INPUTS.get(param, TOOL_INPUTS["str"]).copy() field = TOOL_INPUTS.get(param, TOOL_INPUTS["str"]).copy()
field.name = param field.name = param
field.advanced = False
if param == "aiosession": if param == "aiosession":
field.show = False field.show = False
field.required = False field.required = False
fields.append(field) fields.append(field)
template = Template(fields=fields, type_name=tool_type) template = Template(fields=fields, type_name=tool_type)

View file

@ -91,6 +91,7 @@ class PythonFunctionNode(FrontendNode):
show=True, show=True,
value=DEFAULT_PYTHON_FUNCTION, value=DEFAULT_PYTHON_FUNCTION,
name="code", name="code",
advanced=False,
) )
], ],
) )
@ -217,6 +218,7 @@ class ToolNode(FrontendNode):
multiline=True, multiline=True,
value="", value="",
name="name", name="name",
advanced=False,
), ),
TemplateField( TemplateField(
field_type="str", field_type="str",
@ -227,6 +229,7 @@ class ToolNode(FrontendNode):
multiline=True, multiline=True,
value="", value="",
name="description", name="description",
advanced=False,
), ),
TemplateField( TemplateField(
name="func", name="func",
@ -235,6 +238,7 @@ class ToolNode(FrontendNode):
is_list=False, is_list=False,
show=True, show=True,
multiline=True, multiline=True,
advanced=False,
), ),
TemplateField( TemplateField(
field_type="bool", field_type="bool",
@ -295,12 +299,14 @@ class InitializeAgentNode(FrontendNode):
options=list(NON_CHAT_AGENTS.keys()), options=list(NON_CHAT_AGENTS.keys()),
value=list(NON_CHAT_AGENTS.keys())[0], value=list(NON_CHAT_AGENTS.keys())[0],
name="agent", name="agent",
advanced=False,
), ),
TemplateField( TemplateField(
field_type="BaseChatMemory", field_type="BaseChatMemory",
required=False, required=False,
show=True, show=True,
name="memory", name="memory",
advanced=False,
), ),
TemplateField( TemplateField(
field_type="Tool", field_type="Tool",
@ -308,12 +314,14 @@ class InitializeAgentNode(FrontendNode):
show=True, show=True,
name="tools", name="tools",
is_list=True, is_list=True,
advanced=False,
), ),
TemplateField( TemplateField(
field_type="BaseLanguageModel", field_type="BaseLanguageModel",
required=True, required=True,
show=True, show=True,
name="llm", name="llm",
advanced=False,
), ),
], ],
) )