74 lines
2.7 KiB
Python
74 lines
2.7 KiB
Python
|
from typing import Any
|
||
|
|
|
||
|
|
from fastapi import APIRouter, Depends, Header
|
||
|
|
from pydantic import Field
|
||
|
|
|
||
|
|
from app.api.dependencies.auth import build_request_context
|
||
|
|
from app.api.schemas.admin import StrictPayload
|
||
|
|
from app.core.contracts import RequestContext
|
||
|
|
from app.service.public_platform_service import PublicPlatformService
|
||
|
|
|
||
|
|
router = APIRouter(prefix="/api/v1", tags=["public-platform"])
|
||
|
|
|
||
|
|
|
||
|
|
class SessionCreate(StrictPayload):
|
||
|
|
agent_type: str = Field(pattern=r"^[a-z][a-z0-9_]{1,31}$")
|
||
|
|
|
||
|
|
|
||
|
|
class Cancellation(StrictPayload):
|
||
|
|
reason: str = Field(default="user_cancelled", max_length=128)
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/conversations", status_code=201)
|
||
|
|
async def create_session(
|
||
|
|
payload: SessionCreate,
|
||
|
|
key: str | None = Header(default=None, alias="Idempotency-Key"),
|
||
|
|
context: RequestContext = Depends(build_request_context), # noqa: B008
|
||
|
|
) -> dict[str, Any]:
|
||
|
|
return await PublicPlatformService().write("create", "", context, key, payload.model_dump())
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/conversations/{session_id}")
|
||
|
|
async def get_session(
|
||
|
|
session_id: str, context: RequestContext = Depends(build_request_context), # noqa: B008
|
||
|
|
) -> dict[str, Any]:
|
||
|
|
return await PublicPlatformService().session(session_id, context)
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/conversations/{session_id}/closures")
|
||
|
|
async def close_session(
|
||
|
|
session_id: str, key: str | None = Header(default=None, alias="Idempotency-Key"),
|
||
|
|
context: RequestContext = Depends(build_request_context), # noqa: B008
|
||
|
|
) -> dict[str, Any]:
|
||
|
|
return await PublicPlatformService().write("close", session_id, context, key, {})
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/agent-runs/{run_id}/cancellations", status_code=202)
|
||
|
|
async def cancel_run(
|
||
|
|
run_id: str, payload: Cancellation,
|
||
|
|
key: str | None = Header(default=None, alias="Idempotency-Key"),
|
||
|
|
context: RequestContext = Depends(build_request_context), # noqa: B008
|
||
|
|
) -> dict[str, Any]:
|
||
|
|
return await PublicPlatformService().write("cancel", run_id, context, key, payload.model_dump())
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/handover-requests/{handover_id}")
|
||
|
|
async def get_handover(
|
||
|
|
handover_id: str, context: RequestContext = Depends(build_request_context), # noqa: B008
|
||
|
|
) -> dict[str, Any]:
|
||
|
|
return await PublicPlatformService().handover(handover_id, context)
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/users/me/memory-profile")
|
||
|
|
async def my_memory(
|
||
|
|
context: RequestContext = Depends(build_request_context), # noqa: B008
|
||
|
|
) -> dict[str, Any]:
|
||
|
|
return await PublicPlatformService().memory(int(context.user_id), context)
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/customers/{customer_id}/memory-profile")
|
||
|
|
async def customer_memory(
|
||
|
|
customer_id: int, context: RequestContext = Depends(build_request_context), # noqa: B008
|
||
|
|
) -> dict[str, Any]:
|
||
|
|
return await PublicPlatformService().memory(customer_id, context)
|