From b21bb0a7e13589f67ec1ad7d5154bc361f9f9f8e Mon Sep 17 00:00:00 2001 From: Rajat Ahuja Date: Tue, 5 May 2026 15:31:08 -0400 Subject: [PATCH] fix: comments --- src/crud/message.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/crud/message.py b/src/crud/message.py index 25db88a5..76f9cab7 100644 --- a/src/crud/message.py +++ b/src/crud/message.py @@ -141,7 +141,6 @@ async def _build_merged_snippets( session_matches.setdefault(msg.session_name, []).append(msg) # Build merged ranges per session, then issue a single batched query - # across all sessions to avoid an N+1 over distinct sessions. session_ranges: dict[str, list[tuple[int, int, list[models.Message]]]] = {} for sess_name, matches in session_matches.items(): matches.sort(key=lambda m: m.seq_in_session) @@ -163,8 +162,7 @@ async def _build_merged_snippets( session_ranges[sess_name] = merged_ranges - # One OR-of-ANDs predicate covers every (session, range) pair so we - # hit the DB once regardless of how many sessions matched. + # One OR-of-ANDs predicate covers every (session, range) pair session_predicates = [ and_( models.Message.session_name == sess_name, @@ -193,7 +191,9 @@ async def _build_merged_snippets( for msg in context_result.scalars().all(): by_session.setdefault(msg.session_name, []).append(msg) - snippets: list[tuple[list[models.Message], list[models.Message]]] = [] + snippets: list[ + tuple[list[models.Message], list[models.Message]] + ] = [] # list of tuples, each containing query matches and context messages for sess_name, merged_ranges in session_ranges.items(): all_context_messages = by_session.get(sess_name, []) for start_seq, end_seq, range_matches in merged_ranges: