From 8cabf823e8059cd9d49b607a0c21f2f2129e255b Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Thu, 6 Nov 2025 16:46:20 +0800
Subject: [PATCH] fix: dependency package (#4325)

---
 apps/common/event/__init__.py           | 3 ++-
 apps/knowledge/serializers/document.py  | 2 +-
 apps/knowledge/serializers/knowledge.py | 2 +-
 apps/knowledge/serializers/paragraph.py | 2 +-
 apps/knowledge/task/embedding.py        | 2 +-
 apps/knowledge/task/generate.py         | 3 +--
 6 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/apps/common/event/__init__.py b/apps/common/event/__init__.py
index 59a61e9ab..e8f0b3d81 100644
--- a/apps/common/event/__init__.py
+++ b/apps/common/event/__init__.py
@@ -7,9 +7,10 @@
     @desc:
 """
 from django.core.cache import cache
+from django.db.models import QuerySet
 from django.utils.translation import gettext as _
 
-from .listener_manage import *
+
 from ..constants.cache_version import Cache_Version
 from ..db.sql_execute import update_execute
 from ..utils.lock import RedisLock
diff --git a/apps/knowledge/serializers/document.py b/apps/knowledge/serializers/document.py
index b077f99c6..8f568c110 100644
--- a/apps/knowledge/serializers/document.py
+++ b/apps/knowledge/serializers/document.py
@@ -25,7 +25,7 @@ from rest_framework import serializers
 from xlwt import Utils
 
 from common.db.search import native_search, get_dynamics_model, native_page_search
-from common.event import ListenerManagement
+from common.event.listener_manage import ListenerManagement
 from common.event.common import work_thread_pool
 from common.exception.app_exception import AppApiException
 from common.field.common import UploadedFileField
diff --git a/apps/knowledge/serializers/knowledge.py b/apps/knowledge/serializers/knowledge.py
index 0f98a42a0..9daeb7dad 100644
--- a/apps/knowledge/serializers/knowledge.py
+++ b/apps/knowledge/serializers/knowledge.py
@@ -24,7 +24,7 @@ from common.config.embedding_config import VectorStore
 from common.database_model_manage.database_model_manage import DatabaseModelManage
 from common.db.search import native_search, get_dynamics_model, native_page_search
 from common.db.sql_execute import select_list
-from common.event import ListenerManagement
+from common.event.listener_manage import ListenerManagement
 from common.exception.app_exception import AppApiException
 from common.utils.common import post, get_file_content, parse_image
 from common.utils.fork import Fork, ChildLink
diff --git a/apps/knowledge/serializers/paragraph.py b/apps/knowledge/serializers/paragraph.py
index 28f1aa10b..a1df1bd81 100644
--- a/apps/knowledge/serializers/paragraph.py
+++ b/apps/knowledge/serializers/paragraph.py
@@ -10,7 +10,7 @@ from django.utils.translation import gettext_lazy as _
 from rest_framework import serializers
 
 from common.db.search import page_search
-from common.event import ListenerManagement
+from common.event.listener_manage import ListenerManagement
 from common.exception.app_exception import AppApiException
 from common.utils.common import post
 from knowledge.models import Paragraph, Problem, Document, ProblemParagraphMapping, SourceType, TaskType, State, \
diff --git a/apps/knowledge/task/embedding.py b/apps/knowledge/task/embedding.py
index 8410118d0..750c28e15 100644
--- a/apps/knowledge/task/embedding.py
+++ b/apps/knowledge/task/embedding.py
@@ -8,7 +8,7 @@ from django.db.models import QuerySet
 from django.utils.translation import gettext_lazy as _
 
 from common.config.embedding_config import ModelManage
-from common.event import ListenerManagement, UpdateProblemArgs, UpdateEmbeddingKnowledgeIdArgs, \
+from common.event.listener_manage import ListenerManagement, UpdateProblemArgs, UpdateEmbeddingKnowledgeIdArgs, \
     UpdateEmbeddingDocumentIdArgs
 from common.utils.logger import maxkb_logger
 from knowledge.models import Document, TaskType, State
diff --git a/apps/knowledge/task/generate.py b/apps/knowledge/task/generate.py
index 9a86a7ea2..bf89122a5 100644
--- a/apps/knowledge/task/generate.py
+++ b/apps/knowledge/task/generate.py
@@ -1,4 +1,3 @@
-import logging
 import traceback
 
 from celery_once import QueueOnce
@@ -8,7 +7,7 @@ from django.utils.translation import gettext_lazy as _
 from langchain_core.messages import HumanMessage
 
 from common.config.embedding_config import ModelManage
-from common.event import ListenerManagement
+from common.event.listener_manage import ListenerManagement
 from common.utils.logger import maxkb_logger
 from common.utils.page_utils import page, page_desc
 from knowledge.models import Paragraph, Document, Status, TaskType, State