feat: Add Pandas kwargs support for CSV Agent (#5372)
* ✨ (csv.py): Add support for passing Pandas kwargs to the CSV agent for more customization and flexibility. * 📝 (csv.py): mark 'is_list' attribute as True for the 'Pandas Kwargs' parameter to indicate it can accept a list of values
This commit is contained in:
parent
e657e9de4a
commit
b07bd173b4
1 changed files with 10 additions and 1 deletions
|
|
@ -3,7 +3,7 @@ from langchain_experimental.agents.agent_toolkits.csv.base import create_csv_age
|
|||
from langflow.base.agents.agent import LCAgentComponent
|
||||
from langflow.field_typing import AgentExecutor
|
||||
from langflow.inputs import DropdownInput, FileInput, HandleInput
|
||||
from langflow.inputs.inputs import MessageTextInput
|
||||
from langflow.inputs.inputs import DictInput, MessageTextInput
|
||||
from langflow.schema.message import Message
|
||||
from langflow.template.field.base import Output
|
||||
|
||||
|
|
@ -44,6 +44,13 @@ class CSVAgentComponent(LCAgentComponent):
|
|||
display_name="Text",
|
||||
info="Text to be passed as input and extract info from the CSV File.",
|
||||
),
|
||||
DictInput(
|
||||
name="pandas_kwargs",
|
||||
display_name="Pandas Kwargs",
|
||||
info="Pandas Kwargs to be passed to the agent.",
|
||||
advanced=True,
|
||||
is_list=True,
|
||||
),
|
||||
]
|
||||
|
||||
outputs = [
|
||||
|
|
@ -67,6 +74,7 @@ class CSVAgentComponent(LCAgentComponent):
|
|||
path=self._path(),
|
||||
agent_type=self.agent_type,
|
||||
handle_parsing_errors=self.handle_parsing_errors,
|
||||
pandas_kwargs=self.pandas_kwargs,
|
||||
**agent_kwargs,
|
||||
)
|
||||
|
||||
|
|
@ -84,6 +92,7 @@ class CSVAgentComponent(LCAgentComponent):
|
|||
path=self._path(),
|
||||
agent_type=self.agent_type,
|
||||
handle_parsing_errors=self.handle_parsing_errors,
|
||||
pandas_kwargs=self.pandas_kwargs,
|
||||
**agent_kwargs,
|
||||
)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue