diff --git a/server.py b/server.py index 778d3a1..35c9cc2 100644 --- a/server.py +++ b/server.py @@ -10,7 +10,6 @@ ChatInputCommandInteraction, ) from tc_messageBroker.rabbit_mq.queue import Queue - from utils.credentials import load_rabbitmq_credentials from utils.fetch_community_id import fetch_community_id_by_guild_id from worker.tasks import ask_question_auto_search diff --git a/utils/traceloop.py b/utils/traceloop.py index 4b1c1c9..9704719 100644 --- a/utils/traceloop.py +++ b/utils/traceloop.py @@ -1,7 +1,8 @@ +import logging +import os + from dotenv import load_dotenv from traceloop.sdk import Traceloop -import os -import logging load_dotenv() diff --git a/worker/celery.py b/worker/celery.py index 8c3fdfb..bb67f4b 100644 --- a/worker/celery.py +++ b/worker/celery.py @@ -1,5 +1,4 @@ from celery import Celery - from utils.credentials import load_rabbitmq_credentials rabbit_creds = load_rabbitmq_credentials() diff --git a/worker/tasks.py b/worker/tasks.py index 3bf8ddd..f833a9d 100644 --- a/worker/tasks.py +++ b/worker/tasks.py @@ -4,6 +4,7 @@ from typing import Any from celery.signals import task_postrun, task_prerun +from subquery import query_multiple_source from tc_messageBroker.rabbit_mq.event import Event from tc_messageBroker.rabbit_mq.payload.discord_bot.base_types.interaction_callback_data import ( InteractionCallbackData, @@ -13,8 +14,6 @@ ) from tc_messageBroker.rabbit_mq.payload.payload import Payload from tc_messageBroker.rabbit_mq.queue import Queue - -from subquery import query_multiple_source from utils.data_source_selector import DataSourceSelector from utils.traceloop import init_tracing from worker.celery import app