Spaces:
Sleeping
Sleeping
File size: 1,410 Bytes
8293a2b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
import logging
from typing import List
from fastapi import APIRouter, Depends
from pydantic import BaseModel, Field
from aworld.cmd import SessionModel
from aworld.cmd.utils.agent_server import CURRENT_SERVER
from aworld.cmd.web.web_server import get_user_id_from_jwt
logger = logging.getLogger(__name__)
router = APIRouter()
prefix = "/api/session"
@router.get("/list")
async def list_sessions(
user_id: str = Depends(get_user_id_from_jwt),
) -> List[SessionModel]:
return await CURRENT_SERVER.get_session_service().list_sessions(user_id)
class CommonResponse(BaseModel):
code: int = Field(..., description="The code")
message: str = Field(..., description="The message")
@staticmethod
def success(message: str = "success"):
return CommonResponse(code=0, message=message)
@staticmethod
def error(message: str):
return CommonResponse(code=1, message=message)
class DeleteSessionRequest(BaseModel):
session_id: str = Field(..., description="The session id")
@router.post("/delete")
async def delete_session(
request: DeleteSessionRequest, user_id: str = Depends(get_user_id_from_jwt)
) -> CommonResponse:
try:
await CURRENT_SERVER.get_session_service().delete_session(
user_id, request.session_id
)
return CommonResponse.success()
except Exception as e:
return CommonResponse.error(str(e))
|