Update app.py
Browse files
app.py
CHANGED
@@ -103,7 +103,7 @@ def get_rag_results(prompt):
|
|
103 |
additional_metadata = "None"
|
104 |
else:
|
105 |
response = reranker(query, response) # BERT cross encoder for ranking
|
106 |
-
print("Here is the response from the reranker", response)
|
107 |
|
108 |
|
109 |
return response, additional_metadata
|
@@ -130,15 +130,16 @@ if prompt := st.chat_input("What kind of class are you looking for?"):
|
|
130 |
full_response = ""
|
131 |
|
132 |
messages = [{"role": m["role"], "content": m["content"]}
|
133 |
-
for m in st.session_state.messages]
|
134 |
message_history = " ".join([message["content"] for message in messages])
|
135 |
|
136 |
rag_response, additional_metadata = get_rag_results(prompt)
|
|
|
137 |
st.session_state.rag_responses.append(rag_response)
|
138 |
-
|
139 |
result_query = 'Original Query:' + prompt
|
140 |
# '\n Additional Class Times:' + str(additional_metadata)
|
141 |
-
assistant_response = results_agent(result_query,
|
142 |
# assistant_response = openai.ChatCompletion.create(
|
143 |
# model = "gpt-4",
|
144 |
# messages = [
|
|
|
103 |
additional_metadata = "None"
|
104 |
else:
|
105 |
response = reranker(query, response) # BERT cross encoder for ranking
|
106 |
+
print("##Here is the response from the reranker", response)
|
107 |
|
108 |
|
109 |
return response, additional_metadata
|
|
|
130 |
full_response = ""
|
131 |
|
132 |
messages = [{"role": m["role"], "content": m["content"]}
|
133 |
+
for m in st.session_state.messages[-6:]]
|
134 |
message_history = " ".join([message["content"] for message in messages])
|
135 |
|
136 |
rag_response, additional_metadata = get_rag_results(prompt)
|
137 |
+
rag_response = " ".join([message for message in rag_response])
|
138 |
st.session_state.rag_responses.append(rag_response)
|
139 |
+
all_rag_responses = " ".join([response for response in st.session_state.rag_responses])
|
140 |
result_query = 'Original Query:' + prompt
|
141 |
# '\n Additional Class Times:' + str(additional_metadata)
|
142 |
+
assistant_response = results_agent(result_query, "Class Options from RAG:" + all_rag_responses + "\nMessage_history" + message_history)
|
143 |
# assistant_response = openai.ChatCompletion.create(
|
144 |
# model = "gpt-4",
|
145 |
# messages = [
|