Upload app_federal.py

app_federal.py

@@ -330,12 +330,13 @@ def run_retrieval(query):
     #print (out_dict[:50])
     res = {"result_type":"chatgpt_reranking"}
     res["query"] = query
-    res["input_reranking"] = [i["index"] for i in out_dict]
-    res["scores_input_reranking"] = [i["NV_score"] for i in out_dict]
+    res["input_reranking"] = [int(i["index"]) for i in out_dict]
+    res["scores_input_reranking"] = [float(i["NV_score"]) for i in out_dict]
     out_dict = rerank_with_chatGPT(query, out_dict)[:NUM_RESULTS]
 
-    res["output_reranking"] = [i["index"] for i in out_dict]
-    res["scores_output_reranking"] = [i["NV_score"] for i in out_dict]
+    res["output_reranking"] = [int(i["index"]) for i in out_dict]
+    res["scores_output_reranking"] = [float(i["NV_score"]) for i in out_dict]
+    print (res)
 
     # is that already good?
     with scheduler.lock: