Upload app_federal.py

app_federal.py

@@ -250,7 +250,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[:25]:
+    for i in search_results[:50]:
         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)
@@ -303,7 +303,8 @@ def run_retrieval(query):
         else:
             item["metadata_reranking"] = ""
         out_dict.append(item)
-    print (out_dict[:20])
+    print ("out_dict_before_reranking")
+    print (out_dict[:50])
     out_dict = rerank_with_chatGPT(query, out_dict)[:NUM_RESULTS]
     print ("RETURNING OUT DICT")
     return out_dict