saaketvarma commited on
Commit
c8d18cb
·
1 Parent(s): 158c91f

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +3 -3
utils.py CHANGED
@@ -11,11 +11,11 @@ def query_agent(data, query):
11
  df = pd.read_csv(data)
12
 
13
  llm = OpenAI()
14
- # Set the model to use
15
- #model = "gpt4"
16
 
17
  # Create a Pandas DataFrame agent.
18
- agent = create_pandas_dataframe_agent(llm, df, verbose=True)
19
 
20
  #Python REPL: A Python shell used to evaluating and executing Python commands.
21
  #It takes python code as input and outputs the result. The input python code can be generated from another tool in the LangChain
 
11
  df = pd.read_csv(data)
12
 
13
  llm = OpenAI()
14
+ Set the model to use
15
+ model = "gpt-4-32k"
16
 
17
  # Create a Pandas DataFrame agent.
18
+ agent = create_pandas_dataframe_agent(llm, df, model= model, verbose=True)
19
 
20
  #Python REPL: A Python shell used to evaluating and executing Python commands.
21
  #It takes python code as input and outputs the result. The input python code can be generated from another tool in the LangChain