Update main.py

main.py

@@ -55,7 +55,7 @@ def create_agent(filename: str):
     df = pd.read_csv(filename)
 
     # Create a Pandas DataFrame agent.
-    return create_pandas_dataframe_agent(llm, df, verbose=False, allow_dangerous_code=True, handle_parsing_errors=True, agent_type=AgentType.OPENAI_FUNCTIONS)
+    return create_csv_agent(llm, df, verbose=False, allow_dangerous_code=True, handle_parsing_errors=True, agent_type=AgentType.OPENAI_FUNCTIONS)
 
 def query_agent(agent, query):
     """
@@ -155,7 +155,7 @@ async def set_starters():
 async def on_message(message: cl.Message):
     await cl.Message(f"> SURVEYIA").send()
     agent = create_agent("./public/ExpeCFA_LP_CAA_7-5-2024.csv")
-    result = agent.invoke(message.content)
+    result = agent.run(message.content)
     # Query the agent.
     #response = query_agent(agent=agent, query=message.content)
     # Decode the response.