Update app.py

app.py

@@ -18,7 +18,7 @@ from results import results_agent
 from filter import filter_agent
 from reranker import reranker
 from utils import build_filter, clean_pinecone
-from router import routing_agent
+from keyword import keyword_agent
 
 OPENAI_API = st.secrets["OPENAI_API"]
 PINECONE_API = st.secrets["PINECONE_API"]
@@ -67,7 +67,10 @@ def get_rag_results(prompt):
 3. Rerank the results from vector DB using a BERT-based cross encoder
     '''
     query = filter_agent(prompt, OPENAI_API)
-    print("Here is the response from the filter_agent", query)
+    print("Here is the response from the filter_agent:", query)
+
+    query += keyword_agent(query)
+    print("Here is the new query with keywords added:", query)
 
   ##Get metadata filters  
     days_filter = list()