36 lines
915 B
Python
36 lines
915 B
Python
from celery import Celery
|
|
from app.core.config import settings
|
|
from app.handlers.toxicity_handler import ToxicityHandler
|
|
|
|
# Инициализация Celery
|
|
celery_app = Celery(
|
|
'tasks',
|
|
broker=settings.CELERY_BROKER_URL,
|
|
backend=settings.CELERY_RESULT_BACKEND
|
|
)
|
|
|
|
# Конфигурация Celery
|
|
celery_app.conf.update(
|
|
task_serializer='json',
|
|
result_serializer='json',
|
|
accept_content=['json'],
|
|
timezone='UTC',
|
|
enable_utc=True,
|
|
)
|
|
|
|
# Инициализация обработчика токсичности
|
|
toxicity_handler = ToxicityHandler()
|
|
|
|
@celery_app.task
|
|
def assess_toxicity_task(text: str) -> float:
|
|
"""
|
|
Задача для оценки токсичности текста.
|
|
|
|
Args:
|
|
text (str): Входной текст.
|
|
|
|
Returns:
|
|
float: Оценка токсичности.
|
|
"""
|
|
return toxicity_handler.text_to_toxicity(text)
|