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

fix: test sdk client due to message batch route ordering (#8733)

Browse Source 
* base

* generate
This commit is contained in:
jnjpng
2026-01-14 18:16:39 -08:00
committed by Sarah Wooders
parent aabd58628e
commit 58a5375c19
1 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
letta/server/rest_api/routers/v1/messages.py
View File

@@ -58,22 +58,6 @@ async def list_all_messages(
)
@router.get("/{message_id}", response_model=MessagesResponse, operation_id="retrieve_message")
async def retrieve_message(
message_id: MessageId,
server: SyncServer = Depends(get_letta_server),
headers: HeaderParams = Depends(get_headers),
):
"""
Retrieve a message by ID.
"""
actor = await server.user_manager.get_actor_or_default_async(actor_id=headers.actor_id)
message = await server.message_manager.get_message_by_id_async(message_id=message_id, actor=actor)
if message is None:
raise HTTPException(status_code=404, detail=f"Message with id {message_id} not found.")
return message.to_letta_messages()
@router.post("/search", response_model=List[LettaMessageSearchResult], operation_id="search_all_messages")
async def search_all_messages(
request: SearchAllMessagesRequest = Body(...),
@@ -281,3 +265,19 @@ async def cancel_batch(
# Update all the batch_job statuses
await server.batch_manager.update_llm_batch_status_async(llm_batch_id=llm_batch_job.id, status=JobStatus.cancelled, actor=actor)
@router.get("/{message_id}", response_model=MessagesResponse, operation_id="retrieve_message")
async def retrieve_message(
message_id: MessageId,
server: SyncServer = Depends(get_letta_server),
headers: HeaderParams = Depends(get_headers),
):
"""
Retrieve a message by ID.
"""
actor = await server.user_manager.get_actor_or_default_async(actor_id=headers.actor_id)
message = await server.message_manager.get_message_by_id_async(message_id=message_id, actor=actor)
if message is None:
raise HTTPException(status_code=404, detail=f"Message with id {message_id} not found.")
return message.to_letta_messages()