Refactor code structure for improved readability and maintainability
This commit is contained in:
@@ -17,13 +17,14 @@ import os
|
||||
import sys
|
||||
import time
|
||||
|
||||
sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))
|
||||
PROJECT_ROOT = os.path.dirname(os.path.dirname(__file__))
|
||||
sys.path.insert(0, os.path.join(PROJECT_ROOT, "backend"))
|
||||
|
||||
from loguru import logger
|
||||
from src.config.logging import setup_logging
|
||||
from src.services.document_processor import DocumentProcessor, ProcessingResult
|
||||
from src.services.storage.milvus_client import MilvusClient
|
||||
from src.config.settings import settings
|
||||
from app.config.logging import setup_logging
|
||||
from app.services.document_processor import DocumentProcessor, ProcessingResult
|
||||
from app.services.storage.milvus_client import MilvusClient
|
||||
from app.config.settings import settings
|
||||
|
||||
# 设置日志
|
||||
setup_logging(level="INFO")
|
||||
@@ -66,7 +67,7 @@ def verify_embedding_model():
|
||||
logger.info("=" * 50)
|
||||
|
||||
try:
|
||||
from src.services.embedding.bge_m3_embedder import BGEM3Embedder
|
||||
from app.services.embedding.bge_m3_embedder import BGEM3Embedder
|
||||
|
||||
embedder = BGEM3Embedder()
|
||||
logger.success("嵌入模型加载成功")
|
||||
@@ -126,9 +127,9 @@ def verify_sample_document():
|
||||
"""
|
||||
|
||||
try:
|
||||
from src.services.embedding.text_chunker import RegulationChunker
|
||||
from src.services.embedding.bge_m3_embedder import BGEM3Embedder
|
||||
from src.services.storage.milvus_client import MilvusClient
|
||||
from app.services.embedding.text_chunker import RegulationChunker
|
||||
from app.services.embedding.bge_m3_embedder import BGEM3Embedder
|
||||
from app.services.storage.milvus_client import MilvusClient
|
||||
|
||||
# 1. 分块
|
||||
logger.info("测试分块...")
|
||||
@@ -219,4 +220,4 @@ def main():
|
||||
|
||||
if __name__ == "__main__":
|
||||
success = main()
|
||||
sys.exit(0 if success else 1)
|
||||
sys.exit(0 if success else 1)
|
||||
|
||||
Reference in New Issue
Block a user