Fix sync route for list messages for step (#5764)
* fix: list messages for step route * fix: list messages for step route --------- Co-authored-by: Shubham Naik <shub@memgpt.ai>
This commit is contained in:
committed by
Caren Thomas
parent
e7fff12da0
commit
e3b0398aee
@@ -278,15 +278,14 @@ class StepManager:
|
||||
ascending: bool = False,
|
||||
) -> List[PydanticMessage]:
|
||||
async with db_registry.async_session() as session:
|
||||
messages = MessageModel.list(
|
||||
messages = await MessageModel.list_async(
|
||||
db_session=session,
|
||||
before=before,
|
||||
after=after,
|
||||
ascending=ascending,
|
||||
limit=limit,
|
||||
actor=actor,
|
||||
join_model=StepModel,
|
||||
join_conditions=[MessageModel.step.id == step_id],
|
||||
step_id=step_id,
|
||||
)
|
||||
return [message.to_pydantic() for message in messages]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user