Spaces:
Sleeping
Sleeping
remis
Browse files
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
|
| 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
|