Spaces:
Sleeping
Sleeping
File size: 534 Bytes
545c4ba |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
# from langchain.agents.agent_types import AgentType
import pandas as pd
import sys
sys.path.append('..')
from langchain_experimental.agents.agent_toolkits import create_pandas_dataframe_agent
from langchain_openai import OpenAI, ChatOpenAI
from langchain.agents.agent_types import AgentType
def init_config(df):
llm = ChatOpenAI(model_name="gpt-4")
agent = create_pandas_dataframe_agent(llm=llm, df=df, verbose=True)
return agent
def answer_question(agent, question):
return agent.invoke(question)['output']
|