Update main.py
Browse files
main.py
CHANGED
|
@@ -51,7 +51,7 @@ async def on_message(message: cl.Message):
|
|
| 51 |
)
|
| 52 |
model = await LLMistral()
|
| 53 |
|
| 54 |
-
agent = create_pandas_dataframe_agent(model, df, verbose=True, allow_dangerous_code=True)
|
| 55 |
|
| 56 |
#agent = create_csv_agent(
|
| 57 |
# model,
|
|
|
|
| 51 |
)
|
| 52 |
model = await LLMistral()
|
| 53 |
|
| 54 |
+
agent = create_pandas_dataframe_agent(model, df, verbose=True, allow_dangerous_code=True, handle_parsing_errors=True)
|
| 55 |
|
| 56 |
#agent = create_csv_agent(
|
| 57 |
# model,
|