datacipen commited on
Commit
32d38dd
1 Parent(s): 0667433

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -55,7 +55,7 @@ def create_agent(filename: str):
55
  df = pd.read_csv(filename)
56
 
57
  # Create a Pandas DataFrame agent.
58
- return create_csv_agent(llm, filename, verbose=False, allow_dangerous_code=True, agent_type=AgentType.OPENAI_FUNCTIONS)
59
 
60
  def query_agent(agent, query):
61
  """
 
55
  df = pd.read_csv(filename)
56
 
57
  # Create a Pandas DataFrame agent.
58
+ return create_csv_agent(llm, filename, verbose=False, allow_dangerous_code=True, handle_parsing_errors=True, agent_type=AgentType.OPENAI_FUNCTIONS)
59
 
60
  def query_agent(agent, query):
61
  """