Skip to content

Commit 85bb4b6

Browse files
nit
1 parent 0af5df4 commit 85bb4b6

File tree

1 file changed

+10
-7
lines changed
  • backend/onyx/agents/agent_search/dr

1 file changed

+10
-7
lines changed

backend/onyx/agents/agent_search/dr/utils.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -194,13 +194,16 @@ def get_chat_history_string(chat_history: list[BaseMessage], max_messages: int)
194194
and content_piece.get("type") != "text"
195195
):
196196
removal_indices.append(content_piece_number)
197-
filtered_past_messages[past_message_number].content = [
198-
content_piece
199-
for content_piece_number, content_piece in enumerate(
200-
past_message.content
201-
)
202-
if content_piece_number not in removal_indices
203-
]
197+
198+
# Only rebuild the content list if there are items to remove
199+
if removal_indices:
200+
filtered_past_messages[past_message_number].content = [
201+
content_piece
202+
for content_piece_number, content_piece in enumerate(
203+
past_message.content
204+
)
205+
if content_piece_number not in removal_indices
206+
]
204207

205208
else:
206209
continue

0 commit comments

Comments
 (0)