Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: add null check in context window endpoint [LET-6253] (#6301)

Browse Source 
fix: add null check in context window endpoint
This commit is contained in:
cthomas
2025-11-20 12:51:51 -08:00
committed by Caren Thomas
parent 345ea42630
commit 2d1667a97c
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
letta/services/context_window_calculator/context_window_calculator.py
View File

@@ -125,6 +125,15 @@ class ContextWindowCalculator:
messages = await message_manager.get_messages_by_ids_async(message_ids=agent_state.message_ids[1:], actor=actor)
in_context_messages = [system_message_compiled] + messages
# Filter out None messages (can occur when system message is missing)
original_count = len(in_context_messages)
in_context_messages = [m for m in in_context_messages if m is not None]
if len(in_context_messages) < original_count:
logger.warning(
f"Filtered out {original_count - len(in_context_messages)} None messages for agent {agent_state.id}. "
f"This typically indicates missing system message or corrupted message data."
)
# Convert messages to appropriate format
converted_messages = token_counter.convert_messages(in_context_messages)