Update app.py

app.py

@@ -103,7 +103,7 @@ def get_rag_results(prompt):
         additional_metadata = "None"
     else:
         response = reranker(query, response) # BERT cross encoder for ranking
-        print("Here is the response from the reranker", response)
+        print("##Here is the response from the reranker", response)
     
 
     return response, additional_metadata
@@ -130,15 +130,16 @@ if prompt := st.chat_input("What kind of class are you looking for?"):
         full_response = ""
 
         messages = [{"role": m["role"], "content": m["content"]}
-                for m in st.session_state.messages]
+                for m in st.session_state.messages[-6:]]
         message_history = " ".join([message["content"] for message in messages])
         
         rag_response, additional_metadata = get_rag_results(prompt)
+        rag_response = " ".join([message for message in rag_response])
         st.session_state.rag_responses.append(rag_response)
-        rag_responses = " ".join([message["content"] for message in messages])
+        all_rag_responses = " ".join([response for response in st.session_state.rag_responses])
         result_query = 'Original Query:' + prompt
         # '\n Additional Class Times:' + str(additional_metadata)
-        assistant_response = results_agent(result_query, rag_responses + "\nMessage_history" + message_history)
+        assistant_response = results_agent(result_query, "Class Options from RAG:" + all_rag_responses + "\nMessage_history" + message_history)
             # assistant_response = openai.ChatCompletion.create(
             #     model = "gpt-4",
             #     messages = [