saaketvarma commited on
Commit
60f4dd4
·
1 Parent(s): c1bfd08

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +3 -4
utils.py CHANGED
@@ -10,12 +10,11 @@ def query_agent(data, query):
10
  # Parse the CSV file and create a Pandas DataFrame from its contents.
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
 
10
  # Parse the CSV file and create a Pandas DataFrame from its contents.
11
  df = pd.read_csv(data)
12
 
13
+ llm = ChatOpenAI(model_name="gpt-4", temperature=0.1)
14
+
 
15
 
16
  # Create a Pandas DataFrame agent.
17
+ agent = create_pandas_dataframe_agent(llm, df, verbose=True)
18
 
19
  #Python REPL: A Python shell used to evaluating and executing Python commands.
20
  #It takes python code as input and outputs the result. The input python code can be generated from another tool in the LangChain