40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
from uuid import uuid4
|
|
|
|
from fastapi import APIRouter, Depends, Request, Response
|
|
|
|
from app.core.security import get_current_user
|
|
from app.models.user import User
|
|
from app.schemas.ai import (
|
|
AIProviderStatusResponse,
|
|
SituationalAnalysisRequest,
|
|
SituationalAnalysisResponse,
|
|
)
|
|
from app.services.ai_client import AIProviderClient, get_ai_provider_client
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/provider/status", response_model=AIProviderStatusResponse)
|
|
async def get_ai_provider_status(
|
|
request: Request,
|
|
response: Response,
|
|
current_user: User = Depends(get_current_user),
|
|
provider_client: AIProviderClient = Depends(get_ai_provider_client),
|
|
):
|
|
request_id = request.headers.get("X-Request-ID") or str(uuid4())
|
|
response.headers["X-Request-ID"] = request_id
|
|
return await provider_client.get_status(request_id=request_id)
|
|
|
|
|
|
@router.post("/situational-awareness/analyze", response_model=SituationalAnalysisResponse)
|
|
async def analyze_situational_awareness(
|
|
payload: SituationalAnalysisRequest,
|
|
request: Request,
|
|
response: Response,
|
|
current_user: User = Depends(get_current_user),
|
|
provider_client: AIProviderClient = Depends(get_ai_provider_client),
|
|
):
|
|
request_id = request.headers.get("X-Request-ID") or str(uuid4())
|
|
response.headers["X-Request-ID"] = request_id
|
|
return await provider_client.analyze(payload, request_id=request_id)
|