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

fix: use conversation message_ids when exporting agent with conversation_id (#9294)

Browse Source 
When exporting an agent with a conversation_id, the export function was
setting agent_state.message_ids from the conversation, but from_agent_state
was ignoring this and fetching messages generically via list_messages.

Now from_agent_state checks if message_ids is set and fetches those specific
messages instead.

👾 Generated with [Letta Code](https://letta.com)

Co-authored-by: Letta <noreply@letta.com>
This commit is contained in:
cthomas
2026-02-04 18:08:09 -08:00
committed by Caren Thomas
parent df85ee970b
commit e8db3ac89a
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
letta/schemas/agent_file.py
View File

@@ -191,7 +191,14 @@ class AgentSchema(CreateAgent):
per_file_view_window_char_limit=agent_state.per_file_view_window_char_limit,
)
messages = await message_manager.list_messages(agent_id=agent_state.id, actor=actor, limit=50) # TODO: Expand to get more messages
# If agent_state.message_ids is set (e.g., from conversation export), fetch those specific messages
# Otherwise fall back to listing messages by agent_id
if agent_state.message_ids:
messages = await message_manager.get_messages_by_ids_async(message_ids=agent_state.message_ids, actor=actor)
else:
messages = await message_manager.list_messages(
agent_id=agent_state.id, actor=actor, limit=50
) # TODO: Expand to get more messages
# Convert messages to MessageSchema objects
message_schemas = [MessageSchema.from_message(msg) for msg in messages]