Update main.py

main.py

@@ -5,7 +5,7 @@ import pandas as pd
 import numpy as np
 from typing import List
 from pathlib import Path
-from langchain_openai import ChatOpenAI, OpenAI
+from langchain_openai import ChatOpenAI
 from langchain.schema.runnable.config import RunnableConfig
 from langchain.schema import StrOutputParser
 from langchain_core.prompts import ChatPromptTemplate
@@ -49,13 +49,13 @@ def create_agent(filename: str):
 
     # Create an OpenAI object.
     os.environ['OPENAI_API_KEY'] = os.environ['OPENAI_API_KEY']
-    llm = OpenAI(temperature=0, model="gpt-4o-2024-05-13")
+    llm = ChatOpenAI(temperature=0, model="gpt-4o-2024-05-13")
 
     # Read the CSV file into a Pandas DataFrame.
     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)
+    return create_pandas_dataframe_agent(llm, df, verbose=False, allow_dangerous_code=True, handle_parsing_errors=True, agent_type=AgentType.OPENAI_FUNCTIONS)
 
 def query_agent(agent, query):
     """
@@ -155,12 +155,12 @@ 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)
     # Query the agent.
-    response = query_agent(agent=agent, query=message.content)
+    #response = query_agent(agent=agent, query=message.content)
     # Decode the response.
-    decoded_response = decode_response(response)
+    #decoded_response = decode_response(response)
 
     # Write the response to the Streamlit app.
-    result = write_response(decoded_response)   
+    #result = write_response(decoded_response)   
     await cl.Message(author="COPILOT",content=GoogleTranslator(source='auto', target='fr').translate(result)).send()