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

Update agent field types in agents.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-09 18:11:38 -03:00
commit 0cca8efebd
1 changed files with 20 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

41
src/backend/langflow/template/frontend_node/agents.py
View file

@ -1,7 +1,6 @@
from typing import Optional from typing import Optional
from langchain.agents import types from langchain.agents import types
from langflow.template.field.base import TemplateField from langflow.template.field.base import TemplateField
from langflow.template.frontend_node.base import FrontendNode from langflow.template.frontend_node.base import FrontendNode
from langflow.template.template.base import Template from langflow.template.template.base import Template
@ -29,17 +28,17 @@ class SQLAgentNode(FrontendNode):
type_name="sql_agent", type_name="sql_agent",
fields=[ fields=[
TemplateField( TemplateField(
field_type="str", field_type="str", # pyright: ignore
required=True, required=True,
placeholder="", placeholder="",
is_list=False, is_list=False, # pyright: ignore
show=True, show=True,
multiline=False, multiline=False,
value="", value="",
name="database_uri", name="database_uri",
), ),
TemplateField( TemplateField(
field_type="BaseLanguageModel", field_type="BaseLanguageModel", # pyright: ignore
required=True, required=True,
show=True, show=True,
name="llm", name="llm",
@ -60,14 +59,14 @@ class VectorStoreRouterAgentNode(FrontendNode):
type_name="vectorstorerouter_agent", type_name="vectorstorerouter_agent",
fields=[ fields=[
TemplateField( TemplateField(
field_type="VectorStoreRouterToolkit", field_type="VectorStoreRouterToolkit", # pyright: ignore
required=True, required=True,
show=True, show=True,
name="vectorstoreroutertoolkit", name="vectorstoreroutertoolkit",
display_name="Vector Store Router Toolkit", display_name="Vector Store Router Toolkit",
), ),
TemplateField( TemplateField(
field_type="BaseLanguageModel", field_type="BaseLanguageModel", # pyright: ignore
required=True, required=True,
show=True, show=True,
name="llm", name="llm",
@ -88,14 +87,14 @@ class VectorStoreAgentNode(FrontendNode):
type_name="vectorstore_agent", type_name="vectorstore_agent",
fields=[ fields=[
TemplateField( TemplateField(
field_type="VectorStoreInfo", field_type="VectorStoreInfo", # pyright: ignore
required=True, required=True,
show=True, show=True,
name="vectorstoreinfo", name="vectorstoreinfo",
display_name="Vector Store Info", display_name="Vector Store Info",
), ),
TemplateField( TemplateField(
field_type="BaseLanguageModel", field_type="BaseLanguageModel", # pyright: ignore
required=True, required=True,
show=True, show=True,
name="llm", name="llm",
@ -116,9 +115,9 @@ class SQLDatabaseNode(FrontendNode):
type_name="sql_database", type_name="sql_database",
fields=[ fields=[
TemplateField( TemplateField(
field_type="str", field_type="str", # pyright: ignore
required=True, required=True,
is_list=False, is_list=False, # pyright: ignore
show=True, show=True,
multiline=False, multiline=False,
value="", value="",
@ -139,15 +138,15 @@ class CSVAgentNode(FrontendNode):
type_name="csv_agent", type_name="csv_agent",
fields=[ fields=[
TemplateField( TemplateField(
field_type="file", field_type="file", # pyright: ignore
required=True, required=True,
show=True, show=True,
name="path", name="path",
value="", value="",
file_types=[".csv"], file_types=[".csv"], # pyright: ignore
), ),
TemplateField( TemplateField(
field_type="BaseLanguageModel", field_type="BaseLanguageModel", # pyright: ignore
required=True, required=True,
show=True, show=True,
name="llm", name="llm",
@ -169,9 +168,9 @@ class InitializeAgentNode(FrontendNode):
type_name="initialize_agent", type_name="initialize_agent",
fields=[ fields=[
TemplateField( TemplateField(
field_type="str", field_type="str", # pyright: ignore
required=True, required=True,
is_list=True, is_list=True, # pyright: ignore
show=True, show=True,
multiline=False, multiline=False,
options=list(NON_CHAT_AGENTS.keys()), options=list(NON_CHAT_AGENTS.keys()),
@ -180,22 +179,22 @@ class InitializeAgentNode(FrontendNode):
advanced=False, advanced=False,
), ),
TemplateField( TemplateField(
field_type="BaseChatMemory", field_type="BaseChatMemory", # pyright: ignore
required=False, required=False,
show=True, show=True,
name="memory", name="memory",
advanced=False, advanced=False,
), ),
TemplateField( TemplateField(
field_type="Tool", field_type="Tool", # pyright: ignore
required=True, required=True,
show=True, show=True,
name="tools", name="tools",
is_list=True, is_list=True, # pyright: ignore
advanced=False, advanced=False,
), ),
TemplateField( TemplateField(
field_type="BaseLanguageModel", field_type="BaseLanguageModel", # pyright: ignore
required=True, required=True,
show=True, show=True,
name="llm", name="llm",
@ -222,13 +221,13 @@ class JsonAgentNode(FrontendNode):
type_name="json_agent", type_name="json_agent",
fields=[ fields=[
TemplateField( TemplateField(
field_type="BaseToolkit", field_type="BaseToolkit", # pyright: ignore
required=True, required=True,
show=True, show=True,
name="toolkit", name="toolkit",
), ),
TemplateField( TemplateField(
field_type="BaseLanguageModel", field_type="BaseLanguageModel", # pyright: ignore
required=True, required=True,
show=True, show=True,
name="llm", name="llm",