Upload app_federal.py

app_federal.py

@@ -230,7 +230,7 @@ def rerank_with_chatGPT(query, search_results):
     Return a python list with the ids of the five highest ranking results, nothing else.
     <query>""" + query + "</query>\n\n"
     user_prompt = []
-    for i in search_results[:15]:
+    for i in search_results[:20]:
         user_prompt.append(format_metadata_for_reranking(i["metadata_reranking"], i["text"], i["index"]))
     user_prompt = "\n".join(user_prompt)
     out = text_prompt_call("gpt-4o", system_prompt, user_prompt)
@@ -254,7 +254,7 @@ def run_retrieval(query):
 
     query_embeddings = run_dense_retrieval(query)
     query_embeddings = pca_model.transform(query_embeddings)
-    D, I = faiss_index.search(query_embeddings, 100)
+    D, I = faiss_index.search(query_embeddings, 20)
     scores_embeddings = D[0]
     indices_embeddings = I[0]
     indices_embeddings = [int(i) for i in indices_embeddings]