Spaces:
Runtime error
Runtime error
File size: 2,373 Bytes
66c165b e71dab3 66c165b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
"""Router de testes de processadores"""
from fastapi import APIRouter, HTTPException
from api.models.requests import ProcessorTestRequest
from api.models.responses import ProcessorTestResponse
from api.utils.logger import setup_logger
import time
router = APIRouter()
logger = setup_logger(__name__)
@router.post("/run", response_model=ProcessorTestResponse)
async def test_processor(request: ProcessorTestRequest):
"""Testa um processador específico."""
start_time = time.time()
try:
# Importar processador
processor_map = {
"1": "processor_metadados",
"2": "processor_relatorio",
"metadados": "processor_metadados",
"relatorio": "processor_relatorio"
}
processor_name = processor_map.get(request.processor_name, request.processor_name)
from processors import processor_base
processor = processor_base.get_processor(processor_name)
# Processar
result = await processor.process(
ementa=request.ementa,
integra=request.integra,
tribunal=request.tribunal
)
duration_ms = (time.time() - start_time) * 1000
return ProcessorTestResponse(
processor_name=processor_name,
success=True,
result=result,
duration_ms=duration_ms
)
except Exception as e:
duration_ms = (time.time() - start_time) * 1000
logger.error(f"Processor test error: {e}")
return ProcessorTestResponse(
processor_name=request.processor_name,
success=False,
duration_ms=duration_ms,
error=str(e)
)
@router.get("/list")
async def list_processors():
"""Lista processadores disponíveis."""
return {
"processors": [
{"id": 1, "name": "Metadados"},
{"id": 2, "name": "Relatório"},
{"id": 3, "name": "Fundamentação"},
{"id": 4, "name": "Decisão"},
{"id": 5, "name": "Auditoria"},
{"id": 6, "name": "Arquivo Relacional"},
{"id": 7, "name": "Segmentação Sintática"},
{"id": 8, "name": "Análise Contextual"},
{"id": 9, "name": "Transcrição 3-Partite"}
],
"total": 9
}
|