| from langchain.agents.agent_types import AgentType | |
| from langchain_experimental.agents.agent_toolkits import create_pandas_dataframe_agent | |
| from langchain_openai import ChatOpenAI | |
| from app.utils.data_preparation import load_data | |
| df = load_data() | |
| real_state_agent = create_pandas_dataframe_agent( | |
| llm=ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613"), | |
| df=df, | |
| # prefix = prefix, | |
| verbose=True, | |
| agent_type=AgentType.OPENAI_FUNCTIONS, | |
| ) | |