From 6290837d15727d294ea84b09efc3425cab911820 Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Mon, 15 Jan 2024 17:25:33 -0300 Subject: [PATCH] Refactor CSVAgent to use create_csv_agent function --- src/backend/langflow/components/agents/CSVAgent.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/components/agents/CSVAgent.py b/src/backend/langflow/components/agents/CSVAgent.py index af3766c74..30037dff2 100644 --- a/src/backend/langflow/components/agents/CSVAgent.py +++ b/src/backend/langflow/components/agents/CSVAgent.py @@ -1,6 +1,6 @@ from langflow import CustomComponent from langflow.field_typing import BaseLanguageModel, AgentExecutor - +from langchain_experimental.agents.agent_toolkits.csv.base import create_csv_agent class CSVAgentComponent(CustomComponent): display_name = "CSVAgent" @@ -10,7 +10,7 @@ class CSVAgentComponent(CustomComponent): def build_config(self): return { "llm": {"display_name": "LLM", "type": BaseLanguageModel}, - "path": {"display_name": "Path", "type": "file", "suffixes": [".csv"], "file_types": ["csv"]}, + "path": {"display_name": "Path", "field_type": "file", "suffixes": [".csv"], "file_types": [".csv"]}, } def build( @@ -19,4 +19,4 @@ class CSVAgentComponent(CustomComponent): path: str, ) -> AgentExecutor: # Instantiate and return the CSV agent class with the provided llm and path - return AgentExecutor(llm=llm, path=path) + return create_csv_agent(llm=llm, path=path)