Update results.py
Browse files- 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},
|