Spaces:
Runtime error
Runtime error
File size: 458 Bytes
781e8b6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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,
)
|