jyong
1 month ago
2 changed files with 2 additions and 2 deletions
+ 1
- 1
api/docker/entrypoint.sh
View File
| exec celery -A celery_entrypoint.celery worker -P ${CELERY_WORKER_CLASS:-gevent} $CONCURRENCY_OPTION \ | exec celery -A celery_entrypoint.celery worker -P ${CELERY_WORKER_CLASS:-gevent} $CONCURRENCY_OPTION \ | ||||
| --max-tasks-per-child ${MAX_TASKS_PER_CHILD:-50} --loglevel ${LOG_LEVEL:-INFO} \ | --max-tasks-per-child ${MAX_TASKS_PER_CHILD:-50} --loglevel ${LOG_LEVEL:-INFO} \ | ||||
| -Q ${CELERY_QUEUES:-dataset,mail,ops_trace,app_deletion,plugin,workflow_storage,conversation} | |||||
| -Q ${CELERY_QUEUES:-dataset,pipeline,mail,ops_trace,app_deletion,plugin,workflow_storage,conversation} | |||||
| elif [[ "${MODE}" == "beat" ]]; then | elif [[ "${MODE}" == "beat" ]]; then | ||||
| exec celery -A app.celery beat --loglevel ${LOG_LEVEL:-INFO} | exec celery -A app.celery beat --loglevel ${LOG_LEVEL:-INFO} |
+ 1
- 1
api/tasks/rag_pipeline/rag_pipeline_run_task.py
View File
| from models.workflow import Workflow, WorkflowNodeExecutionTriggeredFrom | from models.workflow import Workflow, WorkflowNodeExecutionTriggeredFrom | ||||
| @shared_task(queue="dataset") | |||||
| @shared_task(queue="pipeline") | |||||
| def rag_pipeline_run_task( | def rag_pipeline_run_task( | ||||
| pipeline_id: str, | pipeline_id: str, | ||||
| application_generate_entity: dict, | application_generate_entity: dict, |
Loading…