fix: patched csv agent

2023-10-30 23:02:13 +03:00 · 2023-10-30 23:02:13 +03:00 · 21c38aaa01
commit 21c38aaa01
parent 42d56f70ed
1 changed files with 2 additions and 2 deletions
--- 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,