chore: initialize project repository
This commit is contained in:
@@ -0,0 +1,120 @@
|
||||
from typing import Any
|
||||
|
||||
from fastapi import APIRouter, Depends, Header, Path, Query, Response
|
||||
from pydantic import BaseModel
|
||||
|
||||
from app.api.dependencies.auth import build_request_context
|
||||
from app.api.schemas.admin import (
|
||||
EmptyPayload,
|
||||
EndpointPayload,
|
||||
IntentPayload,
|
||||
ItemPayload,
|
||||
NegativePayload,
|
||||
PromptPayload,
|
||||
ReleasePayload,
|
||||
ReplyPayload,
|
||||
ReviewPayload,
|
||||
RoutingPayload,
|
||||
)
|
||||
from app.core.contracts import RequestContext
|
||||
from app.service.admin_service import AdminService
|
||||
|
||||
router = APIRouter(prefix="/api/v1/admin", tags=["platform-admin"])
|
||||
|
||||
|
||||
def register_resource(
|
||||
resource: str, schema: type[BaseModel], id_name: str, *, scoped: bool = False,
|
||||
update: bool = True, detail: bool = True,
|
||||
) -> None:
|
||||
prefix = f"/config-releases/{{release_id}}/{resource}" if scoped else f"/{resource}"
|
||||
|
||||
async def create(
|
||||
payload: BaseModel, response: Response, release_id: int | None = None,
|
||||
context: RequestContext = Depends(build_request_context), # noqa: B008
|
||||
key: str | None = Header(default=None, alias="Idempotency-Key"),
|
||||
) -> dict[str, Any]:
|
||||
result = await AdminService().mutate(resource, context, payload.model_dump(mode="json"),
|
||||
key, None, release_id=release_id)
|
||||
response.headers["ETag"] = f'"{result["meta"]["etag"]}"'
|
||||
return result
|
||||
|
||||
create.__annotations__["payload"] = schema
|
||||
router.add_api_route(prefix, create, methods=["POST"], status_code=201,
|
||||
operation_id=f"create_{resource}")
|
||||
|
||||
async def list_rows(
|
||||
release_id: int | None = None, limit: int = Query(default=20, ge=1, le=100),
|
||||
context: RequestContext = Depends(build_request_context), # noqa: B008
|
||||
) -> dict[str, Any]:
|
||||
return await AdminService().query(resource, context, release_id=release_id, limit=limit)
|
||||
|
||||
router.add_api_route(prefix, list_rows, methods=["GET"], operation_id=f"list_{resource}")
|
||||
|
||||
async def get(
|
||||
response: Response, row_id: int = Path(alias=id_name, gt=0),
|
||||
context: RequestContext = Depends(build_request_context), # noqa: B008
|
||||
) -> dict[str, Any]:
|
||||
result = await AdminService().query(resource, context, row_id=row_id)
|
||||
response.headers["ETag"] = f'"{result["meta"]["etag"]}"'
|
||||
return result
|
||||
|
||||
if detail:
|
||||
router.add_api_route(f"{prefix}/{{{id_name}}}", get, methods=["GET"],
|
||||
operation_id=f"get_{resource}")
|
||||
|
||||
async def put(
|
||||
payload: BaseModel, response: Response, row_id: int = Path(alias=id_name, gt=0),
|
||||
release_id: int | None = None,
|
||||
context: RequestContext = Depends(build_request_context), # noqa: B008
|
||||
key: str | None = Header(default=None, alias="Idempotency-Key"),
|
||||
if_match: str | None = Header(default=None, alias="If-Match"),
|
||||
) -> dict[str, Any]:
|
||||
result = await AdminService().mutate(resource, context, payload.model_dump(mode="json"),
|
||||
key, if_match, row_id=row_id, release_id=release_id)
|
||||
response.headers["ETag"] = f'"{result["meta"]["etag"]}"'
|
||||
return result
|
||||
|
||||
put.__annotations__["payload"] = schema
|
||||
if update:
|
||||
router.add_api_route(f"{prefix}/{{{id_name}}}", put, methods=["PUT"],
|
||||
operation_id=f"update_{resource}")
|
||||
|
||||
|
||||
def register_transition(resource: str, id_name: str, action: str) -> None:
|
||||
async def transition(
|
||||
payload: BaseModel, response: Response, row_id: int = Path(alias=id_name, gt=0),
|
||||
context: RequestContext = Depends(build_request_context), # noqa: B008
|
||||
key: str | None = Header(default=None, alias="Idempotency-Key"),
|
||||
if_match: str | None = Header(default=None, alias="If-Match"),
|
||||
) -> dict[str, Any]:
|
||||
result = await AdminService().mutate(resource, context, payload.model_dump(mode="json"),
|
||||
key, if_match, row_id=row_id, action=action)
|
||||
response.headers["ETag"] = f'"{result["meta"]["etag"]}"'
|
||||
return result
|
||||
|
||||
transition.__annotations__["payload"] = ReviewPayload if action == "reviews" else EmptyPayload
|
||||
router.add_api_route(f"/{resource}/{{{id_name}}}/{action}", transition, methods=["POST"],
|
||||
status_code=201 if action == "rollbacks" else 200,
|
||||
operation_id=f"{action}_{resource}")
|
||||
|
||||
|
||||
register_resource("config-releases", ReleasePayload, "release_id", update=False)
|
||||
register_resource("platform-config-items", ItemPayload, "item_id", scoped=True, detail=False)
|
||||
register_resource("model-endpoints", EndpointPayload, "endpoint_id")
|
||||
register_resource("model-routing-rules", RoutingPayload, "rule_id", scoped=True, detail=False)
|
||||
register_resource("prompt-templates", PromptPayload, "prompt_id", update=False)
|
||||
register_resource("agent-intent-configs", IntentPayload, "config_id", detail=False)
|
||||
register_resource("reply-templates", ReplyPayload, "template_id", detail=False)
|
||||
register_resource("negative-word-rules", NegativePayload, "rule_id", detail=False)
|
||||
for action in ("validations", "reviews", "activations", "rollbacks"):
|
||||
register_transition("config-releases", "release_id", action)
|
||||
for action in ("reviews", "activations", "disablements"):
|
||||
register_transition("model-endpoints", "endpoint_id", action)
|
||||
|
||||
|
||||
@router.get("/audit-records")
|
||||
async def audit_records(
|
||||
limit: int = Query(default=20, ge=1, le=100),
|
||||
context: RequestContext = Depends(build_request_context), # noqa: B008
|
||||
) -> dict[str, Any]:
|
||||
return await AdminService().query("audit-records", context, limit=limit)
|
||||
Reference in New Issue
Block a user