Aurele000 commited on
Commit
a404d0c
·
1 Parent(s): c04c6a1
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -146,7 +146,7 @@ def create_agent_and_answer(input: ToolInput) -> str:
146
  """
147
  df = pd.DataFrame(input.data)
148
  question = input.question
149
- agent_excel = create_pandas_dataframe_agent(llm_4o, df, verbose=True)#, allow_dangerous_code=True) todo : remettre
150
  text = agent_excel.run(question)
151
 
152
  return text
 
146
  """
147
  df = pd.DataFrame(input.data)
148
  question = input.question
149
+ agent_excel = create_pandas_dataframe_agent(llm_4o, df, verbose=True, allow_dangerous_code=True)
150
  text = agent_excel.run(question)
151
 
152
  return text