bhulston commited on
Commit
35a89f8
·
1 Parent(s): a18bd26

Update results.py

Browse files
Files changed (1) hide show
  1. results.py +2 -1
results.py CHANGED
@@ -2,7 +2,7 @@ import os
2
  import openai
3
  import json, csv
4
 
5
- def results_agent(query):
6
 
7
  system_prompt = """
8
  You are an academic advisor helping students find classes for the next semester.
@@ -10,6 +10,7 @@ def results_agent(query):
10
  """
11
 
12
  response = openai.ChatCompletion.create(
 
13
  model="gpt-4",
14
  messages=[
15
  {"role": "system", "content": system_prompt},
 
2
  import openai
3
  import json, csv
4
 
5
+ def results_agent(query, key):
6
 
7
  system_prompt = """
8
  You are an academic advisor helping students find classes for the next semester.
 
10
  """
11
 
12
  response = openai.ChatCompletion.create(
13
+ openai.api_key = key,
14
  model="gpt-4",
15
  messages=[
16
  {"role": "system", "content": system_prompt},