Update results.py

results.py

@@ -2,12 +2,13 @@ import os
 import openai
 import json, csv
 
-def results_agent(query, key):
+def results_agent(query, context):
     
     system_prompt = """
-    You are an academic advisor helping students find classes for the next semester. 
+    You are an academic advisor helping students (user role) find classes for the next semester. 
     Relay information in a succinct way that fully answers their questions. 
-    Query Results are sometimes provided to you in naming classes. Do not name classes that do not exist in chat history or in query results!! the scores (lower is better) from the context to help guide a student in finding the best class based on their query (a higher score is not always better). 
+    You will be given a user's query, as well as the chat history and finally, the rag responses in context.
+        Please use your own discretion to determine form the rag responses (which have scores to help you, higher is better) the best way to respond to the user's query
     Please add new lines, bolding/italicizing, formatting like "##" for headers to make it a pretty output as well.
     """
 
@@ -15,7 +16,8 @@ def results_agent(query, key):
         model="gpt-3.5-turbo",
         messages=[
             {"role": "system", "content": system_prompt},
-            {"role": "user", "content": query}
+            {"role": "user", "content": query},
+            {"role": "assistant", "Additional Context": context}
         ]
     )