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

groups [LET-4632] (#5470)

Browse Source 
groups

Co-authored-by: Ari Webb <ari@letta.com>
This commit is contained in:
Ari Webb
2025-10-16 09:34:45 -07:00
committed by Caren Thomas
parent 46dc1c7e34
commit 045890531c
1 changed files with 8 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
letta/server/rest_api/routers/v1/groups.py
View File

@@ -1,11 +1,10 @@
from typing import Annotated, List, Literal, Optional
from fastapi import APIRouter, Body, Depends, Header, HTTPException, Query, status
from fastapi import APIRouter, Body, Depends, Header, Query, status
from fastapi.responses import JSONResponse
from pydantic import Field
from letta.constants import DEFAULT_MESSAGE_TOOL, DEFAULT_MESSAGE_TOOL_KWARG
from letta.orm.errors import NoResultFound
from letta.schemas.group import Group, GroupCreate, GroupUpdate, ManagerType
from letta.schemas.letta_message import LettaMessageUnion, LettaMessageUpdateUnion
from letta.schemas.letta_request import LettaRequest, LettaStreamingRequest
@@ -77,11 +76,7 @@ async def retrieve_group(
Retrieve the group by id.
"""
actor = await server.user_manager.get_actor_or_default_async(actor_id=headers.actor_id)
try:
return await server.group_manager.retrieve_group_async(group_id=group_id, actor=actor)
except NoResultFound as e:
raise HTTPException(status_code=404, detail=str(e))
return await server.group_manager.retrieve_group_async(group_id=group_id, actor=actor)
@router.post("/", response_model=Group, operation_id="create_group")
@@ -96,11 +91,8 @@ async def create_group(
"""
Create a new multi-agent group with the specified configuration.
"""
try:
actor = await server.user_manager.get_actor_or_default_async(actor_id=headers.actor_id)
return await server.group_manager.create_group_async(group, actor=actor)
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
actor = await server.user_manager.get_actor_or_default_async(actor_id=headers.actor_id)
return await server.group_manager.create_group_async(group, actor=actor)
@router.patch("/{group_id}", response_model=Group, operation_id="modify_group")
@@ -116,11 +108,8 @@ async def modify_group(
"""
Create a new multi-agent group with the specified configuration.
"""
try:
actor = await server.user_manager.get_actor_or_default_async(actor_id=headers.actor_id)
return await server.group_manager.modify_group_async(group_id=group_id, group_update=group, actor=actor)
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
actor = await server.user_manager.get_actor_or_default_async(actor_id=headers.actor_id)
return await server.group_manager.modify_group_async(group_id=group_id, group_update=group, actor=actor)
@router.delete("/{group_id}", response_model=None, operation_id="delete_group")
@@ -133,11 +122,8 @@ async def delete_group(
Delete a multi-agent group.
"""
actor = await server.user_manager.get_actor_or_default_async(actor_id=headers.actor_id)
try:
await server.group_manager.delete_group_async(group_id=group_id, actor=actor)
return JSONResponse(status_code=status.HTTP_200_OK, content={"message": f"Group id={group_id} successfully deleted"})
except NoResultFound:
raise HTTPException(status_code=404, detail=f"Group id={group_id} not found for user_id={actor.id}.")
await server.group_manager.delete_group_async(group_id=group_id, actor=actor)
return JSONResponse(status_code=status.HTTP_200_OK, content={"message": f"Group id={group_id} successfully deleted"})
@router.post(