feat: sort search results
This commit is contained in:
parent
957a8853f2
commit
01dc2240a2
1 changed files with 11 additions and 0 deletions
|
|
@ -160,5 +160,16 @@ with search_tab:
|
||||||
st.session_state["search_results"] = vector_search(
|
st.session_state["search_results"] = vector_search(
|
||||||
search_term, 100, search_sources, search_notes
|
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"]:
|
for item in st.session_state["search_results"]:
|
||||||
results_card(item)
|
results_card(item)
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue