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

fix: patched csv agent

Browse source
This commit is contained in:
martins0n 2023-10-30 23:02:13 +03:00
commit 21c38aaa01
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/interface/agents/custom.py
View file

@ -106,9 +106,9 @@ class CSVAgent(CustomAgentExecutor):
tools, tools,
prefix=PANDAS_PREFIX, prefix=PANDAS_PREFIX,
suffix=PANDAS_SUFFIX, 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_chain = LLMChain(
llm=llm, llm=llm,
prompt=partial_prompt, prompt=partial_prompt,