Processamento assíncrono

Colocando IA em Produção com FastAPI

Matt Eckerle

Software and Data Engineering Leader

O que é processamento assíncrono

Uma pessoa servindo uma bebida a alguém sentado à mesa

Permite lidar com várias requisições ao mesmo tempo

Requisições síncronas vs assíncronas

Um gráfico mostrando a diferença nos tempos de processamento para requisições síncronas e assíncronas

Tornando endpoints síncronos em assíncronos

@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}

Implementando tarefas em segundo plano

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 gerencia a fila de processamento de comentários

background_tasks cuida do processamento após a resposta.

add_task agenda process_comments de forma assíncrona.

Adicionando tratamento de erros

@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"]}

Adicionando tratamento de erros

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

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

Vamos praticar!

Colocando IA em Produção com FastAPI