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},
|