admin-valentin/open-notebook
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

review: prevent mutation and remove duplicate final_score calculation

Browse source
This commit is contained in:
LUIS NOVO 2025-06-26 11:56:01 -03:00
parent e3ee803a42
commit 26da01935a
1 changed files with 5 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
pages/3_🔍_Ask_and_Search.py
View file

@ -40,10 +40,9 @@ async def process_ask_query(question, strategy_model, answer_model, final_answer
def results_card(item): def results_card(item):
score = item.get("relevance", item.get("similarity", item.get("score", 0)))
with st.container(border=True): with st.container(border=True):
st.markdown( st.markdown(
f"[{score:.2f}] **[{item['title']}](/?object_id={item['parent_id']})**" f"[{item['final_score']:.2f}] **[{item['title']}](/?object_id={item['parent_id']})**"
) )
if "matches" in item: if "matches" in item:
with st.expander("Matches"): with st.expander("Matches"):
@ -161,15 +160,14 @@ with search_tab:
search_term, 100, search_sources, search_notes search_term, 100, search_sources, search_notes
) )
for item in st.session_state["search_results"]: search_results = st.session_state["search_results"].copy()
for item in search_results:
item["final_score"] = item.get( item["final_score"] = item.get(
"relevance", item.get("similarity", item.get("score", 0)) "relevance", item.get("similarity", item.get("score", 0))
) )
# Sort search results by final_score in descending order # Sort search results by final_score in descending order
st.session_state["search_results"].sort( search_results.sort(key=lambda x: x["final_score"], reverse=True)
key=lambda x: x["final_score"], reverse=True
)
for item in st.session_state["search_results"]: for item in search_results:
results_card(item) results_card(item)