saaketvarma commited on
Commit
bcb43b6
·
1 Parent(s): e751c60

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +5 -1
utils.py CHANGED
@@ -9,7 +9,11 @@ df = pd.read_csv("Tickets_cleaned_v2.csv")
9
 
10
  def query_agent(query):
11
 
12
- llm = OpenAI()
 
 
 
 
13
 
14
 
15
  # Create a Pandas DataFrame agent.
 
9
 
10
  def query_agent(query):
11
 
12
+ llm = ChatOpenAI(
13
+ model_name="gpt-4-320k",
14
+ temperature=0.2,
15
+ openai_api_key=openai_api_key,
16
+ )
17
 
18
 
19
  # Create a Pandas DataFrame agent.