Асинхронная обработка

Развёртывание ИИ в производственной среде с помощью FastAPI

Matt Eckerle

Software and Data Engineering Leader

Что такое асинхронная обработка

Официант подаёт напиток посетителю за столиком

Позволяет обрабатывать несколько запросов одновременно

Синхронные и асинхронные запросы

График сравнения времени обработки синхронных и асинхронных запросов

Преобразование синхронных эндпоинтов в асинхронные

@app.post("/analyze")
def analyze_sync(comment: Comment):
    result = sentiment_model(comment.text)
    return {"sentiment": result}

import asyncio

@app.post("/analyze")
async def analyze_async(comment: Comment):
    result = await asyncio.to_thread(
        sentiment_model, comment.text
    )
    return {"sentiment": result}

Реализация фоновых задач

from fastapi import BackgroundTasks
from typing import List

@app.post("/analyze_batch")
async def analyze_batch(
    comments: Comments,
    background_tasks: BackgroundTasks
):

    async def process_comments(texts: List[str]):
        for text in texts:
            result = await asyncio.to_thread(
              sentiment_model, text)
    background_tasks.add_task(process_comments, 
                              comments.texts)
    return {"message": "Processing started"}

BackgroundTasks управляет очередью обработки комментариев

background_tasks выполняет обработку после отправки ответа.

add_task планирует асинхронное выполнение process_comments.

Добавление обработки ошибок

@app.post("/analyze_comment")
async def analyze_comment(comment: Comment):
    try:
        sentiment_model = SentimentAnalyzer()
        result = await asyncio.wait_for(
            sentiment_model(comment.text),
            timeout=5.0
        )
        return {"sentiment": result["label"]}

Добавление обработки ошибок

    except asyncio.TimeoutError:
        raise HTTPException(
            status_code=408,
            detail="Analysis timed out"
        )

    except Exception:
        raise HTTPException(
            status_code=500,
            detail="Analysis failed"
        )

Давайте потренируемся!

Развёртывание ИИ в производственной среде с помощью FastAPI