From 21c38aaa01ce3acb9406e628cffc97c83a90aa12 Mon Sep 17 00:00:00 2001
From: martins0n <33594071+martins0n@users.noreply.github.com>
Date: Mon, 30 Oct 2023 23:02:13 +0300
Subject: [PATCH] fix: patched csv agent

---
 src/backend/langflow/interface/agents/custom.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/backend/langflow/interface/agents/custom.py b/src/backend/langflow/interface/agents/custom.py
index 735b27917..d84ab8d18 100644
--- a/src/backend/langflow/interface/agents/custom.py
+++ b/src/backend/langflow/interface/agents/custom.py
@@ -106,9 +106,9 @@ class CSVAgent(CustomAgentExecutor):
             tools,
             prefix=PANDAS_PREFIX,
             suffix=PANDAS_SUFFIX,
-            input_variables=["df", "input", "agent_scratchpad"],
+            input_variables=["df_head", "input", "agent_scratchpad"],
         )
-        partial_prompt = prompt.partial(df=str(df.head()))
+        partial_prompt = prompt.partial(df_head=str(df.head()))
         llm_chain = LLMChain(
             llm=llm,
             prompt=partial_prompt,