diff --git a/pages/3_🔍_Ask_and_Search.py b/pages/3_🔍_Ask_and_Search.py index 8c92f61..8930469 100644 --- a/pages/3_🔍_Ask_and_Search.py +++ b/pages/3_🔍_Ask_and_Search.py @@ -160,5 +160,16 @@ with search_tab: st.session_state["search_results"] = vector_search( search_term, 100, search_sources, search_notes ) + + for item in st.session_state["search_results"]: + item["final_score"] = item.get( + "relevance", item.get("similarity", item.get("score", 0)) + ) + + # Sort search results by final_score in descending order + st.session_state["search_results"].sort( + key=lambda x: x["final_score"], reverse=True + ) + for item in st.session_state["search_results"]: results_card(item)