upload
This commit is contained in:
@@ -0,0 +1,92 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from fastapi import APIRouter, Depends, File, HTTPException, Request, UploadFile, status
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.core.database.session import get_session
|
||||
from app.models import Action, ActionIngestStatus, User
|
||||
from app.schemas.capability_sch import ActionIngestWithCapabilitiesResponse
|
||||
from app.services.capability_service import CapabilityService
|
||||
from app.services.openapi_service import OpenAPIService
|
||||
from app.utils.business_logger import log_business_event
|
||||
from app.utils.token_manager import get_current_user
|
||||
|
||||
|
||||
router = APIRouter(tags=["Actions"])
|
||||
|
||||
|
||||
@router.post("/ingest", response_model=ActionIngestWithCapabilitiesResponse, status_code=status.HTTP_201_CREATED)
|
||||
async def ingest_actions(
|
||||
request: Request,
|
||||
file: UploadFile = File(...),
|
||||
session: AsyncSession = Depends(get_session),
|
||||
current_user: User = Depends(get_current_user),
|
||||
):
|
||||
trace_id = getattr(request.state, "traceId", None)
|
||||
payload = await file.read()
|
||||
try:
|
||||
document = OpenAPIService.load_document(payload)
|
||||
ingestion_result = OpenAPIService.extract_actions_with_failures(document, source_filename=file.filename)
|
||||
except ValueError as exc:
|
||||
log_business_event(
|
||||
"actions_ingest_rejected",
|
||||
trace_id=trace_id,
|
||||
user_id=str(current_user.id),
|
||||
source_filename=file.filename,
|
||||
file_size_bytes=len(payload),
|
||||
reason="invalid_openapi_document",
|
||||
details=str(exc),
|
||||
)
|
||||
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(exc)) from exc
|
||||
|
||||
action_payloads = ingestion_result["succeeded"] + ingestion_result["failed"]
|
||||
if not action_payloads:
|
||||
log_business_event(
|
||||
"actions_ingest_rejected",
|
||||
trace_id=trace_id,
|
||||
user_id=str(current_user.id),
|
||||
source_filename=file.filename,
|
||||
file_size_bytes=len(payload),
|
||||
reason="no_supported_operations",
|
||||
)
|
||||
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="No supported HTTP operations found in OpenAPI file")
|
||||
|
||||
actions = [Action(user_id=current_user.id, **action_payload) for action_payload in action_payloads]
|
||||
session.add_all(actions)
|
||||
await session.flush()
|
||||
|
||||
succeeded_actions = [action for action in actions if action.ingest_status == ActionIngestStatus.SUCCEEDED]
|
||||
failed_actions = [action for action in actions if action.ingest_status == ActionIngestStatus.FAILED]
|
||||
|
||||
capability_service = CapabilityService(session)
|
||||
capabilities = await capability_service.create_from_actions(
|
||||
succeeded_actions,
|
||||
owner_user_id=current_user.id,
|
||||
refresh=False,
|
||||
)
|
||||
await session.commit()
|
||||
|
||||
for action in actions:
|
||||
await session.refresh(action)
|
||||
for capability in capabilities:
|
||||
await session.refresh(capability)
|
||||
|
||||
log_business_event(
|
||||
"actions_ingested",
|
||||
trace_id=trace_id,
|
||||
user_id=str(current_user.id),
|
||||
source_filename=file.filename,
|
||||
file_size_bytes=len(payload),
|
||||
succeeded_count=len(succeeded_actions),
|
||||
failed_count=len(failed_actions),
|
||||
created_capabilities_count=len(capabilities),
|
||||
)
|
||||
|
||||
return ActionIngestWithCapabilitiesResponse(
|
||||
succeeded_count=len(succeeded_actions),
|
||||
failed_count=len(failed_actions),
|
||||
created_capabilities_count=len(capabilities),
|
||||
succeeded_actions=succeeded_actions,
|
||||
failed_actions=failed_actions,
|
||||
capabilities=capabilities,
|
||||
)
|
||||
Reference in New Issue
Block a user